feat: restructure code; add size and ext checks (#39)

- Refractor code
- Add file size and extension checkers
- Change repair workflow to use zurg

pkg/arr/content.go

@@ -3,7 +3,9 @@ package arr
 import (
 	"encoding/json"
 	"fmt"
+	"log"
 	"net/http"
+	"strconv"
 )
 
 func (a *Arr) GetMedia(tvId string) ([]Content, error) {
@@ -14,7 +16,7 @@ func (a *Arr) GetMedia(tvId string) ([]Content, error) {
 	}
 	if resp.StatusCode == http.StatusNotFound {
 		// This is Radarr
-		repairLogger.Info().Msg("Radarr detected")
+		log.Println("Radarr detected")
 		a.Type = Radarr
 		return GetMovies(a, tvId)
 	}
@@ -44,11 +46,34 @@ func (a *Arr) GetMedia(tvId string) ([]Content, error) {
 			Title: d.Title,
 			Id:    d.Id,
 		}
-		files := make([]contentFile, 0)
+
+		type episode struct {
+			Id            int `json:"id"`
+			EpisodeFileID int `json:"episodeFileId"`
+		}
+		resp, err = a.Request(http.MethodGet, fmt.Sprintf("api/v3/episode?seriesId=%d", d.Id), nil)
+		if err != nil {
+			continue
+		}
+		defer resp.Body.Close()
+		var episodes []episode
+		if err = json.NewDecoder(resp.Body).Decode(&episodes); err != nil {
+			continue
+		}
+		episodeFileIDMap := make(map[int]int)
+		for _, e := range episodes {
+			episodeFileIDMap[e.EpisodeFileID] = e.Id
+		}
+		files := make([]ContentFile, 0)
 		for _, file := range seriesFiles {
-			files = append(files, contentFile{
-				Id:   file.Id,
-				Path: file.Path,
+			eId, ok := episodeFileIDMap[file.Id]
+			if !ok {
+				eId = 0
+			}
+			files = append(files, ContentFile{
+				FileId: file.Id,
+				Path:   file.Path,
+				Id:     eId,
 			})
 		}
 		ct.Files = files
@@ -73,10 +98,11 @@ func GetMovies(a *Arr, tvId string) ([]Content, error) {
 			Title: movie.Title,
 			Id:    movie.Id,
 		}
-		files := make([]contentFile, 0)
-		files = append(files, contentFile{
-			Id:   movie.MovieFile.Id,
-			Path: movie.MovieFile.Path,
+		files := make([]ContentFile, 0)
+		files = append(files, ContentFile{
+			FileId: movie.MovieFile.Id,
+			Id:     movie.Id,
+			Path:   movie.MovieFile.Path,
 		})
 		ct.Files = files
 		contents = append(contents, ct)
@@ -84,15 +110,69 @@ func GetMovies(a *Arr, tvId string) ([]Content, error) {
 	return contents, nil
 }
 
-func (a *Arr) DeleteFile(id int) error {
+func (a *Arr) SearchMissing(files []ContentFile) error {
+	var payload interface{}
+
+	ids := make([]int, 0)
+	for _, f := range files {
+		ids = append(ids, f.Id)
+	}
+
 	switch a.Type {
 	case Sonarr:
-		_, err := a.Request(http.MethodDelete, fmt.Sprintf("api/v3/episodefile/%d", id), nil)
+		payload = struct {
+			Name       string `json:"name"`
+			EpisodeIds []int  `json:"episodeIds"`
+		}{
+			Name:       "EpisodeSearch",
+			EpisodeIds: ids,
+		}
+	case Radarr:
+		payload = struct {
+			Name     string `json:"name"`
+			MovieIds []int  `json:"movieIds"`
+		}{
+			Name:     "MoviesSearch",
+			MovieIds: ids,
+		}
+	default:
+		return fmt.Errorf("unknown arr type: %s", a.Type)
+	}
+
+	resp, err := a.Request(http.MethodPost, "api/v3/command", payload)
+	if err != nil {
+		return fmt.Errorf("failed to search missing: %v", err)
+	}
+	if statusOk := strconv.Itoa(resp.StatusCode)[0] == '2'; !statusOk {
+		return fmt.Errorf("failed to search missing. Status Code: %s", resp.Status)
+	}
+	return nil
+}
+
+func (a *Arr) DeleteFiles(files []ContentFile) error {
+	ids := make([]int, 0)
+	for _, f := range files {
+		ids = append(ids, f.FileId)
+	}
+	var payload interface{}
+	switch a.Type {
+	case Sonarr:
+		payload = struct {
+			EpisodeFileIds []int `json:"episodeFileIds"`
+		}{
+			EpisodeFileIds: ids,
+		}
+		_, err := a.Request(http.MethodDelete, "api/v3/episodefile/bulk", payload)
 		if err != nil {
 			return err
 		}
 	case Radarr:
-		_, err := a.Request(http.MethodDelete, fmt.Sprintf("api/v3/moviefile/%d", id), nil)
+		payload = struct {
+			MovieFileIds []int `json:"movieFileIds"`
+		}{
+			MovieFileIds: ids,
+		}
+		_, err := a.Request(http.MethodDelete, "api/v3/moviefile/bulk", payload)
 		if err != nil {
 			return err
 		}