Skip to content

Commit 2f1c80a

Browse files
xybydyxybydy
committed
v0.0.1
1 parent 29df0a3 commit 2f1c80a

File tree

9 files changed

+135
-227
lines changed

9 files changed

+135
-227
lines changed

‎cmd/config/flag.go‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
11
package config
22

33
import (
4-
"github.com/spf13/cobra"
54
"jrs/config"
5+
6+
"github.com/spf13/cobra"
67
)
78

89
var (

‎cmd/jackett/flag.go‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
11
package jackett
22

33
import (
4-
"github.com/spf13/cobra"
54
cmd2 "jrs/cmd"
65
"jrs/config"
76
"jrs/pkg/jackett"
7+
8+
"github.com/spf13/cobra"
89
)
910

1011
var (
@@ -67,7 +68,7 @@ var addAllPublicIndexers = &cobra.Command{
6768
}
6869

6970
func init() {
70-
Cmd.PersistentFlags().StringVarP(&url, "url", "u", "http://localhost:9117", "Jackett URL")
71+
Cmd.PersistentFlags().StringVarP(&url, "url", "u", "", "Jackett URL")
7172
Cmd.PersistentFlags().StringVarP(&api, "api", "a", "", "API Key")
7273

7374
addIndexer.Flags().StringVarP(&user, "username", "s", "", "Tracker Username")

‎cmd/radarr/flag.go‎

Lines changed: 47 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,19 @@
11
package radarr
22

33
import (
4-
"github.com/spf13/cobra"
54
cmd2 "jrs/cmd"
65
"jrs/config"
6+
"jrs/pkg/jackett"
77
"jrs/pkg/radarr"
8+
9+
"github.com/spf13/cobra"
810
)
911

1012
var (
1113
url string
1214
api string
13-
app *radarr.Radarr
15+
app *radarr.Client
16+
j *jackett.Jackett
1417
)
1518

1619
var Cmd = &cobra.Command{
@@ -20,23 +23,56 @@ var Cmd = &cobra.Command{
2023
cmd2.CheckConfig("radarr", url, api)
2124
},
2225
PersistentPreRun: func(cmd *cobra.Command, args []string) {
26+
if url != "" {
27+
config.Params.ChangeParams("radarr", "path", url)
28+
}
29+
if api != "" {
30+
config.Params.ChangeParams("radarr", "api", api)
31+
}
32+
2333
createApp()
2434
},
2535
}
2636

37+
var testIndexers = &cobra.Command{
38+
Use: "test",
39+
Short: "Test all indexers added to Radarr",
40+
Run: func(cmd *cobra.Command, args []string) {
41+
app.TestAllIndexers()
42+
},
43+
}
44+
45+
var addAllIndexers = &cobra.Command{
46+
Use: "add",
47+
Short: "Add all available indexers to Radarr",
48+
Run: func(cmd *cobra.Command, args []string) {
49+
app.AddAllIndexers(j)
50+
},
51+
PreRun: func(cmd *cobra.Command, args []string) {
52+
if j == nil {
53+
j = jackett.New(config.Params)
54+
}
55+
},
56+
}
57+
58+
var deleteAllIndexers = &cobra.Command{
59+
Use: "delete",
60+
Short: "Delete all indexers in Radarr",
61+
Run: func(cmd *cobra.Command, args []string) {
62+
app.DeleteAllIndexers()
63+
},
64+
}
65+
2766
func createApp() {
2867
if app == nil {
29-
app = radarr.New(config.Params)
68+
app = radarr.NewClient()
3069
}
3170
}
3271

3372
func init() {
34-
url := Cmd.PersistentFlags().StringP("url", "u", "http://localhost:7878", "Radarr URL")
35-
api := Cmd.PersistentFlags().StringP("api", "a", "", "API Key")
36-
37-
//
38-
// if *url != "" {
39-
// matched := utils.SplitUrl(*url)
40-
//
41-
// }
73+
Cmd.PersistentFlags().StringVarP(&url, "url", "u", "", "Radarr URL")
74+
Cmd.PersistentFlags().StringVarP(&api, "api", "a", "", "API Key")
75+
76+
Cmd.AddCommand(testIndexers, addAllIndexers, deleteAllIndexers)
77+
4278
}

‎cmd/root/flag.go‎

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
11
package root
22

33
import (
4-
"github.com/spf13/cobra"
54
cfg "jrs/cmd/config"
65
"jrs/cmd/jackett"
76
"jrs/cmd/radarr"
87
"jrs/cmd/sonarr"
98
"jrs/config"
9+
10+
"github.com/spf13/cobra"
1011
)
1112

1213
var RootCmd = &cobra.Command{Use: "jrs",
@@ -17,9 +18,5 @@ var RootCmd = &cobra.Command{Use: "jrs",
1718
func init() {
1819
RootCmd.PersistentFlags().StringVarP(&config.ConfPath, "config", "c", "config.toml", "Config file path")
1920
config.ParseConfigFile()
20-
21-
RootCmd.AddCommand(radarr.Cmd)
22-
RootCmd.AddCommand(sonarr.Cmd)
23-
RootCmd.AddCommand(jackett.Cmd)
24-
RootCmd.AddCommand(cfg.Config)
21+
RootCmd.AddCommand(radarr.Cmd, sonarr.Cmd, jackett.Cmd, cfg.Config)
2522
}

‎cmd/sonarr/flag.go‎

Lines changed: 62 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,77 @@
11
package sonarr
22

33
import (
4+
cmd2 "jrs/cmd"
5+
"jrs/config"
6+
"jrs/pkg/jackett"
7+
"jrs/pkg/sonarr"
8+
49
"github.com/spf13/cobra"
510
)
611

12+
var (
13+
url string
14+
api string
15+
app *sonarr.Client
16+
j *jackett.Jackett
17+
)
18+
719
var Cmd = &cobra.Command{
820
Use: "sonarr",
921
Short: "Sonarr commands",
1022
Run: func(cmd *cobra.Command, args []string) {
23+
cmd2.CheckConfig("sonarr", url, api)
24+
},
25+
PersistentPreRun: func(cmd *cobra.Command, args []string) {
26+
if url != "" {
27+
config.Params.ChangeParams("sonarr", "path", url)
28+
}
29+
if api != "" {
30+
config.Params.ChangeParams("sonarr", "api", api)
31+
}
32+
33+
createApp()
1134
},
1235
}
1336

37+
var testIndexers = &cobra.Command{
38+
Use: "test",
39+
Short: "Test all indexers added to Radarr",
40+
Run: func(cmd *cobra.Command, args []string) {
41+
app.TestAllIndexers()
42+
},
43+
}
44+
45+
var addAllIndexers = &cobra.Command{
46+
Use: "add",
47+
Short: "Add all available indexers to Radarr",
48+
Run: func(cmd *cobra.Command, args []string) {
49+
app.AddAllIndexers(j)
50+
},
51+
PreRun: func(cmd *cobra.Command, args []string) {
52+
if j == nil {
53+
j = jackett.New(config.Params)
54+
}
55+
},
56+
}
57+
58+
var deleteAllIndexers = &cobra.Command{
59+
Use: "delete",
60+
Short: "Delete all indexers in Radarr",
61+
Run: func(cmd *cobra.Command, args []string) {
62+
app.DeleteAllIndexers()
63+
},
64+
}
65+
66+
func createApp() {
67+
if app == nil {
68+
app = sonarr.NewClient()
69+
}
70+
}
71+
1472
func init() {
15-
Cmd.Flags().StringP("url", "u", "http://localhost:8989", "Sonarr URL")
16-
Cmd.Flags().StringP("api", "a", "", "API Key")
73+
Cmd.Flags().StringVarP(&url, "url", "u", "", "Sonarr URL")
74+
Cmd.Flags().StringVarP(&api, "api", "a", "", "API Key")
75+
76+
Cmd.AddCommand(testIndexers, addAllIndexers, deleteAllIndexers)
1777
}

0 commit comments

Comments
 (0)