diff --git a/cmd/main.go b/cmd/main.go index ba5923d..c241282 100644 --- a/cmd/main.go +++ b/cmd/main.go @@ -2,6 +2,7 @@ package main import ( "flag" + "fmt" "log" "git.linuxrocker.com/mattburchett/Housekeeper/pkg/communicator" @@ -48,18 +49,17 @@ func main() { } } + files := eraser.LookupTVFileLocation(cfg, ids) + fmt.Printf("%v\n", files) + if delete { if libraryType == "movie" { - files := eraser.LookupFileLocation(cfg, ids) + files := eraser.LookupMovieFileLocation(cfg, ids) err = eraser.DeleteMovies(delete, files) if err != nil { log.Println(err) } - // } else if libraryType == "show" { - // err = eraser.DeleteTVShows() - // if err != nil { - // log.Println(err) - // } + } else if libraryType == "show" { } } } diff --git a/pkg/eraser/eraser.go b/pkg/eraser/eraser.go index 0209faf..ba40f0c 100644 --- a/pkg/eraser/eraser.go +++ b/pkg/eraser/eraser.go @@ -13,8 +13,8 @@ import ( "git.linuxrocker.com/mattburchett/Housekeeper/pkg/model" ) -// LookupFileLocation will gather a list of Information based on IDs returned by locator.GetTitles -func LookupFileLocation(config config.Config, ids []int) []string { +// 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 { @@ -38,13 +38,50 @@ func LookupFileLocation(config config.Config, ids []int) []string { if err != nil { log.Fatal(err) } - plexModel := model.XMLPlexAPI{} + 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) + + 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.XMLPlexTVAPI{} + xml.Unmarshal(body, &plexModel) + + plexTV := plexModel.Video + + for _, i := range plexTV { + fileList = append(fileList, filepath.Dir(filepath.Dir(i.Media.Part.File))) + } + } + return fileList +} + // DeleteMovies will actually perform the deletion. func DeleteMovies(delete bool, files []string) error { var err error diff --git a/pkg/model/eraser_model.go b/pkg/model/eraser_model.go index ed7e883..a79a5f1 100644 --- a/pkg/model/eraser_model.go +++ b/pkg/model/eraser_model.go @@ -2,8 +2,8 @@ package model import "encoding/xml" -// XMLPlexAPI - This is the XML version of the struct below it. -type XMLPlexAPI struct { +// XMLPlexMovieAPI - This is the XML version of the Library. +type XMLPlexMovieAPI struct { XMLName xml.Name `xml:"MediaContainer"` Text string `xml:",chardata"` Size string `xml:"size,attr"` @@ -138,3 +138,95 @@ type XMLPlexAPI struct { } `xml:"Role"` } `xml:"Video"` } + +// XMLPlexTVAPI - This is the XML version of the Library. +type XMLPlexTVAPI struct { + XMLName xml.Name `xml:"MediaContainer"` + Text string `xml:",chardata"` + Size string `xml:"size,attr"` + AllowSync string `xml:"allowSync,attr"` + Art string `xml:"art,attr"` + Banner string `xml:"banner,attr"` + Identifier string `xml:"identifier,attr"` + Key string `xml:"key,attr"` + LibrarySectionID string `xml:"librarySectionID,attr"` + LibrarySectionTitle string `xml:"librarySectionTitle,attr"` + LibrarySectionUUID string `xml:"librarySectionUUID,attr"` + MediaTagPrefix string `xml:"mediaTagPrefix,attr"` + MediaTagVersion string `xml:"mediaTagVersion,attr"` + MixedParents string `xml:"mixedParents,attr"` + Nocache string `xml:"nocache,attr"` + ParentIndex string `xml:"parentIndex,attr"` + ParentTitle string `xml:"parentTitle,attr"` + ParentYear string `xml:"parentYear,attr"` + Theme string `xml:"theme,attr"` + Title1 string `xml:"title1,attr"` + Title2 string `xml:"title2,attr"` + ViewGroup string `xml:"viewGroup,attr"` + ViewMode string `xml:"viewMode,attr"` + Video []struct { + Text string `xml:",chardata"` + RatingKey string `xml:"ratingKey,attr"` + Key string `xml:"key,attr"` + ParentRatingKey string `xml:"parentRatingKey,attr"` + GrandparentRatingKey string `xml:"grandparentRatingKey,attr"` + Studio string `xml:"studio,attr"` + Type string `xml:"type,attr"` + Title string `xml:"title,attr"` + GrandparentKey string `xml:"grandparentKey,attr"` + ParentKey string `xml:"parentKey,attr"` + GrandparentTitle string `xml:"grandparentTitle,attr"` + ParentTitle string `xml:"parentTitle,attr"` + ContentRating string `xml:"contentRating,attr"` + Summary string `xml:"summary,attr"` + Index string `xml:"index,attr"` + ParentIndex string `xml:"parentIndex,attr"` + Rating string `xml:"rating,attr"` + Year string `xml:"year,attr"` + Thumb string `xml:"thumb,attr"` + Art string `xml:"art,attr"` + ParentThumb string `xml:"parentThumb,attr"` + GrandparentThumb string `xml:"grandparentThumb,attr"` + GrandparentArt string `xml:"grandparentArt,attr"` + GrandparentTheme string `xml:"grandparentTheme,attr"` + Duration string `xml:"duration,attr"` + OriginallyAvailableAt string `xml:"originallyAvailableAt,attr"` + AddedAt string `xml:"addedAt,attr"` + UpdatedAt string `xml:"updatedAt,attr"` + TitleSort string `xml:"titleSort,attr"` + Media struct { + Text string `xml:",chardata"` + VideoResolution string `xml:"videoResolution,attr"` + ID string `xml:"id,attr"` + Duration string `xml:"duration,attr"` + Bitrate string `xml:"bitrate,attr"` + Width string `xml:"width,attr"` + Height string `xml:"height,attr"` + AspectRatio string `xml:"aspectRatio,attr"` + AudioChannels string `xml:"audioChannels,attr"` + AudioCodec string `xml:"audioCodec,attr"` + VideoCodec string `xml:"videoCodec,attr"` + Container string `xml:"container,attr"` + VideoFrameRate string `xml:"videoFrameRate,attr"` + VideoProfile string `xml:"videoProfile,attr"` + Part struct { + Text string `xml:",chardata"` + ID string `xml:"id,attr"` + Key string `xml:"key,attr"` + Duration string `xml:"duration,attr"` + File string `xml:"file,attr"` + Size string `xml:"size,attr"` + Container string `xml:"container,attr"` + VideoProfile string `xml:"videoProfile,attr"` + } `xml:"Part"` + } `xml:"Media"` + Director struct { + Text string `xml:",chardata"` + Tag string `xml:"tag,attr"` + } `xml:"Director"` + Writer []struct { + Text string `xml:",chardata"` + Tag string `xml:"tag,attr"` + } `xml:"Writer"` + } `xml:"Video"` +}