feat: edit media details

server/core/integration.go

@@ -10,6 +10,7 @@ import (
 	"polaris/log"
 	"polaris/pkg/notifier"
 	"polaris/pkg/storage"
+	"strings"
 
 	"github.com/pkg/errors"
 )
@@ -55,6 +56,10 @@ func (c *Client) writePlexmatch(seriesId int, episodeId int, targetDir, name str
 	} else {
 		buff.Write(data)
 	}
+	if strings.Contains(buff.String(), name) {
+		log.Debugf("already write plex episode line: %v", name)
+		return nil
+	}
 	buff.WriteString(fmt.Sprintf("\nep: %d: %s\n", ep.EpisodeNumber, name))
 	log.Infof("write season plexmatch file content: %s", buff.String())
 	return st.WriteFile(seasonPlex, buff.Bytes())

server/core/scheduler.go

@@ -116,16 +116,16 @@ func (c *Client) moveCompletedTask(id int) (err1 error) {
 		return err
 	}
 
-	//如果种子是路径，则会把路径展开，只移动文件，类似 move dir/* dir2/, 如果种子是文件，则会直接移动文件，类似 move file dir/
-	if err := stImpl.Copy(filepath.Join(c.db.GetDownloadDir(), torrentName), r.TargetDir); err != nil {
-		return errors.Wrap(err, "move file")
-	}
-
 	// .plexmatch file
 	if err := c.writePlexmatch(r.MediaID, r.EpisodeID, r.TargetDir, torrentName); err != nil {
 		log.Errorf("create .plexmatch file error: %v", err)
 	}
 
+	//如果种子是路径，则会把路径展开，只移动文件，类似 move dir/* dir2/, 如果种子是文件，则会直接移动文件，类似 move file dir/
+	if err := stImpl.Copy(filepath.Join(c.db.GetDownloadDir(), torrentName), r.TargetDir); err != nil {
+		return errors.Wrap(err, "move file")
+	}
+
 	c.db.SetHistoryStatus(r.ID, history.StatusSuccess)
 	if r.EpisodeID != 0 {
 		c.db.SetEpisodeStatus(r.EpisodeID, episode.StatusDownloaded)

server/server.go

@@ -80,6 +80,7 @@ func (s *Server) Serve() error {
 	tv := api.Group("/media")
 	{
 		tv.GET("/search", HttpHandler(s.SearchMedia))
+		tv.POST("/edit", HttpHandler(s.EditMediaMetadata))
 		tv.POST("/tv/watchlist", HttpHandler(s.AddTv2Watchlist))
 		tv.GET("/tv/watchlist", HttpHandler(s.GetTvWatchlist))
 		tv.POST("/torrents", HttpHandler(s.SearchAvailableTorrents))

server/setting.go

@@ -195,8 +195,8 @@ func (s *Server) DeleteDownloadCLient(c *gin.Context) (interface{}, error) {
 }
 
 type episodeMonitoringIn struct {
-	EpisodeID int `json:"episode_id"`
-	Monitor bool `json:"monitor"`
+	EpisodeID int  `json:"episode_id"`
+	Monitor   bool `json:"monitor"`
 }
 
 func (s *Server) ChangeEpisodeMonitoring(c *gin.Context) (interface{}, error) {
@@ -206,4 +206,16 @@ func (s *Server) ChangeEpisodeMonitoring(c *gin.Context) (interface{}, error) {
 	}
 	s.db.SetEpisodeMonitoring(in.EpisodeID, in.Monitor)
 	return "success", nil
+}
+
+func (s *Server) EditMediaMetadata(c *gin.Context) (interface{}, error) {
+	var in db.EditMediaData
+	if err := c.ShouldBindJSON(&in); err != nil {
+		return nil, errors.Wrap(err, "bind")
+	}
+	err := s.db.EditMediaMetadata(in)
+	if err != nil {
+		return nil, errors.Wrap(err, "save db")
+	}
+	return "success", nil
 }