@h-rig/cli 0.0.6-alpha.6 → 0.0.6-alpha.61

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (52) hide show
  1. package/README.md +1 -1
  2. package/dist/bin/rig.js +5033 -1895
  3. package/dist/src/commands/_authority-runs.js +2 -3
  4. package/dist/src/commands/_cli-format.js +369 -0
  5. package/dist/src/commands/_connection-state.js +12 -6
  6. package/dist/src/commands/_doctor-checks.js +79 -34
  7. package/dist/src/commands/_help-catalog.js +446 -0
  8. package/dist/src/commands/_operator-surface.js +220 -0
  9. package/dist/src/commands/_operator-view.js +1124 -64
  10. package/dist/src/commands/_parsers.js +0 -2
  11. package/dist/src/commands/_pi-frontend.js +1080 -0
  12. package/dist/src/commands/_pi-install.js +4 -3
  13. package/dist/src/commands/_pi-remote-session.js +771 -0
  14. package/dist/src/commands/_pi-worker-bridge-extension.js +834 -0
  15. package/dist/src/commands/_policy.js +0 -2
  16. package/dist/src/commands/_preflight.js +98 -116
  17. package/dist/src/commands/_run-driver-helpers.js +46 -19
  18. package/dist/src/commands/_run-replay.js +142 -0
  19. package/dist/src/commands/_server-client.js +225 -48
  20. package/dist/src/commands/_snapshot-upload.js +74 -30
  21. package/dist/src/commands/_spinner.js +63 -0
  22. package/dist/src/commands/_task-picker.js +44 -16
  23. package/dist/src/commands/agent.js +10 -12
  24. package/dist/src/commands/browser.js +4 -6
  25. package/dist/src/commands/connect.js +134 -26
  26. package/dist/src/commands/dist.js +4 -6
  27. package/dist/src/commands/doctor.js +79 -34
  28. package/dist/src/commands/github.js +76 -32
  29. package/dist/src/commands/inbox.js +410 -31
  30. package/dist/src/commands/init.js +398 -90
  31. package/dist/src/commands/inspect.js +296 -23
  32. package/dist/src/commands/inspector.js +2 -4
  33. package/dist/src/commands/pi.js +168 -0
  34. package/dist/src/commands/plugin.js +81 -22
  35. package/dist/src/commands/profile-and-review.js +8 -10
  36. package/dist/src/commands/queue.js +2 -3
  37. package/dist/src/commands/remote.js +18 -20
  38. package/dist/src/commands/repo-git-harness.js +6 -8
  39. package/dist/src/commands/run.js +1600 -131
  40. package/dist/src/commands/server.js +281 -42
  41. package/dist/src/commands/setup.js +84 -45
  42. package/dist/src/commands/task-report-bug.js +5 -7
  43. package/dist/src/commands/task-run-driver.js +736 -93
  44. package/dist/src/commands/task.js +1863 -262
  45. package/dist/src/commands/test.js +3 -5
  46. package/dist/src/commands/workspace.js +4 -6
  47. package/dist/src/commands.js +5675 -2531
  48. package/dist/src/index.js +5654 -2519
  49. package/dist/src/launcher.js +5 -3
  50. package/dist/src/report-bug.js +3 -3
  51. package/dist/src/runner.js +5 -19
  52. package/package.json +7 -5
@@ -5,8 +5,7 @@ import { resolve } from "path";
5
5
  import {
6
6
  readAuthorityRun,
7
7
  readJsonlFile,
8
- resolveAuthorityRunDir,
9
- writeJsonFile
8
+ writeAuthorityRunRecord
10
9
  } from "@rig/runtime/control-plane/authority-files";
11
10
 
12
11
  // packages/cli/src/commands/_paths.ts
@@ -100,7 +99,7 @@ function upsertAgentAuthorityRun(projectRoot, input) {
100
99
  } else if ("errorText" in next) {
101
100
  delete next.errorText;
102
101
  }
103
- writeJsonFile(resolve(resolveAuthorityRunDir(projectRoot, input.runId), "run.json"), next);
102
+ writeAuthorityRunRecord(projectRoot, input.runId, next);
104
103
  return next;
105
104
  }
