rird 2.1.231 → 2.3.0

package/src/cli/cmd/tui/util/terminal.ts

@@ -0,0 +1,146 @@
+import { RGBA } from "@opentui/core"
+
+// Default terminal dimensions when detection fails or returns bogus values
+const DEFAULT_WIDTH = 80
+const DEFAULT_HEIGHT = 24
+
+// Maximum reasonable terminal dimensions (beyond this is likely bogus)
+const MAX_REASONABLE_WIDTH = 500
+const MAX_REASONABLE_HEIGHT = 200
+
+/**
+ * Sanitize terminal dimensions to handle edge cases where detection fails.
+ * Returns sensible defaults (80x24) when dimensions are bogus (e.g., 131072x1).
+ */
+export function sanitizeDimensions(dims: { width: number; height: number }): { width: number; height: number } {
+  const width = dims.width
+  const height = dims.height
+
+  // Check for bogus dimensions: too small, too large, or zero
+  const isBogus =
+    width <= 0 ||
+    height <= 0 ||
+    width > MAX_REASONABLE_WIDTH ||
+    height > MAX_REASONABLE_HEIGHT ||
+    !Number.isFinite(width) ||
+    !Number.isFinite(height)
+
+  if (isBogus) {
+    return { width: DEFAULT_WIDTH, height: DEFAULT_HEIGHT }
+  }
+
+  return { width, height }
+}
+
+export namespace Terminal {
+  export type Colors = Awaited<ReturnType<typeof colors>>
+  /**
+   * Query terminal colors including background, foreground, and palette (0-15).
+   * Uses OSC escape sequences to retrieve actual terminal color values.
+   *
+   * Note: OSC 4 (palette) queries may not work through tmux as responses are filtered.
+   * OSC 10/11 (foreground/background) typically work in most environments.
+   *
+   * Returns an object with background, foreground, and colors array.
+   * Any query that fails will be null/empty.
+   */
+  export async function colors(): Promise<{
+    background: RGBA | null
+    foreground: RGBA | null
+    colors: RGBA[]
+  }> {
+    if (!process.stdin.isTTY) return { background: null, foreground: null, colors: [] }
+
+    return new Promise((resolve) => {
+      let background: RGBA | null = null
+      let foreground: RGBA | null = null
+      const paletteColors: RGBA[] = []
+      let timeout: NodeJS.Timeout
+
+      const cleanup = () => {
+        process.stdin.setRawMode(false)
+        process.stdin.removeListener("data", handler)
+        clearTimeout(timeout)
+      }
+
+      const parseColor = (colorStr: string): RGBA | null => {
+        if (colorStr.startsWith("rgb:")) {
+          const parts = colorStr.substring(4).split("/")
+          return RGBA.fromInts(
+            parseInt(parts[0], 16) >> 8, // Convert 16-bit to 8-bit
+            parseInt(parts[1], 16) >> 8,
+            parseInt(parts[2], 16) >> 8,
+            255,
+          )
+        }
+        if (colorStr.startsWith("#")) {
+          return RGBA.fromHex(colorStr)
+        }
+        if (colorStr.startsWith("rgb(")) {
+          const parts = colorStr.substring(4, colorStr.length - 1).split(",")
+          return RGBA.fromInts(parseInt(parts[0]), parseInt(parts[1]), parseInt(parts[2]), 255)
+        }
+        return null
+      }
+
+      const handler = (data: Buffer) => {
+        const str = data.toString()
+
+        // Match OSC 11 (background color)
+        const bgMatch = str.match(/\x1b]11;([^\x07\x1b]+)/)
+        if (bgMatch) {
+          background = parseColor(bgMatch[1])
+        }
+
+        // Match OSC 10 (foreground color)
+        const fgMatch = str.match(/\x1b]10;([^\x07\x1b]+)/)
+        if (fgMatch) {
+          foreground = parseColor(fgMatch[1])
+        }
+
+        // Match OSC 4 (palette colors)
+        const paletteMatches = str.matchAll(/\x1b]4;(\d+);([^\x07\x1b]+)/g)
+        for (const match of paletteMatches) {
+          const index = parseInt(match[1])
+          const color = parseColor(match[2])
+          if (color) paletteColors[index] = color
+        }
+
+        // Return immediately if we have all 16 palette colors
+        if (paletteColors.filter((c) => c !== undefined).length === 16) {
+          cleanup()
+          resolve({ background, foreground, colors: paletteColors })
+        }
+      }
+
+      process.stdin.setRawMode(true)
+      process.stdin.on("data", handler)
+
+      // Query background (OSC 11)
+      process.stdout.write("\x1b]11;?\x07")
+      // Query foreground (OSC 10)
+      process.stdout.write("\x1b]10;?\x07")
+      // Query palette colors 0-15 (OSC 4)
+      for (let i = 0; i < 16; i++) {
+        process.stdout.write(`\x1b]4;${i};?\x07`)
+      }
+
+      timeout = setTimeout(() => {
+        cleanup()
+        resolve({ background, foreground, colors: paletteColors })
+      }, 1000)
+    })
+  }
+
+  export async function getTerminalBackgroundColor(): Promise<"dark" | "light"> {
+    const result = await colors()
+    if (!result.background) return "dark"
+
+    const { r, g, b } = result.background
+    // Calculate luminance using relative luminance formula
+    const luminance = (0.299 * r + 0.587 * g + 0.114 * b) / 255
+
+    // Determine if dark or light based on luminance threshold
+    return luminance > 0.5 ? "light" : "dark"
+  }
+}

