You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
90 lines
3.2 KiB
90 lines
3.2 KiB
|
|
import * as Winreg from 'winreg';
|
|
import { Logging } from './logging';
|
|
import { toPromise } from './utils';
|
|
|
|
const winreg = new Winreg({
|
|
hive: Winreg.HKCU,
|
|
key: `\\Software\\SimonTatham\\PuTTY\\Sessions\\`,
|
|
});
|
|
|
|
export type NumberAsBoolean = 0 | 1;
|
|
export interface PuttySession {
|
|
//[key: string]: string | number | undefined;
|
|
// General settings
|
|
name: string;
|
|
hostname: string;
|
|
protocol: string;
|
|
portnumber: number;
|
|
username?: string;
|
|
usernamefromenvironment: NumberAsBoolean;
|
|
tryagent: NumberAsBoolean;
|
|
publickeyfile?: string;
|
|
// Proxy settings
|
|
proxyhost?: string;
|
|
proxyport: number;
|
|
proxylocalhost: NumberAsBoolean;
|
|
proxymethod: number; // Key of ['None', 'SOCKS 4', 'SOCKS 5', 'HTTP', 'Telnet', 'Local']
|
|
}
|
|
|
|
function valueFromItem(item: Winreg.RegistryItem) {
|
|
switch (item.type) {
|
|
case 'REG_DWORD':
|
|
return parseInt(item.value, 16);
|
|
case 'REG_SZ':
|
|
return item.value;
|
|
}
|
|
throw new Error(`Unknown RegistryItem type: '${item.type}'`);
|
|
}
|
|
|
|
const FORMATTED_FIELDS: (keyof PuttySession)[] = [
|
|
'name', 'hostname', 'protocol', 'portnumber',
|
|
'username', 'usernamefromenvironment', 'tryagent', 'publickeyfile',
|
|
'proxyhost', 'proxyport', 'proxylocalhost', 'proxymethod',
|
|
];
|
|
export function formatSession(session: PuttySession): string {
|
|
const partial: Partial<PuttySession> = {};
|
|
for (const field of FORMATTED_FIELDS) partial[field] = session[field] as any;
|
|
return JSON.stringify(partial);
|
|
}
|
|
|
|
export async function getSessions() {
|
|
Logging.info`Fetching PuTTY sessions from registry`;
|
|
const values = await toPromise<Winreg.Registry[]>(cb => winreg.keys(cb));
|
|
const sessions: PuttySession[] = [];
|
|
await Promise.all(values.map(regSession => (async () => {
|
|
const name = decodeURIComponent(regSession.key.substr(winreg.key.length));
|
|
const props = await toPromise<Winreg.RegistryItem[]>(cb => regSession.values(cb));
|
|
const properties: { [key: string]: string | number } = {};
|
|
props.forEach(prop => properties[prop.name.toLowerCase()] = valueFromItem(prop));
|
|
sessions.push({ name, ...(properties as any) });
|
|
})()));
|
|
Logging.debug`\tFound ${sessions.length} sessions:`;
|
|
sessions.forEach(s => Logging.debug`\t- ${formatSession(s)}`);
|
|
return sessions;
|
|
}
|
|
|
|
export async function findSession(sessions: PuttySession[], name?: string, host?: string, username?: string, nameOnly = true): Promise<PuttySession | undefined> {
|
|
if (name) {
|
|
name = name.toLowerCase();
|
|
const session = sessions.find(s => s.name.toLowerCase() === name);
|
|
if (nameOnly || session) return session;
|
|
}
|
|
if (!host) return undefined;
|
|
host = host.toLowerCase();
|
|
const hosts = sessions.filter(s => s.hostname && s.hostname.toLowerCase() === host);
|
|
if (!username) return hosts[0] || null;
|
|
username = username.toLowerCase();
|
|
return hosts.find(s => !s.username || s.username.toLowerCase() === username);
|
|
}
|
|
|
|
export async function getSession(name?: string, host?: string, username?: string, nameOnly = true): Promise<PuttySession | undefined> {
|
|
const sessions = await getSessions();
|
|
return findSession(sessions, name, host, username, nameOnly);
|
|
}
|
|
|
|
export async function getCachedFinder(): Promise<typeof getSession> {
|
|
const sessions = await getSessions();
|
|
return (...args) => findSession(sessions, ...args);
|
|
}
|