dvmcp/packages/dvmcp-discovery/index.ts

import { getConfig } from './src/config';
import { DiscoveryServer } from './src/discovery-server';
import type { DVMAnnouncement } from './src/direct-discovery';
import logger from './src/logger';

export interface DirectServerInfo {
  pubkey: string;
  announcement: DVMAnnouncement;
}

async function main(directServerInfo?: DirectServerInfo | null) {
  try {
    const config = getConfig();
    const server = new DiscoveryServer(config);

    if (directServerInfo) {
      // If we have direct server info, register tools from that server only
      logger(`Using direct server with pubkey: ${directServerInfo.pubkey}`);
      await server.registerDirectServerTools(
        directServerInfo.pubkey,
        directServerInfo.announcement
      );
    } else {
      // Otherwise do normal discovery
      await server.start();
    }

    logger(`DVMCP Discovery Server (${config.mcp.version}) started`);
    logger(`Connected to ${config.nostr.relayUrls.length} relays`);

    // Handle shutdown
    const cleanup = () => {
      server.cleanup();
      process.exit(0);
    };

    process.on('SIGINT', cleanup);
    process.on('SIGTERM', cleanup);
  } catch (error) {
    console.error('Failed to start server:', error);
    process.exit(1);
  }
}

export default main;