GuanM/hysteria-dev
1
0
Fork 0
mirror of https://github.com/cmz0228/hysteria-dev.git synced 2025-06-30 16:29:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(cmd): support env

Browse Source 
support env

Signed-off-by: mritd <mritd@linux.com>
This commit is contained in:
mritd 2022-02-06 16:19:04 +08:00
parent fd2e0100c6
commit 8e6a9be5f7
No known key found for this signature in database
GPG Key ID: 150E1851F767CDDD
1 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
cmd/main.go
View File

@ -14,11 +14,7 @@ import (
"github.com/spf13/viper" "github.com/spf13/viper"
) )
var ( const (
appVersion = "Unknown"
appCommit = "Unknown"
appDate = "Unknown"
logo = ` logo = `
██╗ ██╗██╗ ██╗███████╗████████╗███████╗██████╗ ██╗ █████╗ ██╗ ██╗██╗ ██╗███████╗████████╗███████╗██████╗ ██╗ █████╗
██║ ██║╚██╗ ██╔╝██╔════╝╚══██╔══╝██╔════╝██╔══██╗██║██╔══██╗ ██║ ██║╚██╗ ██╔╝██╔════╝╚══██╔══╝██╔════╝██╔══██╗██║██╔══██╗
@ -28,11 +24,19 @@ var (
╚═╝ ╚═╝ ╚═╝ ╚══════╝ ╚═╝ ╚══════╝╚═╝ ╚═╝╚═╝╚═╝ ╚═╝ ╚═╝ ╚═╝ ╚═╝ ╚══════╝ ╚═╝ ╚══════╝╚═╝ ╚═╝╚═╝╚═╝ ╚═╝
` `
desc = "A TCP/UDP relay & SOCKS5/HTTP proxy tool optimized for poor network environments"
authors = "HyNetwork <https://github.com/HyNetwork>"
)
var (
appVersion = "Unknown"
appCommit = "Unknown"
appDate = "Unknown"
) )
var rootCmd = &cobra.Command{ var rootCmd = &cobra.Command{
Use: "hysteria", Use: "hysteria",
Short: logo + "A TCP/UDP relay & SOCKS5/HTTP proxy tool optimized for poor network environments", Short: fmt.Sprintf("%s%s\n\n%s %s %s -- %s", logo, desc, appVersion, appDate, appCommit, authors),
Example: "./hysteria server --config /etc/hysteria.json", Example: "./hysteria server --config /etc/hysteria.json",
Version: fmt.Sprintf("%s%s %s %s", logo, appVersion, appDate, appCommit), Version: fmt.Sprintf("%s%s %s %s", logo, appVersion, appDate, appCommit),
PersistentPreRun: func(cmd *cobra.Command, args []string) { PersistentPreRun: func(cmd *cobra.Command, args []string) {
@ -129,16 +133,20 @@ func init() {
// add to root cmd // add to root cmd
rootCmd.AddCommand(clientCmd, serverCmd) rootCmd.AddCommand(clientCmd, serverCmd)
// bind env // bind flag
_ = viper.BindPFlag("config", rootCmd.PersistentFlags().Lookup("config")) _ = viper.BindPFlag("config", rootCmd.PersistentFlags().Lookup("config"))
_ = viper.BindPFlag("log-level", rootCmd.PersistentFlags().Lookup("log-level")) _ = viper.BindPFlag("log-level", rootCmd.PersistentFlags().Lookup("log-level"))
_ = viper.BindPFlag("log-timestamp", rootCmd.PersistentFlags().Lookup("log-timestamp")) _ = viper.BindPFlag("log-timestamp", rootCmd.PersistentFlags().Lookup("log-timestamp"))
_ = viper.BindPFlag("log-format", rootCmd.PersistentFlags().Lookup("log-format")) _ = viper.BindPFlag("log-format", rootCmd.PersistentFlags().Lookup("log-format"))
_ = viper.BindPFlag("no-check", rootCmd.PersistentFlags().Lookup("log-format")) _ = viper.BindPFlag("no-check", rootCmd.PersistentFlags().Lookup("no-check"))
viper.SetEnvPrefix("HYSTERIA") // bind env
_ = viper.BindEnv("config", "HYSTERIA_CONFIG")
_ = viper.BindEnv("log-level", "HYSTERIA_LOG_LEVEL", "LOGGING_LEVEL")
_ = viper.BindEnv("log-timestamp", "HYSTERIA_LOG_TIMESTAMP", "LOGGING_TIMESTAMP_FORMAT")
_ = viper.BindEnv("log-format", "HYSTERIA_LOG_FORMAT", "LOGGING_FORMATTER")
_ = viper.BindEnv("no-check", "HYSTERIA_NO_CHECK", "HYSTERIA_NO_CHECK_UPDATE")
viper.AutomaticEnv() viper.AutomaticEnv()
_ = viper.ReadInConfig()
} }
func main() { func main() {