Init

2026-09-08 12:43:56 +08:00 · 2023-06-23 04:08:00 +03:00
parent dbcb116e48
commit 8b87e7b4ab
29 changed files with 2480 additions and 8 deletions
@@ -0,0 +1,105 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/karust/googler/core"
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"github.com/spf13/pflag"
+	"github.com/spf13/viper"
+)
+
+const (
+	version                    = "0.1.0"
+	defaultConfigFilename      = "config"
+	envPrefix                  = "OPENSERP"
+	replaceHyphenWithCamelCase = false
+)
+
+type AppConfig struct {
+	Host          string
+	Port          int
+	Timeout       int
+	ConfigPath    string
+	IsBrowserHead bool `mapstructure:"head"`
+	IsLeakless    bool `mapstructure:"leakless"`
+	IsDebug       bool `mapstructure:"debug"`
+	IsVerbose     bool `mapstructure:"verbose"`
+	IsRawRequests bool `mapstructure:"raw_requests"`
+}
+
+var appConf = AppConfig{}
+
+var RootCmd = &cobra.Command{
+	Use:     "openserp",
+	Short:   "Open SERP",
+	Long:    `Search via Google, Yandex and Baidu`,
+	Version: version,
+	PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
+		core.InitLogger(appConf.IsVerbose, appConf.IsDebug)
+		logrus.Debugf("Config: %+v", appConf)
+
+		// Bind cobra and viper
+		return initializeConfig(cmd)
+	},
+	// Run: func(cmd *cobra.Command, args []string) {
+	// 	// Working with OutOrStdout/OutOrStderr allows us to unit test our command easier
+	// 	//out := cmd.OutOrStdout()
+	// 	logrus.Trace("Config:", appConf)
+	// },
+}
+
+// Bind each cobra flag to its associated viper configuration (config file and environment variable)
+func bindFlags(cmd *cobra.Command, vpr *viper.Viper) {
+	cmd.Flags().VisitAll(func(flg *pflag.Flag) {
+		configName := "app." + flg.Name
+		//if replaceHyphenWithCamelCase {
+		//	configName = strings.ReplaceAll(f.Name, "-", "")
+		//}
+
+		// Apply viper config value to the flag if viper has a value
+		if !flg.Changed && vpr.IsSet(configName) {
+			val := vpr.Get(configName)
+			cmd.Flags().Set(flg.Name, fmt.Sprintf("%v", val))
+		}
+	})
+}
+
+// Initialize Viper
+func initializeConfig(cmd *cobra.Command) error {
+	v := viper.New()
+
+	// Base name of the config file, without the file extension
+	v.SetConfigName(defaultConfigFilename)
+	v.AddConfigPath(".")
+
+	// Return an error if we cannot parse the config file.
+	if err := v.ReadInConfig(); err != nil {
+		if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
+			return err
+		}
+	}
+
+	v.SetEnvPrefix(envPrefix)
+
+	// Bind environment variables to their equivalent keys with underscores
+	//v.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
+	v.AutomaticEnv()
+
+	// Bind the current command's flags to viper
+	bindFlags(cmd, v)
+
+	return nil
+}
+
+func init() {
+	RootCmd.PersistentFlags().IntVarP(&appConf.Port, "port", "p", 7070, "Port number to run server")
+	RootCmd.PersistentFlags().StringVarP(&appConf.Host, "host", "a", "127.0.0.1", "Host address to run server")
+	RootCmd.PersistentFlags().StringVarP(&appConf.ConfigPath, "config", "c", "./config.yaml", "Configuration file path")
+	RootCmd.PersistentFlags().BoolVarP(&appConf.IsVerbose, "verbose", "v", false, "Use verbose output")
+	RootCmd.PersistentFlags().BoolVarP(&appConf.IsDebug, "debug", "d", false, "Use debug output. Disable headless browser")
+	RootCmd.PersistentFlags().BoolVarP(&appConf.IsBrowserHead, "head", "", false, "Enable browser UI")
+	RootCmd.PersistentFlags().BoolVarP(&appConf.IsLeakless, "leakless", "l", false, "Use leakless mode to insure browser instances are closed after search")
+	RootCmd.PersistentFlags().BoolVarP(&appConf.IsRawRequests, "raw", "r", false, "Disable browser usage, use HTTP requests")
+}