46 lines
1.4 KiB
Go
46 lines
1.4 KiB
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"flag"
|
|||
|
|
"fmt"
|
|||
|
|
"os"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// CLIArgs 保存从命令行解析得到的参数，对应旧后端 coconut-leaf.py 中 argparse 解析的结果。
|
|||
|
|
type CLIArgs struct {
|
|||
|
|
// Config 是 coconut-leaf 的配置文件路径（即旧后端的 CONFIG_TOML）。
|
|||
|
|
Config string
|
|||
|
|
// Init 指示是否在启动前初始化日历系统。
|
|||
|
|
Init bool
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ParseCLI 使用标准库 flag 包解析命令行参数并返回 *CLIArgs。
|
|||
|
|
// 当必需的 --config 参数缺失时，打印用法并以退出码 2 退出（与 argparse 的退出码一致）。
|
|||
|
|
func ParseCLI() *CLIArgs {
|
|||
|
|
args := &CLIArgs{}
|
|||
|
|
|
|||
|
|
// 自定义用法输出，复刻旧后端 ArgumentParser 的 description 横幅。
|
|||
|
|
flag.Usage = func() {
|
|||
|
|
out := flag.CommandLine.Output()
|
|||
|
|
fmt.Fprintln(out, "The server of light, self-host and multi-account calendar system.")
|
|||
|
|
fmt.Fprintln(out)
|
|||
|
|
fmt.Fprintf(out, "Usage of %s:\n", os.Args[0])
|
|||
|
|
flag.PrintDefaults()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// --config 对应旧后端的 -c/--config（必需），--init 对应旧后端的 -i/--init。
|
|||
|
|
flag.StringVar(&args.Config, "config", "", "The configuration file `CONFIG_TOML` for coconut-leaf")
|
|||
|
|
flag.BoolVar(&args.Init, "init", false, "Set for initialize the calendar system")
|
|||
|
|
|
|||
|
|
flag.Parse()
|
|||
|
|
|
|||
|
|
// 标准库 flag 不支持 required，故解析后手动校验。
|
|||
|
|
if args.Config == "" {
|
|||
|
|
fmt.Fprintln(os.Stderr, "error: the required flag --config is not provided")
|
|||
|
|
flag.Usage()
|
|||
|
|
os.Exit(2)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return args
|
|||
|
|
}
|