feat: support generate .plexmatch

server/integration.go

@@ -0,0 +1,42 @@
+package server
+
+import (
+	"fmt"
+	"path/filepath"
+	"polaris/db"
+	"polaris/ent/media"
+	"polaris/log"
+
+	"github.com/pkg/errors"
+)
+
+func (s *Server) createPlexmatchIfNotExists(seriesId int) error {
+
+	if !s.plexmatchEnabled() {
+		return nil
+	}
+	series, err := s.db.GetMedia(seriesId)
+	if err != nil {
+		return err
+	}
+	if series.MediaType != media.MediaTypeTv {
+		return nil
+	}
+	st, err := s.getStorage(series.StorageID, media.MediaTypeTv)
+	if err != nil {
+		return errors.Wrap(err, "get storage")
+	}
+
+	_, err = st.ReadFile(filepath.Join(series.TargetDir, ".plexmatch"))
+	if err != nil { 
+		//create new
+		log.Warnf(".plexmatch file not found, create new one: %s", series.NameEn)
+		st.WriteFile(filepath.Join(series.TargetDir, ".plexmatch"), []byte(fmt.Sprintf("tmdbid=%d\n",series.TmdbID)))
+		return nil
+	} 
+	return nil
+}
+
+func (s *Server) plexmatchEnabled() bool {
+	return s.db.GetSetting(db.SettingPlexMatchEnabled) == "true"
+}

server/scheduler.go

@@ -71,6 +71,9 @@ func (s *Server) moveCompletedTask(id int) (err1 error) {
 			log.Errorf("no season id: %v", r.TargetDir)
 			seasonNum = -1
 		}
+		if err := s.createPlexmatchIfNotExists(r.MediaID); err != nil {
+			log.Errorf("create .plexmatch file error: %v", err)
+		}
 
 		if err1 != nil {
 			s.db.SetHistoryStatus(r.ID, history.StatusFail)
@@ -143,29 +146,12 @@ func (s *Server) checkDownloadedSeriesFiles(m *ent.Media) error {
 		return nil
 	}
 	log.Infof("check files in directory: %s", m.TargetDir)
-	st := s.db.GetStorage(m.StorageID)
 
-	var storageImpl storage.Storage
-
-	switch st.Implementation {
-	case storage1.ImplementationLocal:
-		ls := st.ToLocalSetting()
-		targetPath := ls.TvPath
-		storageImpl1, err := storage.NewLocalStorage(targetPath)
-		if err != nil {
-			return errors.Wrap(err, "new local")
-		}
-		storageImpl = storageImpl1
-
-	case storage1.ImplementationWebdav:
-		ws := st.ToWebDavSetting()
-		targetPath := ws.TvPath
-		storageImpl1, err := storage.NewWebdavStorage(ws.URL, ws.User, ws.Password, targetPath, ws.ChangeFileHash == "true")
-		if err != nil {
-			return errors.Wrap(err, "new webdav")
-		}
-		storageImpl = storageImpl1
+	var storageImpl, err = s.getStorage(m.StorageID, media.MediaTypeTv)
+	if err != nil {
+		return err
 	}
+
 	files, err := storageImpl.ReadDir(m.TargetDir)
 	if err != nil {
 		return errors.Wrapf(err, "read dir %s", m.TargetDir)

server/setting.go

@@ -14,10 +14,11 @@ import (
 )
 
 type GeneralSettings struct {
-	TmdbApiKey  string `json:"tmdb_api_key"`
-	DownloadDir string `json:"download_dir"`
-	LogLevel    string `json:"log_level"`
-	Proxy       string `json:"proxy"`
+	TmdbApiKey      string `json:"tmdb_api_key"`
+	DownloadDir     string `json:"download_dir"`
+	LogLevel        string `json:"log_level"`
+	Proxy           string `json:"proxy"`
+	EnablePlexmatch bool   `json:"enable_plexmatch"`
 }
 
 func (s *Server) SetSetting(c *gin.Context) (interface{}, error) {
@@ -45,6 +46,13 @@ func (s *Server) SetSetting(c *gin.Context) (interface{}, error) {
 
 	}
 
+	plexmatchEnabled := s.db.GetSetting(db.SettingPlexMatchEnabled)
+	if in.EnablePlexmatch && plexmatchEnabled != "true" {
+		s.db.SetSetting(db.SettingPlexMatchEnabled, "true")
+	} else if !in.EnablePlexmatch && plexmatchEnabled != "false" {
+		s.db.SetSetting(db.SettingPlexMatchEnabled, "false")
+	}
+
 	s.setProxy(in.Proxy)
 	return nil, nil
 }
@@ -72,11 +80,13 @@ func (s *Server) GetSetting(c *gin.Context) (interface{}, error) {
 	tmdb := s.db.GetSetting(db.SettingTmdbApiKey)
 	downloadDir := s.db.GetSetting(db.SettingDownloadDir)
 	logLevel := s.db.GetSetting(db.SettingLogLevel)
+	plexmatchEnabled := s.db.GetSetting(db.SettingPlexMatchEnabled)
 	return &GeneralSettings{
-		TmdbApiKey:  tmdb,
-		DownloadDir: downloadDir,
-		LogLevel:    logLevel,
-		Proxy:       s.db.GetSetting(db.SettingProxy),
+		TmdbApiKey:      tmdb,
+		DownloadDir:     downloadDir,
+		LogLevel:        logLevel,
+		Proxy:           s.db.GetSetting(db.SettingProxy),
+		EnablePlexmatch: plexmatchEnabled == "true",
 	}, nil
 }
 

server/storage.go

@@ -3,6 +3,8 @@ package server
 import (
 	"fmt"
 	"polaris/db"
+	"polaris/ent/media"
+	storage1 "polaris/ent/storage"
 	"polaris/log"
 	"polaris/pkg/storage"
 	"polaris/pkg/utils"
@@ -106,3 +108,35 @@ func (s *Server) SuggestedMovieFolderName(c *gin.Context) (interface{}, error) {
 	log.Infof("tv series of tmdb id %v suggestting name is %v", id, name)
 	return gin.H{"name": name}, nil
 }
+
+
+func (s *Server) getStorage(storageId int, mediaType media.MediaType) (storage.Storage, error) {
+	st := s.db.GetStorage(storageId)
+	switch st.Implementation {
+	case storage1.ImplementationLocal:
+		ls := st.ToLocalSetting()
+		targetPath := ls.TvPath
+		if mediaType == media.MediaTypeMovie {
+			targetPath = ls.MoviePath
+		}
+		storageImpl1, err := storage.NewLocalStorage(targetPath)
+		if err != nil {
+			return nil, errors.Wrap(err, "new local")
+		}
+		return storageImpl1, nil
+
+	case storage1.ImplementationWebdav:
+		ws := st.ToWebDavSetting()
+		targetPath := ws.TvPath
+		if mediaType == media.MediaTypeMovie {
+			targetPath = ws.MoviePath
+		}
+
+		storageImpl1, err := storage.NewWebdavStorage(ws.URL, ws.User, ws.Password, targetPath, ws.ChangeFileHash == "true")
+		if err != nil {
+			return nil, errors.Wrap(err, "new webdav")
+		}
+		return storageImpl1, nil
+	}
+	return nil, errors.New("no storage found")
+}