package/src/cli/cmd/tui/worker.ts

@@ -0,0 +1,63 @@
+import { Installation } from "@/installation"
+import { Server } from "@/server/server"
+import { Log } from "@/util/log"
+import { Instance } from "@/project/instance"
+import { InstanceBootstrap } from "@/project/bootstrap"
+import { Rpc } from "@/util/rpc"
+import { upgrade } from "@/cli/upgrade"
+import type { BunWebSocketData } from "hono/bun"
+
+await Log.init({
+  print: process.argv.includes("--print-logs"),
+  dev: Installation.isLocal(),
+  level: (() => {
+    if (Installation.isLocal()) return "DEBUG"
+    return "INFO"
+  })(),
+})
+
+process.on("unhandledRejection", (e) => {
+  Log.Default.error("rejection", {
+    e: e instanceof Error ? e.message : e,
+  })
+})
+
+process.on("uncaughtException", (e) => {
+  Log.Default.error("exception", {
+    e: e instanceof Error ? e.message : e,
+  })
+})
+
+let server: Bun.Server<BunWebSocketData>
+export const rpc = {
+  async server(input: { port: number; hostname: string }) {
+    if (server) await server.stop(true)
+    try {
+      server = Server.listen(input)
+      return {
+        url: server.url.toString(),
+      }
+    } catch (e) {
+      console.error(e)
+      throw e
+    }
+  },
+  async checkUpgrade(input: { directory: string }) {
+    await Instance.provide({
+      directory: input.directory,
+      init: InstanceBootstrap,
+      fn: async () => {
+        await upgrade().catch(() => {})
+      },
+    })
+  },
+  async shutdown() {
+    Log.Default.info("worker shutting down")
+    await Instance.disposeAll()
+    // TODO: this should be awaited, but ws connections are
+    // causing this to hang, need to revisit this
+    server.stop(true)
+  },
+}
+
+Rpc.listen(rpc)

package/src/cli/cmd/uninstall.ts

