@farhanic017/octocode 3.5.0

package/src/cli/error.ts

@@ -0,0 +1,118 @@
+import { NamedError } from "@octocode-ai/core/util/error"
+import { errorFormat } from "@/util/error"
+import { isRecord } from "@/util/record"
+
+type ConfigIssue = { message: string; path: string[] }
+
+function isTaggedError(error: unknown, tag: string): error is Record<string, unknown> {
+  return isRecord(error) && error._tag === tag
+}
+
+function configData(input: unknown, tag: string): Record<string, unknown> | undefined {
+  if (!isRecord(input)) return undefined
+  if (input.name === tag && isRecord(input.data)) return input.data
+  if (input._tag === tag) return input
+  return undefined
+}
+
+function stringField(input: Record<string, unknown>, key: string): string | undefined {
+  return typeof input[key] === "string" ? input[key] : undefined
+}
+
+function configIssues(input: Record<string, unknown>): ConfigIssue[] {
+  return Array.isArray(input.issues)
+    ? input.issues.filter((issue): issue is ConfigIssue => {
+        if (!isRecord(issue)) return false
+        return (
+          typeof issue.message === "string" &&
+          Array.isArray(issue.path) &&
+          issue.path.every((x) => typeof x === "string")
+        )
+      })
+    : []
+}
+
+export function FormatError(input: unknown): string | undefined {
+  if (input instanceof Error && isRecord(input.cause) && "body" in input.cause) {
+    const formatted = FormatError(input.cause.body)
+    if (formatted) return formatted
+  }
+
+  // CliError: domain failure surfaced from an effectCmd handler via fail("...")
+  if (isTaggedError(input, "CliError")) {
+    if (typeof input.exitCode === "number") process.exitCode = input.exitCode
+    return stringField(input, "message") ?? ""
+  }
+
+  // MCPFailed: { name: string }
+  if (NamedError.hasName(input, "MCPFailed")) {
+    const data = isRecord(input) && isRecord(input.data) ? stringField(input.data, "name") : undefined
+    return `MCP server "${data}" failed. Note, octo code does not support MCP authentication yet.`
+  }
+
+  // AccountServiceError, AccountTransportError: TaggedErrorClass
+  if (isTaggedError(input, "AccountServiceError") || isTaggedError(input, "AccountTransportError")) {
+    return stringField(input, "message") ?? ""
+  }
+
+  // ProviderModelNotFoundError: { providerID: string, modelID: string, suggestions?: string[] }
+  const providerModelNotFound = configData(input, "ProviderModelNotFoundError")
+  if (providerModelNotFound) {
+    const suggestions = Array.isArray(providerModelNotFound.suggestions)
+      ? providerModelNotFound.suggestions.filter((x) => typeof x === "string")
+      : []
+    return [
+      `Model not found: ${stringField(providerModelNotFound, "providerID")}/${stringField(providerModelNotFound, "modelID")}`,
+      ...(suggestions.length ? ["Did you mean: " + suggestions.join(", ")] : []),
+      `Try: \`octo models\` to list available models`,
+      `Or check your config (octo.json) provider/model names`,
+    ].join("\n")
+  }
+
+  // ProviderInitError: { providerID: string }
+  const providerInit = configData(input, "ProviderInitError")
+  if (providerInit) {
+    return `Failed to initialize provider "${stringField(providerInit, "providerID")}". Check credentials and configuration.`
+  }
+
+  // ConfigJsonError: { path: string, message?: string }
+  const configJson = configData(input, "ConfigJsonError")
+  if (configJson) {
+    const message = stringField(configJson, "message")
+    return `Config file at ${stringField(configJson, "path")} is not valid JSON(C)` + (message ? `: ${message}` : "")
+  }
+
+  // ConfigDirectoryTypoError: { dir: string, path: string, suggestion: string }
+  const configDirectoryTypo = configData(input, "ConfigDirectoryTypoError")
+  if (configDirectoryTypo) {
+    return `Directory "${stringField(configDirectoryTypo, "dir")}" in ${stringField(configDirectoryTypo, "path")} is not valid. Rename the directory to "${stringField(configDirectoryTypo, "suggestion")}" or remove it. This is a common typo.`
+  }
+
+  // ConfigFrontmatterError: { message: string }
+  const configFrontmatter = configData(input, "ConfigFrontmatterError")
+  if (configFrontmatter) {
+    return stringField(configFrontmatter, "message") ?? ""
+  }
+
+  // ConfigInvalidError: { path?: string, message?: string, issues?: Array<{ message: string, path: string[] }> }
+  const configInvalid = configData(input, "ConfigInvalidError")
+  if (configInvalid) {
+    const path = stringField(configInvalid, "path")
+    const message = stringField(configInvalid, "message")
+    const issues = configIssues(configInvalid)
+    return [
+      `Configuration is invalid${path && path !== "config" ? ` at ${path}` : ""}` + (message ? `: ${message}` : ""),
+      ...issues.map((issue) => "↳ " + issue.message + " " + issue.path.join(".")),
+    ].join("\n")
+  }
+
+  // UICancelledError: user cancelled an interactive CLI prompt
+  if (isTaggedError(input, "UICancelledError") || NamedError.hasName(input, "UICancelledError")) {
+    return ""
+  }
+  return undefined
+}
+
+export function FormatUnknownError(input: unknown): string {
+  return errorFormat(input)
+}

