DeleteArr/DeleteArr.go

package main

import (
	"bytes"
	"encoding/json"
	"log"
	"net/http"
	"os"
	"path/filepath"
	"slices"
	"strings"
	"time"
)

type GotifyMessage struct {
	Title    string `json:"title"`
	Priority string `json:"priority"`
	Message  string `json:"message"`
}

type MediaFiles struct {
	EventType    string
	SourcePath   string
	SourceFolder string
	FileName     string
	InFolder     bool
	Arr          string
}

func (m *MediaFiles) SendGotify(message string, arr string) {
	url := "https://gotify.rouggy.com/message?token=AKMj5SsZblmpAJ_"
	gm := GotifyMessage{
		Title:    "Deleting Media from " + arr,
		Priority: "10",
		Message:  message,
	}

	marshalled, err := json.Marshal(gm)
	if err != nil {
		log.Fatalf("impossible to marshall Gotify Message: %s", err)
	}

	req, err := http.NewRequest("POST", url, bytes.NewReader(marshalled))
	if err != nil {
		log.Fatalf("impossible to build request: %s", err)
	}

	req.Header.Set("Content-Type", "application/json")

	// create http client
	// do not forget to set timeout; otherwise, no timeout!
	client := http.Client{Timeout: 10 * time.Second}
	// send the request
	res, err := client.Do(req)
	if err != nil {
		log.Fatalf("impossible to send request: %s", err)
	}
	log.Printf("status Code: %d", res.StatusCode)

}

func (m *MediaFiles) IsInFolder() {
	folderList := []string{"Movies", "4K-Movies", "Series", "4K-Series", "Kids", "Animes"}

	SplitFolder := strings.Split(m.SourcePath, "/")
	LastFolder := SplitFolder[len(SplitFolder)-2]
	m.FileName = SplitFolder[len(SplitFolder)-1]

	log.Printf("Found last folder to be: %v", LastFolder)

	ContainsFolder := slices.Contains(folderList, LastFolder)

	if !ContainsFolder {
		m.InFolder = true
		log.Printf("Movie %v is in a Folder\n", m.FileName)
	} else {
		m.InFolder = false
		log.Printf("Movie %v is not in a Folder\n", m.FileName)
	}
}

func main() {

	// os.Setenv("radarr_moviefile_sourcepath", "/mnt/Multimedia/Download/PostProcess/Movies/Butchers.Crossing.2023.MULTi.1080p.WEB.x264-FW.mkv")
	// os.Setenv("radarr_moviefile_sourcefolder", "/mnt/Multimedia/Download/PostProcess/Movies")
	// os.Setenv("radarr_eventtype", "Download")

	ex, err := os.Executable()
	if err != nil {
		panic(err)
	}
	exPath := filepath.Dir(ex)

	f, err := os.OpenFile(exPath+"/log.txt", os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
	if err != nil {
		log.Fatalf("error opening file: %v", err)
	}
	defer f.Close()

	log.SetOutput(f)

	RadarrEventType := os.Getenv("radarr_eventtype")
	SonarrEventType := os.Getenv("sonarr_eventtype")

	m := &MediaFiles{}

	if RadarrEventType != "" {
		m.EventType = os.Getenv("radarr_eventtype")
		m.SourcePath = os.Getenv("radarr_moviefile_sourcepath")
		log.Printf("The Source Path is: %v", m.SourcePath)
		m.SourceFolder = os.Getenv("radarr_moviefile_sourcefolder")
		log.Printf("The Source Folder is: %v", m.SourceFolder)
		m.Arr = "Radarr"

	}

	if SonarrEventType != "" {
		m.EventType = os.Getenv("sonarr_eventtype")
		m.SourcePath = os.Getenv("sonarr_episodefile_sourcepath")
		log.Printf("The Source Path is: %v", m.SourcePath)
		m.SourceFolder = os.Getenv("radarr_moviefile_sourcefolder")
		log.Printf("The Source Folder is: %v", m.SourceFolder)
		m.Arr = "Sonarr"
	}

	m.IsInFolder()

	if m.InFolder {

		f, err := os.Open(m.SourceFolder)
		if err != nil {
			log.Println(err)
			return
		}
		files, err := f.Readdir(0)
		if err != nil {
			log.Println(err)
			return
		}

		mkvCount := 0

		for _, v := range files {
			log.Println("Found file: " + v.Name())
			if filepath.Ext(v.Name()) == ".mkv" {
				mkvCount += 1
			} else {
				os.Remove(m.SourceFolder + "/" + v.Name())
				log.Printf("Deleting file non MKV file: %v", m.SourceFolder+"/"+v.Name())
			}
		}

		if mkvCount > 1 {
			log.Printf("Found %v MKV files in the folder, deleting only %v", mkvCount, m.SourcePath)
			os.Remove(m.SourcePath)
		} else {
			log.Printf("Found only one MKV files in the folder, deleting thoe folder %v", m.SourceFolder)
			os.RemoveAll(m.SourceFolder)
		}

		// if not in folder just delete the file
	} else {
		os.RemoveAll(m.SourcePath)
		log.Printf("Deleting the file %v", m.SourcePath)
	}

	m.SendGotify("Deleting Source Path"+m.SourcePath, m.Arr)

}