dvmcp/packages/dvmcp-commons/config-generator.ts

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;
  }
};