gout 0.6.0

Gout CLI — expose local services to the public internet with one command.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142

/// CLI 命令解析

use clap::{Parser, Subcommand};

#[derive(Parser, Debug)]
#[command(name = "gout", version, about = "轻量内网穿透工具")]
pub struct Cli {
    #[command(subcommand)]
    pub command: Command,
}

#[derive(Subcommand, Debug)]
pub enum Command {
    /// 登录远程服务器（等价于 `server set`）
    Login {
        /// 服务器名称（可选，默认按地址自动生成）
        name: Option<String>,
        /// 服务器地址，如 `server.example.com:8080`
        server: String,
        /// API key
        key: String,
    },
    /// 创建 TCP 隧道
    Tcp {
        /// 本地端口号
        port: u16,
        /// 后台运行
        #[arg(long, short = 'd')]
        detach: bool,
        /// 服务器名称或地址（默认使用配置中的 default_server）
        #[arg(long, short = 's')]
        server: Option<String>,
    },
    /// 创建 UDP 隧道
    Udp {
        /// 本地端口号
        port: u16,
        /// 后台运行
        #[arg(long, short = 'd')]
        detach: bool,
        /// 服务器名称或地址
        #[arg(long, short = 's')]
        server: Option<String>,
    },
    /// 创建 HTTP 隧道（等价于 TCP）
    Http {
        /// 本地端口号
        port: u16,
        /// 后台运行
        #[arg(long, short = 'd')]
        detach: bool,
        /// 服务器名称或地址
        #[arg(long, short = 's')]
        server: Option<String>,
    },
    /// 列出本地后台隧道（按 server 分组）
    #[clap(alias = "ls")]
    List,
    /// 查看后台隧道日志
    Log {
        /// 本地端口号
        port: u16,
        /// 持续跟随（类似 tail -f）
        #[arg(long, short = 'f')]
        follow: bool,
    },
    /// 停止后台隧道
    Kill {
        /// 本地端口号
        port: u16,
    },
    /// 管理 server
    #[command(subcommand)]
    Server(ServerCmd),
}

#[derive(Subcommand, Debug)]
pub enum ServerCmd {
    /// 添加或更新 server
    Set {
        /// server 名称
        name: String,
        /// server 地址，如 `server.example.com:8080`
        host: String,
        /// API key
        key: String,
    },
    /// 设置默认 server
    Default {
        /// server 名称
        name: String,
    },
    /// 删除 server
    Unset {
        /// server 名称
        name: String,
    },
    /// 显示所有 server
    Show,
}

impl Cli {
    pub fn run() -> anyhow::Result<()> {
        let cli = Cli::parse();

        // 子进程模式（由 -d 启动的 daemon），结束后清理 PID 文件
        if let Some(pidfile) = std::env::var("GOUT_DAEMON_PIDFILE").ok() {
            let result = match cli.command {
                Command::Tcp { port, .. } => crate::cmd_tunnel("tcp", port, None),
                Command::Udp { port, .. } => crate::cmd_tunnel("udp", port, None),
                Command::Http { port, .. } => crate::cmd_tunnel("http", port, None),
                _ => anyhow::bail!("daemon mode unsupported for this command"),
            };
            std::fs::remove_file(&pidfile).ok();
            return result;
        }

        match cli.command {
            Command::Login { name, server, key } => {
                let n = name.unwrap_or_else(|| {
                    server.split(':').next().unwrap_or(&server).to_string()
                });
                crate::cmd_server_set(&n, &server, &key)
            }
            Command::Tcp { port, detach: true, server } => crate::cmd_start_daemon("tcp", port, server.as_deref()),
            Command::Tcp { port, detach: false, server } => crate::cmd_tunnel("tcp", port, server.as_deref()),
            Command::Udp { port, detach: true, server } => crate::cmd_start_daemon("udp", port, server.as_deref()),
            Command::Udp { port, detach: false, server } => crate::cmd_tunnel("udp", port, server.as_deref()),
            Command::Http { port, detach: true, server } => crate::cmd_start_daemon("http", port, server.as_deref()),
            Command::Http { port, detach: false, server } => crate::cmd_tunnel("http", port, server.as_deref()),
            Command::Log { port, follow } => crate::cmd_log(port, follow),
            Command::Kill { port } => crate::cmd_kill(port),
            Command::List => crate::cmd_list(),
            Command::Server(cmd) => match cmd {
                ServerCmd::Set { name, host, key } => crate::cmd_server_set(&name, &host, &key),
                ServerCmd::Default { name } => crate::cmd_server_default(&name),
                ServerCmd::Unset { name } => crate::cmd_server_unset(&name),
                ServerCmd::Show => crate::cmd_server_show(),
            },
        }
    }
}