nolo-cli 0.1.6 → 0.1.8

package/tui/readlineWorkspace.ts

@@ -1,6 +1,8 @@
 import { createInterface } from "node:readline";
 import { stdin as defaultInput, stdout as defaultOutput } from "node:process";
 import { runAgentTurn, type RunAgentTurnResult } from "../client/agentRun";
+import { compactDialog, type CompactDialogResult } from "../client/compactDialog";
+import { runSelfUpdate } from "../updateCommands";
 import {
   createInitialTuiState,
   handleTuiInput,
@@ -10,12 +12,22 @@ import {
   type TuiState,
 } from "./session";
 
+export type SelfUpdater = (
+  output: NodeJS.WritableStream
+) => Promise<number>;
+
 type WorkspaceOptions = {
   scriptDir: string;
   env?: NodeJS.ProcessEnv;
   input?: NodeJS.ReadableStream;
   output?: NodeJS.WritableStream;
   agentRunner?: typeof runAgentTurn;
+  compactRunner?: (options: {
+    serverUrl: string;
+    authToken: string;
+    dialogId: string;
+  }) => Promise<CompactDialogResult>;
+  selfUpdater?: SelfUpdater;
 };
 
 async function runAgentChat(
@@ -43,6 +55,8 @@ export async function startTuiWorkspace(options: WorkspaceOptions) {
   let state = createInitialTuiState(options.env ?? process.env);
   const input = options.input ?? defaultInput;
   const output = options.output ?? defaultOutput;
+  const selfUpdater: SelfUpdater =
+    options.selfUpdater ?? ((target) => runSelfUpdate({ output: target }));
   const rl = createInterface({ input, output });
 
   output.write(renderWelcome(state));
@@ -62,6 +76,42 @@ export async function startTuiWorkspace(options: WorkspaceOptions) {
         break;
       }
 
+      if (result.action?.type === "compact") {
+        const runner = options.compactRunner ?? compactDialog;
+        const authToken =
+          options.env?.AUTH_TOKEN ?? options.env?.AUTH ?? options.env?.BENCHMARK_AUTH_TOKEN ?? "";
+        try {
+          const compactResult = await runner({
+            serverUrl: state.serverUrl,
+            authToken,
+            dialogId: result.action.dialogId,
+          });
+          state = {
+            ...state,
+            dialogId: compactResult.dialogId,
+            dialogLabel: compactResult.dialogId,
+          };
+        } catch (error: any) {
+          output.write(
+            `[nolo] Compact failed: ${error?.message ?? String(error)}\n`
+          );
+        }
+      }
+
+      if (result.action?.type === "self-update") {
+        try {
+          const exitCode = await selfUpdater(output);
+          if (exitCode === 0) {
+            output.write("Update finished. Restart nolo to use the new version.\n");
+          } else {
+            output.write("Update failed. Fix the npm error above, then run /update again or use nolo update.\n");
+          }
+        } catch (error) {
+          output.write(`${error instanceof Error ? error.message : String(error)}\n`);
+          output.write("Update failed. Fix the npm error above, then run /update again or use nolo update.\n");
+        }
+      }
+
       if (result.action?.type === "chat") {
         const runResult = await runAgentChat(
           options.scriptDir,

package/tui/session.ts

@@ -1,5 +1,7 @@
+import { DEFAULT_NOLO_SERVER_URL } from "../defaultServer";
+
 export const DEFAULT_TUI_AGENT_KEY = "agent-pub-01NOLOAPPBLD000000019KCKT0";
-export const DEFAULT_TUI_SERVER_URL = "http://127.0.0.1:38123";
+export const DEFAULT_TUI_SERVER_URL = DEFAULT_NOLO_SERVER_URL;
 
 const KNOWN_AGENTS: Array<{
   name: string;
@@ -51,6 +53,13 @@ export type TuiAction =
       agentKey: string;
       continueDialogId?: string;
     }
+  | {
+      type: "compact";
+      dialogId: string;
+    }
+  | {
+      type: "self-update";
+    }
   | {
       type: "exit";
     };
@@ -73,7 +82,10 @@ export function createInitialTuiState(env: EnvLike = process.env): TuiState {
     dialogId: env.NOLO_DIALOG_ID?.trim() || undefined,
     dialogLabel: env.NOLO_DIALOG?.trim() || "new",
     profileName: env.NOLO_PROFILE?.trim() || "local",
-    serverUrl: (env.NOLO_SERVER || env.BASE_URL || DEFAULT_TUI_SERVER_URL).replace(/\/+$/, ""),
+    serverUrl: (env.NOLO_SERVER || env.BASE_URL || DEFAULT_TUI_SERVER_URL).replace(
+      /\/+$/,
+      ""
+    ),
     cliVersion: env.NOLO_CLI_VERSION?.trim() || undefined,
     attachedDocs: [],
   };
@@ -117,6 +129,8 @@ export function renderTuiHelp() {
     "Commands:",
     "  /help                 Show this help",
     "  /new                  Start a fresh dialog",
+    "  /compact              Compact current dialog and fork a new one",
+    "  /context              Show workspace context and next actions",
     "  /agent                Show the current agent",
     "  /agents               List built-in agent shortcuts",
     "  /switch <agent>       Switch the current agent",
@@ -126,6 +140,7 @@ export function renderTuiHelp() {
     "  /customize            Describe how you want to tune nolo",
     "  /login                Show login/profile hint",
     "  /profile              Show active profile",
+    "  /update               Update the global nolo install",
     "  /version              Show version/update hint",
     "  /exit                 Leave the workspace",
     "",
@@ -133,6 +148,26 @@ export function renderTuiHelp() {
   ].join("\n");
 }
 
+export function renderContextPanel(state: TuiState) {
+  const docs = state.attachedDocs.length
+    ? state.attachedDocs.join(", ")
+    : "none";
+  return [
+    "Workspace context",
+    "-----------------",
+    `agent   ${state.agentName}`,
+    `dialog  ${resolveDialogLabel(state)}`,
+    `docs    ${docs}`,
+    `profile ${state.profileName}`,
+    `server  ${state.serverUrl}`,
+    "",
+    "Next:",
+    "  /agents              see specialist shortcuts",
+    "  /doc attach <doc>    add working context",
+    "  /new                 start a clean dialog",
+  ].join("\n");
+}
+
 export function renderKnownAgents() {
   return [
     "Agents:",
@@ -184,6 +219,9 @@ export function handleTuiInput(input: string, state: TuiState): TuiInputResult {
   switch (command) {
     case "/help":
       return { nextState: state, output: renderTuiHelp() };
+    case "/context":
+    case "/ctx":
+      return { nextState: state, output: renderContextPanel(state) };
     case "/exit":
     case "/quit":
       return { nextState: state, output: "Bye.", action: { type: "exit" } };
@@ -197,6 +235,24 @@ export function handleTuiInput(input: string, state: TuiState): TuiInputResult {
         },
         output: "Started a fresh dialog.",
       };
+    case "/compact":
+      if (argText) {
+        return {
+          nextState: state,
+          output: `Unknown command: ${trimmed}\n\n${renderTuiHelp()}`,
+        };
+      }
+      if (!state.dialogId) {
+        return {
+          nextState: state,
+          output: "Current dialog: new (nothing to compact yet)",
+        };
+      }
+      return {
+        nextState: state,
+        output: "Compacting current dialog...",
+        action: { type: "compact", dialogId: state.dialogId },
+      };
     case "/agent":
       return {
         nextState: state,
@@ -278,6 +334,12 @@ export function handleTuiInput(input: string, state: TuiState): TuiInputResult {
         nextState: state,
         output: `Profile: ${state.profileName}`,
       };
+    case "/update":
+      return {
+        nextState: state,
+        output: "Starting self-update...",
+        action: { type: "self-update" },
+      };
     case "/version":
       return {
         nextState: state,

package/updateCommands.ts

@@ -15,6 +15,13 @@ type DoctorInfo = {
   profileName: string;
 };
 
+type RunSelfUpdateOptions = {
+  output?: NodeJS.WritableStream;
+  spawn?: typeof Bun.spawn;
+};
+
+type SpawnOutputChunk = string | ArrayBuffer | ArrayBufferView;
+
 const CLI_DIR = dirname(fileURLToPath(import.meta.url));
 const PACKAGE_JSON_PATH = join(CLI_DIR, "package.json");
 
@@ -53,17 +60,75 @@ export function buildCliDoctorText(info: DoctorInfo) {
   ].join("\n");
 }
 
-export async function runSelfUpdate(output: NodeJS.WritableStream = process.stdout) {
+function isRunSelfUpdateOptions(
+  value: NodeJS.WritableStream | RunSelfUpdateOptions
+): value is RunSelfUpdateOptions {
+  return (
+    typeof value === "object" &&
+    value !== null &&
+    ("output" in value || "spawn" in value)
+  );
+}
+
+async function forwardSpawnOutput(
+  stream: AsyncIterable<SpawnOutputChunk> | undefined,
+  output: NodeJS.WritableStream
+) {
+  if (!stream) {
+    return;
+  }
+
+  for await (const chunk of stream) {
+    output.write(normalizeSpawnChunk(chunk));
+  }
+}
+
+function normalizeSpawnChunk(
+  chunk: SpawnOutputChunk
+): string | Uint8Array<ArrayBufferLike> {
+  if (typeof chunk === "string" || chunk instanceof Uint8Array) {
+    return chunk;
+  }
+
+  if (chunk instanceof ArrayBuffer) {
+    return new Uint8Array(chunk);
+  }
+
+  return new Uint8Array(chunk.buffer, chunk.byteOffset, chunk.byteLength);
+}
+
+export async function runSelfUpdate(
+  outputOrOptions?: NodeJS.WritableStream | RunSelfUpdateOptions
+) {
+  const options =
+    outputOrOptions === undefined
+      ? {}
+      : isRunSelfUpdateOptions(outputOrOptions)
+        ? outputOrOptions
+        : { output: outputOrOptions };
+  const output = options.output ?? process.stdout;
+  const spawn = options.spawn ?? Bun.spawn;
   const command = buildSelfUpdateCommand();
   output.write(`Updating nolo with: ${command.join(" ")}\n`);
 
-  const proc = Bun.spawn({
+  const useCustomSink = options.output !== undefined;
+  const proc = spawn({
     cmd: command,
     stdin: "inherit",
-    stdout: "inherit",
-    stderr: "inherit",
+    stdout: useCustomSink ? "pipe" : "inherit",
+    stderr: useCustomSink ? "pipe" : "inherit",
     env: process.env,
   });
 
-  return await proc.exited;
+  if (!useCustomSink) {
+    return await proc.exited;
+  }
+
+  const [exitCode] = await Promise.all([
+    proc.exited,
+    forwardSpawnOutput(proc.stdout, output),
+    forwardSpawnOutput(proc.stderr, output),
+  ]);
+
+  return exitCode;
 }