106
105
  export {
@@ -0,0 +1,369 @@
1
+ // @bun
2
+ // packages/cli/src/commands/_cli-format.ts
3
+ import { log, note } from "@clack/prompts";
4
+ import pc from "picocolors";
5
+ function stringField(record, key, fallback = "") {
6
+ const value = record[key];
7
+ return typeof value === "string" && value.trim() ? value.trim() : fallback;
8
+ }
9
+ function numberField(record, key) {
10
+ const value = record[key];
11
+ return typeof value === "number" && Number.isFinite(value) ? value : null;
12
+ }
13
+ function arrayField(record, key) {
14
+ const value = record[key];
15
+ return Array.isArray(value) ? value.flatMap((entry) => typeof entry === "string" && entry.trim() ? [entry.trim()] : []) : [];
16
+ }
17
+ function rawObject(record) {
18
+ const raw = record.raw;
19
+ return raw && typeof raw === "object" && !Array.isArray(raw) ? raw : {};
20
+ }
21
+ function truncate(value, width) {
22
+ if (value.length <= width)
23
+ return value;
24
+ if (width <= 1)
25
+ return "\u2026";
26
+ return `${value.slice(0, width - 1)}\u2026`;
27
+ }
28
+ function pad(value, width) {
29
+ return value.length >= width ? value : `${value}${" ".repeat(width - value.length)}`;
30
+ }
31
+ function statusColor(status) {
32
+ const normalized = status.toLowerCase();
33
+ if (["completed", "merged", "closed", "done", "accepted", "pass", "selected", "approved"].includes(normalized))
34
+ return pc.green;
35
+ if (["failed", "needs_attention", "needs-attention", "blocked", "error", "rejected"].includes(normalized))
36
+ return pc.red;
37
+ if (["running", "reviewing", "validating", "in_progress", "in-progress", "remote"].includes(normalized))
38
+ return pc.cyan;
39
+ if (["ready", "open", "queued", "created", "preparing", "local", "pending"].includes(normalized))
40
+ return pc.yellow;
41
+ return pc.dim;
42
+ }
43
+ function compactDate(value) {
44
+ if (!value.trim())
45
+ return "";
46
+ const parsed = Date.parse(value);
47
+ if (!Number.isFinite(parsed))
48
+ return value;
49
+ return new Date(parsed).toISOString().replace("T", " ").replace(/\.\d{3}Z$/, "Z");
50
+ }
51
+ function compactValue(value) {
52
+ if (value === null || value === undefined)
53
+ return "";
54
+ if (typeof value === "string")
55
+ return value;
56
+ if (typeof value === "number" || typeof value === "boolean")
57
+ return String(value);
58
+ if (Array.isArray(value))
59
+ return value.map(compactValue).filter(Boolean).join(", ");
60
+ return JSON.stringify(value);
61
+ }
62
+ function firstString(record, keys, fallback = "") {
63
+ for (const key of keys) {
64
+ const value = stringField(record, key);
65
+ if (value)
66
+ return value;
67
+ }
68
+ return fallback;
69
+ }
70
+ function runIdOf(run) {
71
+ return firstString(run, ["runId", "id"], "(unknown-run)");
72
+ }
73
+ function taskIdOf(run) {
74
+ return firstString(run, ["taskId", "task", "task_id"]);
75
+ }
76
+ function runTitleOf(run) {
77
+ return firstString(run, ["title", "summary", "name"], taskIdOf(run) || "(untitled)");
78
+ }
79
+ function requestIdOf(entry) {
80
+ return firstString(entry, ["requestId", "id", "approvalId", "inputId"], "(unknown-request)");
81
+ }
82
+ function shouldUseClackOutput() {
83
+ return Boolean(process.stdout.isTTY) && process.env.RIG_CLI_PLAIN_HELP !== "1";
84
+ }
85
+ function printFormattedOutput(message, options = {}) {
86
+ if (!shouldUseClackOutput()) {
87
+ console.log(message);
88
+ return;
89
+ }
90
+ if (options.title)
91
+ note(message, options.title);
92
+ else
93
+ log.message(message);
94
+ }
95
+ function formatStatusPill(status) {
96
+ const label = status || "unknown";
97
+ return statusColor(label)(`\u25CF ${label}`);
98
+ }
99
+ function formatSection(title, subtitle) {
100
+ return `${pc.bold(pc.cyan("\u25C6"))} ${pc.bold(title)}${subtitle ? pc.dim(` \u2014 ${subtitle}`) : ""}`;
101
+ }
102
+ function formatSuccessCard(title, rows = []) {
103
+ const body = rows.filter(([, value]) => value !== undefined && value !== null && String(value).length > 0).map(([key, value]) => `${pc.dim("\u2502")} ${pc.dim(key.padEnd(12))} ${value}`);
104
+ return [formatSection(title), ...body].join(`
105
+ `);
106
+ }
107
+ function formatNextSteps(steps) {
108
+ if (steps.length === 0)
109
+ return [];
110
+ return [pc.bold("Next"), ...steps.map((step) => `${pc.dim("\u203A")} ${step}`)];
111
+ }
112
+ function formatTaskList(tasks, options = {}) {
113
+ if (options.raw)
114
+ return tasks.map((task) => JSON.stringify(task)).join(`
115
+ `);
116
+ if (tasks.length === 0)
117
+ return [formatSection("Tasks", "none found"), ...formatNextSteps(["Try `rig server status` to confirm the selected server.", "Relax filters or run `rig task run --title ... --initial-prompt ...` for ad hoc work."])].join(`
118
+ `);
119
+ const rows = tasks.map((task) => {
120
+ const raw = rawObject(task);
121
+ const id = stringField(task, "id", "<unknown>");
122
+ const status = stringField(task, "status", "unknown");
123
+ const title = stringField(task, "title", "Untitled task");
124
+ const source = stringField(task, "source", stringField(raw, "source", ""));
125
+ const labels = arrayField(task, "labels").length > 0 ? arrayField(task, "labels") : arrayField(raw, "labels");
126
+ return { id, status, title, source, labels };
127
+ });
128
+ const idWidth = Math.min(18, Math.max(4, ...rows.map((row) => row.id.length)));
129
+ const statusWidth = Math.min(16, Math.max(6, ...rows.map((row) => row.status.length)));
130
+ const header = `${pc.bold(pad("TASK", idWidth))} ${pc.bold(pad("STATUS", statusWidth))} ${pc.bold("TITLE")}`;
131
+ const body = rows.map((row) => {
132
+ const labels = row.labels.length > 0 ? pc.dim(` ${row.labels.slice(0, 4).map((label) => `#${label}`).join(" ")}`) : "";
133
+ const source = row.source ? pc.dim(` ${row.source}`) : "";
134
+ return [
135
+ pc.bold(pad(truncate(row.id, idWidth), idWidth)),
136
+ statusColor(row.status)(pad(truncate(row.status, statusWidth), statusWidth)),
137
+ `${row.title}${labels}${source}`
138
+ ].join(" ");
139
+ });
140
+ return [formatSection("Tasks", `${rows.length} shown`), header, ...body, "", ...formatNextSteps(["Run one: `rig task run <id>` or `rig task run --next`", "Attach later: `rig run attach <run-id> --follow`"])].join(`
141
+ `);
142
+ }
143
+ function formatTaskCard(task, options = {}) {
144
+ const raw = rawObject(task);
145
+ const id = stringField(task, "id", stringField(raw, "id", "<unknown>"));
146
+ const status = stringField(task, "status", stringField(raw, "status", "unknown"));
147
+ const title = stringField(task, "title", stringField(raw, "title", "Untitled task"));
148
+ const source = stringField(task, "source", stringField(raw, "source", ""));
149
+ const url = stringField(task, "url", stringField(raw, "url", ""));
150
+ const number = numberField(task, "number") ?? numberField(raw, "number");
151
+ const labels = arrayField(task, "labels").length > 0 ? arrayField(task, "labels") : arrayField(raw, "labels");
152
+ const assignees = arrayField(task, "assignees").length > 0 ? arrayField(task, "assignees") : arrayField(raw, "assignees");
153
+ const readiness = compactValue(task.readiness ?? raw.readiness);
154
+ const validators = compactValue(task.validators ?? raw.validators ?? task.validation ?? raw.validation);
155
+ const rows = [
156
+ ["task", pc.bold(id)],
157
+ ["status", formatStatusPill(status)],
158
+ ["title", title],
159
+ ["source", source],
160
+ ["number", number],
161
+ ["labels", labels.length ? labels.map((label) => `#${label}`).join(" ") : ""],
162
+ ["assignees", assignees.join(", ")],
163
+ ["readiness", readiness],
164
+ ["validators", validators],
165
+ ["url", url]
166
+ ];
167
+ return [
168
+ formatSuccessCard(options.title ?? (options.selected ? "Selected task" : "Task"), rows),
169
+ "",
170
+ ...formatNextSteps([`Start: \`rig task run ${id}\``, `Details: \`rig task show ${id} --raw\``])
171
+ ].join(`
172
+ `);
173
+ }
174
+ function formatTaskDetails(task) {
175
+ return formatTaskCard(task, { title: "Task details" });
176
+ }
177
+ function formatRunList(runs, options = {}) {
178
+ if (runs.length === 0) {
179
+ return [
180
+ formatSection("Runs", "none recorded"),
181
+ options.source === "server" ? pc.dim("No runs recorded on the selected Rig server.") : pc.dim("No runs recorded in .rig/runs."),
182
+ "",
183
+ ...formatNextSteps(["Start one: `rig task run --next`", "Check server: `rig server status`"])
184
+ ].join(`
185
+ `);
186
+ }
187
+ const rows = runs.map((run) => {
188
+ const runId = stringField(run, "runId", stringField(run, "id", "(unknown-run)"));
189
+ const status = stringField(run, "status", "unknown");
190
+ const taskId = stringField(run, "taskId", "");
191
+ const title = stringField(run, "title", taskId || "(untitled)");
192
+ const runtime = stringField(run, "runtimeAdapter", "");
193
+ return { runId, status, title, runtime };
194
+ });
195
+ const idWidth = Math.min(36, Math.max(6, ...rows.map((row) => row.runId.length)));
196
+ const statusWidth = Math.min(16, Math.max(6, ...rows.map((row) => row.status.length)));
197
+ const header = `${pc.bold(pad("RUN", idWidth))} ${pc.bold(pad("STATUS", statusWidth))} ${pc.bold("TITLE")}`;
198
+ const body = rows.map((row) => [
199
+ pc.bold(pad(truncate(row.runId, idWidth), idWidth)),
200
+ statusColor(row.status)(pad(truncate(row.status, statusWidth), statusWidth)),
201
+ `${row.title}${row.runtime ? pc.dim(` ${row.runtime}`) : ""}`
202
+ ].join(" "));
203
+ return [formatSection("Runs", options.source === "server" ? "selected server" : "local state"), header, ...body, "", ...formatNextSteps(["Follow live: `rig run attach <run-id> --follow`", "Details: `rig run show <run-id>`"])].join(`
204
+ `);
205
+ }
206
+ function formatSubmittedRun(input) {
207
+ const rows = [["run", pc.bold(input.runId)]];
208
+ if (input.task) {
209
+ const id = stringField(input.task, "id", "<unknown>");
210
+ const status = stringField(input.task, "status", "unknown");
211
+ const title = stringField(input.task, "title", "Untitled task");
212
+ rows.push(["task", `${pc.bold(id)} ${formatStatusPill(status)} ${title}`]);
213
+ }
214
+ const runtime = [input.runtimeAdapter || "pi", input.runtimeMode || "full-access", input.interactionMode || "default"].filter(Boolean).join(" \xB7 ");
215
+ rows.push(["runtime", runtime]);
216
+ return [
217
+ formatSuccessCard("Run submitted", rows),
218
+ "",
219
+ ...formatNextSteps([
220
+ `Attach: \`rig run attach ${input.runId} --follow\``,
221
+ `Inspect: \`rig run show ${input.runId}\``,
222
+ input.detached ? "Submitted detached; attach when you are ready." : "Interactive mode opens the enriched bundled Pi (native UI + Rig layers, worker brain)."
223
+ ])
224
+ ].join(`
225
+ `);
226
+ }
227
+ function formatRunCard(run, options = {}) {
228
+ const raw = rawObject(run);
229
+ const merged = { ...raw, ...run };
230
+ const runId = runIdOf(merged);
231
+ const status = firstString(merged, ["status"], "unknown");
232
+ const taskId = taskIdOf(merged);
233
+ const title = runTitleOf(merged);
234
+ const runtime = firstString(merged, ["runtimeAdapter", "runtime", "adapter"]);
235
+ const mode = firstString(merged, ["runtimeMode", "mode"]);
236
+ const interaction = firstString(merged, ["interactionMode"]);
237
+ const created = compactDate(firstString(merged, ["createdAt"]));
238
+ const started = compactDate(firstString(merged, ["startedAt"]));
239
+ const updated = compactDate(firstString(merged, ["updatedAt"]));
240
+ const completed = compactDate(firstString(merged, ["completedAt", "finishedAt"]));
241
+ const worktree = firstString(merged, ["worktreePath", "cwd", "projectRoot"]);
242
+ const piSession = merged.piSession && typeof merged.piSession === "object" && !Array.isArray(merged.piSession) ? firstString(merged.piSession, ["sessionId", "id"]) : "";
243
+ const timeline = Array.isArray(merged.timeline) ? merged.timeline.length : null;
244
+ const approvals = Array.isArray(merged.approvals) ? merged.approvals.length : null;
245
+ const inputs = Array.isArray(merged.userInputs) ? merged.userInputs.length : null;
246
+ const rows = [
247
+ ["run", pc.bold(runId)],
248
+ ["status", formatStatusPill(status)],
249
+ ["task", taskId],
250
+ ["title", title],
251
+ ["runtime", [runtime, mode, interaction].filter(Boolean).join(" \xB7 ")],
252
+ ["created", created],
253
+ ["started", started],
254
+ ["updated", updated],
255
+ ["completed", completed],
256
+ ["worktree", worktree],
257
+ ["pi", piSession],
258
+ ["timeline", timeline],
259
+ ["approvals", approvals],
260
+ ["inputs", inputs]
261
+ ];
262
+ return [
263
+ formatSuccessCard(options.title ?? "Run details", rows),
264
+ "",
265
+ ...formatNextSteps([`Follow live: \`rig run attach ${runId} --follow\``, `Raw payload: \`rig run show ${runId} --raw\``])
266
+ ].join(`
267
+ `);
268
+ }
269
+ function formatRunStatus(summary, options = {}) {
270
+ const activeRuns = summary.activeRuns ?? [];
271
+ const recentRuns = summary.recentRuns ?? [];
272
+ const lines = [formatSection("Run status", options.source === "server" ? "selected server" : "local state")];
273
+ lines.push("", pc.bold(`Active runs (${activeRuns.length})`));
274
+ if (activeRuns.length === 0) {
275
+ lines.push(pc.dim("No active runs."));
276
+ } else {
277
+ for (const run of activeRuns) {
278
+ lines.push(formatRunSummaryLine(run));
279
+ }
280
+ }
281
+ lines.push("", pc.bold(`Recent runs (${recentRuns.length})`));
282
+ if (recentRuns.length === 0) {
283
+ lines.push(pc.dim("No recent terminal runs."));
284
+ } else {
285
+ for (const run of recentRuns.slice(0, 10)) {
286
+ lines.push(formatRunSummaryLine(run));
287
+ }
288
+ }
289
+ lines.push("", ...formatNextSteps(["Start work: `rig task run --next`", "Attach: `rig run attach <run-id> --follow`", "Details: `rig run show <run-id>`"]));
290
+ return lines.join(`
291
+ `);
292
+ }
293
+ function formatRunSummaryLine(run) {
294
+ const record = run;
295
+ const runId = runIdOf(record);
296
+ const status = firstString(record, ["status"], "unknown");
297
+ const taskId = taskIdOf(record);
298
+ const title = runTitleOf(record);
299
+ const runtime = firstString(record, ["runtimeAdapter", "runtime", "adapter"]);
300
+ const descriptor = [taskId, title].filter(Boolean).join(" \xB7 ");
301
+ return `${pc.dim("\u2502")} ${pc.bold(runId)} ${formatStatusPill(status)} ${descriptor}${runtime ? pc.dim(` ${runtime}`) : ""}`;
302
+ }
303
+ function formatInboxList(kind, entries) {
304
+ const title = kind === "approvals" ? "Approval inbox" : "Input inbox";
305
+ if (entries.length === 0) {
306
+ return [
307
+ formatSection(title, "empty"),
308
+ pc.dim(kind === "approvals" ? "No pending approvals." : "No pending user-input requests."),
309
+ "",
310
+ ...formatNextSteps(["Check runs: `rig run status`", "Start work: `rig task run --next`"])
311
+ ].join(`
312
+ `);
313
+ }
314
+ const lines = [formatSection(title, `${entries.length} pending`)];
315
+ for (const entry of entries) {
316
+ const record = entry.record && typeof entry.record === "object" && !Array.isArray(entry.record) ? entry.record : entry;
317
+ const runId = firstString(entry, ["runId"], firstString(record, ["runId"]));
318
+ const taskId = firstString(entry, ["taskId"], firstString(record, ["taskId", "task"]));
319
+ const requestId = requestIdOf(record);
320
+ const status = firstString(record, ["status", "state"], "pending");
321
+ const prompt = firstString(record, ["prompt", "message", "reason", "title", "summary"], kind === "approvals" ? "Approval requested" : "Input requested");
322
+ lines.push(`${pc.dim("\u2502")} ${pc.bold(requestId)} ${formatStatusPill(status)} ${prompt}`);
323
+ lines.push(`${pc.dim("\u2502")} ${pc.dim("run ")} ${runId || "(unknown-run)"}${taskId ? pc.dim(` task ${taskId}`) : ""}`);
324
+ }
325
+ lines.push("", ...formatNextSteps(kind === "approvals" ? ["Resolve: `rig inbox approve --run <run-id> --request <request-id> --decision approve|reject`", "Rejoin: `rig run attach <run-id> --follow`"] : ["Respond: `rig inbox respond --run <run-id> --request <request-id> --answer key=value`", "Rejoin: `rig run attach <run-id> --follow`"]));
326
+ return lines.join(`
327
+ `);
328
+ }
329
+ function formatConnectionList(connections) {
330
+ const rows = [["local", { kind: "local", mode: "auto" }], ...Object.entries(connections)];
331
+ const aliasWidth = Math.min(24, Math.max(5, ...rows.map(([alias]) => alias.length)));
332
+ const lines = rows.map(([alias, connection]) => [
333
+ pc.bold(pad(truncate(alias, aliasWidth), aliasWidth)),
334
+ formatStatusPill(connection.kind),
335
+ connection.kind === "remote" ? connection.baseUrl ?? "" : connection.mode ?? "local"
336
+ ].join(" "));
337
+ return [formatSection("Rig servers", `${rows.length} available`), `${pc.bold(pad("ALIAS", aliasWidth))} ${pc.bold("KIND")} ${pc.bold("TARGET")}`, ...lines, "", ...formatNextSteps(["Select one: `rig server use <alias|local>`"])].join(`
338
+ `);
339
+ }
340
+ function formatConnectionStatus(selected, connections) {
341
+ const connection = selected === "local" ? { kind: "local", mode: "auto" } : connections[selected];
342
+ const target = !connection ? "not configured" : connection.kind === "remote" ? connection.baseUrl : "local";
343
+ return [
344
+ formatSection("Rig server", "selected for this repo"),
345
+ `${pc.dim("\u2502")} ${pc.dim("selected ")} ${pc.bold(selected)}`,
346
+ `${pc.dim("\u2502")} ${pc.dim("kind ")} ${formatStatusPill(connection?.kind ?? "unknown")}`,
347
+ `${pc.dim("\u2502")} ${pc.dim("target ")} ${target ?? "not configured"}`,
348
+ "",
349
+ ...formatNextSteps(["Change: `rig server use <alias|local>`", "List saved servers: `rig server list`"])
350
+ ].join(`
351
+ `);
352
+ }
353
+ export {
354
+ printFormattedOutput,
355
+ formatTaskList,
356
+ formatTaskDetails,
357
+ formatTaskCard,
358
+ formatSuccessCard,
359
+ formatSubmittedRun,
360
+ formatStatusPill,
361
+ formatSection,
362
+ formatRunStatus,
363
+ formatRunList,
364
+ formatRunCard,
365
+ formatNextSteps,
366
+ formatInboxList,
367
+ formatConnectionStatus,
368
+ formatConnectionList
369
+ };
@@ -8,8 +8,6 @@ import { dirname, resolve } from "path";
8
8
  import { EventBus } from "@rig/runtime/control-plane/runtime/events";
9
9
  import { CliError } from "@rig/runtime/control-plane/errors";
10
10
  import { evaluate, loadPolicy, resolveAction } from "@rig/runtime/control-plane/runtime/guard";
11
- import { PluginManager } from "@rig/runtime/control-plane/runtime/plugins";
12
- import { loadRuntimeContextFromEnv } from "@rig/runtime/control-plane/runtime/context";
13
11
  import { buildBinary } from "@rig/runtime/control-plane/runtime/isolation";
14
12
  import { CliError as CliError2 } from "@rig/runtime/control-plane/errors";
15
13
 
@@ -92,7 +90,8 @@ function readRepoConnection(projectRoot) {
92
90
  return {
93
91
  selected,
94
92
  project: typeof record.project === "string" ? record.project : undefined,
95
- linkedAt: typeof record.linkedAt === "string" ? record.linkedAt : undefined
93
+ linkedAt: typeof record.linkedAt === "string" ? record.linkedAt : undefined,
94
+ serverProjectRoot: typeof record.serverProjectRoot === "string" && record.serverProjectRoot.trim() ? record.serverProjectRoot.trim() : undefined
96
95
  };
97
96
  }
98
97
  function writeRepoConnection(projectRoot, state) {
@@ -103,15 +102,22 @@ function resolveSelectedConnection(projectRoot, options = {}) {
103
102
  if (!repo)
104
103
  return null;
105
104
  if (repo.selected === "local")
106
- return { alias: "local", connection: { kind: "local", mode: "auto" } };
105
+ return { alias: "local", connection: { kind: "local", mode: "auto" }, serverProjectRoot: repo.serverProjectRoot };
107
106
  const global = readGlobalConnections(options);
108
107
  const connection = global.connections[repo.selected];
109
108
  if (!connection) {
110
- throw new CliError2(`Selected Rig connection "${repo.selected}" was not found. Run \`rig connect list\` or \`rig connect use local\`.`, 1);
109
+ throw new CliError2(`Selected Rig server "${repo.selected}" was not found. Run \`rig server list\` or \`rig server use local\`.`, 1);
111
110
  }
112
- return { alias: repo.selected, connection };
111
+ return { alias: repo.selected, connection, serverProjectRoot: repo.serverProjectRoot };
112
+ }
113
+ function writeRepoServerProjectRoot(projectRoot, serverProjectRoot) {
114
+ const repo = readRepoConnection(projectRoot);
115
+ if (!repo)
116
+ return;
117
+ writeRepoConnection(projectRoot, { ...repo, serverProjectRoot });
113
118
  }
114
119
  export {
120
+ writeRepoServerProjectRoot,
115
121
  writeRepoConnection,
116
122
  writeGlobalConnections,
117
123
  upsertGlobalConnection,
@@ -9,8 +9,6 @@ import { resolve as resolve4 } from "path";
9
9
  import { EventBus } from "@rig/runtime/control-plane/runtime/events";
10
10
  import { CliError } from "@rig/runtime/control-plane/errors";
11
11
  import { evaluate, loadPolicy, resolveAction } from "@rig/runtime/control-plane/runtime/guard";
12
- import { PluginManager } from "@rig/runtime/control-plane/runtime/plugins";
13
- import { loadRuntimeContextFromEnv } from "@rig/runtime/control-plane/runtime/context";
14
12
  import { buildBinary } from "@rig/runtime/control-plane/runtime/isolation";
15
13
  import { CliError as CliError2 } from "@rig/runtime/control-plane/errors";
16
14
 
@@ -42,6 +40,11 @@ function readJsonFile(path) {
42
40
  throw new CliError2(`Invalid Rig connection state at ${path}: ${error instanceof Error ? error.message : String(error)}`, 1);
43
41
  }
44
42
  }
43
+ function writeJsonFile(path, value) {
44
+ mkdirSync(dirname(path), { recursive: true });
45
+ writeFileSync(path, `${JSON.stringify(value, null, 2)}
46
+ `, "utf8");
47
+ }
45
48
  function normalizeConnection(value) {
46
49
  if (!value || typeof value !== "object" || Array.isArray(value))
47
50
  return null;
@@ -82,29 +85,38 @@ function readRepoConnection(projectRoot) {
82
85
  return {
83
86
  selected,
84
87
  project: typeof record.project === "string" ? record.project : undefined,
85
- linkedAt: typeof record.linkedAt === "string" ? record.linkedAt : undefined
88
+ linkedAt: typeof record.linkedAt === "string" ? record.linkedAt : undefined,
89
+ serverProjectRoot: typeof record.serverProjectRoot === "string" && record.serverProjectRoot.trim() ? record.serverProjectRoot.trim() : undefined
86
90
  };
87
91
  }
92
+ function writeRepoConnection(projectRoot, state) {
93
+ writeJsonFile(resolveRepoConnectionPath(projectRoot), state);
94
+ }
88
95
  function resolveSelectedConnection(projectRoot, options = {}) {
89
96
  const repo = readRepoConnection(projectRoot);
90
97
  if (!repo)
91
98
  return null;
92
99
  if (repo.selected === "local")
93
- return { alias: "local", connection: { kind: "local", mode: "auto" } };
100
+ return { alias: "local", connection: { kind: "local", mode: "auto" }, serverProjectRoot: repo.serverProjectRoot };
94
101
  const global = readGlobalConnections(options);
95
102
  const connection = global.connections[repo.selected];
96
103
  if (!connection) {
97
- throw new CliError2(`Selected Rig connection "${repo.selected}" was not found. Run \`rig connect list\` or \`rig connect use local\`.`, 1);
104
+ throw new CliError2(`Selected Rig server "${repo.selected}" was not found. Run \`rig server list\` or \`rig server use local\`.`, 1);
98
105
  }
99
- return { alias: repo.selected, connection };
106
+ return { alias: repo.selected, connection, serverProjectRoot: repo.serverProjectRoot };
107
+ }
108
+ function writeRepoServerProjectRoot(projectRoot, serverProjectRoot) {
109
+ const repo = readRepoConnection(projectRoot);
110
+ if (!repo)
111
+ return;
112
+ writeRepoConnection(projectRoot, { ...repo, serverProjectRoot });
100
113
  }
101
114
 
102
115
  // packages/cli/src/commands/_server-client.ts
103
- import { spawnSync } from "child_process";
104
116
  import { existsSync as existsSync2, readFileSync as readFileSync2 } from "fs";
105
117
  import { resolve as resolve2 } from "path";
106
118
  import { ensureLocalRigServerConnection } from "@rig/runtime/local-server";
107
- var cachedGitHubBearerToken;
119
+ var scopedGitHubBearerTokens = new Map;
108
120
  function cleanToken(value) {
109
121
  const trimmed = value?.trim();
110
122
  return trimmed ? trimmed : null;
@@ -121,41 +133,47 @@ function readPrivateRemoteSessionToken(projectRoot) {
121
133
  }
122
134
  }
123
135
  function readGitHubBearerTokenForRemote(projectRoot) {
124
- if (cachedGitHubBearerToken !== undefined)
125
- return cachedGitHubBearerToken;
136
+ const scopedKey = resolve2(projectRoot);
137
+ if (scopedGitHubBearerTokens.has(scopedKey))
138
+ return scopedGitHubBearerTokens.get(scopedKey) ?? null;
126
139
  const privateSession = readPrivateRemoteSessionToken(projectRoot);
127
- if (privateSession) {
128
- cachedGitHubBearerToken = privateSession;
129
- return cachedGitHubBearerToken;
130
- }
131
- const envToken = cleanToken(process.env.RIG_GITHUB_TOKEN) ?? cleanToken(process.env.GITHUB_TOKEN) ?? cleanToken(process.env.GH_TOKEN);
132
- if (envToken) {
133
- cachedGitHubBearerToken = envToken;
134
- return cachedGitHubBearerToken;
140
+ if (privateSession)
141
+ return privateSession;
142
+ return cleanToken(process.env.RIG_SERVER_AUTH_TOKEN) ?? cleanToken(process.env.RIG_REMOTE_AUTH_TOKEN);
143
+ }
144
+ function readStoredGitHubAuthToken(projectRoot) {
145
+ const path = resolve2(projectRoot, ".rig", "state", "github-auth.json");
146
+ if (!existsSync2(path))
147
+ return null;
148
+ try {
149
+ const parsed = JSON.parse(readFileSync2(path, "utf8"));
150
+ return cleanToken(typeof parsed.token === "string" ? parsed.token : undefined);
151
+ } catch {
152
+ return null;
135
153
  }
136
- const result = spawnSync("gh", ["auth", "token"], {
137
- encoding: "utf8",
138
- timeout: 5000,
139
- stdio: ["ignore", "pipe", "ignore"]
140
- });
141
- cachedGitHubBearerToken = result.status === 0 ? cleanToken(result.stdout) : null;
142
- return cachedGitHubBearerToken;
154
+ }
155
+ function readLocalConnectionFallbackToken(projectRoot) {
156
+ return readGitHubBearerTokenForRemote(projectRoot) ?? cleanToken(process.env.RIG_GITHUB_TOKEN) ?? readStoredGitHubAuthToken(projectRoot);
143
157
  }
144
158
  async function ensureServerForCli(projectRoot) {
145
159
  try {
146
160
  const selected = resolveSelectedConnection(projectRoot);
147
161
  if (selected?.connection.kind === "remote") {
162
+ const authToken = readGitHubBearerTokenForRemote(projectRoot);
163
+ const serverProjectRoot = selected.serverProjectRoot ?? await backfillRemoteServerProjectRoot(projectRoot, selected.connection.baseUrl, authToken);
148
164
  return {
149
165
  baseUrl: selected.connection.baseUrl,
150
- authToken: readGitHubBearerTokenForRemote(projectRoot),
151
- connectionKind: "remote"
166
+ authToken,
167
+ connectionKind: "remote",
168
+ serverProjectRoot
152
169
  };
153
170
  }
154
171
  const connection = await ensureLocalRigServerConnection(projectRoot);
155
172
  return {
156
173
  baseUrl: connection.baseUrl,
157
- authToken: connection.authToken,
158
- connectionKind: "local"
174
+ authToken: connection.authToken ?? readLocalConnectionFallbackToken(projectRoot),
175
+ connectionKind: "local",
176
+ serverProjectRoot: resolve2(projectRoot)
159
177
  };
160
178
  } catch (error) {
161
179
  if (error instanceof Error) {
@@ -164,6 +182,29 @@ async function ensureServerForCli(projectRoot) {
164
182
  throw error;
165
183
  }
166
184
  }
185
+ async function backfillRemoteServerProjectRoot(projectRoot, baseUrl, authToken) {
186
+ const repo = readRepoConnection(projectRoot);
187
+ const slug = repo?.project?.trim();
188
+ if (!slug)
189
+ return null;
190
+ try {
191
+ const response = await fetch(`${baseUrl}/api/projects/${encodeURIComponent(slug)}`, {
192
+ headers: mergeHeaders(undefined, authToken)
193
+ });
194
+ if (!response.ok)
195
+ return null;
196
+ const payload = await response.json();
197
+ const project = payload.project && typeof payload.project === "object" && !Array.isArray(payload.project) ? payload.project : null;
198
+ const checkouts = Array.isArray(project?.checkouts) ? project.checkouts : [];
199
+ const latestCheckout = [...checkouts].reverse().find((entry) => Boolean(entry && typeof entry === "object" && !Array.isArray(entry) && typeof entry.path === "string"));
200
+ const path = typeof latestCheckout?.path === "string" && latestCheckout.path.trim() ? latestCheckout.path.trim() : null;
201
+ if (path)
202
+ writeRepoServerProjectRoot(projectRoot, path);
203
+ return path;
204
+ } catch {
205
+ return null;
206
+ }
207
+ }
167
208
  function mergeHeaders(headers, authToken) {
168
209
  const merged = new Headers(headers);
169
210
  if (authToken) {
@@ -188,9 +229,12 @@ function diagnosticMessage(payload) {
188
229
  }
189
230
  async function requestServerJson(context, pathname, init = {}) {
190
231
  const server = await ensureServerForCli(context.projectRoot);
232
+ const headers = mergeHeaders(init.headers, server.authToken);
233
+ if (server.serverProjectRoot)
234
+ headers.set("x-rig-project-root", server.serverProjectRoot);
191
235
  const response = await fetch(`${server.baseUrl}${pathname}`, {
192
236
  ...init,
193
- headers: mergeHeaders(init.headers, server.authToken)
237
+ headers
194
238
  });
195
239
  const text = await response.text();
196
240
  const payload = text.trim().length > 0 ? (() => {
@@ -222,7 +266,8 @@ async function loadRigConfigOrNull(projectRoot) {
222
266
  import { existsSync as existsSync3, readFileSync as readFileSync3, rmSync } from "fs";
223
267
  import { homedir as homedir2 } from "os";
224
268
  import { resolve as resolve3 } from "path";
225
- var PI_RIG_PACKAGE_NAME = "@rig/pi-rig";
269
+ var PI_RIG_PACKAGE_NAME = "@h-rig/pi-rig";
270
+ var LEGACY_PI_RIG_PACKAGE_NAME = "@rig/pi-rig";
226
271
  async function defaultCommandRunner(command, options = {}) {
227
272
  const proc = Bun.spawn(command, { cwd: options.cwd, stdout: "pipe", stderr: "pipe" });
228
273
  const [stdout, stderr, exitCode] = await Promise.all([
@@ -241,7 +286,7 @@ function resolvePiHomeDir(inputHomeDir) {
241
286
  function piListContainsPiRig(output) {
242
287
  return output.split(/\r?\n/).some((line) => {
243
288
  const normalized = line.trim();
244
- return normalized.includes(PI_RIG_PACKAGE_NAME) || /(?:^|[\\/])packages[\\/]pi-rig(?:$|\s)/.test(normalized);
289
+ return normalized.includes(PI_RIG_PACKAGE_NAME) || normalized.includes(LEGACY_PI_RIG_PACKAGE_NAME) || /(?:^|[\\/])packages[\\/]pi-rig(?:$|\s)/.test(normalized);
245
290
  });
246
291
  }
247
292
  async function safeRun(runner, command, options) {
@@ -417,7 +462,7 @@ async function runRigDoctorChecks(options) {
417
462
  const taskSourceKind = config?.taskSource?.kind;
418
463
  checks.push(taskSourceKind ? check("task-source", "task source configured", "pass", taskSourceKind) : check("task-source", "task source configured", "fail", "missing taskSource", "Configure taskSource in rig.config.ts."));
419
464
  const repo = readRepoConnection(projectRoot);
420
- checks.push(repo ? check("project-link", "repo selected Rig connection", repo.project ? "pass" : "warn", `${repo.selected}${repo.project ? ` -> ${repo.project}` : ""}`, "Run `rig init --yes --repo owner/repo` to link this checkout to a GitHub repo slug.") : check("project-link", "repo selected Rig connection", "fail", "missing .rig/state/connection.json", "Run `rig init` or `rig connect use <alias|local>`."));
465
+ checks.push(repo ? check("project-link", "repo selected Rig server", repo.project ? "pass" : "warn", `${repo.selected}${repo.project ? ` -> ${repo.project}` : ""}`, "Run `rig init --yes --repo owner/repo` to link this checkout to a GitHub repo slug.") : check("project-link", "repo selected Rig server", "fail", "missing .rig/state/connection.json", "Run `rig init` or `rig server use <alias|local>`."));
421
466
  const selected = (() => {
422
467
  try {
423
468
  return resolveSelectedConnection(projectRoot);
@@ -425,7 +470,7 @@ async function runRigDoctorChecks(options) {
425
470
  return null;
426
471
  }
427
472
  })();
428
- checks.push(selected ? check("connection", "selected server connection", "pass", selected.connection.kind === "remote" ? selected.connection.baseUrl : "local auto") : check("connection", "selected server connection", repo ? "fail" : "warn", repo ? "selected alias is missing" : "will auto-start local server", repo ? "Run `rig connect list` and `rig connect use <alias|local>`." : undefined));
473
+ checks.push(selected ? check("connection", "selected server connection", "pass", selected.connection.kind === "remote" ? selected.connection.baseUrl : "local auto") : check("connection", "selected server", repo ? "fail" : "warn", repo ? "selected alias is missing" : "will auto-start local server", repo ? "Run `rig server list` and `rig server use <alias|local>`." : undefined));
429
474
  let server = null;
430
475
  try {
431
476
  server = await (options.resolveServer ?? ensureServerForCli)(projectRoot);