import { Client } from '@modelcontextprotocol/sdk/client/index.js';

import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';

const RAG_URL = process.env.ZPM_RAG_SSE || 'http://REDACTED-IP:9876/sse';

const CBM_URL = process.env.ZPM_CBM_SSE || 'http://REDACTED-IP:9877/sse';

function withTimeout<T>(p: Promise<T>, ms: number, label: string): Promise<T> {

  return Promise.race([

    p,

    new Promise<T>((_, rej) => setTimeout(() => rej(new Error(`${label} timeout ${ms}ms`)), ms)),

  ]);

}

async function mcpCall(url: string, name: string, args: Record<string, unknown>, timeoutMs = 60_000): Promise<string> {

  const c = new Client({ name: 'zion-pm', version: '1.0.0' }, { capabilities: {} });

  try {

    await withTimeout(c.connect(new SSEClientTransport(new URL(url))), 15_000, 'mcp connect');

    const r = (await withTimeout(c.callTool({ name, arguments: args }), timeoutMs, 'mcp call')) as { content?: Array<{ text?: string }> };

    return (r?.content || []).map((b) => b.text || '').join('\n');

  } finally {

    try { await c.close(); } catch {  }

  }

}

export async function queryMemory(query: string, limit = 6): Promise<string> {

  try {

    return await mcpCall(RAG_URL, 'homelab_context', { query, limit });

  } catch (e) {

    return `queryMemory unavailable: ${(e as Error).message}`;

  }

}

const looksLikeError = (s: string) => !s || /not found|not indexed|"error"|validation|required property|must have/i.test(s);

export async function queryGraph(query: string): Promise<string> {

  try {

    const arch = await mcpCall(CBM_URL, 'get_architecture', { project: query });

    if (!looksLikeError(arch)) return arch;

  } catch {  }

  try {

    const s = await mcpCall(CBM_URL, 'search_graph', { query });

    if (!looksLikeError(s)) return s;

  } catch {  }

  try {

    const projects = await mcpCall(CBM_URL, 'list_projects', {});

    if (!looksLikeError(projects)) return `No single code-graph project matched the query. Indexed projects available:\n${projects}`;

  } catch {  }

  return 'No matching code-graph project (use queryMemory for prose/docs).';

}