wehub-resource-sync
234 lines
7.3 KiB
JavaScript
234 lines
7.3 KiB
JavaScript
import { readFileSync } from "node:fs";
|
|
import { apiFetch } from "../api.mjs";
|
|
import { emit } from "../output.mjs";
|
|
import { t } from "../i18n.mjs";
|
|
|
|
const AGENTS = ["codex", "devin", "jules"];
|
|
|
|
function truncate(v, len = 35) {
|
|
if (v == null) return "-";
|
|
const s = String(v);
|
|
return s.length > len ? s.slice(0, len - 1) + "…" : s;
|
|
}
|
|
|
|
function fmtTs(v) {
|
|
if (!v) return "-";
|
|
try {
|
|
return new Date(v).toLocaleString();
|
|
} catch {
|
|
return String(v);
|
|
}
|
|
}
|
|
|
|
function fmtStatus(v) {
|
|
if (!v) return "-";
|
|
const colors = {
|
|
running: "\x1b[33m",
|
|
completed: "\x1b[32m",
|
|
failed: "\x1b[31m",
|
|
cancelled: "\x1b[90m",
|
|
};
|
|
const c = colors[v] ?? "";
|
|
return `${c}${v}\x1b[0m`;
|
|
}
|
|
|
|
const taskSchema = [
|
|
{ key: "id", header: "Task ID", width: 22 },
|
|
{ key: "agent", header: "Agent", width: 8 },
|
|
{ key: "status", header: "Status", width: 14, formatter: fmtStatus },
|
|
{ key: "title", header: "Title", width: 35, formatter: truncate },
|
|
{ key: "createdAt", header: "Created", formatter: fmtTs },
|
|
{ key: "updatedAt", header: "Updated", formatter: fmtTs },
|
|
];
|
|
|
|
async function confirm(q) {
|
|
return new Promise((resolve) => {
|
|
process.stdout.write(`${q} (yes/no) `);
|
|
process.stdin.setEncoding("utf8");
|
|
process.stdin.once("data", (c) => resolve(c.toString().trim().toLowerCase().startsWith("y")));
|
|
});
|
|
}
|
|
|
|
function registerTaskCommands(parent, agent) {
|
|
const task = parent.command("task").description(t("cloud.task.description"));
|
|
|
|
task
|
|
.command("create")
|
|
.description(t("cloud.task.create.description"))
|
|
.option("--title <t>", t("cloud.task.create.title"))
|
|
.option("--prompt <p>", t("cloud.task.create.prompt"))
|
|
.option("--prompt-file <path>", t("cloud.task.create.prompt_file"))
|
|
.option("--repo <url>", t("cloud.task.create.repo"))
|
|
.option("--branch <b>", t("cloud.task.create.branch"))
|
|
.option("--metadata <json>", t("cloud.task.create.metadata"))
|
|
.action(async (opts, cmd) => {
|
|
const prompt =
|
|
opts.prompt ?? (opts.promptFile ? readFileSync(opts.promptFile, "utf8") : null);
|
|
if (!prompt) {
|
|
process.stderr.write("--prompt or --prompt-file required\n");
|
|
process.exit(2);
|
|
}
|
|
const body = {
|
|
agent,
|
|
title: opts.title ?? prompt.slice(0, 80),
|
|
prompt,
|
|
...(opts.repo ? { repo: opts.repo } : {}),
|
|
...(opts.branch ? { branch: opts.branch } : {}),
|
|
...(opts.metadata ? { metadata: JSON.parse(opts.metadata) } : {}),
|
|
};
|
|
const res = await apiFetch("/api/v1/agents/tasks", { method: "POST", body });
|
|
if (!res.ok) {
|
|
process.stderr.write(`Error: ${res.status}\n`);
|
|
process.exit(1);
|
|
}
|
|
emit(await res.json(), cmd.optsWithGlobals());
|
|
});
|
|
|
|
task
|
|
.command("list")
|
|
.description(t("cloud.task.list.description"))
|
|
.option("--status <s>", t("cloud.task.list.status"))
|
|
.option("--limit <n>", t("cloud.task.list.limit"), parseInt, 50)
|
|
.action(async (opts, cmd) => {
|
|
const params = new URLSearchParams({ agent, limit: String(opts.limit ?? 50) });
|
|
if (opts.status) params.set("status", opts.status);
|
|
const res = await apiFetch(`/api/v1/agents/tasks?${params}`);
|
|
if (!res.ok) {
|
|
process.stderr.write(`Error: ${res.status}\n`);
|
|
process.exit(1);
|
|
}
|
|
const data = await res.json();
|
|
emit(data.items ?? data, cmd.optsWithGlobals(), taskSchema);
|
|
});
|
|
|
|
task
|
|
.command("get <taskId>")
|
|
.description(t("cloud.task.get.description"))
|
|
.action(async (taskId, opts, cmd) => {
|
|
const res = await apiFetch(`/api/v1/agents/tasks/${taskId}`);
|
|
if (!res.ok) {
|
|
process.stderr.write(`Not found: ${taskId}\n`);
|
|
process.exit(1);
|
|
}
|
|
emit(await res.json(), cmd.optsWithGlobals());
|
|
});
|
|
|
|
task
|
|
.command("status <taskId>")
|
|
.description(t("cloud.task.status.description"))
|
|
.action(async (taskId, opts, cmd) => {
|
|
const res = await apiFetch(`/api/v1/agents/tasks/${taskId}`);
|
|
if (!res.ok) {
|
|
process.stderr.write(`Not found: ${taskId}\n`);
|
|
process.exit(1);
|
|
}
|
|
const data = await res.json();
|
|
const globalOpts = cmd.optsWithGlobals();
|
|
if (globalOpts.output === "json") {
|
|
emit({ status: data.status }, globalOpts);
|
|
} else {
|
|
process.stdout.write(`${data.status}\n`);
|
|
}
|
|
});
|
|
|
|
task
|
|
.command("cancel <taskId>")
|
|
.description(t("cloud.task.cancel.description"))
|
|
.option("--yes", t("cloud.task.cancel.yes"))
|
|
.action(async (taskId, opts, cmd) => {
|
|
if (!opts.yes) {
|
|
const ok = await confirm(`Cancel task ${taskId}?`);
|
|
if (!ok) return;
|
|
}
|
|
const res = await apiFetch(`/api/v1/agents/tasks/${taskId}`, {
|
|
method: "POST",
|
|
body: { op: "cancel" },
|
|
});
|
|
if (!res.ok) {
|
|
process.stderr.write(`Error: ${res.status}\n`);
|
|
process.exit(1);
|
|
}
|
|
process.stdout.write("Cancelled\n");
|
|
});
|
|
|
|
task
|
|
.command("approve <taskId>")
|
|
.description(t("cloud.task.approve.description"))
|
|
.action(async (taskId, opts, cmd) => {
|
|
const res = await apiFetch(`/api/v1/agents/tasks/${taskId}`, {
|
|
method: "POST",
|
|
body: { op: "approve_plan" },
|
|
});
|
|
if (!res.ok) {
|
|
process.stderr.write(`Error: ${res.status}\n`);
|
|
process.exit(1);
|
|
}
|
|
process.stdout.write("Plan approved\n");
|
|
});
|
|
|
|
task
|
|
.command("message <taskId> <message>")
|
|
.description(t("cloud.task.message.description"))
|
|
.action(async (taskId, msg, opts, cmd) => {
|
|
const res = await apiFetch(`/api/v1/agents/tasks/${taskId}`, {
|
|
method: "POST",
|
|
body: { op: "message", message: msg },
|
|
});
|
|
if (!res.ok) {
|
|
process.stderr.write(`Error: ${res.status}\n`);
|
|
process.exit(1);
|
|
}
|
|
process.stdout.write("Message sent\n");
|
|
});
|
|
|
|
parent
|
|
.command("sources <taskId>")
|
|
.description(t("cloud.sources.description"))
|
|
.action(async (taskId, opts, cmd) => {
|
|
const res = await apiFetch(`/api/v1/agents/tasks/${taskId}?op=sources`);
|
|
if (!res.ok) {
|
|
process.stderr.write(`Error: ${res.status}\n`);
|
|
process.exit(1);
|
|
}
|
|
const data = await res.json();
|
|
emit(data.sources ?? data, cmd.optsWithGlobals());
|
|
});
|
|
}
|
|
|
|
export function registerCloud(program) {
|
|
const cloud = program.command("cloud").description(t("cloud.description"));
|
|
|
|
cloud
|
|
.command("agents")
|
|
.description(t("cloud.agents.description"))
|
|
.action(async (opts, cmd) => {
|
|
const res = await apiFetch("/api/v1/agents/tasks?meta=agents");
|
|
if (res.ok) {
|
|
const data = await res.json();
|
|
emit(data.agents ?? AGENTS.map((id) => ({ id })), cmd.optsWithGlobals());
|
|
} else {
|
|
emit(
|
|
AGENTS.map((id) => ({ id })),
|
|
cmd.optsWithGlobals()
|
|
);
|
|
}
|
|
});
|
|
|
|
for (const agent of AGENTS) {
|
|
const agentCmd = cloud
|
|
.command(agent)
|
|
.description(t("cloud.agent.description").replace("{agent}", agent));
|
|
registerTaskCommands(agentCmd, agent);
|
|
|
|
agentCmd
|
|
.command("auth")
|
|
.description(t("cloud.agent.auth.description"))
|
|
.option("--no-browser", "Skip browser open")
|
|
.option("--timeout <ms>", "Auth timeout ms", parseInt, 300000)
|
|
.action(async (opts, cmd) => {
|
|
const { runOAuthStart } = await import("./oauth.mjs");
|
|
await runOAuthStart({ provider: agent, ...opts }, cmd);
|
|
});
|
|
}
|
|
}
|