feat: add to blacklist if torrent has no video file

engine/scheduler.go

@@ -12,6 +12,7 @@ import (
 	"polaris/log"
 	"polaris/pkg"
 	"polaris/pkg/notifier/message"
+	"polaris/pkg/storage"
 	"polaris/pkg/utils"
 	"time"
 
@@ -292,6 +293,10 @@ func (c *Engine) moveCompletedTask(id int) (err1 error) {
 
 	//如果种子是路径，则会把路径展开，只移动文件，类似 move dir/* dir2/, 如果种子是文件，则会直接移动文件，类似 move file dir/
 	if err := stImpl.Copy(filepath.Join(c.db.GetDownloadDir(), torrentName), r.TargetDir, torrent.WalkFunc()); err != nil {
+		if errors.Is(err, &storage.NoVideoFileError{}) {
+			log.Warnf("no video file found in torrent, add torrent to blacklist: %v", torrentName)
+			c.db.AddTorrent2Blacklist(r.Hash, r.SourceTitle, r.MediaID)
+		}
 		return errors.Wrap(err, "move file")
 	}
 	torrent.UploadProgresser = stImpl.UploadProgress

pkg/storage/base.go

@@ -80,10 +80,24 @@ func (b *Base) isFileNeeded(name string) bool {
 
 }
 
+type NoVideoFileError struct {
+	Path string
+}
+
+func (e *NoVideoFileError) Error() string {
+	return "no video file in path: " + e.Path
+}
+
+func (e *NoVideoFileError) Is(target error) bool {
+	_, ok := target.(*NoVideoFileError)
+	return ok
+}
+
 func (b *Base) Upload(destDir string, tryLink, detectMime, changeMediaHash bool, upload uploadFunc, mkdir func(string) error, walkFn WalkFn) error {
 	if !b.checkVideoFilesExist() {
-		return errors.Errorf("torrent has no video file(s)")
+		return &NoVideoFileError{Path: b.src}
 	}
+
 	os.MkdirAll(destDir, os.ModePerm)
 
 	targetBase := filepath.Join(destDir, filepath.Base(b.src)) //文件的场景，要加上文件名, move filename ./dir/

pkg/storage/base_test.go

@@ -0,0 +1,16 @@
+package storage
+
+import (
+	"testing"
+
+	"github.com/pkg/errors"
+)
+
+func TestError(t *testing.T) {
+	err := &NoVideoFileError{Path: "/some/path"}
+	if errors.Is(err, &NoVideoFileError{}) {
+		t.Log("is NoVideoFileError")
+	} else {
+		t.Error("not match")
+	}
+}