Implementing a streaming setup with Usenet

pkg/sabnzbd/sabnzbd.go

@@ -0,0 +1,116 @@
+package sabnzbd
+
+import (
+	"github.com/rs/zerolog"
+	"github.com/sirrobot01/decypharr/internal/config"
+	"github.com/sirrobot01/decypharr/internal/logger"
+	"github.com/sirrobot01/decypharr/pkg/store"
+	"github.com/sirrobot01/decypharr/pkg/usenet"
+	"path/filepath"
+)
+
+type SABnzbd struct {
+	downloadFolder    string
+	config            *Config
+	refreshInterval   int
+	logger            zerolog.Logger
+	usenet            usenet.Usenet
+	defaultCategories []string
+}
+
+func New(usenetClient usenet.Usenet) *SABnzbd {
+	_cfg := config.Get()
+	cfg := _cfg.SABnzbd
+	var defaultCategories []string
+	for _, cat := range _cfg.SABnzbd.Categories {
+		if cat != "" {
+			defaultCategories = append(defaultCategories, cat)
+		}
+	}
+	sb := &SABnzbd{
+		downloadFolder:    cfg.DownloadFolder,
+		refreshInterval:   cfg.RefreshInterval,
+		logger:            logger.New("sabnzbd"),
+		usenet:            usenetClient,
+		defaultCategories: defaultCategories,
+	}
+	sb.SetConfig(_cfg)
+	return sb
+}
+
+func (s *SABnzbd) SetConfig(cfg *config.Config) {
+	sabnzbdConfig := &Config{
+		Misc: MiscConfig{
+			CompleteDir:   s.downloadFolder,
+			DownloadDir:   s.downloadFolder,
+			AdminDir:      s.downloadFolder,
+			WebPort:       cfg.Port,
+			Language:      "en",
+			RefreshRate:   "1",
+			QueueComplete: "0",
+			ConfigLock:    "0",
+			Autobrowser:   "1",
+			CheckNewRel:   "1",
+		},
+		Categories: s.getCategories(),
+	}
+	if cfg.Usenet != nil || len(cfg.Usenet.Providers) == 0 {
+		for _, provider := range cfg.Usenet.Providers {
+			if provider.Host == "" || provider.Port == 0 {
+				continue
+			}
+			sabnzbdConfig.Servers = append(sabnzbdConfig.Servers, Server{
+				Name:        provider.Name,
+				Host:        provider.Host,
+				Port:        provider.Port,
+				Username:    provider.Username,
+				Password:    provider.Password,
+				Connections: provider.Connections,
+				SSL:         provider.SSL,
+			})
+		}
+	}
+	s.config = sabnzbdConfig
+}
+
+func (s *SABnzbd) getCategories() []Category {
+	_store := store.Get()
+	arrs := _store.Arr().GetAll()
+	categories := make([]Category, 0, len(arrs))
+	added := map[string]struct{}{}
+
+	for i, a := range arrs {
+		if _, ok := added[a.Name]; ok {
+			continue // Skip if category already added
+		}
+		categories = append(categories, Category{
+			Name:     a.Name,
+			Order:    i + 1,
+			Pp:       "3",
+			Script:   "None",
+			Dir:      filepath.Join(s.downloadFolder, a.Name),
+			Priority: PriorityNormal,
+		})
+	}
+
+	// Add default categories if not already present
+	for _, defaultCat := range s.defaultCategories {
+		if _, ok := added[defaultCat]; ok {
+			continue // Skip if default category already added
+		}
+		categories = append(categories, Category{
+			Name:     defaultCat,
+			Order:    len(categories) + 1,
+			Pp:       "3",
+			Script:   "None",
+			Dir:      filepath.Join(s.downloadFolder, defaultCat),
+			Priority: PriorityNormal,
+		})
+		added[defaultCat] = struct{}{}
+	}
+
+	return categories
+}
+
+func (s *SABnzbd) Reset() {
+}