@@ -0,0 +1,344 @@
+import type { Argv } from "yargs"
+import { UI } from "../ui"
+import * as prompts from "@clack/prompts"
+import { Installation } from "../../installation"
+import { Global } from "../../global"
+import { $ } from "bun"
+import fs from "fs/promises"
+import path from "path"
+import os from "os"
+
+interface UninstallArgs {
+  keepConfig: boolean
+  keepData: boolean
+  dryRun: boolean
+  force: boolean
+}
+
+interface RemovalTargets {
+  directories: Array<{ path: string; label: string; keep: boolean }>
+  shellConfig: string | null
+  binary: string | null
+}
+
+export const UninstallCommand = {
+  command: "uninstall",
+  describe: "uninstall rird and remove all related files",
+  builder: (yargs: Argv) =>
+    yargs
+      .option("keep-config", {
+        alias: "c",
+        type: "boolean",
+        describe: "keep configuration files",
+        default: false,
+      })
+      .option("keep-data", {
+        alias: "d",
+        type: "boolean",
+        describe: "keep session data and snapshots",
+        default: false,
+      })
+      .option("dry-run", {
+        type: "boolean",
+        describe: "show what would be removed without removing",
+        default: false,
+      })
+      .option("force", {
+        alias: "f",
+        type: "boolean",
+        describe: "skip confirmation prompts",
+        default: false,
+      }),
+
+  handler: async (args: UninstallArgs) => {
+    UI.empty()
+    UI.println(UI.logo("  "))
+    UI.empty()
+    prompts.intro("Uninstall RIRD")
+
+    const method = await Installation.method()
+    prompts.log.info(`Installation method: ${method}`)
+
+    const targets = await collectRemovalTargets(args, method)
+
+    await showRemovalSummary(targets, method)
+
+    if (!args.force && !args.dryRun) {
+      const confirm = await prompts.confirm({
+        message: "Are you sure you want to uninstall?",
+        initialValue: false,
+      })
+      if (!confirm || prompts.isCancel(confirm)) {
+        prompts.outro("Cancelled")
+        return
+      }
+    }
+
+    if (args.dryRun) {
+      prompts.log.warn("Dry run - no changes made")
+      prompts.outro("Done")
+      return
+    }
+
+    await executeUninstall(method, targets)
+
+    prompts.outro("Done")
+  },
+}
+
+async function collectRemovalTargets(args: UninstallArgs, method: Installation.Method): Promise<RemovalTargets> {
+  const directories: RemovalTargets["directories"] = [
+    { path: Global.Path.data, label: "Data", keep: args.keepData },
+    { path: Global.Path.cache, label: "Cache", keep: false },
+    { path: Global.Path.config, label: "Config", keep: args.keepConfig },
+    { path: Global.Path.state, label: "State", keep: false },
+  ]
+
+  const shellConfig = method === "curl" ? await getShellConfigFile() : null
+  const binary = method === "curl" ? process.execPath : null
+
+  return { directories, shellConfig, binary }
+}
+
+async function showRemovalSummary(targets: RemovalTargets, method: Installation.Method) {
+  prompts.log.message("The following will be removed:")
+
+  for (const dir of targets.directories) {
+    const exists = await fs
+      .access(dir.path)
+      .then(() => true)
+      .catch(() => false)
+    if (!exists) continue
+
+    const size = await getDirectorySize(dir.path)
+    const sizeStr = formatSize(size)
+    const status = dir.keep ? UI.Style.TEXT_DIM + "(keeping)" : ""
+    const prefix = dir.keep ? "○" : "✓"
+
+    prompts.log.info(`  ${prefix} ${dir.label}: ${shortenPath(dir.path)} ${UI.Style.TEXT_DIM}(${sizeStr})${status}`)
+  }
+
+  if (targets.binary) {
+    prompts.log.info(`  ✓ Binary: ${shortenPath(targets.binary)}`)
+  }
+
+  if (targets.shellConfig) {
+    prompts.log.info(`  ✓ Shell PATH in ${shortenPath(targets.shellConfig)}`)
+  }
+
+  if (method !== "curl" && method !== "unknown") {
+    const cmds: Record<string, string> = {
+      npm: "npm uninstall -g rird-ai",
+      pnpm: "pnpm uninstall -g rird-ai",
+      bun: "bun remove -g rird-ai",
+      yarn: "yarn global remove rird-ai",
+      brew: "brew uninstall rird",
+    }
+    prompts.log.info(`  ✓ Package: ${cmds[method] || method}`)
+  }
+}
+
+async function executeUninstall(method: Installation.Method, targets: RemovalTargets) {
+  const spinner = prompts.spinner()
+  const errors: string[] = []
+
+  for (const dir of targets.directories) {
+    if (dir.keep) {
+      prompts.log.step(`Skipping ${dir.label} (--keep-${dir.label.toLowerCase()})`)
+      continue
+    }
+
+    const exists = await fs
+      .access(dir.path)
+      .then(() => true)
+      .catch(() => false)
+    if (!exists) continue
+
+    spinner.start(`Removing ${dir.label}...`)
+    const err = await fs.rm(dir.path, { recursive: true, force: true }).catch((e) => e)
+    if (err) {
+      spinner.stop(`Failed to remove ${dir.label}`, 1)
+      errors.push(`${dir.label}: ${err.message}`)
+      continue
+    }
+    spinner.stop(`Removed ${dir.label}`)
+  }
+
+  if (targets.shellConfig) {
+    spinner.start("Cleaning shell config...")
+    const err = await cleanShellConfig(targets.shellConfig).catch((e) => e)
+    if (err) {
+      spinner.stop("Failed to clean shell config", 1)
+      errors.push(`Shell config: ${err.message}`)
+    } else {
+      spinner.stop("Cleaned shell config")
+    }
+  }
+
+  if (method !== "curl" && method !== "unknown") {
+    const cmds: Record<string, string[]> = {
+      npm: ["npm", "uninstall", "-g", "rird-ai"],
+      pnpm: ["pnpm", "uninstall", "-g", "rird-ai"],
+      bun: ["bun", "remove", "-g", "rird-ai"],
+      yarn: ["yarn", "global", "remove", "rird-ai"],
+      brew: ["brew", "uninstall", "rird"],
+    }
+
+    const cmd = cmds[method]
+    if (cmd) {
+      spinner.start(`Running ${cmd.join(" ")}...`)
+      const result = await $`${cmd}`.quiet().nothrow()
+      if (result.exitCode !== 0) {
+        spinner.stop(`Package manager uninstall failed`, 1)
+        prompts.log.warn(`You may need to run manually: ${cmd.join(" ")}`)
+        errors.push(`Package manager: exit code ${result.exitCode}`)
+      } else {
+        spinner.stop("Package removed")
+      }
+    }
+  }
+
+  if (method === "curl" && targets.binary) {
+    UI.empty()
+    prompts.log.message("To finish removing the binary, run:")
+    prompts.log.info(`  rm "${targets.binary}"`)
+
+    const binDir = path.dirname(targets.binary)
+    if (binDir.includes(".RIRD")) {
+      prompts.log.info(`  rmdir "${binDir}" 2>/dev/null`)
+    }
+  }
+
+  if (errors.length > 0) {
+    UI.empty()
+    prompts.log.warn("Some operations failed:")
+    for (const err of errors) {
+      prompts.log.error(`  ${err}`)
+    }
+  }
+
+  UI.empty()
+  prompts.log.success("Thank you for using RIRD!")
+}
+
+async function getShellConfigFile(): Promise<string | null> {
+  const shell = path.basename(process.env.SHELL || "bash")
+  const home = os.homedir()
+  const xdgConfig = process.env.XDG_CONFIG_HOME || path.join(home, ".config")
+
+  const configFiles: Record<string, string[]> = {
+    fish: [path.join(xdgConfig, "fish", "config.fish")],
+    zsh: [
+      path.join(home, ".zshrc"),
+      path.join(home, ".zshenv"),
+      path.join(xdgConfig, "zsh", ".zshrc"),
+      path.join(xdgConfig, "zsh", ".zshenv"),
+    ],
+    bash: [
+      path.join(home, ".bashrc"),
+      path.join(home, ".bash_profile"),
+      path.join(home, ".profile"),
+      path.join(xdgConfig, "bash", ".bashrc"),
+      path.join(xdgConfig, "bash", ".bash_profile"),
+    ],
+    ash: [path.join(home, ".ashrc"), path.join(home, ".profile")],
+    sh: [path.join(home, ".profile")],
+  }
+
+  const candidates = configFiles[shell] || configFiles.bash
+
+  for (const file of candidates) {
+    const exists = await fs
+      .access(file)
+      .then(() => true)
+      .catch(() => false)
+    if (!exists) continue
+
+    const content = await Bun.file(file)
+      .text()
+      .catch(() => "")
+    if (content.includes("# RIRD") || content.includes(".RIRD/bin")) {
+      return file
+    }
+  }
+
+  return null
+}
+
+async function cleanShellConfig(file: string) {
+  const content = await Bun.file(file).text()
+  const lines = content.split("\n")
+
+  const filtered: string[] = []
+  let skip = false
+
+  for (const line of lines) {
+    const trimmed = line.trim()
+
+    if (trimmed === "# RIRD") {
+      skip = true
+      continue
+    }
+
+    if (skip) {
+      skip = false
+      if (trimmed.includes(".RIRD/bin") || trimmed.includes("fish_add_path")) {
+        continue
+      }
+    }
+
+    if (
+      (trimmed.startsWith("export PATH=") && trimmed.includes(".RIRD/bin")) ||
+      (trimmed.startsWith("fish_add_path") && trimmed.includes(".RIRD"))
+    ) {
+      continue
+    }
+
+    filtered.push(line)
+  }
+
+  while (filtered.length > 0 && filtered[filtered.length - 1].trim() === "") {
+    filtered.pop()
+  }
+
+  const output = filtered.join("\n") + "\n"
+  await Bun.write(file, output)
+}
+
+async function getDirectorySize(dir: string): Promise<number> {
+  let total = 0
+
+  const walk = async (current: string) => {
+    const entries = await fs.readdir(current, { withFileTypes: true }).catch(() => [])
+
+    for (const entry of entries) {
+      const full = path.join(current, entry.name)
+      if (entry.isDirectory()) {
+        await walk(full)
+        continue
+      }
+      if (entry.isFile()) {
+        const stat = await fs.stat(full).catch(() => null)
+        if (stat) total += stat.size
+      }
+    }
+  }
+
+  await walk(dir)
+  return total
+}
+
+function formatSize(bytes: number): string {
+  if (bytes < 1024) return `${bytes} B`
+  if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)} KB`
+  if (bytes < 1024 * 1024 * 1024) return `${(bytes / (1024 * 1024)).toFixed(1)} MB`
+  return `${(bytes / (1024 * 1024 * 1024)).toFixed(1)} GB`
+}
+
+function shortenPath(p: string): string {
+  const home = os.homedir()
+  if (p.startsWith(home)) {
+    return p.replace(home, "~")
+  }
+  return p
+}

package/src/cli/cmd/upgrade.ts

@@ -0,0 +1,127 @@
+import type { Argv } from "yargs"
+import { UI } from "../ui"
+import * as prompts from "@clack/prompts"
+import { Installation } from "../../installation"
+import { Log } from "../../util/log"
+import { InstallationError } from "../../lib/errors"
+import { handleError } from "../../lib/error-handler"
+
+const log = Log.create({ service: "upgrade-command" })
+
+export const UpgradeCommand = {
+  command: "upgrade [target]",
+  aliases: ["update"],
+  describe: "upgrade rird to the latest or a specific version",
+  builder: (yargs: Argv) => {
+    return yargs
+      .positional("target", {
+        describe: "version to upgrade to, for ex '0.1.48' or 'v0.1.48'",
+        type: "string",
+      })
+      .option("method", {
+        alias: "m",
+        describe: "installation method to use",
+        type: "string",
+        choices: ["curl", "npm", "pnpm", "bun", "brew"],
+      })
+  },
+  handler: async (args: { target?: string; method?: string }) => {
+    try {
+      UI.empty()
+      UI.println(UI.logo("  "))
+      UI.empty()
+      prompts.intro("Upgrade")
+      const detectedMethod = await Installation.method()
+      const method = (args.method as Installation.Method) ?? detectedMethod
+      if (method === "unknown") {
+        prompts.log.warn(`rird is installed to ${process.execPath} and may be managed by a package manager`)
+        const install = await prompts.select({
+          message: "Install anyways?",
+          options: [
+            { label: "Yes", value: true },
+            { label: "No", value: false },
+          ],
+          initialValue: true,
+        })
+        if (!install) {
+          prompts.outro("Done")
+          return
+        }
+      }
+      prompts.log.info("Using method: " + method)
+
+      let target: string
+      try {
+        target = args.target ? args.target.replace(/^v/, "") : await Installation.latest()
+      } catch (latestErr) {
+        log.warn("Could not fetch latest version from registry", {
+          error: (latestErr as Error).message,
+        })
+        prompts.log.warn("Could not fetch latest version from registry")
+        prompts.log.info("Attempting force reinstall to fix broken installation...")
+        const spinner = prompts.spinner()
+        spinner.start("Reinstalling...")
+        const err = await Installation.forceReinstall().catch((err) => err)
+        if (err) {
+          spinner.stop("Reinstall failed", 1)
+          log.error("Force reinstall failed", {
+            error: err instanceof Error ? err.message : String(err),
+          })
+          if (err instanceof Installation.UpgradeFailedError) prompts.log.error(err.data.stderr)
+          else if (err instanceof Error) prompts.log.error(err.message)
+          prompts.log.info("Manual fix: npm install -g rird-ai@latest")
+        } else {
+          spinner.stop("Reinstall complete")
+          log.info("Force reinstall completed successfully")
+        }
+        prompts.outro("Done")
+        return
+      }
+
+      if (Installation.VERSION === target) {
+        prompts.log.warn(`rird upgrade skipped: ${target} is already installed`)
+        prompts.outro("Done")
+        return
+      }
+
+      prompts.log.info(`From ${Installation.VERSION} -> ${target}`)
+      const spinner = prompts.spinner()
+      spinner.start("Upgrading...")
+      let err = await Installation.upgrade(method, target).catch((err) => err)
+
+      if (err) {
+        log.error("Standard upgrade failed", {
+          method,
+          target,
+          error: err instanceof Error ? err.message : String(err),
+        })
+        spinner.stop("Standard upgrade failed, trying force reinstall...", 1)
+        const forceSpinner = prompts.spinner()
+        forceSpinner.start("Reinstalling...")
+        err = await Installation.forceReinstall().catch((err) => err)
+        if (err) {
+          log.error("Force reinstall also failed", {
+            error: err instanceof Error ? err.message : String(err),
+          })
+          forceSpinner.stop("Reinstall failed", 1)
+          if (err instanceof Installation.UpgradeFailedError) prompts.log.error(err.data.stderr)
+          else if (err instanceof Error) prompts.log.error(err.message)
+          prompts.log.info("Manual fix: npm install -g rird-ai@latest")
+          throw new InstallationError("Upgrade and reinstall both failed", {
+            code: "UPGRADE_FAILED",
+            originalError: err as Error,
+            details: { method, target },
+          })
+        }
+        log.info("Force reinstall completed after initial upgrade failure")
+        forceSpinner.stop("Reinstall complete")
+      } else {
+        log.info("Upgrade completed successfully", { target })
+        spinner.stop("Upgrade complete")
+      }
+      prompts.outro("Done")
+    } catch (error) {
+      await handleError(error, "upgrade-command", { exitOnError: true })
+    }
+  },
+}