Add repair worker

pkg/arr/content.go

@@ -3,27 +3,102 @@ package arr
 import (
 	"encoding/json"
 	"fmt"
+	"log"
 	"net/http"
 )
 
-type ContentRequest struct {
-	ID    string `json:"id"`
-	Title string `json:"name"`
-	Arr   string `json:"arr"`
+func (a *Arr) GetMedia(tvId string) ([]Content, error) {
+	// Get series
+	resp, err := a.Request(http.MethodGet, fmt.Sprintf("api/v3/series?tvdbId=%s", tvId), nil)
+	if err != nil {
+		return nil, err
+	}
+	if resp.StatusCode == http.StatusNotFound {
+		// This is Radarr
+		log.Printf("Radarr detected\n")
+		a.Type = Radarr
+		return GetMovies(a, tvId)
+	}
+	a.Type = Sonarr
+	defer resp.Body.Close()
+	type series struct {
+		Title string `json:"title"`
+		Id    int    `json:"id"`
+	}
+	var data []series
+	if err = json.NewDecoder(resp.Body).Decode(&data); err != nil {
+		return nil, err
+	}
+	// Get series files
+	contents := make([]Content, 0)
+	for _, d := range data {
+		resp, err = a.Request(http.MethodGet, fmt.Sprintf("api/v3/episodefile?seriesId=%d", d.Id), nil)
+		if err != nil {
+			continue
+		}
+		defer resp.Body.Close()
+		var seriesFiles []seriesFile
+		if err = json.NewDecoder(resp.Body).Decode(&seriesFiles); err != nil {
+			continue
+		}
+		ct := Content{
+			Title: d.Title,
+			Id:    d.Id,
+		}
+		files := make([]contentFile, 0)
+		for _, file := range seriesFiles {
+			files = append(files, contentFile{
+				Id:   file.Id,
+				Path: file.Path,
+			})
+		}
+		ct.Files = files
+		contents = append(contents, ct)
+	}
+	return contents, nil
 }
 
-func (a *Arr) GetContents() *ContentRequest {
-	resp, err := a.Request(http.MethodGet, "api/v3/series", nil)
+func GetMovies(a *Arr, tvId string) ([]Content, error) {
+	resp, err := a.Request(http.MethodGet, fmt.Sprintf("api/v3/movie?tmdbId=%s", tvId), nil)
 	if err != nil {
-		return nil
+		return nil, err
 	}
 	defer resp.Body.Close()
-	var data *ContentRequest
-	if err = json.NewDecoder(resp.Body).Decode(&data); err != nil {
-		fmt.Printf("Error: %v\n", err)
-		return nil
+	var movies []Movie
+	if err = json.NewDecoder(resp.Body).Decode(&movies); err != nil {
+		return nil, err
 	}
-	fmt.Printf("Data: %v\n", data)
-	//data.Arr = a.Name
-	return data
+	contents := make([]Content, 0)
+	for _, movie := range movies {
+		ct := Content{
+			Title: movie.Title,
+			Id:    movie.Id,
+		}
+		files := make([]contentFile, 0)
+		files = append(files, contentFile{
+			Id:   movie.MovieFile.Id,
+			Path: movie.MovieFile.Path,
+		})
+		ct.Files = files
+		contents = append(contents, ct)
+	}
+	return contents, nil
+}
+
+func (a *Arr) DeleteFile(id int) error {
+	switch a.Type {
+	case Sonarr:
+		_, err := a.Request(http.MethodDelete, fmt.Sprintf("api/v3/episodefile/%d", id), nil)
+		if err != nil {
+			return err
+		}
+	case Radarr:
+		_, err := a.Request(http.MethodDelete, fmt.Sprintf("api/v3/moviefile/%d", id), nil)
+		if err != nil {
+			return err
+		}
+	default:
+		return fmt.Errorf("unknown arr type: %s", a.Type)
+	}
+	return nil
 }