package/src/cli/heap.ts

@@ -0,0 +1,66 @@
+import path from "path"
+import { writeHeapSnapshot } from "node:v8"
+import { Flag } from "@octocode-ai/core/flag/flag"
+import { Global } from "@octocode-ai/core/global"
+import * as Log from "@octocode-ai/core/util/log"
+
+const log = Log.create({ service: "heap" })
+const MINUTE = 60_000
+const LIMIT = 2 * 1024 * 1024 * 1024
+
+let timer: Timer | undefined
+let lock = false
+let armed = true
+
+export function start() {
+  if (!Flag.OCTOCODE_AUTO_HEAP_SNAPSHOT) return
+  if (timer) return
+
+  const run = async () => {
+    if (lock) return
+
+    const stat = process.memoryUsage()
+    if (stat.rss <= LIMIT) {
+      armed = true
+      return
+    }
+    if (!armed) return
+
+    lock = true
+    armed = false
+    const file = path.join(
+      Global.Path.log,
+      `heap-${process.pid}-${new Date().toISOString().replace(/[:.]/g, "")}.heapsnapshot`,
+    )
+    log.warn("heap usage exceeded limit", {
+      rss: stat.rss,
+      heap: stat.heapUsed,
+      file,
+    })
+
+    await Promise.resolve()
+      .then(() => writeHeapSnapshot(file))
+      .catch((err) => {
+        log.error("failed to write heap snapshot", {
+          error: err instanceof Error ? err.message : String(err),
+          file,
+        })
+      })
+
+    lock = false
+  }
+
+  timer = setInterval(() => {
+    void run()
+  }, MINUTE)
+  timer.unref?.()
+}
+
+export function stop() {
+  if (timer) {
+    clearInterval(timer)
+    timer = undefined
+  }
+}
+
+export * as Heap from "./heap"

package/src/cli/logo.ts

@@ -0,0 +1 @@
+export * from "@octocode-ai/tui/logo"

package/src/cli/network.ts

