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" {
+		} else if libraryType == "show" {
-			// 	err = eraser.DeleteTVShows()
-			// 	if err != nil {
-			// 		log.Println(err)
-			// 	}
 		}
 	}
 }

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
+// LookupMovieFileLocation will gather a list of Information based on IDs returned by locator.GetTitles
-func LookupFileLocation(config config.Config, ids []int) []string {
+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

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.
+// XMLPlexMovieAPI - This is the XML version of the Library.
-type XMLPlexAPI struct {
+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"`
+}