55 lines
1.7 KiB
TypeScript
55 lines
1.7 KiB
TypeScript
|
import type { KnownHost } from "./models";
|
||
|
|
import { createKnownHostFromHostKeyInfo } from "./knownHosts";
|
||
|
|
import {
|
||
|
|
toHostKeyInfo,
|
||
|
|
type HostKeyInfo,
|
||
|
|
type HostKeyVerificationRequest,
|
||
|
|
} from "./hostKey";
|
||
|
|
|
||
|
|
export type { HostKeyInfo, HostKeyVerificationRequest };
|
||
|
|
|
||
|
|
export const isPortForwardHostKeySessionId = (sessionId?: string): boolean => {
|
||
|
|
return typeof sessionId === "string" && sessionId.startsWith("pf-");
|
||
|
|
};
|
||
|
|
|
||
|
|
export type PortForwardHostKeyRequest = HostKeyVerificationRequest & {
|
||
|
|
requestId: string;
|
||
|
|
sessionId?: string;
|
||
|
|
};
|
||
|
|
|
||
|
|
export interface PendingPortForwardHostKeyVerification {
|
||
|
|
requestId: string;
|
||
|
|
hostKeyInfo: HostKeyInfo;
|
||
|
|
}
|
||
|
|
|
||
|
|
export const toPendingPortForwardHostKeyVerification = (
|
||
|
|
request: PortForwardHostKeyRequest,
|
||
|
|
): PendingPortForwardHostKeyVerification | null => {
|
||
|
|
if (!isPortForwardHostKeySessionId(request.sessionId)) return null;
|
||
|
|
return {
|
||
|
|
requestId: request.requestId,
|
||
|
|
hostKeyInfo: toHostKeyInfo(request),
|
||
|
|
};
|
||
|
|
};
|
||
|
|
|
||
|
|
export const enqueuePortForwardHostKeyVerification = (
|
||
|
|
queue: PendingPortForwardHostKeyVerification[],
|
||
|
|
pending: PendingPortForwardHostKeyVerification,
|
||
|
|
): PendingPortForwardHostKeyVerification[] => [...queue, pending];
|
||
|
|
|
||
|
|
export const removePortForwardHostKeyVerification = (
|
||
|
|
queue: PendingPortForwardHostKeyVerification[],
|
||
|
|
requestId: string,
|
||
|
|
): PendingPortForwardHostKeyVerification[] => {
|
||
|
|
if (queue[0]?.requestId === requestId) {
|
||
|
|
return queue.slice(1);
|
||
|
|
}
|
||
|
|
return queue.filter((pending) => pending.requestId !== requestId);
|
||
|
|
};
|
||
|
|
|
||
|
|
export const createKnownHostFromPortForwardHostKeyInfo = (
|
||
|
|
hostKeyInfo: HostKeyInfo,
|
||
|
|
now = Date.now(),
|
||
|
|
idSuffix = Math.random().toString(36).slice(2, 11),
|
||
|
|
): KnownHost => createKnownHostFromHostKeyInfo(hostKeyInfo, { now, idSuffix });
|