mirror of
https://codeberg.org/cunicu/cunicu.git
synced 2025-10-31 11:46:32 +08:00
86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
"go.uber.org/zap/zapcore"
|
|
"riasc.eu/wice/internal"
|
|
)
|
|
|
|
const (
|
|
usageTemplate = `Usage:{{if .Runnable}}
|
|
{{.UseLine}}{{end}}{{if .HasAvailableSubCommands}}
|
|
{{.CommandPath}} [command]{{end}}{{if gt (len .Aliases) 0}}
|
|
|
|
Aliases:
|
|
{{.NameAndAliases}}{{end}}{{if .HasExample}}
|
|
|
|
Examples:
|
|
{{.Example}}{{end}}{{if .HasAvailableSubCommands}}
|
|
|
|
Available Commands:{{range .Commands}}{{if (or .IsAvailableCommand (eq .Name "help"))}}
|
|
{{rpad .Name .NamePadding }} {{.Short}}{{end}}{{end}}{{end}}{{if .HasAvailableLocalFlags}}
|
|
|
|
Flags:
|
|
{{.LocalFlags.FlagUsages | trimTrailingWhitespaces}}{{end}}{{if .HasAvailableInheritedFlags}}
|
|
|
|
Global Flags:
|
|
{{.InheritedFlags.FlagUsages | trimTrailingWhitespaces}}{{end}}{{if .HasHelpSubCommands}}
|
|
|
|
Additional help topics:{{range .Commands}}{{if .IsAdditionalHelpTopicCommand}}
|
|
{{rpad .CommandPath .CommandPathPadding}} {{.Short}}{{end}}{{end}}{{end}}{{if .HasAvailableSubCommands}}
|
|
|
|
Use "{{.CommandPath}} [command] --help" for more information about a command.{{end}}
|
|
|
|
Credits:
|
|
Steffen Vogel <post@steffenvogel.de>
|
|
|
|
Website:
|
|
https://github.com/stv0g/wice
|
|
`
|
|
)
|
|
|
|
var (
|
|
rootCmd = &cobra.Command{
|
|
Use: "wice",
|
|
Short: "WICE",
|
|
Long: "Wireguard Interactive Connectitivty Establishment",
|
|
|
|
// The main wice command is just an alias for "wice daemon"
|
|
Run: daemon,
|
|
DisableAutoGenTag: true,
|
|
Version: version,
|
|
}
|
|
|
|
version string //lint:ignore U1000 set via ldflags -X / goreleaser
|
|
commit string //lint:ignore U1000 set via ldflags -X / goreleaser
|
|
date string //lint:ignore U1000 set via ldflags -X / goreleaser
|
|
|
|
logLevel = level{zapcore.InfoLevel}
|
|
logFile string
|
|
)
|
|
|
|
type level struct {
|
|
zapcore.Level
|
|
}
|
|
|
|
func (l *level) Type() string {
|
|
return "string"
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.SetUsageTemplate(usageTemplate)
|
|
|
|
cobra.OnInitialize(
|
|
internal.SetupRand,
|
|
setupLogging,
|
|
)
|
|
|
|
pf := rootCmd.PersistentFlags()
|
|
pf.VarP(&logLevel, "log-level", "d", "log level (one of \"debug\", \"info\", \"warn\", \"error\", \"dpanic\", \"panic\", and \"fatal\")")
|
|
pf.StringVarP(&logFile, "log-file", "l", "", "path of a file to write logs to")
|
|
}
|
|
|
|
func setupLogging() {
|
|
logger = internal.SetupLogging(logLevel.Level, logFile)
|
|
}
|