tg-agent 0.1.0 → 1.0.1

package/dist/mcp.js

@@ -1,427 +1,3 @@
-import fs from "node:fs";
-import fsPromises from "node:fs/promises";
-import path from "node:path";
-import readline from "node:readline";
-import { spawn } from "node:child_process";
-const DEFAULT_TIMEOUT_MS = 60_000;
-const DEFAULT_MAX_BYTES = 200_000;
-export function getMcpConfigPath(agentDir) {
-    return path.join(agentDir, "config.toml");
-}
-export function loadMcpServersSync(agentDir) {
-    const configPath = getMcpConfigPath(agentDir);
-    try {
-        const content = fs.readFileSync(configPath, "utf8");
-        return parseMcpToml(content);
-    }
-    catch {
-        return [];
-    }
-}
-export async function loadMcpServers(agentDir) {
-    const configPath = getMcpConfigPath(agentDir);
-    try {
-        const content = await fsPromises.readFile(configPath, "utf8");
-        return parseMcpToml(content);
-    }
-    catch {
-        return [];
-    }
-}
-export function formatMcpTarget(server) {
-    if (server.type === "http") {
-        return server.url ? `url=${server.url}` : "url=(missing)";
-    }
-    const args = server.args && server.args.length > 0 ? ` ${server.args.join(" ")}` : "";
-    return server.command ? `command=${server.command}${args}` : "command=(missing)";
-}
-export async function probeMcpServer(server, timeoutMs = 5000) {
-    const started = Date.now();
-    try {
-        await callMcpServer(server, "tools/list", {}, { timeoutMs });
-        return { ok: true, durationMs: Date.now() - started };
-    }
-    catch (error) {
-        const message = error instanceof Error ? error.message : String(error);
-        return { ok: false, durationMs: Date.now() - started, error: message };
-    }
-}
-export async function callMcpServer(server, method, params, options = {}) {
-    const timeoutMs = clampNumber(options.timeoutMs ?? DEFAULT_TIMEOUT_MS, 1000, 120_000);
-    const maxBytes = clampNumber(options.maxBytes ?? DEFAULT_MAX_BYTES, 1024, 5_000_000);
-    const controller = new AbortController();
-    const onAbort = () => controller.abort();
-    options.signal?.addEventListener("abort", onAbort, { once: true });
-    const timeout = setTimeout(() => controller.abort(), timeoutMs);
-    try {
-        if (server.type === "http") {
-            return await callHttpMcp(server, method, params, maxBytes, controller.signal);
-        }
-        return await callStdioMcp(server, method, params, maxBytes, controller.signal);
-    }
-    finally {
-        clearTimeout(timeout);
-        options.signal?.removeEventListener("abort", onAbort);
-    }
-}
-function clampNumber(value, min, max) {
-    if (Number.isNaN(value))
-        return min;
-    return Math.min(max, Math.max(min, value));
-}
-async function callHttpMcp(server, method, params, maxBytes, signal) {
-    if (!server.url) {
-        throw new Error("MCP server url is missing.");
-    }
-    const payload = {
-        jsonrpc: "2.0",
-        id: `tg-agent-${Date.now()}`,
-        method,
-        params: params ?? {},
-    };
-    const headers = {
-        "content-type": "application/json",
-    };
-    if (server.auth) {
-        headers.authorization = server.auth.startsWith("Bearer ")
-            ? server.auth
-            : `Bearer ${server.auth}`;
-    }
-    const started = Date.now();
-    const response = await fetch(server.url, {
-        method: "POST",
-        headers,
-        body: JSON.stringify(payload),
-        signal,
-    });
-    const readResult = await readResponseText(response, maxBytes, signal);
-    const contentType = response.headers.get("content-type");
-    let bodyText = readResult.text;
-    try {
-        const parsed = JSON.parse(readResult.text);
-        bodyText = JSON.stringify(parsed, null, 2);
-    }
-    catch {
-        // keep raw text
-    }
-    return {
-        ok: response.ok,
-        output: bodyText,
-        durationMs: Date.now() - started,
-        bytes: readResult.bytes,
-        truncated: readResult.truncated,
-        statusCode: response.status,
-        statusText: response.statusText,
-        contentType,
-    };
-}
-async function callStdioMcp(server, method, params, maxBytes, signal) {
-    if (!server.command) {
-        throw new Error("MCP server command is missing.");
-    }
-    const payload = {
-        jsonrpc: "2.0",
-        id: `tg-agent-${Date.now()}`,
-        method,
-        params: params ?? {},
-    };
-    const started = Date.now();
-    const child = spawn(server.command, server.args ?? [], {
-        stdio: ["pipe", "pipe", "pipe"],
-    });
-    const cleanup = () => {
-        if (!child.killed) {
-            child.kill("SIGTERM");
-        }
-    };
-    if (signal.aborted) {
-        cleanup();
-        throw new Error("MCP request aborted.");
-    }
-    signal.addEventListener("abort", cleanup, { once: true });
-    let stdoutBuffer = "";
-    let stderrBuffer = "";
-    let truncated = false;
-    let parsedJson;
-    const stdoutRl = readline.createInterface({ input: child.stdout, terminal: false });
-    stdoutRl.on("line", (line) => {
-        if (stdoutBuffer.length + line.length + 1 > maxBytes) {
-            truncated = true;
-            return;
-        }
-        stdoutBuffer += line + "\n";
-        if (parsedJson === undefined) {
-            try {
-                parsedJson = JSON.parse(line);
-            }
-            catch {
-                // ignore non-json line
-            }
-        }
-    });
-    child.stderr?.on("data", (data) => {
-        const text = data.toString();
-        if (stderrBuffer.length + text.length > maxBytes) {
-            truncated = true;
-            return;
-        }
-        stderrBuffer += text;
-    });
-    const payloadLine = `${JSON.stringify(payload)}\n`;
-    child.stdin?.write(payloadLine);
-    child.stdin?.end();
-    const exitResult = await new Promise((resolve) => {
-        child.on("exit", (code, exitSignal) => resolve({ code, signal: exitSignal }));
-    });
-    stdoutRl.close();
-    signal.removeEventListener("abort", cleanup);
-    if (signal.aborted) {
-        throw new Error("MCP request aborted.");
-    }
-    if (!parsedJson && stdoutBuffer.trim()) {
-        try {
-            parsedJson = JSON.parse(stdoutBuffer.trim());
-        }
-        catch {
-            // ignore
-        }
-    }
-    if (exitResult.code !== 0 && !parsedJson) {
-        const stderrText = stderrBuffer.trim() || "stdio process failed";
-        throw new Error(stderrText);
-    }
-    if (!parsedJson) {
-        const stderrText = stderrBuffer.trim();
-        if (stderrText) {
-            throw new Error(stderrText);
-        }
-        throw new Error("MCP stdio returned no JSON.");
-    }
-    const output = JSON.stringify(parsedJson, null, 2);
-    return {
-        ok: true,
-        output,
-        durationMs: Date.now() - started,
-        bytes: Math.min(output.length, maxBytes),
-        truncated,
-    };
-}
-async function readResponseText(response, maxBytes, signal) {
-    const reader = response.body?.getReader?.();
-    if (!reader) {
-        const text = await response.text();
-        const bytes = Math.min(text.length, maxBytes);
-        return { text: text.slice(0, maxBytes), bytes, truncated: text.length > maxBytes };
-    }
-    const decoder = new TextDecoder("utf-8");
-    const chunks = [];
-    let bytes = 0;
-    let truncated = false;
-    while (true) {
-        if (signal.aborted) {
-            try {
-                await reader.cancel();
-            }
-            catch {
-                // ignore
-            }
-            throw new Error("MCP request aborted.");
-        }
-        const { done, value } = await reader.read();
-        if (done)
-            break;
-        if (!value)
-            continue;
-        const nextBytes = bytes + value.byteLength;
-        if (nextBytes > maxBytes) {
-            const sliceSize = Math.max(0, maxBytes - bytes);
-            if (sliceSize > 0) {
-                chunks.push(value.slice(0, sliceSize));
-                bytes += sliceSize;
-            }
-            truncated = true;
-            try {
-                await reader.cancel();
-            }
-            catch {
-                // ignore
-            }
-            break;
-        }
-        chunks.push(value);
-        bytes = nextBytes;
-    }
-    const buffer = new Uint8Array(bytes);
-    let offset = 0;
-    for (const chunk of chunks) {
-        buffer.set(chunk, offset);
-        offset += chunk.byteLength;
-    }
-    const text = decoder.decode(buffer);
-    return { text, bytes, truncated };
-}
-function parseMcpToml(content) {
-    const servers = [];
-    let currentName = null;
-    let current = {};
-    const flush = () => {
-        if (!currentName)
-            return;
-        const type = current.type ??
-            (current.url ? "http" : current.command ? "stdio" : undefined);
-        if (!type)
-            return;
-        const normalizedType = type === "http" || type === "stdio" ? type : undefined;
-        if (!normalizedType)
-            return;
-        if (normalizedType === "http" && !current.url)
-            return;
-        if (normalizedType === "stdio" && !current.command)
-            return;
-        servers.push({
-            name: currentName,
-            type: normalizedType,
-            url: current.url,
-            command: current.command,
-            args: current.args,
-            auth: current.auth,
-        });
-    };
-    for (const rawLine of content.split(/\r?\n/)) {
-        const line = stripInlineComment(rawLine).trim();
-        if (!line)
-            continue;
-        if (line.startsWith("[") && line.endsWith("]")) {
-            flush();
-            const section = line.slice(1, -1).trim();
-            if (!section.startsWith("mcp_servers.")) {
-                currentName = null;
-                current = {};
-                continue;
-            }
-            let name = section.slice("mcp_servers.".length).trim();
-            name = stripQuotes(name);
-            if (!name) {
-                currentName = null;
-                current = {};
-                continue;
-            }
-            currentName = name;
-            current = {};
-            continue;
-        }
-        if (!currentName)
-            continue;
-        const eq = line.indexOf("=");
-        if (eq === -1)
-            continue;
-        const key = line.slice(0, eq).trim();
-        const value = line.slice(eq + 1).trim();
-        if (!key)
-            continue;
-        switch (key) {
-            case "type":
-                current.type = parseTomlString(value).toLowerCase();
-                break;
-            case "url":
-                current.url = parseTomlString(value);
-                break;
-            case "command":
-                current.command = parseTomlString(value);
-                break;
-            case "args":
-                current.args = parseTomlStringArray(value);
-                break;
-            case "auth":
-            case "authorization":
-            case "token":
-                current.auth = parseTomlString(value);
-                break;
-            default:
-                break;
-        }
-    }
-    flush();
-    return servers;
-}
-function stripQuotes(value) {
-    if (value.startsWith("\"") && value.endsWith("\"")) {
-        return value.slice(1, -1);
-    }
-    if (value.startsWith("'") && value.endsWith("'")) {
-        return value.slice(1, -1);
-    }
-    return value;
-}
-function parseTomlString(value) {
-    const trimmed = value.trim();
-    if (!trimmed)
-        return "";
-    if (trimmed.startsWith("\"") && trimmed.endsWith("\"")) {
-        return trimmed
-            .slice(1, -1)
-            .replace(/\\\\/g, "\\")
-            .replace(/\\"/g, "\"");
-    }
-    if (trimmed.startsWith("'") && trimmed.endsWith("'")) {
-        return trimmed.slice(1, -1);
-    }
-    return trimmed;
-}
-function parseTomlStringArray(value) {
-    const trimmed = value.trim();
-    if (!trimmed.startsWith("[") || !trimmed.endsWith("]")) {
-        return [];
-    }
-    const inner = trimmed.slice(1, -1).trim();
-    if (!inner)
-        return [];
-    const entries = [];
-    let current = "";
-    let inQuotes = false;
-    let quoteChar = "";
-    for (let i = 0; i < inner.length; i += 1) {
-        const char = inner[i];
-        if (!inQuotes && (char === "\"" || char === "'")) {
-            inQuotes = true;
-            quoteChar = char;
-            continue;
-        }
-        if (inQuotes && char === quoteChar) {
-            inQuotes = false;
-            continue;
-        }
-        if (!inQuotes && char === ",") {
-            const entry = current.trim();
-            if (entry)
-                entries.push(stripQuotes(entry));
-            current = "";
-            continue;
-        }
-        current += char;
-    }
-    const last = current.trim();
-    if (last)
-        entries.push(stripQuotes(last));
-    return entries;
-}
-function stripInlineComment(line) {
-    let inQuotes = false;
-    let quoteChar = "";
-    for (let i = 0; i < line.length; i += 1) {
-        const char = line[i];
-        if (!inQuotes && (char === "\"" || char === "'")) {
-            inQuotes = true;
-            quoteChar = char;
-            continue;
-        }
-        if (inQuotes && char === quoteChar) {
-            inQuotes = false;
-            continue;
-        }
-        if (!inQuotes && (char === "#" || char === ";")) {
-            return line.slice(0, i);
-        }
-    }
-    return line;
-}
+import E from"node:fs";import W from"node:fs/promises";import C from"node:path";import N from"node:readline";import{spawn as D}from"node:child_process";const P=6e4,L=2e5;function M(t){return C.join(t,"config.toml")}function U(t){const n=M(t);try{const o=E.readFileSync(n,"utf8");return x(o)}catch{return[]}}async function Q(t){const n=M(t);try{const o=await W.readFile(n,"utf8");return x(o)}catch{return[]}}function B(t){if(t.type==="http")return t.url?`url=${t.url}`:"url=(missing)";const n=t.args&&t.args.length>0?` ${t.args.join(" ")}`:"";return t.command?`command=${t.command}${n}`:"command=(missing)"}async function G(t,n=5e3){const o=Date.now();try{return await O(t,"tools/list",{},{timeoutMs:n}),{ok:!0,durationMs:Date.now()-o}}catch(e){const i=e instanceof Error?e.message:String(e);return{ok:!1,durationMs:Date.now()-o,error:i}}}async function O(t,n,o,e={}){const i=S(e.timeoutMs??P,1e3,12e4),a=S(e.maxBytes??L,1024,5e6),r=new AbortController,c=()=>r.abort();e.signal?.addEventListener("abort",c,{once:!0});const u=setTimeout(()=>r.abort(),i);try{return t.type==="http"?await _(t,n,o,a,r.signal):await J(t,n,o,a,r.signal)}finally{clearTimeout(u),e.signal?.removeEventListener("abort",c)}}function S(t,n,o){return Number.isNaN(t)?n:Math.min(o,Math.max(n,t))}async function _(t,n,o,e,i){if(!t.url)throw new Error("MCP server url is missing.");const a={jsonrpc:"2.0",id:`tg-agent-${Date.now()}`,method:n,params:o??{}},r={"content-type":"application/json"};t.auth&&(r.authorization=t.auth.startsWith("Bearer ")?t.auth:`Bearer ${t.auth}`);const c=Date.now(),u=await fetch(t.url,{method:"POST",headers:r,body:JSON.stringify(a),signal:i}),s=await $(u,e,i),d=u.headers.get("content-type");let l=s.text;try{const f=JSON.parse(s.text);l=JSON.stringify(f,null,2)}catch{}return{ok:u.ok,output:l,durationMs:Date.now()-c,bytes:s.bytes,truncated:s.truncated,statusCode:u.status,statusText:u.statusText,contentType:d}}async function J(t,n,o,e,i){if(!t.command)throw new Error("MCP server command is missing.");const a={jsonrpc:"2.0",id:`tg-agent-${Date.now()}`,method:n,params:o??{}},r=Date.now(),c=D(t.command,t.args??[],{stdio:["pipe","pipe","pipe"]}),u=()=>{c.killed||c.kill("SIGTERM")};if(i.aborted)throw u(),new Error("MCP request aborted.");i.addEventListener("abort",u,{once:!0});let s="",d="",l=!1,f;const h=N.createInterface({input:c.stdout,terminal:!1});h.on("line",m=>{if(s.length+m.length+1>e){l=!0;return}if(s+=m+`
+`,f===void 0)try{f=JSON.parse(m)}catch{}}),c.stderr?.on("data",m=>{const g=m.toString();if(d.length+g.length>e){l=!0;return}d+=g});const p=`${JSON.stringify(a)}
+`;c.stdin?.write(p),c.stdin?.end();const T=await new Promise(m=>{c.on("exit",(g,k)=>m({code:g,signal:k}))});if(h.close(),i.removeEventListener("abort",u),i.aborted)throw new Error("MCP request aborted.");if(!f&&s.trim())try{f=JSON.parse(s.trim())}catch{}if(T.code!==0&&!f){const m=d.trim()||"stdio process failed";throw new Error(m)}if(!f){const m=d.trim();throw m?new Error(m):new Error("MCP stdio returned no JSON.")}const b=JSON.stringify(f,null,2);return{ok:!0,output:b,durationMs:Date.now()-r,bytes:Math.min(b.length,e),truncated:l}}async function $(t,n,o){const e=t.body?.getReader?.();if(!e){const l=await t.text(),f=Math.min(l.length,n);return{text:l.slice(0,n),bytes:f,truncated:l.length>n}}const i=new TextDecoder("utf-8"),a=[];let r=0,c=!1;for(;;){if(o.aborted){try{await e.cancel()}catch{}throw new Error("MCP request aborted.")}const{done:l,value:f}=await e.read();if(l)break;if(!f)continue;const h=r+f.byteLength;if(h>n){const p=Math.max(0,n-r);p>0&&(a.push(f.slice(0,p)),r+=p),c=!0;try{await e.cancel()}catch{}break}a.push(f),r=h}const u=new Uint8Array(r);let s=0;for(const l of a)u.set(l,s),s+=l.byteLength;return{text:i.decode(u),bytes:r,truncated:c}}function x(t){const n=[];let o=null,e={};const i=()=>{if(!o)return;const a=e.type??(e.url?"http":e.command?"stdio":void 0);if(!a)return;const r=a==="http"||a==="stdio"?a:void 0;r&&(r==="http"&&!e.url||r==="stdio"&&!e.command||n.push({name:o,type:r,url:e.url,command:e.command,args:e.args,auth:e.auth}))};for(const a of t.split(/\r?\n/)){const r=q(a).trim();if(!r)continue;if(r.startsWith("[")&&r.endsWith("]")){i();const d=r.slice(1,-1).trim();if(!d.startsWith("mcp_servers.")){o=null,e={};continue}let l=d.slice(12).trim();if(l=w(l),!l){o=null,e={};continue}o=l,e={};continue}if(!o)continue;const c=r.indexOf("=");if(c===-1)continue;const u=r.slice(0,c).trim(),s=r.slice(c+1).trim();if(u)switch(u){case"type":e.type=y(s).toLowerCase();break;case"url":e.url=y(s);break;case"command":e.command=y(s);break;case"args":e.args=A(s);break;case"auth":case"authorization":case"token":e.auth=y(s);break;default:break}}return i(),n}function w(t){return t.startsWith('"')&&t.endsWith('"')||t.startsWith("'")&&t.endsWith("'")?t.slice(1,-1):t}function y(t){const n=t.trim();return n?n.startsWith('"')&&n.endsWith('"')?n.slice(1,-1).replace(/\\\\/g,"\\").replace(/\\"/g,'"'):n.startsWith("'")&&n.endsWith("'")?n.slice(1,-1):n:""}function A(t){const n=t.trim();if(!n.startsWith("[")||!n.endsWith("]"))return[];const o=n.slice(1,-1).trim();if(!o)return[];const e=[];let i="",a=!1,r="";for(let u=0;u<o.length;u+=1){const s=o[u];if(!a&&(s==='"'||s==="'")){a=!0,r=s;continue}if(a&&s===r){a=!1;continue}if(!a&&s===","){const d=i.trim();d&&e.push(w(d)),i="";continue}i+=s}const c=i.trim();return c&&e.push(w(c)),e}function q(t){let n=!1,o="";for(let e=0;e<t.length;e+=1){const i=t[e];if(!n&&(i==='"'||i==="'")){n=!0,o=i;continue}if(n&&i===o){n=!1;continue}if(!n&&(i==="#"||i===";"))return t.slice(0,e)}return t}export{O as callMcpServer,B as formatMcpTarget,M as getMcpConfigPath,Q as loadMcpServers,U as loadMcpServersSync,G as probeMcpServer};