import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
import { z } from "zod";
import { ChangeSetsApi } from "@systeminit/api-client";
import { apiConfig, WORKSPACE_ID } from "../si_client.ts";
import {
errorResponse,
generateDescription,
successResponse,
withAnalytics,
} from "./commonBehavior.ts";
const name = "change-set-force-apply";
const title = "Force apply a change set";
const description = `<description>Force apply a change set. Returns 'success' if the status was changed. On failure, returns error details</description><usage>Use this tool to Force apply a change set. This tool will avoid *all* workspace approval flows and apply directly to HEAD. You may *never* force apply the HEAD change set.</usage>`;
const ForceApplyChangeSetInputSchemaRaw = {
changeSetId: z.string().describe("the ID of the change set to force apply"),
};
const ForceApplyChangeSetOutputSchemaRaw = {
status: z.enum(["success", "failure"]),
errorMessage: z
.string()
.optional()
.describe(
"If the status is failure, the error message will contain information about what went wrong",
),
data: z
.object({ success: z.boolean().describe("will always be true") })
.describe("will be true if the request succeeds"),
};
const ForceApplyChangeSetOutputSchema = z.object(
ForceApplyChangeSetOutputSchemaRaw,
);
export function changeSetForceApplyTool(server: McpServer) {
server.registerTool(
name,
{
title,
description: generateDescription(
description,
"changeSetForceApplyResponse",
ForceApplyChangeSetOutputSchema,
),
annotations: {
destructiveHint: true,
},
inputSchema: ForceApplyChangeSetInputSchemaRaw,
outputSchema: ForceApplyChangeSetOutputSchemaRaw,
},
async ({ changeSetId }): Promise<CallToolResult> => {
return await withAnalytics(name, async () => {
if (!changeSetId) {
return errorResponse({
message:
"Must provide a change set id; ensure you get one from the user!",
});
}
const siApi = new ChangeSetsApi(apiConfig);
try {
const response = await siApi.getChangeSet({
workspaceId: WORKSPACE_ID,
changeSetId,
});
if (response.data.changeSet.isHead) {
return errorResponse({
message:
"You may not force-apply the HEAD change set. Inform the user that HEAD is immutable, and they should not try and make any changes directly to it. Call them a cheeky monkey.",
});
}
} catch (error) {
return errorResponse(error);
}
try {
const response = await siApi.forceApply({
workspaceId: WORKSPACE_ID,
changeSetId,
});
return successResponse(response.data);
} catch (error) {
return errorResponse(error);
}
});
},
);
}
