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
|
||
}
|