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

Adding the ability for multiple section IDs, title sorting, completes #7

Browse Source
This commit is contained in:
Matt Burchett 2020-12-11 15:13:26 -06:00
parent b1bbda788c
commit f3424d83bc
1 changed files with 39 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
cmd/main.go
View File

@ -3,6 +3,9 @@ package main
import ( import (
"flag" "flag"
"log" "log"
"sort"
"strconv"
"strings"
"git.linuxrocker.com/mattburchett/Housekeeper/pkg/communicator" "git.linuxrocker.com/mattburchett/Housekeeper/pkg/communicator"
"git.linuxrocker.com/mattburchett/Housekeeper/pkg/config" "git.linuxrocker.com/mattburchett/Housekeeper/pkg/config"
@ -13,14 +16,14 @@ import (
func main() { func main() {
var c string var c string
var days int var days int
var sectionID int var sectionID string
var check bool var check bool
var text bool var text bool
var delete bool var delete bool
flag.StringVar(&c, "config", "", "Configuration to load") flag.StringVar(&c, "config", "", "Configuration to load")
flag.IntVar(&days, "days", 0, "How many days of inactivity to look for on Plex.") flag.IntVar(&days, "days", 0, "How many days of inactivity to look for on Plex.")
flag.IntVar(&sectionID, "sectionid", 0, "Plex Section ID") flag.StringVar(&sectionID, "sectionid", "", "Plex Section ID. Multiples can be specified (separated by a comma). Ex: 1,2")
flag.BoolVar(&check, "check", true, "Perform only a check. This will send the message out to Telegram with what can be removed. Does not delete.") flag.BoolVar(&check, "check", true, "Perform only a check. This will send the message out to Telegram with what can be removed. Does not delete.")
flag.BoolVar(&text, "text", false, "This will override the communication to Telegram and print to stdout.") flag.BoolVar(&text, "text", false, "This will override the communication to Telegram and print to stdout.")
flag.BoolVar(&delete, "delete", false, "Perform the delete task.") flag.BoolVar(&delete, "delete", false, "Perform the delete task.")
@ -30,7 +33,7 @@ func main() {
if c == "" { if c == "" {
log.Fatal("You need to specify a configuration file.") log.Fatal("You need to specify a configuration file.")
} }
if sectionID == 0 { if sectionID == "" {
log.Fatal("You need to specify a section ID for Plex.") log.Fatal("You need to specify a section ID for Plex.")
} }
@ -39,36 +42,47 @@ func main() {
log.Fatal(err) log.Fatal(err)
} }
libraryType := locator.GetLibraryType(cfg, sectionID) sectionIds := strings.Split(sectionID, ",")
titlesFullList := make([]string, 0)
ids, titles := locator.GetTitles(cfg, sectionID, days) for _, section := range sectionIds {
sectionIDConv, _ := strconv.Atoi(section)
libraryType := locator.GetLibraryType(cfg, sectionIDConv)
ids, titles := locator.GetTitles(cfg, sectionIDConv, days)
for _, title := range titles {
titlesFullList = append(titlesFullList, title)
}
if delete {
if libraryType == "movie" {
files := eraser.LookupMovieFileLocation(cfg, ids)
err = eraser.DeleteFiles(delete, files)
if err != nil {
log.Println(err)
}
} else if libraryType == "show" {
// files := eraser.LookupTVFileLocation(cfg, ids)
sonarrIDs := locator.GetSonarrIDs(cfg, titles)
eraser.DeleteSeriesFromSonarr(cfg, sonarrIDs)
// err = eraser.DeleteFiles(delete, files)
// if err != nil {
// log.Println(err)
// }
}
}
}
if check { if check {
sort.Strings(titlesFullList)
if text { if text {
communicator.StdoutPost(titles) communicator.StdoutPost(titlesFullList)
} else { } else {
err = communicator.TelegramPost(cfg, titles) err = communicator.TelegramPost(cfg, titlesFullList)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
} }
} }
if delete {
if libraryType == "movie" {
files := eraser.LookupMovieFileLocation(cfg, ids)
err = eraser.DeleteFiles(delete, files)
if err != nil {
log.Println(err)
}
} else if libraryType == "show" {
// files := eraser.LookupTVFileLocation(cfg, ids)
sonarrIDs := locator.GetSonarrIDs(cfg, titles)
eraser.DeleteSeriesFromSonarr(cfg, sonarrIDs)
// err = eraser.DeleteFiles(delete, files)
// if err != nil {
// log.Println(err)
// }
}
}
} }