mattburchett/Housekeeper
1
0
Fork 0
Code Issues 2 Pull Requests Releases Wiki Activity

TV Show Fix #6

Merged
mattburchett merged 17 commits from matt/#2 into master 2018-11-20 23:08:30 +00:00
Conversation 0 Commits 17 Files Changed 5 +140 -11
3 changed files with 140 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit b175b02b87 - Show all commits

12
cmd/main.go
View File

@ -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" {
}
}
}

43
pkg/eraser/eraser.go
View File

@ -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

96
pkg/model/eraser_model.go
View File

@ -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"`
}