-
Notifications
You must be signed in to change notification settings - Fork 18
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* feat(sonarr): add downloadclientconfig * feat(sonarr): add downloadclient * feat(sonarr): add remotepathmapping * fix: align comments fix typos and path joining * style: forgot a path join * fix: int64 type and few more typo * fix: wrong assertion in test
- Loading branch information
Showing
21 changed files
with
1,283 additions
and
80 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,143 @@ | ||
| package sonarr | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "path" | ||
|
|
||
| "golift.io/starr" | ||
| ) | ||
|
|
||
| // Define Base Path for download client calls. | ||
| const bpDownloadClient = APIver + "/downloadClient" | ||
|
|
||
| // DownloadClientInput is the input for a new or updated download client. | ||
| type DownloadClientInput struct { | ||
| Enable bool `json:"enable"` | ||
| RemoveCompletedDownloads bool `json:"removeCompletedDownloads"` | ||
| RemoveFailedDownloads bool `json:"removeFailedDownloads"` | ||
| Priority int `json:"priority"` | ||
| ID int64 `json:"id,omitempty"` | ||
| ConfigContract string `json:"configContract"` | ||
| Implementation string `json:"implementation"` | ||
| Name string `json:"name"` | ||
| Protocol string `json:"protocol"` | ||
| Tags []int `json:"tags"` | ||
| Fields []*starr.FieldInput `json:"fields"` | ||
| } | ||
|
|
||
| // DownloadClientOutput is the output from the download client methods. | ||
| type DownloadClientOutput struct { | ||
| Enable bool `json:"enable"` | ||
| RemoveCompletedDownloads bool `json:"removeCompletedDownloads"` | ||
| RemoveFailedDownloads bool `json:"removeFailedDownloads"` | ||
| Priority int `json:"priority"` | ||
| ID int64 `json:"id,omitempty"` | ||
| ConfigContract string `json:"configContract"` | ||
| Implementation string `json:"implementation"` | ||
| ImplementationName string `json:"implementationName"` | ||
| InfoLink string `json:"infoLink"` | ||
| Name string `json:"name"` | ||
| Protocol string `json:"protocol"` | ||
| Tags []int `json:"tags"` | ||
| Fields []*starr.FieldOutput `json:"fields"` | ||
| } | ||
|
|
||
| // GetDownloadClients returns all configured download clients. | ||
| func (s *Sonarr) GetDownloadClients() ([]*DownloadClientOutput, error) { | ||
| return s.GetDownloadClientsContext(context.Background()) | ||
| } | ||
|
|
||
| // GetDownloadClientsContext returns all configured download clients. | ||
| func (s *Sonarr) GetDownloadClientsContext(ctx context.Context) ([]*DownloadClientOutput, error) { | ||
| var output []*DownloadClientOutput | ||
|
|
||
| req := starr.Request{URI: bpDownloadClient} | ||
| if err := s.GetInto(ctx, req, &output); err != nil { | ||
| return nil, fmt.Errorf("api.Get(%s): %w", &req, err) | ||
| } | ||
|
|
||
| return output, nil | ||
| } | ||
|
|
||
| // GetDownloadClient returns a single download client. | ||
| func (s *Sonarr) GetDownloadClient(downloadclientID int64) (*DownloadClientOutput, error) { | ||
| return s.GetDownloadClientContext(context.Background(), downloadclientID) | ||
| } | ||
|
|
||
| // GetDownloadClientContext returns a single download client. | ||
| func (s *Sonarr) GetDownloadClientContext(ctx context.Context, downloadclientID int64) (*DownloadClientOutput, error) { | ||
| var output DownloadClientOutput | ||
|
|
||
| req := starr.Request{URI: path.Join(bpDownloadClient, fmt.Sprint(downloadclientID))} | ||
| if err := s.GetInto(ctx, req, &output); err != nil { | ||
| return nil, fmt.Errorf("api.Get(%s): %w", &req, err) | ||
| } | ||
|
|
||
| return &output, nil | ||
| } | ||
|
|
||
| // AddDownloadClient creates a download client. | ||
| func (s *Sonarr) AddDownloadClient(downloadclient *DownloadClientInput) (*DownloadClientOutput, error) { | ||
| return s.AddDownloadClientContext(context.Background(), downloadclient) | ||
| } | ||
|
|
||
| // AddDownloadClientContext creates a download client. | ||
| func (s *Sonarr) AddDownloadClientContext(ctx context.Context, | ||
| client *DownloadClientInput, | ||
| ) (*DownloadClientOutput, error) { | ||
| var output DownloadClientOutput | ||
|
|
||
| var body bytes.Buffer | ||
| if err := json.NewEncoder(&body).Encode(client); err != nil { | ||
| return nil, fmt.Errorf("json.Marshal(%s): %w", bpDownloadClient, err) | ||
| } | ||
|
|
||
| req := starr.Request{URI: bpDownloadClient, Body: &body} | ||
| if err := s.PostInto(ctx, req, &output); err != nil { | ||
| return nil, fmt.Errorf("api.Post(%s): %w", &req, err) | ||
| } | ||
|
|
||
| return &output, nil | ||
| } | ||
|
|
||
| // UpdateDownloadClient updates the download client. | ||
| func (s *Sonarr) UpdateDownloadClient(downloadclient *DownloadClientInput) (*DownloadClientOutput, error) { | ||
| return s.UpdateDownloadClientContext(context.Background(), downloadclient) | ||
| } | ||
|
|
||
| // UpdateDownloadClientContext updates the download client. | ||
| func (s *Sonarr) UpdateDownloadClientContext(ctx context.Context, | ||
| client *DownloadClientInput, | ||
| ) (*DownloadClientOutput, error) { | ||
| var output DownloadClientOutput | ||
|
|
||
| var body bytes.Buffer | ||
| if err := json.NewEncoder(&body).Encode(client); err != nil { | ||
| return nil, fmt.Errorf("json.Marshal(%s): %w", bpDownloadClient, err) | ||
| } | ||
|
|
||
| req := starr.Request{URI: path.Join(bpDownloadClient, fmt.Sprint(client.ID)), Body: &body} | ||
| if err := s.PutInto(ctx, req, &output); err != nil { | ||
| return nil, fmt.Errorf("api.Put(%s): %w", &req, err) | ||
| } | ||
|
|
||
| return &output, nil | ||
| } | ||
|
|
||
| // DeleteDownloadClient removes a single download client. | ||
| func (s *Sonarr) DeleteDownloadClient(downloadclientID int64) error { | ||
| return s.DeleteDownloadClientContext(context.Background(), downloadclientID) | ||
| } | ||
|
|
||
| // DeleteDownloadClientContext removes a single download client. | ||
| func (s *Sonarr) DeleteDownloadClientContext(ctx context.Context, downloadclientID int64) error { | ||
| req := starr.Request{URI: path.Join(bpDownloadClient, fmt.Sprint(downloadclientID))} | ||
| if err := s.DeleteAny(ctx, req); err != nil { | ||
| return fmt.Errorf("api.Delete(%s): %w", &req, err) | ||
| } | ||
|
|
||
| return nil | ||
| } |
Oops, something went wrong.