node build fixed
This commit is contained in:
ra_ma
231
seanime-2.9.10/internal/torrentstream/repository.go
Normal file
231
seanime-2.9.10/internal/torrentstream/repository.go
Normal file
@@ -0,0 +1,231 @@
|
||||
package torrentstream
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"seanime/internal/api/anilist"
|
||||
"seanime/internal/api/metadata"
|
||||
"seanime/internal/database/db"
|
||||
"seanime/internal/database/models"
|
||||
"seanime/internal/directstream"
|
||||
"seanime/internal/events"
|
||||
hibiketorrent "seanime/internal/extension/hibike/torrent"
|
||||
"seanime/internal/library/anime"
|
||||
"seanime/internal/library/playbackmanager"
|
||||
"seanime/internal/mediaplayers/mediaplayer"
|
||||
"seanime/internal/nativeplayer"
|
||||
"seanime/internal/platforms/platform"
|
||||
"seanime/internal/torrents/torrent"
|
||||
"seanime/internal/util"
|
||||
"seanime/internal/util/result"
|
||||
|
||||
"github.com/rs/zerolog"
|
||||
"github.com/samber/mo"
|
||||
)
|
||||
|
||||
type (
|
||||
Repository struct {
|
||||
client *Client
|
||||
handler *handler
|
||||
playback playback
|
||||
settings mo.Option[Settings] // None by default, set and refreshed by [SetSettings]
|
||||
|
||||
selectionHistoryMap *result.Map[int, *hibiketorrent.AnimeTorrent] // Key: AniList media ID
|
||||
|
||||
// Injected dependencies
|
||||
torrentRepository *torrent.Repository
|
||||
baseAnimeCache *anilist.BaseAnimeCache
|
||||
completeAnimeCache *anilist.CompleteAnimeCache
|
||||
platform platform.Platform
|
||||
wsEventManager events.WSEventManagerInterface
|
||||
metadataProvider metadata.Provider
|
||||
playbackManager *playbackmanager.PlaybackManager
|
||||
mediaPlayerRepository *mediaplayer.Repository
|
||||
mediaPlayerRepositorySubscriber *mediaplayer.RepositorySubscriber
|
||||
nativePlayerSubscriber *nativeplayer.Subscriber
|
||||
directStreamManager *directstream.Manager
|
||||
nativePlayer *nativeplayer.NativePlayer
|
||||
logger *zerolog.Logger
|
||||
db *db.Database
|
||||
|
||||
onEpisodeCollectionChanged func(ec *anime.EpisodeCollection)
|
||||
|
||||
previousStreamOptions mo.Option[*StartStreamOptions]
|
||||
}
|
||||
|
||||
Settings struct {
|
||||
models.TorrentstreamSettings
|
||||
Host string
|
||||
Port int
|
||||
}
|
||||
|
||||
NewRepositoryOptions struct {
|
||||
Logger *zerolog.Logger
|
||||
TorrentRepository *torrent.Repository
|
||||
BaseAnimeCache *anilist.BaseAnimeCache
|
||||
CompleteAnimeCache *anilist.CompleteAnimeCache
|
||||
Platform platform.Platform
|
||||
MetadataProvider metadata.Provider
|
||||
PlaybackManager *playbackmanager.PlaybackManager
|
||||
WSEventManager events.WSEventManagerInterface
|
||||
Database *db.Database
|
||||
DirectStreamManager *directstream.Manager
|
||||
NativePlayer *nativeplayer.NativePlayer
|
||||
}
|
||||
)
|
||||
|
||||
// NewRepository creates a new injectable Repository instance
|
||||
func NewRepository(opts *NewRepositoryOptions) *Repository {
|
||||
ret := &Repository{
|
||||
client: nil,
|
||||
handler: nil,
|
||||
settings: mo.Option[Settings]{},
|
||||
selectionHistoryMap: result.NewResultMap[int, *hibiketorrent.AnimeTorrent](),
|
||||
torrentRepository: opts.TorrentRepository,
|
||||
baseAnimeCache: opts.BaseAnimeCache,
|
||||
completeAnimeCache: opts.CompleteAnimeCache,
|
||||
platform: opts.Platform,
|
||||
wsEventManager: opts.WSEventManager,
|
||||
metadataProvider: opts.MetadataProvider,
|
||||
playbackManager: opts.PlaybackManager,
|
||||
mediaPlayerRepository: nil,
|
||||
mediaPlayerRepositorySubscriber: nil,
|
||||
logger: opts.Logger,
|
||||
db: opts.Database,
|
||||
directStreamManager: opts.DirectStreamManager,
|
||||
nativePlayer: opts.NativePlayer,
|
||||
previousStreamOptions: mo.None[*StartStreamOptions](),
|
||||
}
|
||||
ret.client = NewClient(ret)
|
||||
ret.handler = newHandler(ret)
|
||||
return ret
|
||||
}
|
||||
|
||||
func (r *Repository) IsEnabled() bool {
|
||||
return r.settings.IsPresent() && r.settings.MustGet().Enabled && r.client != nil
|
||||
}
|
||||
|
||||
func (r *Repository) GetPreviousStreamOptions() (*StartStreamOptions, bool) {
|
||||
return r.previousStreamOptions.OrElse(nil), r.previousStreamOptions.IsPresent()
|
||||
}
|
||||
|
||||
// SetMediaPlayerRepository sets the mediaplayer repository and listens to events.
|
||||
// This MUST be called after instantiating the repository and will run even if the module is disabled.
|
||||
//
|
||||
// // Note: This is also used for Debrid streaming
|
||||
func (r *Repository) SetMediaPlayerRepository(mediaPlayerRepository *mediaplayer.Repository) {
|
||||
r.mediaPlayerRepository = mediaPlayerRepository
|
||||
r.listenToMediaPlayerEvents()
|
||||
}
|
||||
|
||||
// InitModules sets the settings for the torrentstream module.
|
||||
// It should be called before any other method, to ensure the module is active.
|
||||
func (r *Repository) InitModules(settings *models.TorrentstreamSettings, host string, port int) (err error) {
|
||||
r.client.Shutdown()
|
||||
|
||||
defer util.HandlePanicInModuleWithError("torrentstream/InitModules", &err)
|
||||
|
||||
if settings == nil {
|
||||
r.logger.Error().Msg("torrentstream: Cannot initialize module, no settings provided")
|
||||
r.settings = mo.None[Settings]()
|
||||
return errors.New("torrentstream: Cannot initialize module, no settings provided")
|
||||
}
|
||||
|
||||
s := *settings
|
||||
|
||||
if s.Enabled == false {
|
||||
r.logger.Info().Msg("torrentstream: Module is disabled")
|
||||
r.Shutdown()
|
||||
r.settings = mo.None[Settings]()
|
||||
return nil
|
||||
}
|
||||
|
||||
// Set default download directory, which is a temporary directory
|
||||
if s.DownloadDir == "" {
|
||||
s.DownloadDir = r.getDefaultDownloadPath()
|
||||
_ = os.MkdirAll(s.DownloadDir, os.ModePerm) // Create the directory if it doesn't exist
|
||||
}
|
||||
|
||||
// DEVNOTE: Commented code below causes error log after initializing the client
|
||||
//// Empty the download directory
|
||||
//_ = os.RemoveAll(s.DownloadDir)
|
||||
|
||||
if s.StreamingServerPort == 0 {
|
||||
s.StreamingServerPort = 43214
|
||||
}
|
||||
if s.TorrentClientPort == 0 {
|
||||
s.TorrentClientPort = 43213
|
||||
}
|
||||
if s.StreamingServerHost == "" {
|
||||
s.StreamingServerHost = "127.0.0.1"
|
||||
}
|
||||
|
||||
// Set the settings
|
||||
r.settings = mo.Some(Settings{
|
||||
TorrentstreamSettings: s,
|
||||
Host: host,
|
||||
Port: port,
|
||||
})
|
||||
|
||||
// Initialize the torrent client
|
||||
err = r.client.initializeClient()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// Start listening to native player events
|
||||
r.listenToNativePlayerEvents()
|
||||
|
||||
r.logger.Info().Msg("torrentstream: Module initialized")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *Repository) HTTPStreamHandler() http.Handler {
|
||||
return r.handler
|
||||
}
|
||||
|
||||
func (r *Repository) FailIfNoSettings() error {
|
||||
if r.settings.IsAbsent() {
|
||||
return errors.New("torrentstream: no settings provided, the module is dormant")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Shutdown closes the torrent client and streaming server
|
||||
// TEST-ONLY
|
||||
func (r *Repository) Shutdown() {
|
||||
r.logger.Debug().Msg("torrentstream: Shutting down module")
|
||||
r.client.Shutdown()
|
||||
}
|
||||
|
||||
//// Cleanup shuts down the module and removes the download directory
|
||||
//func (r *Repository) Cleanup() {
|
||||
// if r.settings.IsAbsent() {
|
||||
// return
|
||||
// }
|
||||
// r.client.Close()
|
||||
//
|
||||
// // Remove the download directory
|
||||
// downloadDir := r.GetDownloadDir()
|
||||
// _ = os.RemoveAll(downloadDir)
|
||||
//}
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
func (r *Repository) GetDownloadDir() string {
|
||||
if r.settings.IsAbsent() {
|
||||
return r.getDefaultDownloadPath()
|
||||
}
|
||||
if r.settings.MustGet().DownloadDir == "" {
|
||||
return r.getDefaultDownloadPath()
|
||||
}
|
||||
return r.settings.MustGet().DownloadDir
|
||||
}
|
||||
|
||||
func (r *Repository) getDefaultDownloadPath() string {
|
||||
tempDir := os.TempDir()
|
||||
downloadDirPath := filepath.Join(tempDir, "seanime", "torrentstream")
|
||||
return downloadDirPath
|
||||
}
|
||||
Reference in New Issue
Block a user