143 lines
3.3 KiB
Go
143 lines
3.3 KiB
Go
package eraser
|
|
|
|
import (
|
|
"encoding/json"
|
|
"encoding/xml"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"git.linuxrocker.com/mattburchett/Housekeeper/pkg/config"
|
|
"git.linuxrocker.com/mattburchett/Housekeeper/pkg/model"
|
|
)
|
|
|
|
// LookupMovieFileLocation will gather a list of Information based on IDs returned by locator.GetTitles
|
|
func LookupMovieFileLocation(config config.Config, ids []int) []string {
|
|
fileList := make([]string, 0)
|
|
|
|
for _, i := range ids {
|
|
plexURL := fmt.Sprintf("%s:%d%s%d%s%s", config.PlexHost, config.PlexPort, "/library/metadata/", i, "/?X-Plex-Token=", config.PlexToken)
|
|
|
|
req, err := http.NewRequest(http.MethodGet, plexURL, nil)
|
|
|
|
httpClient := http.Client{}
|
|
req.Header.Set("User-Agent", "Housekeeper")
|
|
|
|
res, getErr := httpClient.Do(req)
|
|
if getErr != nil {
|
|
log.Fatal(getErr)
|
|
}
|
|
|
|
body, readErr := ioutil.ReadAll(res.Body)
|
|
if readErr != nil {
|
|
log.Fatal(readErr)
|
|
}
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
plexModel := model.XMLPlexMovieAPI{}
|
|
xml.Unmarshal(body, &plexModel)
|
|
fileList = append(fileList, filepath.Dir(plexModel.Video.Media.Part.File))
|
|
}
|
|
return fileList
|
|
}
|
|
|
|
// LookupTVFileLocation will gather a list of Information based on IDs returned by locator.GetTitles
|
|
func LookupTVFileLocation(config config.Config, ids []int) []string {
|
|
fileList := make([]string, 0)
|
|
m := make(map[string]bool)
|
|
results := make([]string, 0)
|
|
|
|
for _, i := range ids {
|
|
plexURL := fmt.Sprintf("%s:%d%s%d%s%s", config.PlexHost, config.PlexPort, "/library/metadata/", i, "/allLeaves/?X-Plex-Token=", config.PlexToken)
|
|
|
|
req, err := http.NewRequest(http.MethodGet, plexURL, nil)
|
|
|
|
httpClient := http.Client{}
|
|
req.Header.Set("User-Agent", "Housekeeper")
|
|
|
|
res, getErr := httpClient.Do(req)
|
|
if getErr != nil {
|
|
log.Fatal(getErr)
|
|
}
|
|
|
|
body, readErr := ioutil.ReadAll(res.Body)
|
|
if readErr != nil {
|
|
log.Fatal(readErr)
|
|
}
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
plexModel := model.XMLPlexTVAPI{}
|
|
xml.Unmarshal(body, &plexModel)
|
|
|
|
plexTV := plexModel.Video
|
|
|
|
for _, i := range plexTV {
|
|
fileList = append(fileList, filepath.Dir(i.Media.Part.File))
|
|
}
|
|
|
|
for _, r := range fileList {
|
|
if _, ok := m[r]; !ok {
|
|
m[r] = true
|
|
results = append(results, r)
|
|
}
|
|
}
|
|
|
|
}
|
|
return results
|
|
}
|
|
|
|
func DeleteSeriesFromSonarr(config config.Config, ids []int) {
|
|
for _, i := range ids {
|
|
sonarrURL := fmt.Sprintf("%s%s%s%d%s%s", config.BaseURL, config.SonarrContext, "/api/series/", i, "/?deleteFiles=true&apikey=", config.SonarrAPIKey)
|
|
req, err := http.NewRequest(http.MethodDelete, sonarrURL, nil)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
httpClient := http.Client{}
|
|
req.Header.Set("User-Agent", "Housekeeper")
|
|
|
|
res, getErr := httpClient.Do(req)
|
|
if getErr != nil {
|
|
log.Fatal(getErr)
|
|
}
|
|
|
|
body, readErr := ioutil.ReadAll(res.Body)
|
|
if readErr != nil {
|
|
log.Fatal(readErr)
|
|
}
|
|
|
|
deleteModel := model.SonarrResponse{}
|
|
jsonErr := json.Unmarshal(body, &deleteModel)
|
|
if jsonErr != nil {
|
|
log.Fatal(jsonErr)
|
|
}
|
|
|
|
// if strings.Contains("does not exist", deleteModel.Message) {
|
|
// log.Printf("The following ID does not exist: %v", i)
|
|
// }
|
|
}
|
|
}
|
|
|
|
// DeleteFiles will actually perform the deletion.
|
|
func DeleteFiles(delete bool, files []string) error {
|
|
var err error
|
|
if delete {
|
|
for _, i := range files {
|
|
fmt.Printf("Removing %v\n", i)
|
|
err = os.RemoveAll(i)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
return err
|
|
}
|