Test
This commit is contained in:
Matt Burchett
parent
b6b5814588
commit
b175b02b87
12
cmd/main.go
12
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" {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@ -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
|
||||
|
||||
@ -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"`
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user