feat: add PairingManager and gateway lock tests (Tier 4 feature 4 foundation)

This commit is contained in:
William Valentin
2026-02-09 13:32:59 -08:00
parent 4413c4dc7c
commit 9d4d440ecf
4 changed files with 389 additions and 0 deletions
+133
View File
@@ -0,0 +1,133 @@
import { randomBytes } from 'crypto';
export interface PairingConfig {
enabled: boolean;
codeTtl: number; // milliseconds
codeLength: number; // number of characters
}
interface PendingCode {
code: string;
createdAt: number;
expiresAt: number;
/** Optional label for the code (e.g. "for alice"). */
label?: string;
}
interface ApprovedSender {
channel: string;
senderId: string;
approvedAt: number;
/** The code that was used. */
codeUsed: string;
}
/**
* Manages DM pairing codes for authenticating unknown senders.
*
* Flow:
* 1. Admin generates a pairing code via gateway API or TUI command.
* 2. Unknown sender DMs the bot with the code as their first message.
* 3. If the code is valid and not expired, the sender is approved.
* 4. Approved senders bypass the allowlist check for subsequent messages.
*/
export class PairingManager {
private config: PairingConfig;
private pendingCodes: Map<string, PendingCode> = new Map();
private approvedSenders: Map<string, ApprovedSender> = new Map();
constructor(config: PairingConfig) {
this.config = config;
}
/** Generate a new pairing code. Returns the code string. */
generateCode(label?: string): string {
this.cleanup();
const code = randomBytes(Math.ceil(this.config.codeLength / 2))
.toString('hex')
.slice(0, this.config.codeLength)
.toUpperCase();
const now = Date.now();
this.pendingCodes.set(code, {
code,
createdAt: now,
expiresAt: now + this.config.codeTtl,
label,
});
return code;
}
/**
* Validate a code for a given channel+sender.
* If valid, adds the sender to the approved list and removes the code.
* Returns true if the code was valid.
*/
validateCode(channel: string, senderId: string, code: string): boolean {
this.cleanup();
const normalizedCode = code.trim().toUpperCase();
const pending = this.pendingCodes.get(normalizedCode);
if (!pending) return false;
if (Date.now() > pending.expiresAt) {
this.pendingCodes.delete(normalizedCode);
return false;
}
// Code is valid — approve the sender
const key = `${channel}:${senderId}`;
this.approvedSenders.set(key, {
channel,
senderId,
approvedAt: Date.now(),
codeUsed: normalizedCode,
});
// Remove the used code
this.pendingCodes.delete(normalizedCode);
return true;
}
/** Check if a sender is already approved. */
isApproved(channel: string, senderId: string): boolean {
const key = `${channel}:${senderId}`;
return this.approvedSenders.has(key);
}
/** Revoke approval for a sender. Returns true if the sender was found and removed. */
revokeApproval(channel: string, senderId: string): boolean {
const key = `${channel}:${senderId}`;
return this.approvedSenders.delete(key);
}
/** List all currently approved senders. */
listApproved(): ApprovedSender[] {
return Array.from(this.approvedSenders.values());
}
/** List all pending (non-expired) codes. */
listPendingCodes(): Array<{ code: string; expiresAt: number; label?: string }> {
this.cleanup();
return Array.from(this.pendingCodes.values()).map(p => ({
code: p.code,
expiresAt: p.expiresAt,
label: p.label,
}));
}
/** Remove expired codes. */
cleanup(): void {
const now = Date.now();
for (const [code, pending] of this.pendingCodes) {
if (now > pending.expiresAt) {
this.pendingCodes.delete(code);
}
}
}
/** Whether pairing is enabled. */
get enabled(): boolean {
return this.config.enabled;
}
}