main.ts

import { parseArgs, type ParseOptions } from '@std/cli/parse-args';
import { NostrMCPGateway } from '@contextvm/sdk/gateway';
import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
import { PrivateKeySigner } from '@contextvm/sdk/signer/private-key-signer';
import meta from './deno.json' with { type: 'json' };
import { loadConfig, YAML_CONFIG_PATH } from './src/config.ts';
import type { EncryptionMode } from '@contextvm/sdk';
import { initCommand } from './src/commands/init.ts';
import { ApplesauceRelayPool } from '@contextvm/sdk/relay';
function printUsage() {
  console.log(`
Usage: gateway-cli [command|OPTIONS]

Commands:
  init                                Run the interactive configuration wizard.

Options:
  --server <command> [...args]         The command and arguments to start the MCP server. (Env: GW_SERVER)
  --private-key <hex-key>             The Nostr private key in hex format. (Env: GW_PRIVATE_KEY)
  --relays <urls...>                  A list of Nostr relay URLs, separated by commas. (Env: GW_RELAYS)
  --public                            Announce the server publicly for discovery. (Env: GW_PUBLIC)
  --server-info-name <name>           The name of the server to be displayed in the server info. (Env: GW_SERVER_INFO_NAME)
  --server-info-picture <url>         The URL of the server's picture. (Env: GW_SERVER_INFO_PICTURE)
  --server-info-website <url>         The URL of the server's website. (Env: GW_SERVER_INFO_WEBSITE)
  --allowed-public-keys <keys...>     A list of allowed public keys. (Env: GW_ALLOWED_PUBLIC_KEYS)
  --encryption-mode <mode>            The encryption mode to use. Can be 'none', 'nip04', or 'nip44'. (Env: GW_ENCRYPTION_MODE)
  -e, --env <KEY=VALUE>               Environment variables to pass to the MCP server. Can be specified multiple times. (Env: GW_ENVIRONMENT as JSON)
  -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) {
  // Read from environment variable or fallback to deno.json
  const version = Deno.env.get('GATEWAY_CLI_VERSION') ||
    ((meta as { version?: string }).version ?? '0.1.0');
  console.log(version);
  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 [command, ...commandArgs] = config.server;
  console.log('Starting MCP server:', command, ...commandArgs);
  const mcpClientTransport = new StdioClientTransport({
    command: command,
    args: commandArgs,
    env: config.environment,
  });

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

  const gateway = new NostrMCPGateway({
    mcpClientTransport,
    nostrTransportOptions: {
      signer,
      relayHandler: relayPool,
      isPublicServer: config.public,
      serverInfo: config.serverInfo,
      allowedPublicKeys: config.allowedPublicKeys,
      encryptionMode: config.encryptionMode as EncryptionMode,
    },
  });

  await gateway.start();
  const shutdown = async () => {
    if (gateway) {
      try {
        await gateway.stop();
      } catch (error) {
        console.error('Error stopping NostrMCPGateway:', 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);
  });
}