@h-rig/cli 0.0.6-alpha.5 → 0.0.6-alpha.50

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 (49) hide show
  1. package/dist/bin/rig.js +4072 -1204
  2. package/dist/src/commands/_cli-format.js +369 -0
  3. package/dist/src/commands/_connection-state.js +12 -6
  4. package/dist/src/commands/_doctor-checks.js +65 -34
  5. package/dist/src/commands/_help-catalog.js +445 -0
  6. package/dist/src/commands/_operator-surface.js +220 -0
  7. package/dist/src/commands/_operator-view.js +1109 -63
  8. package/dist/src/commands/_parsers.js +0 -2
  9. package/dist/src/commands/_pi-frontend.js +1066 -0
  10. package/dist/src/commands/_pi-install.js +4 -3
  11. package/dist/src/commands/_pi-remote-session.js +757 -0
  12. package/dist/src/commands/_pi-worker-bridge-extension.js +820 -0
  13. package/dist/src/commands/_policy.js +0 -2
  14. package/dist/src/commands/_preflight.js +84 -116
  15. package/dist/src/commands/_run-driver-helpers.js +2 -2
  16. package/dist/src/commands/_server-client.js +211 -48
  17. package/dist/src/commands/_snapshot-upload.js +60 -30
  18. package/dist/src/commands/_spinner.js +63 -0
  19. package/dist/src/commands/_task-picker.js +44 -16
  20. package/dist/src/commands/agent.js +8 -9
  21. package/dist/src/commands/browser.js +4 -6
  22. package/dist/src/commands/connect.js +134 -26
  23. package/dist/src/commands/dist.js +4 -6
  24. package/dist/src/commands/doctor.js +65 -34
  25. package/dist/src/commands/github.js +62 -32
  26. package/dist/src/commands/inbox.js +396 -31
  27. package/dist/src/commands/init.js +342 -88
  28. package/dist/src/commands/inspect.js +282 -23
  29. package/dist/src/commands/inspector.js +2 -4
  30. package/dist/src/commands/pi.js +168 -0
  31. package/dist/src/commands/plugin.js +81 -22
  32. package/dist/src/commands/profile-and-review.js +8 -10
  33. package/dist/src/commands/queue.js +2 -3
  34. package/dist/src/commands/remote.js +18 -20
  35. package/dist/src/commands/repo-git-harness.js +6 -8
  36. package/dist/src/commands/run.js +1407 -130
  37. package/dist/src/commands/server.js +266 -40
  38. package/dist/src/commands/setup.js +69 -44
  39. package/dist/src/commands/task-report-bug.js +5 -7
  40. package/dist/src/commands/task-run-driver.js +662 -70
  41. package/dist/src/commands/task.js +1846 -259
  42. package/dist/src/commands/test.js +3 -5
  43. package/dist/src/commands/workspace.js +4 -6
  44. package/dist/src/commands.js +4054 -1180
  45. package/dist/src/index.js +4065 -1200
  46. package/dist/src/launcher.js +5 -3
  47. package/dist/src/report-bug.js +3 -3
  48. package/dist/src/runner.js +5 -19
  49. package/package.json +7 -4
@@ -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,33 @@ 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;
135
- }
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;
140
+ if (privateSession)
141
+ return privateSession;
142
+ return cleanToken(process.env.RIG_SERVER_AUTH_TOKEN) ?? cleanToken(process.env.RIG_REMOTE_AUTH_TOKEN);
143
143
  }
