@inkeep/agents-work-apps 0.0.0-dev-20260203033642

package/dist/github/mcp/index.js

@@ -0,0 +1,670 @@
+import runDbClient_default from "../../db/runDbClient.js";
+import { githubMcpAuth } from "./auth.js";
+import { commitFileChanges, commitNewFile, fetchPrFileDiffs, fetchPrFiles, fetchPrInfo, formatFileDiff, generatePrMarkdown, getGitHubClientFromRepo, visualizeUpdateOperations } from "./utils.js";
+import { getMcpToolRepositoryAccessWithDetails } from "@inkeep/agents-core";
+import { Hono } from "hono";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import { toFetchResponse, toReqRes } from "fetch-to-node";
+import { z } from "zod/v3";
+
+//#region src/github/mcp/index.ts
+const updateOperationSchema = z.object({
+	operation: z.enum([
+		"replace_lines",
+		"insert_after",
+		"insert_before",
+		"delete_lines"
+	]).describe("Operation type"),
+	lineStart: z.number().min(1).describe("Starting line number (1-indexed)"),
+	lineEnd: z.number().min(1).optional().describe("Ending line number (1-indexed, inclusive). Required for 'replace_lines' and 'delete_lines'"),
+	content: z.string().optional().describe("New content to insert/replace"),
+	reason: z.string().describe("Explanation of why this change is needed")
+}).describe("Update operation to apply to a file");
+const updateOperationsSchema = z.array(updateOperationSchema).describe("List of update operations to apply to a file");
+const getAvailableRepositoryString = (repositoryAccess) => {
+	if (repositoryAccess.length === 0) return "No repositories available";
+	return `Available repositories: ${repositoryAccess.map((r) => `"${r.repositoryFullName}"`).join(", ")}`;
+};
+/**
+* Creates and configures an MCP server for the given context
+*/
+const getServer = async (toolId) => {
+	const repositoryAccess = await getMcpToolRepositoryAccessWithDetails(runDbClient_default)(toolId);
+	const installationIdMap = /* @__PURE__ */ new Map();
+	for (const repo of repositoryAccess) installationIdMap.set(repo.repositoryFullName, repo.installationId);
+	if (repositoryAccess.length === 0) throw new Error("No repository access found for tool");
+	const server = new McpServer({
+		name: "inkeep-github-mcp-server",
+		version: "1.0.0",
+		description: "A GitHub MCP server with access to the following repositories:\n" + repositoryAccess.map((r) => `• ${r.repositoryFullName}`).join("\n")
+	}, { capabilities: { logging: {} } });
+	server.tool("search-files-in-repo", `Find files in a repository using GitHub's search API. ${getAvailableRepositoryString(repositoryAccess)}`, {
+		owner: z.string().describe("The owner of the repository"),
+		repo: z.string().describe("The name of the repository"),
+		query: z.string().describe("The query to search for"),
+		path: z.string().optional().describe("The path to the files to search in relative to the root of the repository"),
+		limit: z.number().default(30).describe("The maximum number of files to return")
+	}, async ({ owner, repo, query, path, limit }) => {
+		try {
+			const repoFullName = `${owner}/${repo}`;
+			let githubClient;
+			try {
+				githubClient = getGitHubClientFromRepo(owner, repo, installationIdMap);
+			} catch (error) {
+				return {
+					content: [{
+						type: "text",
+						text: `Error accessing GitHub: ${error instanceof Error ? error.message : "Unknown error"}`
+					}],
+					isError: true
+				};
+			}
+			let searchQuery = `${query} repo:${repoFullName}`;
+			if (path) searchQuery += ` path:${path}`;
+			const files = (await githubClient.rest.search.code({
+				q: searchQuery,
+				per_page: Math.min(limit, 100)
+			})).data.items.map((item) => ({
+				name: item.name,
+				path: item.path,
+				url: item.html_url,
+				repository: repoFullName,
+				sha: item.sha,
+				score: item.score
+			}));
+			return { content: [{
+				type: "text",
+				text: `Found ${files.length} files matching "${query}" in ${repoFullName}:\n\n${files.map((file) => `• ${file.name} (${file.path})\n  URL: ${file.url}\n  Score: ${file.score}`).join("\n\n")}`
+			}] };
+		} catch (error) {
+			return {
+				content: [{
+					type: "text",
+					text: `Error searching files: ${error instanceof Error ? error.message : "Unknown error"}`
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("get-file-content", `Get the content of a specific file from a repository. Returns a mapping of line number to line content. ${getAvailableRepositoryString(repositoryAccess)}`, {
+		owner: z.string().describe("Repository owner name"),
+		repo: z.string().describe("Repository name"),
+		file_path: z.string().describe("Path to the file. the path is relative to the root of the repository")
+	}, async ({ owner, repo, file_path }) => {
+		try {
+			let githubClient;
+			try {
+				githubClient = getGitHubClientFromRepo(owner, repo, installationIdMap);
+			} catch (error) {
+				return {
+					content: [{
+						type: "text",
+						text: `Error accessing GitHub: ${error instanceof Error ? error.message : "Unknown error"}`
+					}],
+					isError: true
+				};
+			}
+			const response = await githubClient.rest.repos.getContent({
+				owner,
+				repo,
+				path: file_path
+			});
+			if ("content" in response.data && !Array.isArray(response.data)) {
+				const fileData = response.data;
+				const output_mapping = Buffer.from(fileData.content, "base64").toString("utf-8").split("\n").map((line, index) => ({
+					line: index + 1,
+					content: line
+				}));
+				return { content: [{
+					type: "text",
+					text: JSON.stringify(output_mapping)
+				}] };
+			}
+			return {
+				content: [{
+					type: "text",
+					text: `The path "${file_path}" is not a file or could not be retrieved as file content.`
+				}],
+				isError: true
+			};
+		} catch (error) {
+			return {
+				content: [{
+					type: "text",
+					text: `Error getting file content: ${error instanceof Error ? error.message : "Unknown error"}`
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("get-pull-request-details", `Get the details of a pull request from a repository including the pull request details, the commits, and the files that were changed. ${getAvailableRepositoryString(repositoryAccess)}`, {
+		owner: z.string().describe("Repository owner name"),
+		repo: z.string().describe("Repository name"),
+		pull_request_number: z.number().describe("Pull request number")
+	}, async ({ owner, repo, pull_request_number }) => {
+		try {
+			let githubClient;
+			try {
+				githubClient = getGitHubClientFromRepo(owner, repo, installationIdMap);
+			} catch (error) {
+				return {
+					content: [{
+						type: "text",
+						text: `Error accessing GitHub: ${error instanceof Error ? error.message : "Unknown error"}`
+					}],
+					isError: true
+				};
+			}
+			return { content: [{
+				type: "text",
+				text: generatePrMarkdown(await fetchPrInfo(githubClient, owner, repo, pull_request_number), await fetchPrFileDiffs(githubClient, owner, repo, pull_request_number), owner, repo)
+			}] };
+		} catch (error) {
+			if (error instanceof Error && "status" in error) {
+				const apiError = error;
+				if (apiError.status === 404) return {
+					content: [{
+						type: "text",
+						text: `Pull request #${pull_request_number} not found in ${owner}/${repo}. Please check the repository exists and the pull request number is correct.`
+					}],
+					isError: true
+				};
+				if (apiError.status === 403) return {
+					content: [{
+						type: "text",
+						text: `Access denied to pull request #${pull_request_number} in ${owner}/${repo}. Your GitHub App may not have sufficient permissions.`
+					}],
+					isError: true
+				};
+			}
+			return {
+				content: [{
+					type: "text",
+					text: `Error getting pull request: ${error instanceof Error ? error.message : "Unknown error"}`
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("get-file-patches", `Get the patch/diff text for specific files in a pull request. Use this to fetch detailed changes for one or more files without retrieving the entire PR. ${getAvailableRepositoryString(repositoryAccess)}`, {
+		owner: z.string().describe("Repository owner name"),
+		repo: z.string().describe("Repository name"),
+		pull_request_number: z.number().describe("Pull request number"),
+		file_paths: z.array(z.string()).min(1).describe("List of file paths to get patches for (exact paths or glob patterns)"),
+		include_contents: z.boolean().default(false).describe("Whether to include full file contents in addition to patches")
+	}, async ({ owner, repo, pull_request_number, file_paths, include_contents }) => {
+		try {
+			let githubClient;
+			try {
+				githubClient = getGitHubClientFromRepo(owner, repo, installationIdMap);
+			} catch (error) {
+				return {
+					content: [{
+						type: "text",
+						text: `Error accessing GitHub: ${error instanceof Error ? error.message : "Unknown error"}`
+					}],
+					isError: true
+				};
+			}
+			const results = await fetchPrFiles(githubClient, owner, repo, pull_request_number, file_paths, include_contents, true);
+			if (results.length === 0) return { content: [{
+				type: "text",
+				text: `No files found matching the specified paths in PR #${pull_request_number}.\n\nRequested paths: ${file_paths.join(", ")}`
+			}] };
+			return { content: [{
+				type: "text",
+				text: await formatFileDiff(pull_request_number, results, include_contents)
+			}] };
+		} catch (error) {
+			if (error instanceof Error && "status" in error) {
+				if (error.status === 404) return {
+					content: [{
+						type: "text",
+						text: `Pull request #${pull_request_number} not found in ${owner}/${repo}.`
+					}],
+					isError: true
+				};
+			}
+			return {
+				content: [{
+					type: "text",
+					text: `Error getting file patches: ${error instanceof Error ? error.message : "Unknown error"}`
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("create-branch", `Create a new branch in a repository. ${getAvailableRepositoryString(repositoryAccess)}`, {
+		owner: z.string().describe("Repository owner name"),
+		repo: z.string().describe("Repository name"),
+		from_branch: z.string().optional().describe("Branch to create from (defaults to default branch)")
+	}, async ({ owner, repo, from_branch }) => {
+		const branch_name = `docs-writer-ai-update-${(/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-")}`;
+		try {
+			const githubClient = getGitHubClientFromRepo(owner, repo, installationIdMap);
+			const repoInfo = await githubClient.rest.repos.get({
+				owner,
+				repo
+			});
+			const sourceBranch = from_branch || repoInfo.data.default_branch;
+			const sourceSha = (await githubClient.rest.git.getRef({
+				owner,
+				repo,
+				ref: `heads/${sourceBranch}`
+			})).data.object.sha;
+			await githubClient.rest.git.createRef({
+				owner,
+				repo,
+				ref: `refs/heads/${branch_name}`,
+				sha: sourceSha
+			});
+			return { content: [{
+				type: "text",
+				text: `Successfully created branch "${branch_name}" in ${owner}/${repo}`
+			}] };
+		} catch (error) {
+			if (error instanceof Error && "status" in error) {
+				const apiError = error;
+				if (apiError.status === 422) return {
+					content: [{
+						type: "text",
+						text: `Branch "${branch_name}" already exists in ${owner}/${repo}`
+					}],
+					isError: true
+				};
+				if (apiError.status === 404) return {
+					content: [{
+						type: "text",
+						text: `Repository ${owner}/${repo} not found or source branch "${from_branch || "default"}" does not exist`
+					}],
+					isError: true
+				};
+			}
+			return {
+				content: [{
+					type: "text",
+					text: `Error creating branch: ${error instanceof Error ? error.message : "Unknown error"}`
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("commit-file-changes", `Commit changes to a files in a repository. ${getAvailableRepositoryString(repositoryAccess)}`, {
+		owner: z.string().describe("Repository owner name"),
+		repo: z.string().describe("Repository name"),
+		branch_name: z.string().describe("Branch to commit to"),
+		file_path: z.string().describe("Path to the file to commit"),
+		update_operations: updateOperationsSchema,
+		commit_message: z.string().describe("Commit message")
+	}, async ({ owner, repo, branch_name, file_path, update_operations, commit_message }) => {
+		try {
+			let githubClient;
+			try {
+				githubClient = getGitHubClientFromRepo(owner, repo, installationIdMap);
+			} catch (error) {
+				return {
+					content: [{
+						type: "text",
+						text: `Error accessing GitHub: ${error instanceof Error ? error.message : "Unknown error"}`
+					}],
+					isError: true
+				};
+			}
+			const fileResponse = await githubClient.rest.repos.getContent({
+				owner,
+				repo,
+				path: file_path
+			});
+			if (!("content" in fileResponse.data) || Array.isArray(fileResponse.data)) throw new Error(`File ${file_path} not found or is not a file`);
+			const currentFileContent = Buffer.from(fileResponse.data.content, "base64").toString("utf-8");
+			const updateOperations = update_operations.map((op) => ({
+				operation: op.operation,
+				lineStart: op.lineStart,
+				lineEnd: op.lineEnd,
+				content: op.content,
+				reason: op.reason
+			}));
+			await commitFileChanges({
+				githubClient,
+				owner,
+				repo,
+				fileContent: currentFileContent,
+				filePath: file_path,
+				branchName: branch_name,
+				operations: updateOperations,
+				commitMessage: commit_message
+			});
+			return { content: [{
+				type: "text",
+				text: `Successfully committed changes to ${owner}/${repo} on branch "${branch_name}"`
+			}] };
+		} catch (error) {
+			if (error instanceof Error && "status" in error) {
+				const apiError = error;
+				if (apiError.status === 404) return {
+					content: [{
+						type: "text",
+						text: `Repository ${owner}/${repo} or branch "${branch_name}" not found`
+					}],
+					isError: true
+				};
+				if (apiError.status === 422) return {
+					content: [{
+						type: "text",
+						text: "Invalid commit data. Please check file path and update operations format."
+					}],
+					isError: true
+				};
+			}
+			return {
+				content: [{
+					type: "text",
+					text: `Error creating commit: ${error instanceof Error ? error.message : "Unknown error"}`
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("commit-new-file", `Create and commit a new file in a repository. ${getAvailableRepositoryString(repositoryAccess)}`, {
+		owner: z.string().describe("Repository owner name"),
+		repo: z.string().describe("Repository name"),
+		branch_name: z.string().describe("Branch to commit to"),
+		file_path: z.string().describe("Path for the new file (relative to repository root)"),
+		content: z.string().describe("Content for the new file"),
+		commit_message: z.string().describe("Commit message")
+	}, async ({ owner, repo, branch_name, file_path, content, commit_message }) => {
+		try {
+			let githubClient;
+			try {
+				githubClient = getGitHubClientFromRepo(owner, repo, installationIdMap);
+			} catch (error) {
+				return {
+					content: [{
+						type: "text",
+						text: `Error accessing GitHub: ${error instanceof Error ? error.message : "Unknown error"}`
+					}],
+					isError: true
+				};
+			}
+			return { content: [{
+				type: "text",
+				text: `Successfully created and committed new file "${file_path}" to ${owner}/${repo} on branch "${branch_name}"\n\nCommit SHA: ${await commitNewFile({
+					githubClient,
+					owner,
+					repo,
+					filePath: file_path,
+					branchName: branch_name,
+					content,
+					commitMessage: commit_message
+				})}`
+			}] };
+		} catch (error) {
+			if (error instanceof Error && "status" in error) {
+				const apiError = error;
+				if (apiError.status === 404) return {
+					content: [{
+						type: "text",
+						text: `Repository ${owner}/${repo} or branch "${branch_name}" not found`
+					}],
+					isError: true
+				};
+				if (apiError.status === 422) return {
+					content: [{
+						type: "text",
+						text: `File "${file_path}" may already exist or the path is invalid.`
+					}],
+					isError: true
+				};
+			}
+			return {
+				content: [{
+					type: "text",
+					text: `Error creating file: ${error instanceof Error ? error.message : "Unknown error"}`
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("create-pull-request", `Create a pull request in a repository. ${getAvailableRepositoryString(repositoryAccess)}`, {
+		owner: z.string().describe("Repository owner name"),
+		repo: z.string().describe("Repository name"),
+		title: z.string().describe("Pull request title"),
+		body: z.string().describe("Pull request description"),
+		head_branch: z.string().describe("Branch containing the changes")
+	}, async ({ owner, repo, title, body, head_branch }) => {
+		try {
+			let githubClient;
+			try {
+				githubClient = getGitHubClientFromRepo(owner, repo, installationIdMap);
+			} catch (error) {
+				return {
+					content: [{
+						type: "text",
+						text: `Error accessing GitHub: ${error instanceof Error ? error.message : "Unknown error"}`
+					}],
+					isError: true
+				};
+			}
+			const targetBaseBranch = (await githubClient.rest.repos.get({
+				owner,
+				repo
+			})).data.default_branch;
+			const pullRequestResponse = await githubClient.rest.pulls.create({
+				owner,
+				repo,
+				title,
+				body,
+				head: head_branch,
+				base: targetBaseBranch,
+				draft: false
+			});
+			return { content: [{
+				type: "text",
+				text: `Successfully created pull request in ${owner}/${repo}\n\nPull Request details:\n• Number: #${pullRequestResponse.data.number}\n• Title: ${title}\n• Head: ${head_branch}\n• Base: ${targetBaseBranch}\n• Status: Open\n• URL: ${pullRequestResponse.data.html_url}\n\nDescription:\n${body}`
+			}] };
+		} catch (error) {
+			if (error instanceof Error && "status" in error) {
+				const apiError = error;
+				if (apiError.status === 404) return {
+					content: [{
+						type: "text",
+						text: `Repository ${owner}/${repo} not found or branch "${head_branch}" does not exist`
+					}],
+					isError: true
+				};
+				if (apiError.status === 422) return {
+					content: [{
+						type: "text",
+						text: "Pull request validation failed. This could be due to: no commits between branches, pull request already exists, or invalid branch names."
+					}],
+					isError: true
+				};
+				if (apiError.status === 403) return {
+					content: [{
+						type: "text",
+						text: "Permission denied. Your GitHub App may not have sufficient permissions to create pull requests in this repository."
+					}],
+					isError: true
+				};
+			}
+			return {
+				content: [{
+					type: "text",
+					text: `Error creating pull request: ${error instanceof Error ? error.message : "Unknown error"}`
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("leave-comment-on-pull-request", `Leave a comment on a pull request. This creates a general comment on the PR, not a line-specific review comment. ${getAvailableRepositoryString(repositoryAccess)}`, {
+		owner: z.string().describe("Repository owner name"),
+		repo: z.string().describe("Repository name"),
+		pull_request_number: z.number().describe("Pull request number"),
+		body: z.string().describe("The comment body text (supports GitHub markdown)")
+	}, async ({ owner, repo, pull_request_number, body }) => {
+		try {
+			let githubClient;
+			try {
+				githubClient = getGitHubClientFromRepo(owner, repo, installationIdMap);
+			} catch (error) {
+				return {
+					content: [{
+						type: "text",
+						text: `Error accessing GitHub: ${error instanceof Error ? error.message : "Unknown error"}`
+					}],
+					isError: true
+				};
+			}
+			const commentResponse = await githubClient.rest.issues.createComment({
+				owner,
+				repo,
+				issue_number: pull_request_number,
+				body
+			});
+			return { content: [{
+				type: "text",
+				text: `Successfully posted comment on PR #${pull_request_number} in ${owner}/${repo}\n\nComment details:\n• ID: ${commentResponse.data.id}\n• URL: ${commentResponse.data.html_url}\n• Created at: ${commentResponse.data.created_at}`
+			}] };
+		} catch (error) {
+			if (error instanceof Error && "status" in error) {
+				const apiError = error;
+				if (apiError.status === 404) return {
+					content: [{
+						type: "text",
+						text: `Pull request #${pull_request_number} not found in ${owner}/${repo}. Please check the repository exists and the pull request number is correct.`
+					}],
+					isError: true
+				};
+				if (apiError.status === 403) return {
+					content: [{
+						type: "text",
+						text: `Access denied to PR #${pull_request_number} in ${owner}/${repo}. Your GitHub App may not have sufficient permissions to comment on pull requests.`
+					}],
+					isError: true
+				};
+				if (apiError.status === 422) return {
+					content: [{
+						type: "text",
+						text: "Invalid comment data. Please check the comment body is not empty and is valid."
+					}],
+					isError: true
+				};
+			}
+			return {
+				content: [{
+					type: "text",
+					text: `Error posting comment: ${error instanceof Error ? error.message : "Unknown error"}`
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("visualize-update-operations", `Apply a list of operations to a piece of documentation and return a mapping of line number to line content. ${getAvailableRepositoryString(repositoryAccess)}`, {
+		owner: z.string().describe("Repository owner name"),
+		repo: z.string().describe("Repository name"),
+		file_path: z.string().describe("The path of the file to visualize the update operations for"),
+		operations: updateOperationsSchema
+	}, async ({ owner, repo, file_path, operations }) => {
+		try {
+			let githubClient;
+			try {
+				githubClient = getGitHubClientFromRepo(owner, repo, installationIdMap);
+			} catch (error) {
+				return {
+					content: [{
+						type: "text",
+						text: `Error accessing GitHub: ${error instanceof Error ? error.message : "Unknown error"}`
+					}],
+					isError: true
+				};
+			}
+			const response = await githubClient.rest.repos.getContent({
+				owner,
+				repo,
+				path: file_path
+			});
+			if ("content" in response.data && !Array.isArray(response.data)) {
+				const fileData = response.data;
+				const result = visualizeUpdateOperations(Buffer.from(fileData.content, "base64").toString("utf-8"), operations.map((op) => ({
+					operation: op.operation,
+					lineStart: op.lineStart,
+					lineEnd: op.lineEnd,
+					content: op.content,
+					reason: op.reason
+				})));
+				if (typeof result === "string") return {
+					content: [{
+						type: "text",
+						text: result
+					}],
+					isError: true
+				};
+				return { content: [{
+					type: "text",
+					text: `${Object.entries(result).sort(([a], [b]) => Number(a) - Number(b)).map(([lineNum, content]) => `${lineNum.padStart(3, " ")}| ${content}`).join("\n")}`
+				}] };
+			}
+			return {
+				content: [{
+					type: "text",
+					text: `The path "${file_path}" is not a file or could not be retrieved as file content.`
+				}],
+				isError: true
+			};
+		} catch (error) {
+			return {
+				content: [{
+					type: "text",
+					text: `Error visualizing update operations: ${error instanceof Error ? error.message : "Unknown error"}`
+				}],
+				isError: true
+			};
+		}
+	});
+	return server;
+};
+const app = new Hono();
+app.use("/", githubMcpAuth());
+app.post("/", async (c) => {
+	if (!process.env.GITHUB_APP_ID || !process.env.GITHUB_APP_PRIVATE_KEY) return c.json({ error: "GITHUB_APP_ID and GITHUB_APP_PRIVATE_KEY must be set" }, 500);
+	const toolId = c.get("toolId");
+	const body = await c.req.json();
+	const server = await getServer(toolId);
+	const transport = new StreamableHTTPServerTransport({ sessionIdGenerator: void 0 });
+	server.connect(transport);
+	const { req, res } = toReqRes(c.req.raw);
+	await transport.handleRequest(req, res, body);
+	return toFetchResponse(res);
+});
+app.delete("/", async (c) => {
+	return c.json({
+		jsonrpc: "2.0",
+		error: {
+			code: -32001,
+			message: "Method Not Allowed"
+		},
+		id: null
+	}, { status: 405 });
+});
+app.get("/", async (c) => {
+	return c.json({
+		jsonrpc: "2.0",
+		error: {
+			code: -32e3,
+			message: "Method not allowed."
+		},
+		id: null
+	}, { status: 405 });
+});
+app.get("/health", async (c) => {
+	return c.json({
+		status: "healthy",
+		timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+		service: "GitHub MCP Server"
+	});
+});
+var mcp_default = app;
+
+//#endregion
+export { mcp_default as default };