node build fixed

seanime-2.9.10/internal/torrents/torrent/repository.go

@@ -0,0 +1,142 @@
+package torrent
+
+import (
+	"seanime/internal/api/metadata"
+	"seanime/internal/extension"
+	"seanime/internal/util/result"
+	"sync"
+
+	"github.com/rs/zerolog"
+)
+
+type (
+	Repository struct {
+		logger                         *zerolog.Logger
+		extensionBank                  *extension.UnifiedBank
+		animeProviderSearchCaches      *result.Map[string, *result.Cache[string, *SearchData]]
+		animeProviderSmartSearchCaches *result.Map[string, *result.Cache[string, *SearchData]]
+		settings                       RepositorySettings
+		metadataProvider               metadata.Provider
+		mu                             sync.Mutex
+	}
+
+	RepositorySettings struct {
+		DefaultAnimeProvider string // Default torrent provider
+	}
+)
+
+type NewRepositoryOptions struct {
+	Logger           *zerolog.Logger
+	MetadataProvider metadata.Provider
+}
+
+func NewRepository(opts *NewRepositoryOptions) *Repository {
+	ret := &Repository{
+		logger:                         opts.Logger,
+		metadataProvider:               opts.MetadataProvider,
+		extensionBank:                  extension.NewUnifiedBank(),
+		animeProviderSearchCaches:      result.NewResultMap[string, *result.Cache[string, *SearchData]](),
+		animeProviderSmartSearchCaches: result.NewResultMap[string, *result.Cache[string, *SearchData]](),
+		settings:                       RepositorySettings{},
+		mu:                             sync.Mutex{},
+	}
+
+	return ret
+}
+
+func (r *Repository) InitExtensionBank(bank *extension.UnifiedBank) {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+	r.extensionBank = bank
+
+	go func() {
+		for {
+			select {
+			case <-bank.OnExtensionAdded():
+				//r.logger.Debug().Msg("torrent repo: Anime provider extension added")
+				r.OnExtensionReloaded()
+			}
+		}
+	}()
+
+	go func() {
+		for {
+			select {
+			case <-bank.OnExtensionRemoved():
+				r.OnExtensionReloaded()
+			}
+		}
+	}()
+
+	r.logger.Debug().Msg("torrent repo: Initialized anime provider extension bank")
+}
+
+func (r *Repository) OnExtensionReloaded() {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+	r.reloadExtensions()
+}
+
+// This is called each time a new extension is added or removed
+func (r *Repository) reloadExtensions() {
+	// Clear the search caches
+	r.animeProviderSearchCaches = result.NewResultMap[string, *result.Cache[string, *SearchData]]()
+	r.animeProviderSmartSearchCaches = result.NewResultMap[string, *result.Cache[string, *SearchData]]()
+
+	go func() {
+		// Create new caches for each provider
+		extension.RangeExtensions(r.extensionBank, func(provider string, value extension.AnimeTorrentProviderExtension) bool {
+			r.animeProviderSearchCaches.Set(provider, result.NewCache[string, *SearchData]())
+			r.animeProviderSmartSearchCaches.Set(provider, result.NewCache[string, *SearchData]())
+			return true
+		})
+	}()
+
+	// Check if the default provider is in the list of providers
+	//if r.settings.DefaultAnimeProvider != "" && r.settings.DefaultAnimeProvider != "none" {
+	//	if _, ok := r.extensionBank.Get(r.settings.DefaultAnimeProvider); !ok {
+	//		//r.logger.Error().Str("defaultProvider", r.settings.DefaultAnimeProvider).Msg("torrent repo: Default torrent provider not found in extensions")
+	//		// Set the default provider to empty
+	//		r.settings.DefaultAnimeProvider = ""
+	//	}
+	//}
+
+	//r.logger.Trace().Str("defaultProvider", r.settings.DefaultAnimeProvider).Msg("torrent repo: Reloaded extensions")
+}
+
+// SetSettings should be called after the repository is created and settings are refreshed
+func (r *Repository) SetSettings(s *RepositorySettings) {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+
+	r.logger.Trace().Msg("torrent repo: Setting settings")
+
+	if s != nil {
+		r.settings = *s
+	} else {
+		r.settings = RepositorySettings{
+			DefaultAnimeProvider: "",
+		}
+	}
+
+	if r.settings.DefaultAnimeProvider == "none" {
+		r.settings.DefaultAnimeProvider = ""
+	}
+
+	// Reload extensions after settings change
+	r.reloadExtensions()
+}
+
+func (r *Repository) GetDefaultAnimeProviderExtension() (extension.AnimeTorrentProviderExtension, bool) {
+	r.mu.Lock()
+	defer r.mu.Unlock()
+
+	if r.settings.DefaultAnimeProvider == "" {
+		return nil, false
+	}
+	return extension.GetExtension[extension.AnimeTorrentProviderExtension](r.extensionBank, r.settings.DefaultAnimeProvider)
+}
+
+func (r *Repository) GetAnimeProviderExtension(id string) (extension.AnimeTorrentProviderExtension, bool) {
+	return extension.GetExtension[extension.AnimeTorrentProviderExtension](r.extensionBank, id)
+}