1use clap::Parser;
 2use remote_server::Commands;
 3use std::io::Write as _;
 4use std::path::PathBuf;
 5
 6#[derive(Parser)]
 7#[command(disable_version_flag = true)]
 8struct Cli {
 9    #[command(subcommand)]
10    command: Option<Commands>,
11    /// Used for SSH/Git password authentication, to remove the need for netcat as a dependency,
12    /// by having Zed act like netcat communicating over a Unix socket.
13    #[arg(long, hide = true)]
14    askpass: Option<String>,
15    /// Used for recording minidumps on crashes by having the server run a separate
16    /// process communicating over a socket.
17    #[arg(long, hide = true)]
18    crash_handler: Option<PathBuf>,
19    /// Used for loading the environment from the project.
20    #[arg(long, hide = true)]
21    printenv: bool,
22}
23
24fn main() -> anyhow::Result<()> {
25    let cli = Cli::parse();
26
27    if let Some(socket_path) = &cli.askpass {
28        askpass::main(socket_path);
29        return Ok(());
30    }
31
32    if let Some(socket) = &cli.crash_handler {
33        crashes::crash_server(socket.as_path());
34        return Ok(());
35    }
36
37    if cli.printenv {
38        util::shell_env::print_env();
39        return Ok(());
40    }
41
42    if let Some(command) = cli.command {
43        use remote_server::ExecuteProxyError;
44
45        let res = remote_server::run(command);
46        if let Err(e) = &res
47            && let Some(e) = e.downcast_ref::<ExecuteProxyError>()
48        {
49            std::io::stderr().write_fmt(format_args!("{e:#}\n")).ok();
50            // It is important for us to report the proxy spawn exit code here
51            // instead of the generic 1 that result returns
52            // The client reads the exit code to determine if the server process has died when trying to reconnect
53            // signaling that it needs to try spawning a new server
54            std::process::exit(e.to_exit_code());
55        }
56        res
57    } else {
58        std::io::stderr()
59            .write_all(b"usage: remote <run|proxy|version>\n")
60            .ok();
61        std::process::exit(1);
62    }
63}