Colin McDonnell
129 lines
3.2 KiB
TypeScript
129 lines
3.2 KiB
TypeScript
import "./arkConfig.ts";
|
|
// this must be imported first
|
|
import { createServer } from "node:net";
|
|
import { FastMCP, type Tool } from "fastmcp";
|
|
import { ghPullfrogMcpName } from "../external.ts";
|
|
import { GetCheckSuiteLogsTool } from "./checkSuite.ts";
|
|
import {
|
|
CreateCommentTool,
|
|
EditCommentTool,
|
|
ReplyToReviewCommentTool,
|
|
ReportProgressTool,
|
|
} from "./comment.ts";
|
|
import { DebugShellCommandTool } from "./debug.ts";
|
|
import { CommitFilesTool, CreateBranchTool, PushBranchTool } from "./git.ts";
|
|
import { IssueTool } from "./issue.ts";
|
|
import { GetIssueCommentsTool } from "./issueComments.ts";
|
|
import { GetIssueEventsTool } from "./issueEvents.ts";
|
|
import { IssueInfoTool } from "./issueInfo.ts";
|
|
import { AddLabelsTool } from "./labels.ts";
|
|
import { PullRequestTool } from "./pr.ts";
|
|
import { PullRequestInfoTool } from "./prInfo.ts";
|
|
import { ReviewTool } from "./review.ts";
|
|
import { GetReviewCommentsTool, ListPullRequestReviewsTool } from "./reviewComments.ts";
|
|
import { SelectModeTool } from "./selectMode.ts";
|
|
import {
|
|
addTools,
|
|
getMcpContext,
|
|
initMcpContext,
|
|
isProgressCommentDisabled,
|
|
type McpInitContext,
|
|
} from "./shared.ts";
|
|
|
|
/**
|
|
* Find an available port starting from the given port
|
|
*/
|
|
async function findAvailablePort(startPort: number): Promise<number> {
|
|
const checkPort = (port: number): Promise<boolean> => {
|
|
return new Promise((resolve) => {
|
|
const server = createServer();
|
|
server.once("error", () => {
|
|
server.close();
|
|
resolve(false);
|
|
});
|
|
server.listen(port, () => {
|
|
server.close(() => {
|
|
resolve(true);
|
|
});
|
|
});
|
|
});
|
|
};
|
|
|
|
let port = startPort;
|
|
while (port < startPort + 100) {
|
|
if (await checkPort(port)) {
|
|
return port;
|
|
}
|
|
port++;
|
|
}
|
|
throw new Error(`Could not find available port starting from ${startPort}`);
|
|
}
|
|
|
|
/**
|
|
* Start the MCP HTTP server and return the URL and close function
|
|
*/
|
|
export async function startMcpHttpServer(
|
|
state: McpInitContext
|
|
): Promise<{ url: string; close: () => Promise<void> }> {
|
|
initMcpContext(state);
|
|
|
|
const server = new FastMCP({
|
|
name: ghPullfrogMcpName,
|
|
version: "0.0.1",
|
|
});
|
|
|
|
// get context for dynamic tool creation
|
|
const ctx = await getMcpContext();
|
|
|
|
const tools: Tool<any, any>[] = [
|
|
SelectModeTool,
|
|
CreateCommentTool,
|
|
EditCommentTool,
|
|
ReplyToReviewCommentTool,
|
|
IssueTool,
|
|
IssueInfoTool,
|
|
GetIssueCommentsTool,
|
|
GetIssueEventsTool,
|
|
PullRequestTool,
|
|
ReviewTool,
|
|
PullRequestInfoTool,
|
|
GetReviewCommentsTool,
|
|
ListPullRequestReviewsTool,
|
|
GetCheckSuiteLogsTool,
|
|
DebugShellCommandTool,
|
|
AddLabelsTool,
|
|
CreateBranchTool(ctx),
|
|
CommitFilesTool,
|
|
PushBranchTool(ctx),
|
|
];
|
|
|
|
// only include ReportProgressTool if progress comment is not disabled
|
|
if (!isProgressCommentDisabled()) {
|
|
tools.push(ReportProgressTool);
|
|
}
|
|
|
|
addTools(server, tools);
|
|
|
|
const port = await findAvailablePort(3764);
|
|
const host = "127.0.0.1";
|
|
const endpoint = "/mcp";
|
|
|
|
await server.start({
|
|
transportType: "httpStream",
|
|
httpStream: {
|
|
port,
|
|
host,
|
|
endpoint,
|
|
},
|
|
});
|
|
|
|
const url = `http://${host}:${port}${endpoint}`;
|
|
|
|
return {
|
|
url,
|
|
close: async () => {
|
|
await server.stop();
|
|
},
|
|
};
|
|
}
|