parameters

Author: xybydy

Gopkg.lock

@@ -1,15 +1,17 @@
 package config
 
 import (
+	"fmt"
 	"github.com/spf13/cobra"
+	"jrs/config"
 )
 
 var (
 	url string
 	api string
 )
 
-var config = &cobra.Command{
+var Config = &cobra.Command{
 	Use: "config",
 }
 
@@ -31,12 +33,16 @@ var jackett = &cobra.Command{
 
 var save = &cobra.Command{
 	Use: "save",
+	Run: func(cmd *cobra.Command, args []string) {
+		fmt.Println(args[0])
+		config.Params.SaveFile(args[0])
+	},
 }
 
 func init() {
 	radarr.AddCommand(save)
 	sonarr.AddCommand(save)
 	jackett.AddCommand(save)
 	set.AddCommand(radarr, sonarr, jackett)
-	config.AddCommand(set)
+	Config.AddCommand(set)
 }

cmd/config/flag.go

@@ -1,23 +1,25 @@
 package cmd
 
 import (
-	"flag"
 	"github.com/spf13/cobra"
+	cfg "jrs/cmd/config"
+	"jrs/cmd/jackett"
+	"jrs/cmd/radarr"
+	"jrs/cmd/sonarr"
+	"jrs/config"
 )
 
-func ParseOptions() {
-	var configPath string
-	flag.StringVar(&configPath, "config", "config.toml", "Config file path")
-
-}
-
-var RadarrCmd = &cobra.Command{
-	Use:   "radarr",
-	Short: "Radarr commands",
+var RootCmd = &cobra.Command{Use: "jrs",
+	Run: func(cmd *cobra.Command, args []string) {
+	},
 }
 
-var RootCmd = &cobra.Command{Use: "jrs"}
-
 func init() {
-	RootCmd.AddCommand(RadarrCmd)
+	RootCmd.PersistentFlags().StringVarP(&config.ConfPath, "config", "c", "config.toml", "Config file path")
+	config.ParseConfigFile()
+
+	RootCmd.AddCommand(radarr.Cmd)
+	RootCmd.AddCommand(sonarr.Cmd)
+	RootCmd.AddCommand(jackett.Cmd)
+	RootCmd.AddCommand(cfg.Config)
 }

cmd/flag.go

@@ -0,0 +1,15 @@
+package jackett
+
+import (
+	"github.com/spf13/cobra"
+)
+
+var Cmd = &cobra.Command{
+	Use:   "jackett",
+	Short: "Jackett commands",
+}
+
+func init() {
+	Cmd.Flags().StringP("url", "u", "http://localhost:9117", "Sonarr URL")
+	Cmd.Flags().StringP("api", "a", "", "API Key")
+}

cmd/jackett/flag.go

@@ -1,23 +1,24 @@
 package radarr
 
 import (
-	"flag"
+	"fmt"
 	"github.com/spf13/cobra"
 )
 
-func ParseOptions() {
-	var configPath string
-	flag.StringVar(&configPath, "config", "config.toml", "Config file path")
-
-}
-
-var RadarrCmd = &cobra.Command{
+var Cmd = &cobra.Command{
 	Use:   "radarr",
 	Short: "Radarr commands",
+	Run: func(cmd *cobra.Command, args []string) {
+		fmt.Println(cmd.Flag("config").Value)
+	},
 }
 
-var RootCmd = &cobra.Command{Use: "jrs"}
-
 func init() {
-	RootCmd.AddCommand(RadarrCmd)
+	// url := Cmd.Flags().StringP("url", "u", "http://localhost:7878", "Radarr URL")
+	// api := Cmd.Flags().StringP("api", "a", "", "API Key")
+	//
+	// if *url != "" {
+	// 	matched := utils.SplitUrl(*url)
+	//
+	// }
 }

cmd/radarr/flag.go

@@ -0,0 +1,15 @@
+package sonarr
+
+import (
+	"github.com/spf13/cobra"
+)
+
+var Cmd = &cobra.Command{
+	Use:   "sonarr",
+	Short: "Sonarr commands",
+}
+
+func init() {
+	Cmd.Flags().StringP("url", "u", "http://localhost:8989", "Sonarr URL")
+	Cmd.Flags().StringP("api", "a", "", "API Key")
+}

cmd/sonarr/flag.go

@@ -1,23 +1,28 @@
 package config
 
-import "strings"
+import (
+	"github.com/burntsushi/toml"
+	"log"
+	"os"
+	"path/filepath"
+	"reflect"
+	"strings"
+)
+
+var (
+	ConfPath = "config.toml"
+	Params   *Config
+)
 
 type Config struct {
-	Jackett Jackett
-	Dest    []Destination `toml:"destinations"`
+	Dest []Destination `toml:"destinations"`
 }
 
 type Destination struct {
 	Name string
-	IP   string
-	Port int
-	API  string
-}
-
-type Jackett struct {
-	IP   string
-	Port int
-	API  string
+	Ip   string
+	Port uint16
+	Api  string
 }
 
 func (c *Config) GetDestination(name string) *Destination {
@@ -29,4 +34,80 @@ func (c *Config) GetDestination(name string) *Destination {
 	return nil
 }
 
-var Params *Config
+func (c *Config) SaveFile(path string) error {
+	ext := filepath.Ext(path)
+	if ext == "" {
+		path = path + ".toml"
+	} else if ext != ".toml" {
+		log.Fatal("Config file does not have toml extension, please correct it")
+	}
+
+	f, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
+
+	defer f.Close()
+	if err != nil {
+		return err
+	}
+
+	if err := toml.NewEncoder(f).Encode(c); err != nil {
+		log.Fatal(err)
+		return err
+	}
+
+	return nil
+}
+
+func (c *Config) ChangeParams(dest, param string, value interface{}) {
+	main := reflect.ValueOf(c)
+	mainElem := main.Elem()
+	items := mainElem.FieldByName("Dest")
+
+	for i := 0; i < items.Len(); i++ {
+		item := items.Index(i)
+		if item.FieldByName("Name").String() == strings.Title(dest) {
+			field := item.FieldByName(strings.Title(param))
+			if field.IsValid() {
+				if field.CanSet() {
+					if field.Kind() == reflect.Uint16 {
+						if reflect.TypeOf(value).String() == "int" {
+							newValue := uint64(value.(int))
+							field.SetUint(newValue)
+						}
+					}
+					if field.Kind() == reflect.String {
+						newValue := value.(string)
+						field.SetString(newValue)
+					}
+				}
+			}
+
+		}
+	}
+}
+
+func ParseConfigFile() {
+	if _, err := toml.DecodeFile(ConfPath, &Params); err != nil {
+		log.Fatal(err)
+	}
+
+	if len(Params.Dest) < 1 {
+		log.Fatal("There is no Destination configured.")
+	}
+
+	for _, dest := range Params.Dest {
+		if dest.Api == "" {
+			param := os.Getenv(strings.ToUpper(dest.Name))
+
+			if param != "" {
+				Params.GetDestination(dest.Name).Api = param
+			} else {
+				if len(Params.Dest) == 1 {
+					log.Fatalf("There is no %s API configured.", dest.Name)
+				} else {
+					log.Printf("There is no %s API configured.", dest.Name)
+				}
+			}
+		}
+
+	}
+}