implement download feature

server/scheduler.go

@@ -0,0 +1,49 @@
+package server
+
+import (
+	"path/filepath"
+	"polaris/db"
+	"polaris/log"
+	"polaris/pkg/storage"
+
+	"github.com/pkg/errors"
+)
+
+func (s *Server) scheduler() {
+	s.cron.AddFunc("@every 1m", s.checkTasks)
+}
+
+func (s *Server) checkTasks() {
+	for id, t := range s.tasks {
+		log.Infof("task %s percentage done: %d%%", t.Name(), t.Progress())
+		if t.Progress() == 100 {
+			log.Infof("task is done: %v", t.Name())
+			s.moveCompletedTask(id)
+		}
+	}
+}
+
+func (s *Server) moveCompletedTask(id int) error {
+	torrent := s.tasks[id]
+	r := s.db.GetHistory(id)
+	series := s.db.GetSeriesDetails(r.SeriesID)
+	st := s.db.GetStorage(series.StorageID)
+	if st.Implementation == db.ImplWebdav {
+		storageImpl, err := storage.NewWebdavStorage(st.Path, st.User, st.Password)
+		if err != nil {
+			return errors.Wrap(err, "new webdav")
+		}
+		if err := storageImpl.Move(filepath.Join(s.db.GetDownloadDir(), torrent.Name()), r.TargetDir); err != nil {
+			return errors.Wrap(err, "move webdav")
+		}
+	} else if st.Implementation == db.ImplLocal {
+		storageImpl := storage.NewLocalStorage(st.Path)
+
+		if err := storageImpl.Move(filepath.Join(s.db.GetDownloadDir(), torrent.Name()), r.TargetDir); err != nil {
+			return errors.Wrap(err, "move webdav")
+		}
+
+	}
+	log.Infof("move downloaded files to target dir success, file: %v, target dir: %v", torrent.Name(), r.TargetDir)
+	return nil
+}