always-on-slash.js 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420
  1. import {
  2. extractProjectDirectory,
  3. getProjectCronJobsOverview,
  4. } from './projects.js';
  5. import {
  6. getProjectDiscoveryPlansOverview,
  7. rerunDiscoveryPlan,
  8. } from './discovery-plans.js';
  9. import { getPilotDeckGateway } from './pilotdeck-bridge.js';
  10. const TARGET_ALIASES = new Map([
  11. ['cron', 'cron'],
  12. ['crons', 'cron'],
  13. ['job', 'cron'],
  14. ['jobs', 'cron'],
  15. ['cron-job', 'cron'],
  16. ['cron-jobs', 'cron'],
  17. ['plan', 'plan'],
  18. ['plans', 'plan'],
  19. ]);
  20. function normalizeTargetType(value) {
  21. if (typeof value !== 'string') {
  22. return null;
  23. }
  24. return TARGET_ALIASES.get(value.trim().toLowerCase()) || null;
  25. }
  26. function formatDateTime(value, fallback = '-') {
  27. if (value === undefined || value === null || value === '') {
  28. return fallback;
  29. }
  30. const date = new Date(value);
  31. if (Number.isNaN(date.getTime())) {
  32. return fallback;
  33. }
  34. return date.toLocaleString();
  35. }
  36. function formatText(value, fallback = '-') {
  37. if (typeof value !== 'string') {
  38. return fallback;
  39. }
  40. const trimmed = value.trim();
  41. return trimmed.length > 0 ? trimmed : fallback;
  42. }
  43. function summarizeText(value, maxLength = 140, fallback = '-') {
  44. const text = formatText(value, '');
  45. if (!text) {
  46. return fallback;
  47. }
  48. const collapsed = text.replace(/\s+/g, ' ');
  49. return collapsed.length > maxLength
  50. ? `${collapsed.slice(0, maxLength - 3)}...`
  51. : collapsed;
  52. }
  53. function buildUsageMarkdown() {
  54. return [
  55. '# Always-On Slash',
  56. '',
  57. 'Usage:',
  58. '- `/ao list [cron|plan]`',
  59. '- `/ao status <cron|plan> <id>`',
  60. '- `/ao run <cron|plan> <id>`',
  61. '- `/ao help`',
  62. ].join('\n');
  63. }
  64. function buildResponse(content, extraData = {}) {
  65. return {
  66. type: 'builtin',
  67. action: 'ao',
  68. data: {
  69. mode: 'message',
  70. content,
  71. ...extraData,
  72. },
  73. };
  74. }
  75. export function parseAlwaysOnSlashArgs(args = []) {
  76. const [actionRaw = 'help', targetRaw, idRaw, ...extra] = Array.isArray(args) ? args : [];
  77. const action = String(actionRaw).trim().toLowerCase();
  78. if (!action || action === 'help') {
  79. return { action: 'help' };
  80. }
  81. if (action === 'list') {
  82. if (!targetRaw) {
  83. return { action: 'list', target: 'all' };
  84. }
  85. const target = normalizeTargetType(targetRaw);
  86. if (!target || extra.length > 0 || idRaw) {
  87. return {
  88. action: 'help',
  89. error: 'Usage: `/ao list [cron|plan]`',
  90. };
  91. }
  92. return { action: 'list', target };
  93. }
  94. if (action === 'status' || action === 'run') {
  95. const target = normalizeTargetType(targetRaw);
  96. const id = typeof idRaw === 'string' ? idRaw.trim() : '';
  97. if (!target || !id || extra.length > 0) {
  98. return {
  99. action: 'help',
  100. error: `Usage: \`/ao ${action} <cron|plan> <id>\``,
  101. };
  102. }
  103. return { action, target, id };
  104. }
  105. return {
  106. action: 'help',
  107. error: `Unknown /ao action: \`${action}\``,
  108. };
  109. }
  110. function getProjectContext(context) {
  111. const projectName =
  112. typeof context?.projectName === 'string' ? context.projectName.trim() : '';
  113. const projectPath =
  114. typeof context?.projectPath === 'string' ? context.projectPath.trim() : '';
  115. if (!projectName) {
  116. return null;
  117. }
  118. return {
  119. projectName,
  120. projectPath,
  121. };
  122. }
  123. function buildCronListMarkdown(projectPath, jobs) {
  124. const lines = [
  125. `## Cron jobs (${jobs.length})`,
  126. '',
  127. ];
  128. if (projectPath) {
  129. lines.push(`Workspace: \`${projectPath}\``);
  130. lines.push('');
  131. }
  132. if (jobs.length === 0) {
  133. lines.push('No cron jobs found.');
  134. return lines.join('\n');
  135. }
  136. for (const job of jobs) {
  137. const kind = [
  138. job.durable === false ? 'session' : 'durable',
  139. job.recurring ? 'recurring' : 'one-shot',
  140. job.manualOnly ? 'manual-only' : null,
  141. ].filter(Boolean).join(', ');
  142. lines.push(`- \`${job.id}\` - ${summarizeText(job.prompt)}`);
  143. lines.push(` - Status: \`${job.status}\``);
  144. lines.push(` - Kind: ${kind}`);
  145. lines.push(` - Schedule: \`${job.cron}\``);
  146. lines.push(` - Last fired: ${formatDateTime(job.lastFiredAt)}`);
  147. if (job.latestRun?.summary) {
  148. lines.push(` - Latest summary: ${summarizeText(job.latestRun.summary, 180)}`);
  149. }
  150. }
  151. return lines.join('\n');
  152. }
  153. function buildPlanListMarkdown(projectPath, plans) {
  154. const lines = [
  155. `## Discovery plans (${plans.length})`,
  156. '',
  157. ];
  158. if (projectPath) {
  159. lines.push(`Workspace: \`${projectPath}\``);
  160. lines.push('');
  161. }
  162. if (plans.length === 0) {
  163. lines.push('No discovery plans found.');
  164. return lines.join('\n');
  165. }
  166. for (const plan of plans) {
  167. lines.push(`- \`${plan.id}\` - ${formatText(plan.title)}`);
  168. lines.push(` - Status: \`${plan.status}\``);
  169. lines.push(` - Updated: ${formatDateTime(plan.updatedAt)}`);
  170. lines.push(` - Summary: ${summarizeText(plan.summary, 180)}`);
  171. }
  172. return lines.join('\n');
  173. }
  174. function buildCombinedListMarkdown(projectPath, { jobs, plans }) {
  175. return [
  176. '# Always-On',
  177. '',
  178. buildPlanListMarkdown(projectPath, plans),
  179. '',
  180. buildCronListMarkdown(projectPath, jobs),
  181. ].join('\n');
  182. }
  183. function buildCronStatusMarkdown(projectPath, job) {
  184. const latestRun = job.latestRun || null;
  185. return [
  186. `# Cron job \`${job.id}\``,
  187. '',
  188. projectPath ? `Workspace: \`${projectPath}\`` : '',
  189. projectPath ? '' : '',
  190. `- Status: \`${job.status}\``,
  191. `- Schedule: \`${job.cron}\``,
  192. `- Scope: \`${job.durable === false ? 'session' : 'durable'}\``,
  193. `- Type: \`${job.recurring ? 'recurring' : 'one-shot'}\``,
  194. `- Manual only: \`${job.manualOnly ? 'yes' : 'no'}\``,
  195. `- Created: ${formatDateTime(job.createdAt)}`,
  196. `- Last fired: ${formatDateTime(job.lastFiredAt)}`,
  197. `- Origin session: ${formatText(job.originSessionId)}`,
  198. `- Transcript key: ${formatText(job.transcriptKey)}`,
  199. '',
  200. '## Prompt',
  201. '',
  202. formatText(job.prompt),
  203. '',
  204. '## Latest run',
  205. '',
  206. `- Last activity: ${formatDateTime(latestRun?.lastActivity)}`,
  207. `- Summary: ${formatText(latestRun?.summary)}`,
  208. `- Task ID: ${formatText(latestRun?.taskId)}`,
  209. `- Transcript: ${formatText(latestRun?.relativeTranscriptPath)}`,
  210. `- Output file: ${formatText(latestRun?.outputFile)}`,
  211. ].filter(Boolean).join('\n');
  212. }
  213. function buildPlanStatusMarkdown(projectPath, plan) {
  214. return [
  215. `# Discovery plan \`${plan.id}\``,
  216. '',
  217. projectPath ? `Workspace: \`${projectPath}\`` : '',
  218. projectPath ? '' : '',
  219. `- Title: ${formatText(plan.title)}`,
  220. `- Status: \`${plan.status}\``,
  221. `- Updated: ${formatDateTime(plan.updatedAt)}`,
  222. `- Execution session: ${formatText(plan.executionSessionId)}`,
  223. `- Execution started: ${formatDateTime(plan.executionStartedAt)}`,
  224. `- Last activity: ${formatDateTime(plan.executionLastActivityAt)}`,
  225. `- Plan file: ${formatText(plan.planFilePath)}`,
  226. '',
  227. '## Summary',
  228. '',
  229. formatText(plan.summary),
  230. '',
  231. '## Rationale',
  232. '',
  233. formatText(plan.rationale),
  234. '',
  235. '## Latest summary',
  236. '',
  237. formatText(plan.latestSummary),
  238. ].filter(Boolean).join('\n');
  239. }
  240. function buildCronRunMarkdown(jobId, result) {
  241. if (result?.reason === 'already_running' || result?.started === false) {
  242. return [
  243. '# Always-On',
  244. '',
  245. `Cron job \`${jobId}\` is already running.`,
  246. '',
  247. `Use \`/ao status cron ${jobId}\` to inspect the current state.`,
  248. ].join('\n');
  249. }
  250. return [
  251. '# Always-On',
  252. '',
  253. `Started cron job \`${jobId}\` immediately.`,
  254. '',
  255. `Use \`/ao status cron ${jobId}\` to inspect the latest state.`,
  256. ].join('\n');
  257. }
  258. function buildNotFoundMarkdown(target, id) {
  259. return [
  260. '# Always-On',
  261. '',
  262. `No ${target} found with id \`${id}\`.`,
  263. ].join('\n');
  264. }
  265. function buildErrorMarkdown(message) {
  266. return [
  267. '# Always-On',
  268. '',
  269. message,
  270. ].join('\n');
  271. }
  272. function sortCronJobs(jobs) {
  273. return [...jobs].sort((left, right) => right.createdAt - left.createdAt);
  274. }
  275. export async function executeAlwaysOnSlashCommand(args = [], context = {}) {
  276. const project = getProjectContext(context);
  277. if (!project) {
  278. return buildResponse(
  279. 'Please select a project before using `/ao`.',
  280. );
  281. }
  282. const parsed = parseAlwaysOnSlashArgs(args);
  283. if (parsed.action === 'help') {
  284. const content = parsed.error
  285. ? `${buildErrorMarkdown(parsed.error)}\n\n${buildUsageMarkdown()}`
  286. : buildUsageMarkdown();
  287. return buildResponse(content);
  288. }
  289. try {
  290. if (parsed.action === 'list') {
  291. if (parsed.target === 'cron') {
  292. const overview = await getProjectCronJobsOverview(project.projectName);
  293. return buildResponse(
  294. buildCronListMarkdown(project.projectPath, sortCronJobs(overview.jobs || [])),
  295. );
  296. }
  297. if (parsed.target === 'plan') {
  298. const overview = await getProjectDiscoveryPlansOverview(project.projectName);
  299. return buildResponse(
  300. buildPlanListMarkdown(project.projectPath, overview.plans || []),
  301. );
  302. }
  303. const [cronOverview, planOverview] = await Promise.all([
  304. getProjectCronJobsOverview(project.projectName),
  305. getProjectDiscoveryPlansOverview(project.projectName),
  306. ]);
  307. return buildResponse(
  308. buildCombinedListMarkdown(project.projectPath, {
  309. jobs: sortCronJobs(cronOverview.jobs || []),
  310. plans: planOverview.plans || [],
  311. }),
  312. );
  313. }
  314. if (parsed.action === 'status' && parsed.target === 'cron') {
  315. const overview = await getProjectCronJobsOverview(project.projectName);
  316. const job = (overview.jobs || []).find((candidate) => candidate.id === parsed.id);
  317. if (!job) {
  318. return buildResponse(buildNotFoundMarkdown('cron job', parsed.id));
  319. }
  320. return buildResponse(buildCronStatusMarkdown(project.projectPath, job));
  321. }
  322. if (parsed.action === 'status' && parsed.target === 'plan') {
  323. const overview = await getProjectDiscoveryPlansOverview(project.projectName);
  324. const plan = (overview.plans || []).find((candidate) => candidate.id === parsed.id);
  325. if (!plan) {
  326. return buildResponse(buildNotFoundMarkdown('discovery plan', parsed.id));
  327. }
  328. return buildResponse(buildPlanStatusMarkdown(project.projectPath, plan));
  329. }
  330. if (parsed.action === 'run' && parsed.target === 'cron') {
  331. const gateway = await getPilotDeckGateway();
  332. const result = await gateway.cronRunNow({ taskId: parsed.id });
  333. if (result.reason === 'not_found') {
  334. return buildResponse(buildNotFoundMarkdown('cron job', parsed.id));
  335. }
  336. return buildResponse(buildCronRunMarkdown(parsed.id, result));
  337. }
  338. if (parsed.action === 'run' && parsed.target === 'plan') {
  339. const result = await rerunDiscoveryPlan(project.projectName, parsed.id);
  340. return buildResponse(
  341. `Plan \`${parsed.id}\` has been queued for re-execution (runId: \`${result.runId}\`).`,
  342. );
  343. }
  344. return buildResponse(buildUsageMarkdown());
  345. } catch (error) {
  346. const message =
  347. error instanceof Error && error.message.trim().length > 0
  348. ? error.message
  349. : 'Always-On slash command failed.';
  350. if (parsed.action === 'run' && parsed.target === 'plan' && /not found/i.test(message)) {
  351. return buildResponse(buildNotFoundMarkdown('discovery plan', parsed.id));
  352. }
  353. return buildResponse(buildErrorMarkdown(message));
  354. }
  355. }
  356. export {
  357. buildUsageMarkdown as getAlwaysOnSlashUsage,
  358. };