From ba6437844c180861b67133077c0d1569f5c4baff Mon Sep 17 00:00:00 2001 From: ErdemOzgen <12345678+username@users.noreply.github.com> Date: Wed, 9 Oct 2024 17:45:23 +0300 Subject: [PATCH] Cobra and viper --- .../Golang Cobra and Viper.md | 234 ++++++++++++++++++ 1 file changed, 234 insertions(+) create mode 100644 content/SoftwareEngineering/Golang Cobra and Viper.md diff --git a/content/SoftwareEngineering/Golang Cobra and Viper.md b/content/SoftwareEngineering/Golang Cobra and Viper.md new file mode 100644 index 000000000..bfea7c815 --- /dev/null +++ b/content/SoftwareEngineering/Golang Cobra and Viper.md @@ -0,0 +1,234 @@ +#viper #cobra #go #golang + + +The `cobra` and `viper` packages are two popular libraries in the Go programming language ecosystem, often used together for building CLI applications and managing configuration. Here's an overview of each package, their primary features, and how they integrate: + +## Cobra + +`Cobra` is a powerful library used to create CLI (Command-Line Interface) applications in Go. It provides an easy way to define commands, flags, and subcommands, making it ideal for building complex CLI tools. + +### Key Features of Cobra + +1. **Command and Subcommand Structure**: + - `cobra` supports creating multiple commands and subcommands, similar to how tools like `kubectl` or `git` work. + - Each command can have its own set of flags and arguments. + +2. **Built-in Help and Usage Messages**: + - `cobra` automatically generates usage messages and help documentation for your commands, so you don't have to manually code these features. + +3. **Command Hierarchy**: + - Commands are organized hierarchically. For example, `myapp` might have a `myapp create` subcommand, and `myapp delete` subcommand. This structure allows for organized and logical grouping of CLI operations. + +4. **Command Aliases**: + - You can define aliases for commands so that multiple command strings trigger the same command handler. + +5. **Persistent Flags and Local Flags**: + - `Persistent Flags` are flags that work for this command and all subcommands, while `Local Flags` are flags that only work for this specific command. + +6. **Argument Parsing**: + - `cobra` helps you handle command arguments (positional and named) in a structured manner. + +### Example of Cobra Usage + +```go +package main + +import ( + "fmt" + "github.com/spf13/cobra" + "os" +) + +func main() { + var rootCmd = &cobra.Command{ + Use: "myapp", + Short: "MyApp is a CLI tool", + Long: "MyApp is a command-line application written in Go", + Run: func(cmd *cobra.Command, args []string) { + fmt.Println("Hello from MyApp") + }, + } + + rootCmd.AddCommand(versionCmd) + + if err := rootCmd.Execute(); err != nil { + fmt.Println(err) + os.Exit(1) + } +} + +var versionCmd = &cobra.Command{ + Use: "version", + Short: "Print the version number of MyApp", + Run: func(cmd *cobra.Command, args []string) { + fmt.Println("MyApp v1.0.0") + }, +} +``` + +In this example: +- `myapp` is the root command. +- `version` is a subcommand that prints the version of the application. + +### Cobra Folder Structure + +When using `cobra`, you can structure your project as follows: + +``` +myapp/ +│ +├── cmd/ +│ ├── root.go +│ ├── create.go +│ └── delete.go +├── main.go +└── go.mod +``` + +Each file under the `cmd` folder can represent a command or subcommand, making it easy to manage large CLI tools. + +## Viper + +`Viper` is a configuration management library for Go. It is designed to handle configurations in a variety of formats and integrate seamlessly with `cobra`. + +### Key Features of Viper + +1. **Supports Multiple Configuration Formats**: + - `viper` can read configurations from JSON, YAML, TOML, HCL, and Java properties files. + +2. **Environment Variables**: + - `viper` supports reading configuration from environment variables. You can bind environment variables to specific configuration keys, allowing your application to be easily configured through the environment. + +3. **Remote Configuration**: + - `viper` can fetch configurations from remote systems such as Consul or ETCD. + +4. **Default Values**: + - You can set default values for your configuration keys, making it easier to handle cases where the configuration file or environment variable might be missing. + +5. **Configuration Overriding**: + - Configuration settings can be overridden in a layered approach. For example, settings in a config file can be overridden by environment variables or command-line flags. + +6. **Hot Reloading**: + - `viper` supports monitoring configuration files for changes and automatically reloading the values. + +### Example of Viper Usage + +```go +package main + +import ( + "fmt" + "github.com/spf13/viper" +) + +func main() { + // Set default values + viper.SetDefault("app.name", "MyApp") + viper.SetDefault("app.version", "1.0.0") + + // Read configuration from a file + viper.SetConfigName("config") // name of config file (without extension) + viper.SetConfigType("yaml") // config file type + viper.AddConfigPath(".") // look for config in the current directory + + err := viper.ReadInConfig() // Find and read the config file + if err != nil { + fmt.Printf("Error reading config file, %s", err) + } + + // Access configuration values + fmt.Println("App Name:", viper.GetString("app.name")) + fmt.Println("App Version:", viper.GetString("app.version")) +} +``` + +In this example: +- `viper` is used to read a configuration file called `config.yaml`. +- If the `config.yaml` file is present, the application reads the `app.name` and `app.version` from it. + +### Viper Configuration File Example (`config.yaml`) + +```yaml +app: + name: MyApp + version: 1.0.0 +``` + +### Using Cobra and Viper Together + +`cobra` and `viper` are often used together to build powerful CLI applications with easy configuration management. When used together: + +1. **Flags and Configuration Binding**: + - `cobra` flags can be bound to `viper` keys, making it easy to set configuration values through command-line flags. + +2. **Environment Variable Integration**: + - `viper` can automatically use environment variables for configuration values, and `cobra` can parse these variables as flags. + +3. **Handling Multiple Configurations**: + - Use `viper` for the overall configuration and `cobra` for command-specific flags and options. + +### Example: Using Cobra and Viper Together + +```go +package main + +import ( + "fmt" + "github.com/spf13/cobra" + "github.com/spf13/viper" +) + +func main() { + var cfgFile string + + var rootCmd = &cobra.Command{ + Use: "myapp", + Short: "MyApp is a CLI tool", + Run: func(cmd *cobra.Command, args []string) { + fmt.Println("Hello from MyApp") + fmt.Println("Config File Used:", viper.ConfigFileUsed()) + }, + } + + // Persistent flag for configuration file + rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.myapp.yaml)") + cobra.OnInitialize(initConfig(cfgFile)) + + // Bind the config flag to viper + viper.BindPFlag("config", rootCmd.PersistentFlags().Lookup("config")) + + if err := rootCmd.Execute(); err != nil { + fmt.Println(err) + } +} + +// Initialize configuration +func initConfig(cfgFile string) func() { + return func() { + if cfgFile != "" { + viper.SetConfigFile(cfgFile) + } else { + viper.AddConfigPath("$HOME") + viper.SetConfigName(".myapp") + } + + if err := viper.ReadInConfig(); err != nil { + fmt.Println("Error reading config file:", err) + } + } +} +``` + +In this example: +- The `config` flag is used to specify the configuration file location. +- `viper` reads the configuration file and makes it available to the application. + +## Best Practices + +- Use `cobra` for CLI structure and argument parsing. +- Use `viper` for configuration management (from file, environment variables, or remote systems). +- Bind `cobra` flags to `viper` keys for unified configuration management. + +These two packages together make building complex, user-friendly, and configurable CLI applications straightforward and maintainable in Go. + +