David Blass
127 lines
3.4 KiB
TypeScript
127 lines
3.4 KiB
TypeScript
import { flatMorph } from "@ark/util";
|
|
import { type } from "arktype";
|
|
import { agents } from "./agents/index.ts";
|
|
import { createMcpConfigs } from "./mcp/config.ts";
|
|
import packageJson from "./package.json" with { type: "json" };
|
|
import { fetchRepoSettings } from "./utils/api.ts";
|
|
import { log } from "./utils/cli.ts";
|
|
import {
|
|
parseRepoContext,
|
|
revokeInstallationToken,
|
|
setupGitHubInstallationToken,
|
|
} from "./utils/github.ts";
|
|
import { setupGitAuth, setupGitConfig } from "./utils/setup.ts";
|
|
|
|
export const AgentName = type.enumerated(...Object.values(agents).map((agent) => agent.name));
|
|
export type AgentName = typeof AgentName.infer;
|
|
|
|
export const AgentInputKey = type.enumerated(
|
|
...Object.values(agents).map((agent) => agent.inputKey)
|
|
);
|
|
export type AgentInputKey = typeof AgentInputKey.infer;
|
|
|
|
const keyInputDefs = flatMorph(
|
|
agents,
|
|
(_, agent) => [agent.inputKey, "string | undefined?"] as const
|
|
);
|
|
|
|
export const Inputs = type({
|
|
prompt: "string",
|
|
...keyInputDefs,
|
|
"agent?": AgentName,
|
|
});
|
|
|
|
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> {
|
|
let tokenToRevoke: string | null = null;
|
|
|
|
try {
|
|
log.info(`🐸 Running pullfrog/action@${packageJson.version}...`);
|
|
|
|
Inputs.assert(inputs);
|
|
setupGitConfig();
|
|
|
|
const { githubInstallationToken, wasAcquired } = await setupGitHubInstallationToken();
|
|
if (wasAcquired) {
|
|
tokenToRevoke = githubInstallationToken;
|
|
}
|
|
const repoContext = parseRepoContext();
|
|
|
|
const repoSettings = await fetchRepoSettings({
|
|
token: githubInstallationToken,
|
|
repoContext,
|
|
});
|
|
|
|
const agentName: AgentName = inputs.agent || repoSettings.defaultAgent || "claude";
|
|
|
|
const agent = agents[agentName];
|
|
|
|
setupGitAuth(githubInstallationToken, repoContext);
|
|
|
|
const mcpServers = createMcpConfigs(githubInstallationToken);
|
|
|
|
log.debug(`📋 MCP Config: ${JSON.stringify(mcpServers, null, 2)}`);
|
|
|
|
// Install agent CLI before running
|
|
const cliPath = await agent.install();
|
|
|
|
log.info(`Running ${agentName}...`);
|
|
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
|
|
|
|
// Get API key based on agent type
|
|
|
|
const apiKey = inputs[agent.inputKey];
|
|
|
|
if (!apiKey) throw new Error(`${agent.inputKey} is required for ${agentName} agent`);
|
|
|
|
const result = await agent.run({
|
|
prompt: inputs.prompt,
|
|
mcpServers,
|
|
githubInstallationToken,
|
|
apiKey,
|
|
cliPath,
|
|
});
|
|
|
|
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,
|
|
};
|
|
} finally {
|
|
if (tokenToRevoke) {
|
|
await revokeInstallationToken(tokenToRevoke);
|
|
}
|
|
}
|
|
}
|