openAgent

1import yargs from "yargs"
2import { hideBin } from "yargs/helpers"
3import { RunCommand } from "./cli/cmd/run"
4import { GenerateCommand } from "./cli/cmd/generate"
5import { ConsoleCommand } from "./cli/cmd/account"
6import { ProvidersCommand } from "./cli/cmd/providers"
7import { AgentCommand } from "./cli/cmd/agent"
8import { UpgradeCommand } from "./cli/cmd/upgrade"
9import { UninstallCommand } from "./cli/cmd/uninstall"
10import { ModelsCommand } from "./cli/cmd/models"
11import { UI } from "./cli/ui"
12import { InstallationVersion } from "@opencode-ai/core/installation/version"
13import { FormatError } from "./cli/error"
14import { ServeCommand } from "./cli/cmd/serve"
15import { DebugCommand } from "./cli/cmd/debug"
16import { StatsCommand } from "./cli/cmd/stats"
17import { McpCommand } from "./cli/cmd/mcp"
18import { GithubCommand } from "./cli/cmd/github"
19import { ExportCommand } from "./cli/cmd/export"
20import { ImportCommand } from "./cli/cmd/import"
21import { AttachCommand } from "./cli/cmd/attach"
22import { TuiThreadCommand } from "./cli/cmd/tui"
23import { AcpCommand } from "./cli/cmd/acp"
24import { EOL } from "os"
25import { WebCommand } from "./cli/cmd/web"
26import { PrCommand } from "./cli/cmd/pr"
27import { SessionCommand } from "./cli/cmd/session"
28import { DbCommand } from "./cli/cmd/db"
29import { errorMessage } from "./util/error"
30import { PluginCommand } from "./cli/cmd/plug"
31import { ToolsCommand } from "./cli/cmd/tools"
32import { CredentialsCommand } from "./cli/cmd/credentials"
33import { Heap } from "./cli/heap"
34 
35const args = hideBin(process.argv)
36 
37function show(out: string) {
38  const text = out.trimStart()
39  if (!text.startsWith("openagent ") && !text.startsWith("opencode ")) {
40    process.stderr.write(UI.logo() + EOL + EOL)
41    process.stderr.write(text + EOL)
42    return
43  }
44  process.stderr.write(out)
45}
46 
47const cli = yargs(args)
48  .parserConfiguration({ "populate--": true })
49  .scriptName("openagent")
50  .wrap(100)
51  .help("help", "show help")
52  .alias("help", "h")
53  .version("version", "show version number", InstallationVersion)
54  .alias("version", "v")
55  .option("print-logs", {
56    describe: "print logs to stderr",
57    type: "boolean",
58  })
59  .option("log-level", {
60    describe: "log level",
61    type: "string",
62    choices: ["DEBUG", "INFO", "WARN", "ERROR"],
63  })
64  .option("pure", {
65    describe: "run without external plugins",
66    type: "boolean",
67  })
68  .middleware(async (opts) => {
69    if (opts.printLogs) {
70      process.env.OPENAGENT_PRINT_LOGS = "1"
71      process.env.OPENCODE_PRINT_LOGS = "1"
72    }
73    if (opts.logLevel) {
74      process.env.OPENAGENT_LOG_LEVEL = opts.logLevel
75      process.env.OPENCODE_LOG_LEVEL = opts.logLevel
76    }
77    if (opts.pure) {
78      process.env.OPENAGENT_PURE = "1"
79      process.env.OPENCODE_PURE = "1"
80    }
81 
82    Heap.start()
83 
84    process.env.AGENT = "1"
85    process.env.OPENAGENT = "1"
86    process.env.OPENCODE = "1"
87    process.env.OPENAGENT_PID = String(process.pid)
88    process.env.OPENCODE_PID = String(process.pid)
89  })
90  .usage("")
91  .completion("completion", "generate shell completion script")
92  .command(AcpCommand)
93  .command(McpCommand)
94  .command(TuiThreadCommand)
95  .command(AttachCommand)
96  .command(RunCommand)
97  .command(GenerateCommand)
98  .command(DebugCommand)
99  .command(ConsoleCommand)
100  .command(ProvidersCommand)
101  .command(AgentCommand)
102  .command(UpgradeCommand)
103  .command(UninstallCommand)
104  .command(ServeCommand)
105  .command(WebCommand)
106  .command(ModelsCommand)
107  .command(StatsCommand)
108  .command(ExportCommand)
109  .command(ImportCommand)
110  .command(GithubCommand)
111  .command(PrCommand)
112  .command(SessionCommand)
113  .command(PluginCommand)
114  .command(ToolsCommand)
115  .command(CredentialsCommand)
116  .command(DbCommand)
117  .fail((msg, err) => {
118    if (
119      msg?.startsWith("Unknown argument") ||
120      msg?.startsWith("Not enough non-option arguments") ||
121      msg?.startsWith("Invalid values:")
122    ) {
123      if (err) throw err
124      cli.showHelp(show)
125    }
126    if (err) throw err
127    process.exit(1)
128  })
129  .strict()
130 
131try {
132  if (args.includes("-h") || args.includes("--help")) {
133    await cli.parse(args, (err: Error | undefined, _argv: unknown, out: string) => {
134      if (err) throw err
135      if (!out) return
136      show(out)
137    })
138  } else {
139    await cli.parse()
140  }
141} catch (e) {
142  const formatted = FormatError(e)
143  if (formatted) UI.error(formatted)
144  if (formatted === undefined) {
145    UI.error("Unexpected error" + EOL)
146    process.stderr.write(errorMessage(e) + EOL)
147  }
148  process.exitCode = 1
149} finally {
150  // Some subprocesses don't react properly to SIGTERM and similar signals.
151  // Most notably, some docker-container-based MCP servers don't handle such signals unless
152  // run using `docker run --init`.
153  // Explicitly exit to avoid any hanging subprocesses.
154  process.exit()
155}
156