90 lines
2.9 KiB
Go
90 lines
2.9 KiB
Go
// Package config 解析各子命令的命令行开关。
|
||
package config
|
||
|
||
import (
|
||
"flag"
|
||
"fmt"
|
||
"net"
|
||
"strings"
|
||
)
|
||
|
||
// Config 运行时配置，serve 与维护子命令共用。
|
||
type Config struct {
|
||
Addr string
|
||
DataDir string
|
||
LogLevel string
|
||
LogFormat string // text|json
|
||
BehindProxy bool
|
||
Dev bool
|
||
AllowNewer bool
|
||
}
|
||
|
||
// DBPath SQLite 数据库文件路径。
|
||
func (c *Config) DBPath() string { return c.DataDir + "/pn.db" }
|
||
|
||
func addCommonFlags(fs *flag.FlagSet, c *Config) {
|
||
fs.StringVar(&c.DataDir, "data-dir", "./data", "数据目录（SQLite 数据库所在）")
|
||
}
|
||
|
||
func addServeFlags(fs *flag.FlagSet, c *Config) {
|
||
addCommonFlags(fs, c)
|
||
fs.StringVar(&c.Addr, "addr", ":8080", "HTTP 监听地址")
|
||
fs.StringVar(&c.LogLevel, "log-level", "info", "日志级别（debug|info|warn|error）")
|
||
fs.StringVar(&c.LogFormat, "log-format", "text", "日志格式（text|json）")
|
||
fs.BoolVar(&c.BehindProxy, "behind-proxy", false, "位于可信反向代理之后（取 X-Forwarded-For 最右条目作为客户端 IP）")
|
||
fs.BoolVar(&c.Dev, "dev", false, "开发模式：允许非 Secure Cookie，仅允许监听 loopback 地址")
|
||
fs.BoolVar(&c.AllowNewer, "allow-newer", false, "允许在更新的数据库 schema 版本上运行（跳过版本上界守卫）")
|
||
}
|
||
|
||
// ParseStart 解析 `pn start` 参数。
|
||
func ParseStart(args []string) (*Config, error) {
|
||
c := &Config{}
|
||
fs := flag.NewFlagSet("start", flag.ContinueOnError)
|
||
addServeFlags(fs, c)
|
||
if err := fs.Parse(args); err != nil {
|
||
return nil, err
|
||
}
|
||
if c.Dev && !isLoopbackAddr(c.Addr) {
|
||
return nil, fmt.Errorf("--dev 仅允许监听 loopback 地址（如 127.0.0.1:8080），当前为 %q", c.Addr)
|
||
}
|
||
return c, nil
|
||
}
|
||
|
||
// ParseInit 解析 `pn init` 参数。
|
||
func ParseInit(args []string) (*Config, error) {
|
||
c := &Config{}
|
||
fs := flag.NewFlagSet("init", flag.ContinueOnError)
|
||
addCommonFlags(fs, c)
|
||
fs.BoolVar(&c.AllowNewer, "allow-newer", false, "允许在更新的数据库 schema 版本上运行（跳过版本上界守卫）")
|
||
if err := fs.Parse(args); err != nil {
|
||
return nil, err
|
||
}
|
||
return c, nil
|
||
}
|
||
|
||
// ParseMaint 解析 `pn backup` / `pn gc` / `pn passwd` 参数（仅数据目录；
|
||
// 维护命令经 store.OpenData 打开，不迁移、无版本守卫，故无 --allow-newer，D6）。
|
||
// 返回 flag 解析后的剩余位置参数（如 backup 的输出路径）。
|
||
func ParseMaint(cmd string, args []string) (*Config, []string, error) {
|
||
c := &Config{}
|
||
fs := flag.NewFlagSet(cmd, flag.ContinueOnError)
|
||
addCommonFlags(fs, c)
|
||
if err := fs.Parse(args); err != nil {
|
||
return nil, nil, err
|
||
}
|
||
return c, fs.Args(), nil
|
||
}
|
||
|
||
func isLoopbackAddr(addr string) bool {
|
||
host, _, err := net.SplitHostPort(addr)
|
||
if err != nil {
|
||
// 无端口：整体视为主机名
|
||
host = addr
|
||
}
|
||
if host == "" || strings.EqualFold(host, "localhost") {
|
||
return host != ""
|
||
}
|
||
ip := net.ParseIP(host)
|
||
return ip != nil && ip.IsLoopback()
|
||
}
|