gout 0.4.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

/// 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 {
    /// 登录远程服务器，保存凭据
    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>,
    },
    /// 列出本地后台隧道
    List,
    /// 查看后台隧道日志
    Log {
        /// 本地端口号
        port: u16,
        /// 持续跟随（类似 tail -f）
        #[arg(long, short = 'f')]
        follow: bool,
    },
    /// 停止后台隧道
    Kill {
        /// 本地端口号
        port: u16,
    },
    /// 设置默认 server
    Default {
        /// 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_login(&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::Default { name } => crate::cmd_set_default(&name),
            Command::Show => crate::cmd_show(),
        }
    }
}