feat: create flag env

README.md

@@ -20,6 +20,30 @@ Check if the plugin of rabbitmq-manager is available in your RabbitMQ
 
 Rabbit Tools for now only has two functions show List of Queue or Purge Queue
 
+### Flags
+
+
+#### Env
+
+```
+--env=Prefix
+```
+
+You pass in flag env the prefix to use more than one configuration. Example:
+
+```
+./rabbit-tools list all --env=PROD
+```
+
+The configuration look like:
+
+```
+PROD_RABBIT_URL=http://your_url/api/queues/your_virtual_host
+PROD_RABBIT_USER=your_user
+PROD_RABBIT_PASS=your_pass
+```
+
+
 ### List
 
 ```

cmd/list.go

@@ -1,6 +1,7 @@
 package cmd
 
 import (
+
 	"github.com/daltondiaz/rabbit-tools/internal"
 	"github.com/jedib0t/go-pretty/v6/table"
 	"github.com/spf13/cobra"
@@ -12,7 +13,8 @@ var listCmd = &cobra.Command{
 	Args:  cobra.ExactArgs(1),
 	Run: func(cmd *cobra.Command, args []string) {
 		queue := args[0]
-		result := internal.GetQueues(queue)
+        env, _ := cmd.Flags().GetString("env")
+		result := internal.GetQueues(queue, env)
         var prettyResult internal.PrettyModelResult
         prettyResult.Title = "List of Queues"
         prettyResult.Header = table.Row{"Queue", "Messages"} 
@@ -30,4 +32,5 @@ var listCmd = &cobra.Command{
 
 func init() {
 	rootCmd.AddCommand(listCmd)
+    listCmd.PersistentFlags().String("env", "", "Prefix of environment in config.env")
 }

cmd/purge.go

@@ -11,10 +11,12 @@ var purgeCmd = &cobra.Command{
     Args: cobra.ExactArgs(1),
     Run: func(cmd *cobra.Command, args []string) {
         queue := args[0]
-        internal.PurgeQueue(queue)
+        env, _ := cmd.Flags().GetString("env")
+        internal.PurgeQueue(queue, env)
     },
 }
 
 func init(){
     rootCmd.AddCommand(purgeCmd)
+    rootCmd.PersistentFlags().String("env","", "Prefix of environment in config.env")
 }

internal/list.go

@@ -14,9 +14,9 @@ type ParamAccess struct {
     User string
 }
 
-func GetQueues(queue string) map[string]float64 {
+func GetQueues(queue string, env string) map[string]float64 {
 
-    params := LoadEnvVariables()
+    params := LoadEnvVariables(env)
     client := &http.Client{}
     req, err := http.NewRequest("GET", params.Url, nil)
     req.SetBasicAuth(params.User, params.Pass)

internal/load_env.go

@@ -7,15 +7,23 @@ import (
 	"github.com/joho/godotenv"
 )
 
-func LoadEnvVariables() ParamAccess {
+func LoadEnvVariables(env string) ParamAccess {
 
     err := godotenv.Load("./config.env")
     if err != nil {
         log.Fatal("Error loading .env file")
     }
+    url := "RABBIT_URL"
+    user := "RABBIT_USER"
+    pass := "RABBIT_PASS"
+    if env != "" {
+        url = env + "_" + url
+        user = env + "_" + user 
+        pass = env + "_" + pass 
+    }
     var params ParamAccess
-    params.Url = os.Getenv("RABBIT_URL")
-    params.User = os.Getenv("RABBIT_USER")
-    params.Pass = os.Getenv("RABBIT_PASS") 
+    params.Url = os.Getenv(url)
+    params.User = os.Getenv(user)
+    params.Pass = os.Getenv(pass) 
     return params
 }

internal/purge.go

@@ -8,10 +8,10 @@ import (
 )
 
 // Do purge of the queues found from queue argument
-func PurgeQueue(queue string) {
-    params := LoadEnvVariables()
+func PurgeQueue(queue string, env string) {
+    params := LoadEnvVariables(env)
     client := &http.Client{}
-    queues := GetQueues(queue)
+    queues := GetQueues(queue, env)
     sum := 0.0
     for queueName, value := range queues {
         url := (params.Url + "/" + queueName + "/contents")