polaris/server/scheduler.go

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
}