/
manualimport.go
107 lines (90 loc) · 3.86 KB
/
manualimport.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package radarr
import (
"bytes"
"context"
"encoding/json"
"fmt"
"net/url"
"golift.io/starr"
)
// Define Base Path for Manual Import calls.
const bpManualImport = APIver + "/manualimport"
// ManualImportInput is the input data for a manual import request using a POST request.
type ManualImportInput struct {
ID int64 `json:"id"`
Path string `json:"path"`
MovieID int64 `json:"movieId"`
Movie *Movie `json:"movie"`
Quality *starr.Quality `json:"quality"`
Languages []*starr.Value `json:"languages"`
ReleaseGroup string `json:"releaseGroup"`
DownloadID string `json:"downloadId"`
CustomFormats []*CustomFormatInput `json:"customFormats"`
CustomFormatScore int64 `json:"customFormatScore"`
Rejections []*Rejection `json:"rejections"`
}
// ManualImportOutput is the output data for a manual import request.
type ManualImportOutput struct {
ID int64 `json:"id"`
Path string `json:"path"`
RelativePath string `json:"relativePath"`
FolderName string `json:"folderName"`
Name string `json:"name"`
Size int `json:"size"`
Movie *Movie `json:"movie"`
Quality *starr.Quality `json:"quality"`
Languages []*starr.Value `json:"languages"`
ReleaseGroup string `json:"releaseGroup"`
QualityWeight int64 `json:"qualityWeight"`
DownloadID string `json:"downloadId"`
CustomFormats []*CustomFormatOutput `json:"customFormats"`
CustomFormatScore int64 `json:"customFormatScore"`
Rejections []*Rejection `json:"rejections"`
}
// Rejection is part of the manual import payload.
type Rejection struct {
Reason string `json:"reason"`
// permanent or temporary
Type string `json:"type"`
}
// ManualImportParams provides the input parameters for the GET /manualimport API.
type ManualImportParams struct {
Folder string
DownloadID string
MovieID int64
FilterExistingFiles bool
}
// ManualImport initiates a manual import (GET).
func (r *Radarr) ManualImport(params *ManualImportParams) (*ManualImportOutput, error) {
return r.ManualImportContext(context.Background(), params)
}
// ManualImportContext initiates a manual import (GET).
func (r *Radarr) ManualImportContext(ctx context.Context, params *ManualImportParams) (*ManualImportOutput, error) {
var output ManualImportOutput
req := starr.Request{URI: bpManualImport, Query: make(url.Values)}
req.Query.Add("folder", params.Folder)
req.Query.Add("downloadId", params.DownloadID)
req.Query.Add("movieId", fmt.Sprint(params.MovieID))
req.Query.Add("filterExistingFiles", fmt.Sprint(params.FilterExistingFiles))
if err := r.GetInto(ctx, req, &output); err != nil {
return nil, fmt.Errorf("api.Get(%s): %w", &req, err)
}
return &output, nil
}
// ManualImportReprocess reprocesses a manual import (POST).
func (r *Radarr) ManualImportReprocess(manualimport *ManualImportInput) error {
return r.ManualImportReprocessContext(context.Background(), manualimport)
}
// ManualImportReprocessContext reprocesses a manual import (POST).
func (r *Radarr) ManualImportReprocessContext(ctx context.Context, manualimport *ManualImportInput) error {
var output interface{}
var body bytes.Buffer
if err := json.NewEncoder(&body).Encode(manualimport); err != nil {
return fmt.Errorf("json.Marshal(%s): %w", bpManualImport, err)
}
req := starr.Request{URI: bpManualImport, Body: &body}
if err := r.PostInto(ctx, req, &output); err != nil {
return fmt.Errorf("api.Post(%s): %w", &req, err)
}
return nil
}