@@ -0,0 +1,64 @@
+import type { Argv, InferredOptionTypes } from "yargs"
+import { ConfigV1 } from "@octocode-ai/core/v1/config/config"
+import type { Config } from "@/config/config"
+import { Effect } from "effect"
+
+const options = {
+  port: {
+    type: "number" as const,
+    describe: "port to listen on",
+    default: 0,
+  },
+  hostname: {
+    type: "string" as const,
+    describe: "hostname to listen on",
+    default: "127.0.0.1",
+  },
+  mdns: {
+    type: "boolean" as const,
+    describe: "enable mDNS service discovery (defaults hostname to 0.0.0.0)",
+    default: false,
+  },
+  "mdns-domain": {
+    type: "string" as const,
+    describe: "custom domain name for mDNS service (default: octo.local)",
+    default: "octo.local",
+  },
+  cors: {
+    type: "string" as const,
+    array: true,
+    describe: "additional domains to allow for CORS",
+    default: [] as string[],
+  },
+}
+
+export type NetworkOptions = InferredOptionTypes<typeof options>
+
+export function withNetworkOptions<T>(yargs: Argv<T>) {
+  return yargs.options(options)
+}
+export const resolveNetworkOptions = Effect.fn("Cli.resolveNetworkOptions")(function* (args: NetworkOptions) {
+  const { Config } = yield* Effect.promise(() => import("@/config/config"))
+  const config = yield* Config.Service.use((cfg) => cfg.getGlobal())
+  return resolveNetworkOptionsNoConfig(args, config)
+})
+
+export function resolveNetworkOptionsNoConfig(args: NetworkOptions, config?: ConfigV1.Info) {
+  const portExplicitlySet = process.argv.includes("--port")
+  const hostnameExplicitlySet = process.argv.includes("--hostname")
+  const mdnsExplicitlySet = process.argv.includes("--mdns")
+  const mdnsDomainExplicitlySet = process.argv.includes("--mdns-domain")
+  const mdns = mdnsExplicitlySet ? args.mdns : (config?.server?.mdns ?? args.mdns)
+  const mdnsDomain = mdnsDomainExplicitlySet ? args["mdns-domain"] : (config?.server?.mdnsDomain ?? args["mdns-domain"])
+  const port = portExplicitlySet ? args.port : (config?.server?.port ?? args.port)
+  const hostname = hostnameExplicitlySet
+    ? args.hostname
+    : mdns && !config?.server?.hostname
+      ? "0.0.0.0"
+      : (config?.server?.hostname ?? args.hostname)
+  const configCors = config?.server?.cors ?? []
+  const argsCors = Array.isArray(args.cors) ? args.cors : args.cors ? [args.cors] : []
+  const cors = [...configCors, ...argsCors]
+
+  return { hostname, port, mdns, mdnsDomain, cors }
+}

package/src/cli/tui/layer.ts

@@ -0,0 +1,7 @@
+import { run as runTui, type TuiInput } from "@octocode-ai/tui"
+import { Global } from "@octocode-ai/core/global"
+import { Effect } from "effect"
+
+export function run(input: TuiInput) {
+  return runTui(input).pipe(Effect.provide(Global.defaultLayer))
+}

package/src/cli/tui/worker.ts

@@ -0,0 +1,99 @@
+import { Installation } from "@/installation"
+import { Server } from "@/server/server"
+import * as Log from "@octocode-ai/core/util/log"
+import { InstanceRuntime } from "@/project/instance-runtime"
+import { Rpc } from "@/util/rpc"
+import { upgrade } from "@/cli/upgrade"
+import { Config } from "@/config/config"
+import { GlobalBus } from "@/bus/global"
+import { ServerAuth } from "@/server/auth"
+import { writeHeapSnapshot } from "node:v8"
+import { Heap } from "@/cli/heap"
+import { AppRuntime } from "@/effect/app-runtime"
+import { ensureProcessMetadata } from "@octocode-ai/core/util/octocode-process"
+import { Effect } from "effect"
+import { disposeAllInstancesAndEmitGlobalDisposed } from "@/server/global-lifecycle"
+
+ensureProcessMetadata("worker")
+
+await Log.init({
+  print: process.argv.includes("--print-logs"),
+  dev: Installation.isLocal(),
+  level: (() => {
+    if (Installation.isLocal()) return "DEBUG"
+    return "INFO"
+  })(),
+})
+
+Heap.start()
+
+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,
+  })
+})
+
+// Subscribe to global events and forward them via RPC
+GlobalBus.on("event", (event) => {
+  Rpc.emit("global.event", event)
+})
+
+let server: Awaited<ReturnType<typeof Server.listen>> | undefined
+
+export const rpc = {
+  async fetch(input: { url: string; method: string; headers: Record<string, string>; body?: string }) {
+    const headers = { ...input.headers }
+    const auth = ServerAuth.header()
+    if (auth && !headers["authorization"] && !headers["Authorization"]) {
+      headers["Authorization"] = auth
+    }
+    const request = new Request(input.url, {
+      method: input.method,
+      headers,
+      body: input.body,
+    })
+    const response = await Server.Default().app.fetch(request)
+    const body = await response.text()
+    return {
+      status: response.status,
+      headers: Object.fromEntries(response.headers.entries()),
+      body,
+    }
+  },
+  snapshot() {
+    const result = writeHeapSnapshot("server.heapsnapshot")
+    return result
+  },
+  async server(input: { port: number; hostname: string; mdns?: boolean; cors?: string[] }) {
+    if (server) await server.stop(true)
+    server = await Server.listen(input)
+    return { url: server.url.toString() }
+  },
+  async checkUpgrade(input: { directory: string }) {
+    await InstanceRuntime.load({ directory: input.directory })
+    await upgrade().catch(() => {})
+  },
+  async reload() {
+    await AppRuntime.runPromise(
+      Effect.gen(function* () {
+        const cfg = yield* Config.Service
+        yield* cfg.invalidate()
+        yield* disposeAllInstancesAndEmitGlobalDisposed({ swallowErrors: true })
+      }),
+    )
+  },
+  async shutdown() {
+    Log.Default.info("worker shutting down")
+
+    await InstanceRuntime.disposeAllInstances()
+    if (server) await server.stop(true)
+  },
+}
+
+Rpc.listen(rpc)