144
144
  async function ensureServerForCli(projectRoot) {
145
145
  try {
146
146
  const selected = resolveSelectedConnection(projectRoot);
147
147
  if (selected?.connection.kind === "remote") {
148
+ const authToken = readGitHubBearerTokenForRemote(projectRoot);
149
+ const serverProjectRoot = selected.serverProjectRoot ?? await backfillRemoteServerProjectRoot(projectRoot, selected.connection.baseUrl, authToken);
148
150
  return {
149
151
  baseUrl: selected.connection.baseUrl,
150
- authToken: readGitHubBearerTokenForRemote(projectRoot),
151
- connectionKind: "remote"
152
+ authToken,
153
+ connectionKind: "remote",
154
+ serverProjectRoot
152
155
  };
153
156
  }
154
157
  const connection = await ensureLocalRigServerConnection(projectRoot);
155
158
  return {
156
159
  baseUrl: connection.baseUrl,
157
160
  authToken: connection.authToken,
158
- connectionKind: "local"
161
+ connectionKind: "local",
162
+ serverProjectRoot: resolve2(projectRoot)
159
163
  };
160
164
  } catch (error) {
161
165
  if (error instanceof Error) {
@@ -164,6 +168,29 @@ async function ensureServerForCli(projectRoot) {
164
168
  throw error;
165
169
  }
166
170
  }
171
+ async function backfillRemoteServerProjectRoot(projectRoot, baseUrl, authToken) {
172
+ const repo = readRepoConnection(projectRoot);
173
+ const slug = repo?.project?.trim();
174
+ if (!slug)
175
+ return null;
176
+ try {
177
+ const response = await fetch(`${baseUrl}/api/projects/${encodeURIComponent(slug)}`, {
178
+ headers: mergeHeaders(undefined, authToken)
179
+ });
180
+ if (!response.ok)
181
+ return null;
182
+ const payload = await response.json();
183
+ const project = payload.project && typeof payload.project === "object" && !Array.isArray(payload.project) ? payload.project : null;
184
+ const checkouts = Array.isArray(project?.checkouts) ? project.checkouts : [];
185
+ const latestCheckout = [...checkouts].reverse().find((entry) => Boolean(entry && typeof entry === "object" && !Array.isArray(entry) && typeof entry.path === "string"));
186
+ const path = typeof latestCheckout?.path === "string" && latestCheckout.path.trim() ? latestCheckout.path.trim() : null;
187
+ if (path)
188
+ writeRepoServerProjectRoot(projectRoot, path);
189
+ return path;
190
+ } catch {
191
+ return null;
192
+ }
193
+ }
167
194
  function mergeHeaders(headers, authToken) {
168
195
  const merged = new Headers(headers);
169
196
  if (authToken) {
@@ -188,9 +215,12 @@ function diagnosticMessage(payload) {
188
215
  }
189
216
  async function requestServerJson(context, pathname, init = {}) {
190
217
  const server = await ensureServerForCli(context.projectRoot);
218
+ const headers = mergeHeaders(init.headers, server.authToken);
219
+ if (server.serverProjectRoot)
220
+ headers.set("x-rig-project-root", server.serverProjectRoot);
191
221
  const response = await fetch(`${server.baseUrl}${pathname}`, {
192
222
  ...init,
193
- headers: mergeHeaders(init.headers, server.authToken)
223
+ headers
194
224
  });
195
225
  const text = await response.text();
196
226
  const payload = text.trim().length > 0 ? (() => {
@@ -222,7 +252,8 @@ async function loadRigConfigOrNull(projectRoot) {
222
252
  import { existsSync as existsSync3, readFileSync as readFileSync3, rmSync } from "fs";
223
253
  import { homedir as homedir2 } from "os";
224
254
  import { resolve as resolve3 } from "path";
225
- var PI_RIG_PACKAGE_NAME = "@rig/pi-rig";
255
+ var PI_RIG_PACKAGE_NAME = "@h-rig/pi-rig";
256
+ var LEGACY_PI_RIG_PACKAGE_NAME = "@rig/pi-rig";
226
257
  async function defaultCommandRunner(command, options = {}) {
227
258
  const proc = Bun.spawn(command, { cwd: options.cwd, stdout: "pipe", stderr: "pipe" });
228
259
  const [stdout, stderr, exitCode] = await Promise.all([
@@ -241,7 +272,7 @@ function resolvePiHomeDir(inputHomeDir) {
241
272
  function piListContainsPiRig(output) {
242
273
  return output.split(/\r?\n/).some((line) => {
243
274
  const normalized = line.trim();
244
- return normalized.includes(PI_RIG_PACKAGE_NAME) || /(?:^|[\\/])packages[\\/]pi-rig(?:$|\s)/.test(normalized);
275
+ return normalized.includes(PI_RIG_PACKAGE_NAME) || normalized.includes(LEGACY_PI_RIG_PACKAGE_NAME) || /(?:^|[\\/])packages[\\/]pi-rig(?:$|\s)/.test(normalized);
245
276
  });
246
277
  }
247
278
  async function safeRun(runner, command, options) {
@@ -417,7 +448,7 @@ async function runRigDoctorChecks(options) {
417
448
  const taskSourceKind = config?.taskSource?.kind;
418
449
  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
450
  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>`."));
451
+ 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
452
  const selected = (() => {
422
453
  try {
423
454
  return resolveSelectedConnection(projectRoot);
@@ -425,7 +456,7 @@ async function runRigDoctorChecks(options) {
425
456
  return null;
426
457
  }
427
458
  })();
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));
459
+ 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
460
  let server = null;
430
461
  try {
431
462
  server = await (options.resolveServer ?? ensureServerForCli)(projectRoot);