🎉first commit

readers/cli.go

@@ -0,0 +1,85 @@
+package readers
+
+import (
+	"fmt"
+
+	"github.com/aykhans/dodo/config"
+	customerrors "github.com/aykhans/dodo/custom_errors"
+	"github.com/aykhans/dodo/utils"
+	"github.com/spf13/cobra"
+	"github.com/spf13/pflag"
+)
+
+func CLIConfigReader() (*config.CLIConfig, error) {
+	var (
+		returnNil    = false
+		cliConfig    = &config.CLIConfig{}
+		dodosCount   int
+		requestCount int
+		timeout      int
+		rootCmd      = &cobra.Command{
+			Use: "dodo [flags]",
+			Example: `  Simple usage only with URL:
+    dodo -u https://example.com
+
+  Simple usage with config file:
+    dodo -c /path/to/config/file/config.json
+
+  Usage with all flags:
+    dodo -c /path/to/config/file/config.json -u https://example.com -m POST -d 10 -r 1000 -t 2000`,
+			Short: `
+██████████      ███████    ██████████      ███████   
+░░███░░░░███   ███░░░░░███ ░░███░░░░███   ███░░░░░███ 
+ ░███   ░░███ ███     ░░███ ░███   ░░███ ███     ░░███
+ ░███    ░███░███      ░███ ░███    ░███░███      ░███
+ ░███    ░███░███      ░███ ░███    ░███░███      ░███
+ ░███    ███ ░░███     ███  ░███    ███ ░░███     ███ 
+ ██████████   ░░░███████░   ██████████   ░░░███████░  
+░░░░░░░░░░      ░░░░░░░    ░░░░░░░░░░      ░░░░░░░    
+`,
+			Run:           func(cmd *cobra.Command, args []string) {},
+			SilenceErrors: true,
+			SilenceUsage:  true,
+			Version:       config.VERSION,
+		}
+	)
+
+	rootCmd.Flags().StringVarP(&cliConfig.ConfigFile, "config-file", "c", "", "Path to the config file")
+	rootCmd.Flags().StringVarP(&cliConfig.Method, "method", "m", "", fmt.Sprintf("HTTP Method (default %s)", config.DefaultMethod))
+	rootCmd.Flags().StringVarP(&cliConfig.URL, "url", "u", "", "URL for stress testing")
+	rootCmd.Flags().IntVarP(&dodosCount, "dodos-count", "d", config.DefaultDodosCount, "Number of dodos(threads)")
+	rootCmd.Flags().IntVarP(&requestCount, "request-count", "r", config.DefaultRequestCount, "Number of total requests")
+	rootCmd.Flags().IntVarP(&timeout, "timeout", "t", config.DefaultTimeout, "Timeout for each request in milliseconds")
+	if err := rootCmd.Execute(); err != nil {
+		utils.PrintErr(err)
+		rootCmd.Println(rootCmd.UsageString())
+		return nil, customerrors.CobraErrorFormater(err)
+	}
+	rootCmd.Flags().Visit(func(f *pflag.Flag) {
+		switch f.Name {
+		case "help":
+			returnNil = true
+		case "version":
+			returnNil = true
+		case "dodos-count":
+			cliConfig.DodosCount = dodosCount
+		case "request-count":
+			cliConfig.RequestCount = requestCount
+		case "timeout":
+			cliConfig.Timeout = timeout
+		}
+	})
+	if returnNil {
+		return nil, nil
+	}
+	return cliConfig, nil
+}
+
+func CLIYesOrNoReader(message string) bool {
+	var answer string
+	fmt.Printf("%s [y/N]: ", message)
+	if _, err := fmt.Scanln(&answer); err != nil {
+		return false
+	}
+	return answer == "y" || answer == "Y"
+}

readers/json.go

@@ -0,0 +1,33 @@
+package readers
+
+import (
+	"encoding/json"
+	"os"
+
+	"github.com/aykhans/dodo/config"
+	"github.com/aykhans/dodo/custom_errors"
+)
+
+func JSONConfigReader(filePath string) (*config.JSONConfig, error) {
+	data, err := os.ReadFile(filePath)
+	if err != nil {
+		return nil, customerrors.OSErrorFormater(err)
+	}
+	jsonConf := &config.JSONConfig{}
+	err = json.Unmarshal(data, &jsonConf)
+
+	if err != nil {
+		switch err := err.(type) {
+		case *json.UnmarshalTypeError:
+			return nil,
+				customerrors.NewTypeError(
+					err.Type.String(),
+					err.Value,
+					err.Field,
+					err,
+				)
+		}
+		return nil, customerrors.NewInvalidFileError(filePath, err)
+	}
+	return jsonConf, nil
+}