package/src/cli/ui.ts

@@ -0,0 +1,144 @@
+import { EOL } from "os"
+import { Schema } from "effect"
+import { logo as glyphs } from "./logo"
+
+const wordmark = [
+  `_______  _______ _________ _______  _______  _______  ______   _______`,
+  `(  ___  )(  ____ \\__   __/(  ___  )(  ____ \\(  ___  )(  __  \\ (  ____ \\`,
+  `| (   ) || (    \\/   ) (   | (   ) || (    \\/| (   ) || (  \\  )| (    \\/`,
+  `| |   | || |         | |   | |   | || |      | |   | || |   ) || (__    `,
+  `| |   | || |         | |   | |   | || |      | |   | || |   | ||  __)   `,
+  `| |   | || |         | |   | |   | || |      | |   | || |   ) || (      `,
+  `| (___) || (____/\\   | |   | (___) || (____/\\| (___) || (__/  )| (____/\\`,
+  `(_______)(_______/   )_(   (_______)(_______/(_______)(______/ (_______/`,
+]
+
+export class CancelledError extends Schema.TaggedErrorClass<CancelledError>()("UICancelledError", {}) {}
+
+export const Style = {
+  TEXT_HIGHLIGHT: "\x1b[96m",
+  TEXT_HIGHLIGHT_BOLD: "\x1b[96m\x1b[1m",
+  TEXT_DIM: "\x1b[90m",
+  TEXT_DIM_BOLD: "\x1b[90m\x1b[1m",
+  TEXT_NORMAL: "\x1b[0m",
+  TEXT_NORMAL_BOLD: "\x1b[1m",
+  TEXT_WARNING: "\x1b[93m",
+  TEXT_WARNING_BOLD: "\x1b[93m\x1b[1m",
+  TEXT_DANGER: "\x1b[91m",
+  TEXT_DANGER_BOLD: "\x1b[91m\x1b[1m",
+  TEXT_SUCCESS: "\x1b[92m",
+  TEXT_SUCCESS_BOLD: "\x1b[92m\x1b[1m",
+  TEXT_INFO: "\x1b[94m",
+  TEXT_INFO_BOLD: "\x1b[94m\x1b[1m",
+}
+
+export function println(...message: string[]) {
+  print(...message)
+  process.stderr.write(EOL)
+}
+
+export function print(...message: string[]) {
+  blank = false
+  process.stderr.write(message.join(" "))
+}
+
+let blank = false
+export function empty() {
+  if (blank) return
+  println("" + Style.TEXT_NORMAL)
+  blank = true
+}
+
+export function logo(pad?: string) {
+  if (!process.stdout.isTTY && !process.stderr.isTTY) {
+    const result = []
+    for (const row of wordmark) {
+      if (pad) result.push(pad)
+      result.push(row)
+      result.push(EOL)
+    }
+    return result.join("").trimEnd()
+  }
+
+  const result: string[] = []
+  const reset = "\x1b[0m"
+  const left = {
+    fg: "\x1b[90m",
+    shadow: "\x1b[38;5;235m",
+    bg: "\x1b[48;5;235m",
+  }
+  const right = {
+    fg: reset,
+    shadow: "\x1b[38;5;238m",
+    bg: "\x1b[48;5;238m",
+  }
+  const gap = " "
+  const draw = (line: string, fg: string, shadow: string, bg: string) => {
+    const parts: string[] = []
+    for (const char of line) {
+      if (char === "_") {
+        parts.push(bg, " ", reset)
+        continue
+      }
+      if (char === "^") {
+        parts.push(fg, bg, "▀", reset)
+        continue
+      }
+      if (char === "~") {
+        parts.push(shadow, "▀", reset)
+        continue
+      }
+      if (char === ",") {
+        parts.push(shadow, "▄", reset)
+        continue
+      }
+      if (char === "█") {
+        parts.push(fg, "█", reset)
+        continue
+      }
+      if (char === "▀") {
+        parts.push(fg, "▀", reset)
+        continue
+      }
+      if (char === "▄") {
+        parts.push(fg, "▄", reset)
+        continue
+      }
+      parts.push(fg, char, reset)
+    }
+    return parts.join("")
+  }
+
+  for (const line of wordmark) {
+    const leftPart = line.slice(0, glyphs.left[0]?.length ?? 0)
+    const rightPart = line.slice((glyphs.left[0]?.length ?? 0) + 1)
+    if (pad) result.push(pad)
+    result.push(draw(leftPart, left.fg, left.shadow, left.bg))
+    result.push(gap)
+    result.push(draw(rightPart, right.fg, right.shadow, right.bg))
+    result.push(EOL)
+  }
+  return result.join("").trimEnd()
+}
+
+export function error(message: string) {
+  println(Style.TEXT_DANGER, "Error:", Style.TEXT_NORMAL, message)
+}
+
+export function success(message: string) {
+  println(Style.TEXT_SUCCESS, message, Style.TEXT_NORMAL)
+}
+
+export function warning(message: string) {
+  println(Style.TEXT_WARNING, message, Style.TEXT_NORMAL)
+}
+
+export function info(message: string) {
+  println(Style.TEXT_INFO, message, Style.TEXT_NORMAL)
+}
+
+export function markdown(text: string): string {
+  return text
+}
+
+export * as UI from "./ui"

