mirror of
https://github.com/aljazceru/dvmcp.git
synced 2025-12-17 05:14:24 +01:00
379 lines
10 KiB
TypeScript
379 lines
10 KiB
TypeScript
import { createInterface } from 'node:readline';
|
||
import { parse, stringify } from 'yaml';
|
||
import { writeFileSync, existsSync, readFileSync } from 'node:fs';
|
||
import { randomBytes } from 'node:crypto';
|
||
import { HEX_KEYS_REGEX } from './constants';
|
||
|
||
export const CONFIG_EMOJIS = {
|
||
NOSTR: '🔑',
|
||
RELAY: '🔄',
|
||
SERVICE: '🤖',
|
||
SERVER: '🖥️',
|
||
WHITELIST: '🔒',
|
||
SETUP: '⚙️',
|
||
SUCCESS: '✅',
|
||
INFO: 'ℹ️',
|
||
PROMPT: '❯',
|
||
} as const;
|
||
|
||
export type FieldType =
|
||
| 'string'
|
||
| 'array'
|
||
| 'boolean'
|
||
| 'hex'
|
||
| 'url'
|
||
| 'nested'
|
||
| 'object-array'
|
||
| 'set';
|
||
|
||
interface ArrayItem {
|
||
name: string;
|
||
command: string;
|
||
args: string[];
|
||
[key: string]: any;
|
||
}
|
||
|
||
export interface FieldConfig {
|
||
type: FieldType;
|
||
description?: string;
|
||
default?: any;
|
||
validation?: (value: any) => boolean;
|
||
generator?: () => any;
|
||
required?: boolean;
|
||
itemType?: FieldConfig;
|
||
fields?: Record<string, FieldConfig>;
|
||
comment?: string;
|
||
emoji?: string;
|
||
}
|
||
|
||
export class ConfigGenerator<T extends Record<string, any>> {
|
||
private currentConfig: T | null = null;
|
||
constructor(
|
||
private configPath: string,
|
||
private fields: Record<string, FieldConfig>
|
||
) {
|
||
if (existsSync(configPath)) {
|
||
try {
|
||
this.currentConfig = parse(readFileSync(configPath, 'utf8'));
|
||
} catch (error) {
|
||
console.warn(
|
||
`${CONFIG_EMOJIS.INFO} Could not parse existing configuration`
|
||
);
|
||
}
|
||
}
|
||
}
|
||
|
||
private async prompt(question: string, defaultValue = ''): Promise<string> {
|
||
const rl = createInterface({
|
||
input: process.stdin,
|
||
output: process.stdout,
|
||
});
|
||
|
||
const answer = await new Promise<string>((resolve) => {
|
||
rl.question(
|
||
`${question}${defaultValue ? ` (${defaultValue})` : ''} `,
|
||
(answer) => resolve(answer || defaultValue)
|
||
);
|
||
});
|
||
|
||
rl.close();
|
||
return answer;
|
||
}
|
||
|
||
private async promptYesNo(
|
||
question: string,
|
||
defaultValue = false
|
||
): Promise<boolean> {
|
||
const answer = await this.prompt(
|
||
`${question} (${defaultValue ? 'Y/n' : 'y/N'})`
|
||
);
|
||
return answer.trim() === ''
|
||
? defaultValue
|
||
: answer.toLowerCase().startsWith('y');
|
||
}
|
||
|
||
private async handleObjectArrayItem(
|
||
fields: Record<string, FieldConfig>
|
||
): Promise<Record<string, any> | null> {
|
||
const item: Record<string, any> = { name: '' };
|
||
|
||
const name = await this.prompt(
|
||
`${CONFIG_EMOJIS.SERVER} Name (empty to finish):`
|
||
);
|
||
if (!name) return null;
|
||
|
||
item.name = name;
|
||
|
||
for (const [fieldName, fieldConfig] of Object.entries(fields)) {
|
||
if (fieldName === 'name') continue;
|
||
|
||
const value = await this.handleField(fieldName, fieldConfig, false);
|
||
if (value === null) return null;
|
||
item[fieldName] = value;
|
||
}
|
||
|
||
return item;
|
||
}
|
||
|
||
private async handleField(
|
||
fieldName: string,
|
||
config: FieldConfig,
|
||
showFieldName = true,
|
||
currentValue?: any
|
||
): Promise<any> {
|
||
const emoji = config.emoji || CONFIG_EMOJIS.PROMPT;
|
||
|
||
if (currentValue !== undefined) {
|
||
if (config.type !== 'nested' && config.type !== 'object-array') {
|
||
console.log(
|
||
`${CONFIG_EMOJIS.INFO} Current value: ${
|
||
typeof currentValue === 'object'
|
||
? JSON.stringify(currentValue)
|
||
: currentValue
|
||
}`
|
||
);
|
||
}
|
||
|
||
const keepCurrent = await this.promptYesNo(
|
||
`${emoji} Keep current ${fieldName}?`,
|
||
true
|
||
);
|
||
|
||
if (keepCurrent) {
|
||
return currentValue;
|
||
}
|
||
}
|
||
|
||
if (config.type === 'nested') {
|
||
console.log(`\n${emoji} ${config.description || fieldName}`);
|
||
}
|
||
|
||
const handlers: Record<FieldType, () => Promise<any>> = {
|
||
string: async () =>
|
||
this.promptWithValidation(
|
||
`${emoji} ${config.description || (showFieldName ? fieldName : '')}:`,
|
||
config,
|
||
currentValue || config.default
|
||
),
|
||
|
||
hex: async () => {
|
||
if (
|
||
config.generator &&
|
||
(await this.promptYesNo(`${emoji} Generate new ${fieldName}?`))
|
||
) {
|
||
const value = config.generator();
|
||
console.log(`${CONFIG_EMOJIS.PROMPT} ${value}`);
|
||
return value;
|
||
}
|
||
|
||
const existingValue = currentValue || config.default;
|
||
existingValue &&
|
||
console.log(`${CONFIG_EMOJIS.PROMPT} ${existingValue}`);
|
||
return this.promptWithValidation(
|
||
`${emoji} Enter ${fieldName}:`,
|
||
config,
|
||
existingValue
|
||
);
|
||
},
|
||
|
||
array: async () => {
|
||
const array: string[] = currentValue || [];
|
||
if (array.length > 0) {
|
||
console.log('\nCurrent items:');
|
||
array.forEach((item: string, index: number) => {
|
||
console.log(`${CONFIG_EMOJIS.INFO} ${index + 1}. ${item}`);
|
||
});
|
||
|
||
if (await this.promptYesNo(`${emoji} Remove any items?`, false)) {
|
||
while (true) {
|
||
const index =
|
||
parseInt(
|
||
await this.prompt('Enter index to remove (0 to finish):')
|
||
) - 1;
|
||
if (isNaN(index) || index < 0) break;
|
||
if (index < array.length) {
|
||
array.splice(index, 1);
|
||
console.log('Item removed');
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
if (await this.promptYesNo(`${emoji} Add ${fieldName}?`, true)) {
|
||
while (true) {
|
||
const item = await this.promptWithValidation(
|
||
`${emoji} Value (empty to finish):`,
|
||
config,
|
||
'',
|
||
true
|
||
);
|
||
if (!item) break;
|
||
array.push(item);
|
||
}
|
||
}
|
||
return array;
|
||
},
|
||
|
||
set: async () => {
|
||
const items: string[] = Array.from(currentValue || []);
|
||
if (await this.promptYesNo('Would you like to add items?', false)) {
|
||
while (true) {
|
||
const item = await this.promptWithValidation(
|
||
'Enter item (empty to finish):',
|
||
config,
|
||
'',
|
||
true
|
||
);
|
||
if (!item) break;
|
||
items.push(item);
|
||
}
|
||
}
|
||
return new Set(items);
|
||
},
|
||
|
||
'object-array': async () => {
|
||
let array: ArrayItem[] = currentValue || [];
|
||
if (array.length > 0) {
|
||
console.log('\nCurrent servers:');
|
||
array.forEach((item: ArrayItem, index: number) => {
|
||
console.log(
|
||
`${CONFIG_EMOJIS.INFO} ${index + 1}. ${item.name} (${item.command} ${item.args.join(' ')})`
|
||
);
|
||
});
|
||
console.log('');
|
||
const keepCurrent = await this.promptYesNo(
|
||
`${emoji} Keep current ${fieldName}?`,
|
||
true
|
||
);
|
||
if (!keepCurrent) {
|
||
array = [];
|
||
console.log(`${CONFIG_EMOJIS.INFO} Cleared existing servers.`);
|
||
} else {
|
||
if (
|
||
await this.promptYesNo(
|
||
`${emoji} Remove any existing servers?`,
|
||
false
|
||
)
|
||
) {
|
||
while (true) {
|
||
const index =
|
||
parseInt(
|
||
await this.prompt(
|
||
'Enter server number to remove (0 to finish):'
|
||
)
|
||
) - 1;
|
||
if (isNaN(index) || index < 0) break;
|
||
if (index < array.length) {
|
||
console.log(
|
||
`${CONFIG_EMOJIS.INFO} Removed server: ${array[index].name}`
|
||
);
|
||
array.splice(index, 1);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
if (await this.promptYesNo(`${emoji} Add new ${fieldName}?`, true)) {
|
||
while (true) {
|
||
const item = await this.handleObjectArrayItem(config.fields!);
|
||
if (!item) break;
|
||
array.push(item as ArrayItem);
|
||
console.log(`${CONFIG_EMOJIS.INFO} Added new server: ${item.name}`);
|
||
console.log('');
|
||
}
|
||
}
|
||
|
||
return array;
|
||
},
|
||
|
||
nested: async () => {
|
||
const nestedObj: Record<string, any> = {};
|
||
if (config.fields) {
|
||
for (const [key, fieldConfig] of Object.entries(config.fields)) {
|
||
nestedObj[key] = await this.handleField(
|
||
key,
|
||
fieldConfig,
|
||
true,
|
||
currentValue?.[key]
|
||
);
|
||
}
|
||
}
|
||
return nestedObj;
|
||
},
|
||
|
||
boolean: async () => false,
|
||
url: async () =>
|
||
this.promptWithValidation(
|
||
`${emoji} Enter URL:`,
|
||
config,
|
||
currentValue || config.default
|
||
),
|
||
};
|
||
|
||
return handlers[config.type]();
|
||
}
|
||
|
||
private async validateInput(
|
||
value: any,
|
||
config: FieldConfig,
|
||
allowEmpty = true
|
||
): Promise<boolean> {
|
||
if (allowEmpty && value === '') return true;
|
||
|
||
if (!config.validation) return true;
|
||
|
||
const isValid = config.validation(value);
|
||
if (!isValid) {
|
||
console.log('❌ Invalid input. Please try again.');
|
||
}
|
||
return isValid;
|
||
}
|
||
|
||
private async promptWithValidation(
|
||
question: string,
|
||
config: FieldConfig,
|
||
defaultValue = '',
|
||
allowEmpty = true
|
||
): Promise<string> {
|
||
while (true) {
|
||
const value = await this.prompt(question, defaultValue);
|
||
if (!value && !config.required && allowEmpty) return value;
|
||
if (await this.validateInput(value, config, allowEmpty)) return value;
|
||
}
|
||
}
|
||
|
||
async generate(): Promise<T> {
|
||
const config: Record<string, any> = {};
|
||
for (const [fieldName, fieldConfig] of Object.entries(this.fields)) {
|
||
const currentValue = this.currentConfig?.[fieldName];
|
||
config[fieldName] = await this.handleField(
|
||
fieldName,
|
||
fieldConfig,
|
||
true,
|
||
currentValue
|
||
);
|
||
}
|
||
|
||
writeFileSync(this.configPath, stringify(config));
|
||
console.log(`\n${CONFIG_EMOJIS.SUCCESS} Configuration saved successfully!`);
|
||
|
||
return config as T;
|
||
}
|
||
}
|
||
|
||
export const generateHexKey = () =>
|
||
Buffer.from(randomBytes(32)).toString('hex');
|
||
|
||
export const validateHexKey = (value: string) => HEX_KEYS_REGEX.test(value);
|
||
|
||
export const validateRelayUrl = (url: string) => {
|
||
try {
|
||
const trimmedUrl = url.trim();
|
||
new URL(trimmedUrl);
|
||
return trimmedUrl.startsWith('ws://') || trimmedUrl.startsWith('wss://');
|
||
} catch {
|
||
return false;
|
||
}
|
||
};
|