@arii/boomtick-mcp 0.6.1 → 0.6.2

package/dist/mcp/http.js

@@ -7,7 +7,7 @@ export function createMcpHttpServer(options) {
     const server = http.createServer(async (req, res) => {
         // CORS Headers
         res.setHeader("Access-Control-Allow-Origin", "*");
-        res.setHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
+        res.setHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS, HEAD");
         res.setHeader("Access-Control-Allow-Headers", "Content-Type, Authorization");
         if (req.method === "OPTIONS") {
             res.writeHead(204);
@@ -15,19 +15,27 @@ export function createMcpHttpServer(options) {
             return;
         }
         const url = new URL(req.url || "/", `http://${req.headers.host || "localhost"}`);
-        // Endpoint 1: OAuth Protected Resource metadata
-        if (url.pathname === "/.well-known/oauth-protected-resource" && req.method === "GET") {
-            res.writeHead(200, { "Content-Type": "application/json" });
-            res.end(JSON.stringify({
-                resource: "mcp://boomtick-mcp",
-                "authorization_servers": [
+        // Endpoint 1: RFC 9728 OAuth Protected Resource Metadata Discovery
+        if (url.pathname === "/.well-known/oauth-protected-resource" && (req.method === "GET" || req.method === "HEAD")) {
+            const metadata = {
+                resource: process.env.MCP_RESOURCE_URI || process.env.BOOMTICK_RESOURCE_URI || "mcp://boomtick-mcp",
+                authorization_servers: [
                     process.env.MCP_OAUTH_AUTH_SERVER || "https://auth.boomtick.dev",
                 ],
                 scopes_supported: ["mcp:read", "mcp:write"],
-            }));
+                bearer_methods_supported: ["header"],
+                resource_documentation: process.env.MCP_RESOURCE_DOCS || "https://github.com/arii/boomtick",
+            };
+            res.writeHead(200, { "Content-Type": "application/json" });
+            if (req.method === "HEAD") {
+                res.end();
+            }
+            else {
+                res.end(JSON.stringify(metadata, null, 2));
+            }
             return;
         }
-        // Bearer Token Validation for SSE & Messages endpoints
+        // Bearer Token / OAuth 2.1 Validation for Protected Routes (/sse & /messages)
         if (url.pathname === "/sse" || url.pathname === "/messages") {
             if (!bearerToken) {
                 res.writeHead(500, { "Content-Type": "application/json" });
@@ -44,7 +52,7 @@ export function createMcpHttpServer(options) {
             if (!isMatch) {
                 res.writeHead(401, {
                     "Content-Type": "application/json",
-                    "WWW-Authenticate": 'Bearer realm="boomtick-mcp", error="invalid_token"',
+                    "WWW-Authenticate": 'Bearer realm="boomtick-mcp", error="invalid_token", error_description="Invalid or missing Bearer token"',
                 });
                 res.end(JSON.stringify({ error: "unauthorized", message: "Invalid or missing Bearer token" }));
                 return;

package/dist/mcp/http.test.js

@@ -1,6 +1,7 @@
 import { describe, it, expect, beforeEach, afterEach } from "vitest";
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { createMcpHttpServer } from "./http.js";
+import { BoomtickMCPServer } from "./server.js";
 describe("createMcpHttpServer", () => {
     let mcpServer;
     let httpServerObj;
@@ -13,29 +14,54 @@ describe("createMcpHttpServer", () => {
             httpServerObj.server.close();
         }
     });
-    it("should respond with OAuth protected resource metadata on /.well-known/oauth-protected-resource", async () => {
+    it("should respond with RFC 9728 OAuth protected resource metadata on /.well-known/oauth-protected-resource", async () => {
         httpServerObj = createMcpHttpServer({ port: 0, mcpServer });
         port = await httpServerObj.listen(0);
         const res = await fetch(`http://localhost:${port}/.well-known/oauth-protected-resource`);
         expect(res.status).toBe(200);
+        expect(res.headers.get("content-type")).toContain("application/json");
         const body = await res.json();
         expect(body.resource).toBe("mcp://boomtick-mcp");
-        expect(body.authorization_servers).toBeDefined();
+        expect(body.authorization_servers).toEqual(["https://auth.boomtick.dev"]);
         expect(body.scopes_supported).toEqual(["mcp:read", "mcp:write"]);
+        expect(body.bearer_methods_supported).toEqual(["header"]);
+        expect(body.resource_documentation).toBe("https://github.com/arii/boomtick");
     });
-    it("should reject requests when Bearer token is missing or invalid", async () => {
+    it("should handle HEAD requests for /.well-known/oauth-protected-resource", async () => {
+        httpServerObj = createMcpHttpServer({ port: 0, mcpServer });
+        port = await httpServerObj.listen(0);
+        const res = await fetch(`http://localhost:${port}/.well-known/oauth-protected-resource`, {
+            method: "HEAD",
+        });
+        expect(res.status).toBe(200);
+        const text = await res.text();
+        expect(text).toBe("");
+    });
+    it("should handle OPTIONS preflight CORS requests", async () => {
+        httpServerObj = createMcpHttpServer({ port: 0, mcpServer });
+        port = await httpServerObj.listen(0);
+        const res = await fetch(`http://localhost:${port}/sse`, {
+            method: "OPTIONS",
+        });
+        expect(res.status).toBe(204);
+        expect(res.headers.get("access-control-allow-origin")).toBe("*");
+        expect(res.headers.get("access-control-allow-methods")).toContain("GET");
+        expect(res.headers.get("access-control-allow-headers")).toContain("Authorization");
+    });
+    it("should reject protected requests when Bearer token is missing or invalid", async () => {
         httpServerObj = createMcpHttpServer({ port: 0, bearerToken: "secret123", mcpServer });
         port = await httpServerObj.listen(0);
         // Request without token
         const res1 = await fetch(`http://localhost:${port}/sse`);
         expect(res1.status).toBe(401);
         expect(res1.headers.get("www-authenticate")).toContain('Bearer realm="boomtick-mcp"');
+        expect(res1.headers.get("www-authenticate")).toContain('error="invalid_token"');
         // Request with wrong token
         const res2 = await fetch(`http://localhost:${port}/sse`, {
             headers: { Authorization: "Bearer wrong" },
         });
         expect(res2.status).toBe(401);
-        // Request with valid token
+        // Public resource metadata accessible with or without token
         const res3 = await fetch(`http://localhost:${port}/.well-known/oauth-protected-resource`, {
             headers: { Authorization: "Bearer secret123" },
         });
@@ -55,4 +81,28 @@ describe("createMcpHttpServer", () => {
         const body = await res.json();
         expect(body.error).toContain("Server misconfiguration");
     });
+    it("should return 400 for message requests without session or invalid session", async () => {
+        httpServerObj = createMcpHttpServer({ port: 0, bearerToken: "secret123", mcpServer });
+        port = await httpServerObj.listen(0);
+        const res = await fetch(`http://localhost:${port}/messages?sessionId=invalid_session`, {
+            method: "POST",
+            headers: {
+                Authorization: "Bearer secret123",
+                "Content-Type": "application/json",
+            },
+            body: JSON.stringify({ jsonrpc: "2.0", method: "ping", id: 1 }),
+        });
+        expect(res.status).toBe(400);
+        const body = await res.json();
+        expect(body.error).toContain("Session not found or expired");
+    });
+    it("should allow BoomtickMCPServer to start HTTP server via run() when transport is http", async () => {
+        process.env.MCP_TRANSPORT = "http";
+        process.env.PORT = "0";
+        process.env.BOOMTICK_MCP_BEARER_TOKEN = "test-token";
+        const boomtickServer = new BoomtickMCPServer();
+        await boomtickServer.run();
+        // Verify instance initialized properly
+        expect(boomtickServer).toBeInstanceOf(BoomtickMCPServer);
+    });
 });

package/dist/mcp/server.js

@@ -20,6 +20,7 @@ import { getWorkflowRunsHandler, GetWorkflowRunsInputSchema } from "../tools/rep
 import { createRepairBranchHandler, CreateRepairBranchInputSchema } from "../tools/repo.create_repair_branch.js";
 import { createBranchHandler, CreateBranchInputSchema } from "../tools/repo.create_branch.js";
 import { runTestsHandler, RunTestsInputSchema } from "../tools/repo.run_tests.js";
+import { checkJulesStatusHandler, CheckJulesStatusInputSchema } from "../tools/repo.check_jules_status.js";
 import { commitPatchHandler, CommitPatchInputSchema } from "../tools/repo.commit_patch.js";
 import { openReplacementPrHandler, OpenReplacementPrInputSchema } from "../tools/github.open_replacement_pr.js";
 import { commentTriageSummaryHandler, CommentTriageSummaryInputSchema } from "../tools/github.comment_triage_summary.js";
@@ -120,6 +121,14 @@ export class BoomtickMCPServer {
                     contents: [{ uri, mimeType: "application/json", text: JSON.stringify(changedFiles, null, 2) }],
                 };
             }
+            if (uri === "mcp://ci/latest-run" ||
+                uri === "repo://ci/latest-run" ||
+                uri === "repo://ci-latest-run") {
+                const workflowRuns = await getWorkflowRunsHandler(GetWorkflowRunsInputSchema.parse({}));
+                return {
+                    contents: [{ uri, mimeType: "application/json", text: JSON.stringify(workflowRuns, null, 2) }],
+                };
+            }
             if (uri === "repo://package-json" || uri === "mcp://workspace/package-json") {
                 const content = await fs.readFile(path.join(config.repoPath, "package.json"), "utf-8");
                 return {
@@ -201,6 +210,9 @@ export class BoomtickMCPServer {
                     case "repo.create.repair.branch":
                     case "repo.create_repair_branch":
                         return createSuccessResult(await createRepairBranchHandler(CreateRepairBranchInputSchema.parse(normArgs)));
+                    case "repo.check.jules.status":
+                    case "repo.check_jules_status":
+                        return createSuccessResult(await checkJulesStatusHandler(CheckJulesStatusInputSchema.parse(normArgs)));
                     case "repo.run.tests":
                     case "repo.run_tests":
                         return createSuccessResult(await runTestsHandler(RunTestsInputSchema.parse(normArgs)));
@@ -335,20 +347,33 @@ export class BoomtickMCPServer {
             process.exit(1);
         }
     }
-    async startHttp(port = parseInt(process.env.PORT || process.env.HTTP_PORT || "3000", 10), bearerToken) {
-        this.prepareEnvironmentAndPreflight();
-        const mcpHttp = createMcpHttpServer({ port, bearerToken, mcpServer: this.server });
-        return await mcpHttp.listen(port);
-    }
     async run() {
         this.prepareEnvironmentAndPreflight();
+        const args = process.argv.slice(2);
+        let port = parseInt(process.env.PORT || process.env.HTTP_PORT || "3000", 10);
+        let bearerToken = process.env.BOOMTICK_MCP_BEARER_TOKEN || process.env.MCP_BEARER_TOKEN || process.env.AUTH_TOKEN;
+        for (let i = 0; i < args.length; i++) {
+            if (args[i] === "--port" && i + 1 < args.length) {
+                port = parseInt(args[i + 1], 10);
+                i++;
+            }
+            else if (args[i].startsWith("--port=")) {
+                port = parseInt(args[i].split("=")[1], 10);
+            }
+            else if (args[i] === "--bearer-token" && i + 1 < args.length) {
+                bearerToken = args[i + 1];
+                i++;
+            }
+            else if (args[i].startsWith("--bearer-token=")) {
+                bearerToken = args[i].split("=")[1];
+            }
+        }
         const isHttp = process.env.MCP_TRANSPORT === "http" ||
-            process.argv.includes("--http") ||
-            (process.env.PORT !== undefined && !process.argv.includes("--stdio")) ||
-            (process.env.HTTP_PORT !== undefined && !process.argv.includes("--stdio"));
+            args.includes("--http") ||
+            (process.env.PORT !== undefined && !args.includes("--stdio")) ||
+            (process.env.HTTP_PORT !== undefined && !args.includes("--stdio"));
         if (isHttp) {
-            const port = parseInt(process.env.PORT || process.env.HTTP_PORT || "3000", 10);
-            const mcpHttp = createMcpHttpServer({ port, mcpServer: this.server });
+            const mcpHttp = createMcpHttpServer({ port, bearerToken, mcpServer: this.server });
             await mcpHttp.listen(port);
             return;
         }

package/dist/mcp/server.test.js

@@ -79,12 +79,16 @@ describe("BoomtickMCPServer", () => {
         });
     });
     describe("ListResources", () => {
-        it("should return resources list", async () => {
+        it("should return resources list containing workspace resources", async () => {
             const handler = requestHandlers.get("resources/list");
             expect(handler).toBeDefined();
             const result = await handler({ method: "resources/list" });
             expect(result.resources).toBeDefined();
             expect(Array.isArray(result.resources)).toBe(true);
+            const uris = result.resources.map((r) => r.uri);
+            expect(uris).toContain("mcp://workspace/agent-contract");
+            expect(uris).toContain("mcp://git/changed-files");
+            expect(uris).toContain("mcp://ci/latest-run");
         });
         it("should return resource templates list", async () => {
             const handler = requestHandlers.get("resources/templates/list");
@@ -150,6 +154,29 @@ describe("BoomtickMCPServer", () => {
             expect(result.contents[0].uri).toBe("repo://ci/42");
             expect(JSON.parse(result.contents[0].text)).toEqual({ status: "success" });
         });
+        it("should handle mcp://git/changed-files", async () => {
+            const handler = requestHandlers.get("resources/read");
+            const changedFilesModule = await import("../tools/repo.get_changed_files.js");
+            vi.spyOn(changedFilesModule, "getChangedFilesHandler").mockResolvedValue({ files: ["src/index.ts"] });
+            const result = await handler({
+                method: "resources/read",
+                params: { uri: "mcp://git/changed-files" },
+            });
+            expect(result.contents[0].uri).toBe("mcp://git/changed-files");
+            expect(result.contents[0].mimeType).toBe("application/json");
+            expect(JSON.parse(result.contents[0].text)).toEqual({ files: ["src/index.ts"] });
+        });
+        it("should handle mcp://ci/latest-run", async () => {
+            const handler = requestHandlers.get("resources/read");
+            vi.spyOn(workflowRunsModule, "getWorkflowRunsHandler").mockResolvedValue({ workflowRuns: [{ id: 1, status: "completed", conclusion: "success" }] });
+            const result = await handler({
+                method: "resources/read",
+                params: { uri: "mcp://ci/latest-run" },
+            });
+            expect(result.contents[0].uri).toBe("mcp://ci/latest-run");
+            expect(result.contents[0].mimeType).toBe("application/json");
+            expect(JSON.parse(result.contents[0].text)).toEqual({ workflowRuns: [{ id: 1, status: "completed", conclusion: "success" }] });
+        });
         it("should throw when resource is unknown", async () => {
             const handler = requestHandlers.get("resources/read");
             await expect(handler({

package/dist/tools/contract.js

@@ -1564,6 +1564,49 @@ export const GetMergeConflictFilesInputJsonSchema = {
     "title": "GetMergeConflictFilesInput",
     "type": "object"
 };
+export const GetMergeConflictFilesResponseSchema = z.object({ "hasConflicts": z.boolean(), "conflictedFiles": z.array(z.string()).optional(), "prNumber": z.union([z.number().int(), z.null()]).default(null), "baseBranch": z.string(), "headRef": z.string() });
+export const GetMergeConflictFilesResponseJsonSchema = {
+    "properties": {
+        "hasConflicts": {
+            "title": "Hasconflicts",
+            "type": "boolean"
+        },
+        "conflictedFiles": {
+            "items": {
+                "type": "string"
+            },
+            "title": "Conflictedfiles",
+            "type": "array"
+        },
+        "prNumber": {
+            "anyOf": [
+                {
+                    "type": "integer"
+                },
+                {
+                    "type": "null"
+                }
+            ],
+            "default": null,
+            "title": "Prnumber"
+        },
+        "baseBranch": {
+            "title": "Basebranch",
+            "type": "string"
+        },
+        "headRef": {
+            "title": "Headref",
+            "type": "string"
+        }
+    },
+    "required": [
+        "hasConflicts",
+        "baseBranch",
+        "headRef"
+    ],
+    "title": "GetMergeConflictFilesResponse",
+    "type": "object"
+};
 export const GetPackageScriptsInputSchema = z.object({ "filter": z.union([z.string(), z.null()]).describe("Optional glob pattern to filter script names.").default(null) });
 export const GetPackageScriptsInputJsonSchema = {
     "properties": {
@@ -3086,6 +3129,40 @@ export const RunTestsInputJsonSchema = {
     "title": "RunTestsInput",
     "type": "object"
 };
+export const RunTestsResponseSchema = z.object({ "passed": z.boolean(), "exitCode": z.number().int(), "failedTests": z.array(z.string()).optional(), "results": z.array(z.record(z.string(), z.any())).optional() });
+export const RunTestsResponseJsonSchema = {
+    "properties": {
+        "passed": {
+            "title": "Passed",
+            "type": "boolean"
+        },
+        "exitCode": {
+            "title": "Exitcode",
+            "type": "integer"
+        },
+        "failedTests": {
+            "items": {
+                "type": "string"
+            },
+            "title": "Failedtests",
+            "type": "array"
+        },
+        "results": {
+            "items": {
+                "additionalProperties": true,
+                "type": "object"
+            },
+            "title": "Results",
+            "type": "array"
+        }
+    },
+    "required": [
+        "passed",
+        "exitCode"
+    ],
+    "title": "RunTestsResponse",
+    "type": "object"
+};
 export const SearchDdgsInputSchema = z.object({ "query": z.string().describe("The search query."), "maxResults": z.union([z.number().int(), z.null()]).describe("Maximum number of results to return.").default(null) });
 export const SearchDdgsInputJsonSchema = {
     "properties": {

package/dist/tools/github.comment_triage_summary.js

@@ -1,23 +1,16 @@
 import { z } from "zod";
-import { runCommand } from "../lib/shell.js";
+import { runTDCli } from "./jules/cli.js";
 export const CommentTriageSummaryInputSchema = z.object({
     prNumber: z.number(),
     body: z.string(),
 });
 export async function commentTriageSummaryHandler(args) {
-    const result = await runCommand("td", [
+    const output = await runTDCli([
         "gh",
         "post-comment",
         "--pr", args.prNumber.toString(),
         "--body", args.body
-    ]);
-    if (result.exitCode !== 0) {
-        throw new Error(`Failed to comment on PR: ${result.stderr}`);
-    }
-    const output = JSON.parse(result.stdout);
-    if (output.status === "error") {
-        throw new Error(`Failed to comment on PR: ${output.message}`);
-    }
+    ], "comment on PR");
     return {
         success: true,
         commentUrl: output.html_url || output.url

package/dist/tools/github.get_merge_conflict_files.js

@@ -9,10 +9,12 @@ export async function getMergeConflictFilesHandler(args) {
     const baseBranch = args.baseBranch || "main";
     const headRef = args.headRef || "HEAD";
     const conflictFiles = await getMergeConflictFiles(baseBranch, headRef);
+    const hasConflicts = conflictFiles.length > 0;
     return {
-        prNumber: args.prNumber,
+        hasConflicts,
+        conflictedFiles: conflictFiles,
+        prNumber: args.prNumber ?? null,
         baseBranch,
         headRef,
-        conflictFiles,
     };
 }

package/dist/tools/github.get_pr_diff.js

@@ -1,6 +1,6 @@
 import { z } from "zod";
-import { runCommand } from "../lib/shell.js";
 import { resolveRepo } from "../lib/git.js";
+import { runTDCli } from "./jules/cli.js";
 export const GetPrDiffInputSchema = z.object({
     prNumber: z.number().describe("The number of the pull request to get the diff for."),
     repo: z.string().optional().describe("The target repository override (e.g. org/repo)."),
@@ -8,17 +8,9 @@ export const GetPrDiffInputSchema = z.object({
 export async function getPrDiffHandler(args) {
     const params = GetPrDiffInputSchema.parse(args);
     const targetRepo = resolveRepo(params.repo);
-    const result = await runCommand("td", [
+    const output = await runTDCli([
         "gh", "pr-diff", params.prNumber.toString(), "--repo", targetRepo
-    ]);
-    if (result.exitCode !== 0) {
-        throw new Error(`Failed to get PR diff: ${result.stderr}`);
-    }
-    const output = JSON.parse(result.stdout);
-    if (output.status === "error") {
-        throw new Error(`Failed to get PR diff: ${output.message}`);
-    }
-    // Every TypeScript tool file must be a pure routing shim
+    ], "get PR diff");
     return {
         prNumber: output.prNumber,
         files: output.files,

package/dist/tools/jules/cancel-session.js

@@ -1,17 +1,10 @@
 import { z } from "zod";
-import { runCommand } from "../../lib/shell.js";
+import { runTDCli } from "./cli.js";
 export const CancelJulesSessionInputSchema = z.object({
     sessionId: z.string(),
 });
 export async function cancelJulesSessionHandler(input) {
-    const result = await runCommand("td", ["agent", "cancel", input.sessionId]);
-    if (result.exitCode !== 0) {
-        throw new Error(`Failed to cancel session: ${result.stderr}`);
-    }
-    const output = JSON.parse(result.stdout);
-    if (output.status === "error") {
-        throw new Error(`Failed to cancel session: ${output.message}`);
-    }
+    await runTDCli(["agent", "cancel", input.sessionId], "cancel session");
     return {
         id: input.sessionId.replace("sessions/", ""),
         status: "FAILED",

package/dist/tools/jules/cli.js

@@ -0,0 +1,31 @@
+import { runCommand } from "../../lib/shell.js";
+function tryParseJson(text) {
+    try {
+        return JSON.parse(text);
+    }
+    catch {
+        return null;
+    }
+}
+export async function runTDCli(args, actionName) {
+    const result = await runCommand("td", args);
+    const stdout = result.stdout.trim();
+    const output = stdout.startsWith("{") ? tryParseJson(stdout) : null;
+    if (output?.status === "error") {
+        const errorMsg = output.message ?? "Unknown error";
+        if (actionName) {
+            throw new Error(`Failed to ${actionName}: ${errorMsg}`);
+        }
+        throw new Error(`td returned error: ${errorMsg}`);
+    }
+    if (result.exitCode !== 0) {
+        if (actionName) {
+            throw new Error(`Failed to ${actionName}: ${result.stderr || stdout}`);
+        }
+        throw new Error(`td command failed (${args.join(" ")}): ${result.stderr || stdout}`);
+    }
+    if (!output) {
+        throw new Error(`td returned non-JSON output with exit code 0: ${stdout}`);
+    }
+    return output;
+}

package/dist/tools/jules/get-messages.js

@@ -1,17 +1,10 @@
 import { z } from "zod";
-import { runCommand } from "../../lib/shell.js";
+import { runTDCli } from "./cli.js";
 export const GetJulesMessagesInputSchema = z.object({
     sessionId: z.string(),
 });
 export async function getJulesMessagesHandler(input) {
-    const result = await runCommand("td", ["agent", "messages", input.sessionId]);
-    if (result.exitCode !== 0) {
-        throw new Error(`Failed to get messages: ${result.stderr}`);
-    }
-    const output = JSON.parse(result.stdout);
-    if (output.status === "error") {
-        throw new Error(`Failed to get messages: ${output.message}`);
-    }
+    const output = await runTDCli(["agent", "messages", input.sessionId], "get messages");
     return {
         id: input.sessionId.replace("sessions/", ""),
         messages: output.messages,

package/dist/tools/jules/list-sessions.js

@@ -1,43 +1,12 @@
 import { z } from "zod";
-import { runCommand } from "../../lib/shell.js";
+import { runTDCli } from "./cli.js";
+import { parseJulesSession } from "./session.js";
 export const ListJulesSessionsInputSchema = z.object({
     pageSize: z.number().optional(),
     pageToken: z.string().optional(),
 });
 export async function listJulesSessionsHandler(input) {
-    const result = await runCommand("td", ["agent", "sync"]);
-    if (result.exitCode !== 0) {
-        throw new Error(`Failed to list sessions: ${result.stderr}`);
-    }
-    const output = JSON.parse(result.stdout);
-    if (output.status === "error") {
-        throw new Error(`Failed to list sessions: ${output.message}`);
-    }
+    const output = await runTDCli(["agent", "sync"], "list sessions");
     const sessions = output.sessions || [];
-    return sessions.map((session) => {
-        const name = session.name || "";
-        const id = name.startsWith("sessions/") ? name.substring(9) : name;
-        let status = "PENDING";
-        if (session.state === "SUCCEEDED" || session.state === "COMPLETED")
-            status = "COMPLETED";
-        else if (session.state === "FAILED" || session.state === "CANCELLED" || session.state === "TERMINATED")
-            status = "FAILED";
-        else if (session.state === "IN_PROGRESS")
-            status = "IN_PROGRESS";
-        let pullRequestUrl;
-        if (session.outputs && Array.isArray(session.outputs)) {
-            for (const out of session.outputs) {
-                if (out.pullRequest && out.pullRequest.url) {
-                    pullRequestUrl = out.pullRequest.url;
-                    break;
-                }
-            }
-        }
-        return {
-            id,
-            status,
-            createdAt: session.createTime ? new Date(session.createTime) : new Date(),
-            pullRequestUrl,
-        };
-    });
+    return sessions.map((session) => parseJulesSession(session));
 }

package/dist/tools/jules/send-message.js

@@ -1,19 +1,12 @@
 import { z } from "zod";
-import { runCommand } from "../../lib/shell.js";
+import { runTDCli } from "./cli.js";
 export const SendJulesMessageInputSchema = z.object({
     sessionId: z.union([z.string(), z.array(z.string())]),
     message: z.string(),
 });
 export async function sendJulesMessageHandler(input) {
     const sessionIds = Array.isArray(input.sessionId) ? input.sessionId.join(",") : input.sessionId;
-    const result = await runCommand("td", ["agent", "send", sessionIds, input.message]);
-    if (result.exitCode !== 0) {
-        throw new Error(`Failed to send message: ${result.stderr}`);
-    }
-    const output = JSON.parse(result.stdout);
-    if (output.status === "error") {
-        throw new Error(`Failed to send message: ${output.message}`);
-    }
+    await runTDCli(["agent", "send", sessionIds, input.message], "send message");
     return {
         id: Array.isArray(input.sessionId) ? input.sessionId.map(id => id.replace("sessions/", "")) : input.sessionId.replace("sessions/", ""),
         status: "success",

package/dist/tools/jules/session.js

@@ -0,0 +1,26 @@
+export function parseJulesSession(data) {
+    const name = data.name || "";
+    const id = name.startsWith("sessions/") ? name.substring(9) : name;
+    let status = "PENDING";
+    if (data.state === "SUCCEEDED" || data.state === "COMPLETED")
+        status = "COMPLETED";
+    else if (data.state === "FAILED" || data.state === "CANCELLED" || data.state === "TERMINATED")
+        status = "FAILED";
+    else if (data.state === "IN_PROGRESS")
+        status = "IN_PROGRESS";
+    let pullRequestUrl;
+    if (data.outputs && Array.isArray(data.outputs)) {
+        for (const out of data.outputs) {
+            if (out.pullRequest && out.pullRequest.url) {
+                pullRequestUrl = out.pullRequest.url;
+                break;
+            }
+        }
+    }
+    return {
+        id,
+        status,
+        createdAt: data.createTime ? new Date(data.createTime) : new Date(),
+        pullRequestUrl,
+    };
+}