generated from bilalcaliskan/golang-cli-template
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #21 from bilalcaliskan/devel
Devel
- Loading branch information
Showing
16 changed files
with
256 additions
and
49 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,23 +1,95 @@ | ||
| package options | ||
|
|
||
| import "github.com/spf13/cobra" | ||
| import ( | ||
| "log" | ||
| "os" | ||
| "path/filepath" | ||
|
|
||
| "github.com/spf13/viper" | ||
|
|
||
| "github.com/pkg/errors" | ||
| "github.com/spf13/cobra" | ||
| ) | ||
|
|
||
| var rootOptions = &RootOptions{} | ||
|
|
||
| // RootOptions contains frequent command line and application options. | ||
| type RootOptions struct { | ||
| DnsServers string | ||
| CheckIntervalMin int | ||
| Verbose bool | ||
| Workspace string // This field will be managed by the command line argument | ||
| ConfigFile string // This field will be managed by the command line argument | ||
| DnsServers string `toml:"dns-servers"` // This field will be managed by the config file | ||
| CheckIntervalMin int `toml:"check-interval-min"` // This field will be managed by the config file | ||
| SocketPath string `toml:"socket-path"` // This field will be managed by the config file | ||
| Verbose bool `toml:"verbose"` // This field will be managed by the config file | ||
| } | ||
|
|
||
| // GetRootOptions returns the pointer of RootOptions | ||
| func GetRootOptions() *RootOptions { | ||
| return rootOptions | ||
| } | ||
|
|
||
| func (opts *RootOptions) InitFlags(cmd *cobra.Command) { | ||
| func (opts *RootOptions) InitFlags(cmd *cobra.Command) error { | ||
| if err := opts.setFlags(cmd); err != nil { | ||
| return errors.Wrap(err, "failed to set flags") | ||
| } | ||
|
|
||
| if err := viper.BindPFlags(cmd.Flags()); err != nil { | ||
| return errors.Wrap(err, "failed to bind flags") | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| //func (opts *RootOptions) setDefaultWorkspace() error { | ||
| // homeDir, err := os.UserHomeDir() | ||
| // if err != nil { | ||
| // return errors.Wrap(err, "failed to get user home directory") | ||
| // } | ||
| // | ||
| // ws := filepath.Join(homeDir, ".split-the-tunnel") | ||
| // opts.Workspace = ws | ||
| // opts.ConfigFile = filepath.Join(ws, "config.toml") | ||
| // | ||
| // return nil | ||
| //} | ||
|
|
||
| func (opts *RootOptions) setFlags(cmd *cobra.Command) error { | ||
| homeDir, err := os.UserHomeDir() | ||
| if err != nil { | ||
| return errors.Wrap(err, "failed to get user home directory") | ||
| } | ||
|
|
||
| cmd.Flags().StringVarP(&opts.Workspace, "workspace", "w", filepath.Join(homeDir, ".split-the-tunnel"), "workspace directory path") | ||
| cmd.Flags().StringVarP(&opts.ConfigFile, "config-file", "c", "config.toml", "config file path, will search in workspace") | ||
| cmd.Flags().StringVarP(&opts.SocketPath, "socket-path", "", "ipc.sock", "unix domain socket path in workspace") | ||
| cmd.Flags().BoolVarP(&opts.Verbose, "verbose", "", false, "verbose logging output") | ||
| cmd.Flags().StringVarP(&opts.DnsServers, "dns-servers", "", "", "comma separated dns servers to be used for DNS resolving") | ||
| cmd.Flags().IntVarP(&opts.CheckIntervalMin, "check-interval-min", "", 5, "routing table check interval with collected state, in minutes") | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (opts *RootOptions) ReadConfig() error { | ||
| viper.SetConfigFile(opts.ConfigFile) | ||
| viper.SetConfigType("toml") | ||
|
|
||
| opts.SocketPath = filepath.Join(opts.Workspace, opts.SocketPath) | ||
| opts.ConfigFile = filepath.Join(opts.Workspace, opts.ConfigFile) | ||
|
|
||
| if _, err := os.Stat(opts.ConfigFile); os.IsNotExist(err) { | ||
| log.Printf("config file not found in %s, will use default values\n", opts.ConfigFile) | ||
| return nil | ||
| } else if err != nil { | ||
| return errors.Wrap(err, "failed to access config file") | ||
| } | ||
|
|
||
| if err := viper.ReadInConfig(); err != nil { | ||
| return errors.Wrap(err, "failed to read config file") | ||
| } | ||
|
|
||
| if err := viper.Unmarshal(opts); err != nil { | ||
| return errors.Wrap(err, "failed to unmarshal config file") | ||
| } | ||
|
|
||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.