Housekeeper/cmd/main.go

package main

import (
	"flag"
	"log"
	"sort"
	"strconv"
	"strings"

	"git.linuxrocker.com/mattburchett/Housekeeper/pkg/communicator"
	"git.linuxrocker.com/mattburchett/Housekeeper/pkg/config"
	"git.linuxrocker.com/mattburchett/Housekeeper/pkg/eraser"
	"git.linuxrocker.com/mattburchett/Housekeeper/pkg/locator"
)

func main() {
	var c string
	var days int
	var sectionID string
	var check bool
	var text bool
	var delete bool

	flag.StringVar(&c, "config", "", "Configuration to load")
	flag.IntVar(&days, "days", 0, "How many days of inactivity to look for on Plex.")
	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(&text, "text", false, "This will override the communication to Telegram and print to stdout.")
	flag.BoolVar(&delete, "delete", false, "Perform the delete task.")
	flag.Parse()

	// Stop the app if they're missing required flags.
	if c == "" {
		log.Fatal("You need to specify a configuration file.")
	}
	if sectionID == "" {
		log.Fatal("You need to specify a section ID for Plex.")
	}

	cfg, err := config.GetConfig(c)
	if err != nil {
		log.Fatal(err)
	}

	sectionIds := strings.Split(sectionID, ",")
	titlesFullList := make([]string, 0)

	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 {
		sort.Strings(titlesFullList)
		if text {
			communicator.StdoutPost(titlesFullList)
		} else {
			err = communicator.TelegramPost(cfg, titlesFullList)
			if err != nil {
				log.Fatal(err)
			}
		}
	}
}