David Blass
86 lines
2.6 KiB
TypeScript
86 lines
2.6 KiB
TypeScript
import { type } from "arktype";
|
|
import { claude } from "./agents/claude.ts";
|
|
import { createMcpConfigs } from "./mcp/config.ts";
|
|
import packageJson from "./package.json" with { type: "json" };
|
|
import { log } from "./utils/cli.ts";
|
|
import { parseRepoContext, setupGitHubInstallationToken } from "./utils/github.ts";
|
|
import { setupGitAuth, setupGitConfig } from "./utils/setup.ts";
|
|
|
|
export const Inputs = type({
|
|
prompt: "string",
|
|
"anthropic_api_key?": "string | undefined",
|
|
});
|
|
|
|
export type Inputs = typeof Inputs.infer;
|
|
|
|
export interface MainResult {
|
|
success: boolean;
|
|
output?: string | undefined;
|
|
error?: string | undefined;
|
|
}
|
|
|
|
export type PromptJSON = {};
|
|
|
|
export async function main(inputs: Inputs): Promise<MainResult> {
|
|
try {
|
|
log.info(`🐸 Running pullfrog/action@${packageJson.version}...`);
|
|
|
|
// Change to GITHUB_WORKSPACE if set (this is where actions/checkout puts the repo)
|
|
// JavaScript actions run from the action's directory, not the checked-out repo
|
|
if (process.env.GITHUB_WORKSPACE && process.cwd() !== process.env.GITHUB_WORKSPACE) {
|
|
log.debug(`Changing to GITHUB_WORKSPACE: ${process.env.GITHUB_WORKSPACE}`);
|
|
process.chdir(process.env.GITHUB_WORKSPACE);
|
|
log.debug(`New working directory: ${process.cwd()}`);
|
|
}
|
|
|
|
setupGitConfig();
|
|
|
|
const githubInstallationToken = await setupGitHubInstallationToken();
|
|
const repoContext = parseRepoContext();
|
|
|
|
setupGitAuth(githubInstallationToken, repoContext);
|
|
|
|
const mcpServers = createMcpConfigs(githubInstallationToken);
|
|
|
|
log.debug(`📋 MCP Config: ${JSON.stringify(mcpServers, null, 2)}`);
|
|
|
|
log.info("Running Claude Agent SDK...");
|
|
log.box(inputs.prompt, { title: "Prompt" });
|
|
|
|
// TODO: check if `inputs.prompts` is JSON
|
|
// if yes, check if it's a webhook payload or toJSON(github.event)
|
|
// for webhook payloads, check the specified `agent` field
|
|
|
|
const result = await claude.run({
|
|
prompt: inputs.prompt,
|
|
mcpServers,
|
|
githubInstallationToken,
|
|
apiKey: inputs.anthropic_api_key!,
|
|
});
|
|
|
|
if (!result.success) {
|
|
return {
|
|
success: false,
|
|
error: result.error || "Agent execution failed",
|
|
output: result.output!,
|
|
};
|
|
}
|
|
|
|
log.success("Task complete.");
|
|
await log.writeSummary();
|
|
|
|
return {
|
|
success: true,
|
|
output: result.output || "",
|
|
};
|
|
} catch (error) {
|
|
const errorMessage = error instanceof Error ? error.message : "Unknown error occurred";
|
|
log.error(errorMessage);
|
|
await log.writeSummary();
|
|
return {
|
|
success: false,
|
|
error: errorMessage,
|
|
};
|
|
}
|
|
}
|