package/src/cli/upgrade.ts

@@ -0,0 +1,53 @@
+import { Config } from "@/config/config"
+import { AppRuntime } from "@/effect/app-runtime"
+import { Flag } from "@octocode-ai/core/flag/flag"
+import { Installation } from "@/installation"
+import { InstallationVersion } from "@octocode-ai/core/installation/version"
+import { GlobalBus } from "@/bus/global"
+
+export async function upgrade() {
+  const config = await AppRuntime.runPromise(Config.Service.use((cfg) => cfg.getGlobal()))
+  if (config.autoupdate === false || Flag.OCTOCODE_DISABLE_AUTOUPDATE) return
+  const method = await Installation.method()
+  const latest = await Installation.latest(method).catch(() => {})
+  if (!latest) return
+
+  if (Flag.OCTOCODE_ALWAYS_NOTIFY_UPDATE) {
+    GlobalBus.emit("event", {
+      directory: "global",
+      payload: {
+        type: Installation.Event.UpdateAvailable.type,
+        properties: { version: latest },
+      },
+    })
+    return
+  }
+
+  if (InstallationVersion === latest) return
+
+  const kind = Installation.getReleaseType(InstallationVersion, latest)
+
+  if (config.autoupdate === "notify" || kind !== "patch") {
+    GlobalBus.emit("event", {
+      directory: "global",
+      payload: {
+        type: Installation.Event.UpdateAvailable.type,
+        properties: { version: latest },
+      },
+    })
+    return
+  }
+
+  if (method === "unknown") return
+  await Installation.upgrade(method, latest)
+    .then(() =>
+      GlobalBus.emit("event", {
+        directory: "global",
+        payload: {
+          type: Installation.Event.Updated.type,
+          properties: { version: latest },
+        },
+      }),
+    )
+    .catch(() => {})
+}