main.ts

import { parseArgs, type ParseOptions } from '@std/cli/parse-args';
import { NostrMCPProxy } from '@contextvm/sdk/proxy';
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
import { PrivateKeySigner } from '@contextvm/sdk/signer/private-key-signer';
import { loadConfig } from './src/config.ts';
import { initCommand } from './src/commands/init.ts';
import meta from './deno.json' with { type: 'json' };
import { YAML_CONFIG_PATH } from './src/config.ts';
import { ApplesauceRelayPool } from '@contextvm/sdk/relay';

function printUsage() {
  console.log(`
Usage: proxy-cli [command|OPTIONS]

Commands:
  init                                Run the interactive configuration wizard.

Options:
  --private-key <hex-key>             Your Nostr private key in hex format. (Env: PROXY_PRIVATE_KEY)
  --relays <urls...>                  Comma-separated list of Nostr relay URLs. (Env: PROXY_RELAYS)
  --server-pubkey <hex-key>            The public key of the server to connect to. (Env: PROXY_SERVER_PUBKEY)
  --encryption-mode <mode>            The encryption mode to use (optional, required, disabled). (Env: PROXY_ENCRYPTION_MODE)
  -h, --help                          Show this help message.
  -v, --version                       Show the version number.

Configuration can also be provided via a '${YAML_CONFIG_PATH}' file or environment variables.
Priority: CLI flags > ${YAML_CONFIG_PATH} > environment variables.
  `);
}

const options: ParseOptions = {
  boolean: ['help', 'version', 'init'],
  alias: { h: 'help', v: 'version' },
};
const args = parseArgs(Deno.args, options);
if (args.help) {
  printUsage();
  Deno.exit(0);
}

if (args.version) {
  console.log((meta as { version?: string }).version ?? '0.1.0');
  Deno.exit(0);
}

async function main() {
  if (args.init || Deno.args[0] === 'init') {
    await initCommand();
    Deno.exit(0);
  }

  const config = await loadConfig(Deno.args);

  const mcpHostTransport = new StdioServerTransport();

  const signer = new PrivateKeySigner(config.privateKey);
  const relayPool = new ApplesauceRelayPool(config.relays);

  const proxy = new NostrMCPProxy({
    mcpHostTransport,
    nostrTransportOptions: {
      signer,
      relayHandler: relayPool,
      serverPubkey: config.serverPubkey,
      encryptionMode: config.encryptionMode,
    },
  });

  await proxy.start();
  const shutdown = async () => {
    if (proxy) {
      try {
        await proxy.stop();
      } catch (error) {
        console.error('Error stopping NostrMCPProxy:', error);
      }
    }
    Deno.exit(0);
  };

  Deno.addSignalListener('SIGINT', shutdown);
  Deno.addSignalListener('SIGTERM', shutdown);
}

if (import.meta.main) {
  main().catch((error) => {
    console.error('Failed to start gateway:', error);
    Deno.exit(1);
  });
}