First Release

2024-08-25 05:36:47 +01:00
commit d54706fed6
20 changed files with 1682 additions and 0 deletions
@@ -0,0 +1,169 @@
+package cmd
+
+import (
+	"fmt"
+	"github.com/fsnotify/fsnotify"
+	"goBlack/common"
+	"goBlack/debrid"
+	"log"
+	"os"
+	"path/filepath"
+	"sync"
+	"time"
+)
+
+func fileReady(path string) bool {
+	_, err := os.Stat(path)
+	return !os.IsNotExist(err) // Returns true if the file exists
+}
+
+func checkFileLoop(wg *sync.WaitGroup, dir string, file debrid.TorrentFile, ready chan<- debrid.TorrentFile) {
+	defer wg.Done()
+	ticker := time.NewTicker(1 * time.Second) // Check every second
+	defer ticker.Stop()
+	path := filepath.Join(dir, file.Path)
+	for {
+		select {
+		case <-ticker.C:
+			if fileReady(path) {
+				ready <- file
+				return
+			}
+		}
+	}
+}
+
+func ProcessFiles(arr *debrid.Arr, torrent *debrid.Torrent) {
+	var wg sync.WaitGroup
+	files := torrent.Files
+	ready := make(chan debrid.TorrentFile, len(files))
+
+	log.Println("Checking files...")
+
+	for _, file := range files {
+		wg.Add(1)
+		go checkFileLoop(&wg, arr.Debrid.Folder, file, ready)
+	}
+
+	go func() {
+		wg.Wait()
+		close(ready)
+	}()
+
+	for r := range ready {
+		log.Println("File is ready:", r.Name)
+		CreateSymLink(arr, torrent)
+
+	}
+	go torrent.Cleanup(true)
+	fmt.Printf("%s downloaded", torrent.Name)
+}
+
+func CreateSymLink(config *debrid.Arr, torrent *debrid.Torrent) {
+	path := filepath.Join(config.CompletedFolder, torrent.Folder)
+	err := os.MkdirAll(path, os.ModePerm)
+	if err != nil {
+		log.Printf("Failed to create directory: %s\n", path)
+	}
+	for _, file := range torrent.Files {
+		// Combine the directory and filename to form a full path
+		fullPath := filepath.Join(config.CompletedFolder, file.Path)
+
+		// Create a symbolic link if file doesn't exist
+		_ = os.Symlink(filepath.Join(config.Debrid.Folder, file.Path), fullPath)
+	}
+}
+
+func watchFiles(watcher *fsnotify.Watcher, events map[string]time.Time) {
+	for {
+		select {
+		case event, ok := <-watcher.Events:
+			if !ok {
+				return
+			}
+			if event.Op&fsnotify.Write == fsnotify.Write {
+				if filepath.Ext(event.Name) == ".torrent" || filepath.Ext(event.Name) == ".magnet" {
+					events[event.Name] = time.Now()
+				}
+
+			}
+		case err, ok := <-watcher.Errors:
+			if !ok {
+				return
+			}
+			log.Println("ERROR:", err)
+		}
+	}
+}
+
+func processFilesDebounced(arr *debrid.Arr, db debrid.Service, events map[string]time.Time, debouncePeriod time.Duration) {
+	ticker := time.NewTicker(1 * time.Second) // Check every second
+	defer ticker.Stop()
+
+	for range ticker.C {
+		for file, lastEventTime := range events {
+			if time.Since(lastEventTime) >= debouncePeriod {
+				log.Printf("Torrent file detected: %s", file)
+				// Process the torrent file
+				torrent, err := db.Process(arr, file)
+				if err != nil && torrent != nil {
+					// remove torrent file
+					torrent.Cleanup(true)
+					_ = torrent.MarkAsFailed()
+					log.Printf("Error processing torrent file: %s", err)
+				}
+				if err == nil && torrent != nil && len(torrent.Files) > 0 {
+					go ProcessFiles(arr, torrent)
+				}
+				delete(events, file) // remove file from channel
+
+			}
+		}
+	}
+}
+
+func StartArr(conf *debrid.Arr, db debrid.Service) {
+	log.Printf("Watching: %s", conf.WatchFolder)
+	w, err := fsnotify.NewWatcher()
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer func(w *fsnotify.Watcher) {
+		err := w.Close()
+		if err != nil {
+			log.Fatal(err)
+		}
+	}(w)
+	events := make(map[string]time.Time)
+
+	go watchFiles(w, events)
+	if err = w.Add(conf.WatchFolder); err != nil {
+		log.Println("Error Watching folder:", err)
+		return
+	}
+
+	processFilesDebounced(conf, db, events, 1*time.Second)
+}
+
+func StartBlackhole(config *common.Config, deb debrid.Service) {
+	var wg sync.WaitGroup
+	for _, conf := range config.Arrs {
+		wg.Add(1)
+		defer wg.Done()
+		headers := map[string]string{
+			"X-Api-Key": conf.Token,
+		}
+		client := common.NewRLHTTPClient(nil, headers)
+
+		arr := &debrid.Arr{
+			Debrid:          config.Debrid,
+			WatchFolder:     conf.WatchFolder,
+			CompletedFolder: conf.CompletedFolder,
+			Token:           conf.Token,
+			URL:             conf.URL,
+			Client:          client,
+		}
+		go StartArr(arr, deb)
+	}
+	wg.Wait()
+}