mirror of
https://github.com/simon-ding/polaris.git
synced 2026-02-25 13:10:46 +08:00
346 lines
9.0 KiB
Go
346 lines
9.0 KiB
Go
package server
|
||
|
||
import (
|
||
"path/filepath"
|
||
"polaris/ent"
|
||
"polaris/ent/episode"
|
||
"polaris/ent/history"
|
||
"polaris/ent/media"
|
||
storage1 "polaris/ent/storage"
|
||
"polaris/log"
|
||
"polaris/pkg"
|
||
"polaris/pkg/storage"
|
||
"polaris/pkg/utils"
|
||
"polaris/server/core"
|
||
"time"
|
||
|
||
"github.com/pkg/errors"
|
||
)
|
||
|
||
func (s *Server) scheduler() {
|
||
s.mustAddCron("@every 1m", s.checkTasks)
|
||
s.mustAddCron("@every 1h", func() {
|
||
s.downloadTvSeries()
|
||
s.downloadMovie()
|
||
})
|
||
s.mustAddCron("@every 12h", s.checkAllSeriesNewSeason)
|
||
s.cron.Start()
|
||
}
|
||
|
||
func (s *Server) mustAddCron(spec string, cmd func()) {
|
||
if err := s.cron.AddFunc(spec, cmd); err != nil {
|
||
log.Errorf("add func error: %v", err)
|
||
panic(err)
|
||
}
|
||
}
|
||
|
||
func (s *Server) checkTasks() {
|
||
log.Debug("begin check tasks...")
|
||
for id, t := range s.tasks {
|
||
if !t.Exists() {
|
||
log.Infof("task no longer exists: %v", id)
|
||
continue
|
||
}
|
||
log.Infof("task (%s) percentage done: %d%%", t.Name(), t.Progress())
|
||
if t.Progress() == 100 {
|
||
log.Infof("task is done: %v", t.Name())
|
||
go func() {
|
||
if err := s.moveCompletedTask(id); err != nil {
|
||
log.Infof("post tasks for id %v fail: %v", id, err)
|
||
}
|
||
}()
|
||
}
|
||
}
|
||
}
|
||
|
||
func (s *Server) moveCompletedTask(id int) (err1 error) {
|
||
torrent := s.tasks[id]
|
||
r := s.db.GetHistory(id)
|
||
if r.Status == history.StatusUploading {
|
||
log.Infof("task %d is already uploading, skip", id)
|
||
return nil
|
||
}
|
||
s.db.SetHistoryStatus(r.ID, history.StatusUploading)
|
||
|
||
defer func() {
|
||
seasonNum, err := utils.SeasonId(r.TargetDir)
|
||
if err != nil {
|
||
log.Errorf("no season id: %v", r.TargetDir)
|
||
seasonNum = -1
|
||
}
|
||
|
||
if err1 != nil {
|
||
s.db.SetHistoryStatus(r.ID, history.StatusFail)
|
||
if r.EpisodeID != 0 {
|
||
s.db.SetEpisodeStatus(r.EpisodeID, episode.StatusMissing)
|
||
} else {
|
||
s.db.SetSeasonAllEpisodeStatus(r.MediaID, seasonNum, episode.StatusMissing)
|
||
}
|
||
|
||
} else {
|
||
delete(s.tasks, r.ID)
|
||
s.db.SetHistoryStatus(r.ID, history.StatusSuccess)
|
||
if r.EpisodeID != 0 {
|
||
s.db.SetEpisodeStatus(r.EpisodeID, episode.StatusDownloaded)
|
||
} else {
|
||
s.db.SetSeasonAllEpisodeStatus(r.MediaID, seasonNum, episode.StatusDownloaded)
|
||
}
|
||
|
||
torrent.Remove()
|
||
}
|
||
}()
|
||
|
||
series := s.db.GetMediaDetails(r.MediaID)
|
||
if series == nil {
|
||
return nil
|
||
}
|
||
st := s.db.GetStorage(series.StorageID)
|
||
log.Infof("move task files to target dir: %v", r.TargetDir)
|
||
var stImpl storage.Storage
|
||
if st.Implementation == storage1.ImplementationWebdav {
|
||
ws := st.ToWebDavSetting()
|
||
targetPath := ws.TvPath
|
||
if series.MediaType == media.MediaTypeMovie {
|
||
targetPath = ws.MoviePath
|
||
}
|
||
storageImpl, err := storage.NewWebdavStorage(ws.URL, ws.User, ws.Password, targetPath, ws.ChangeFileHash == "true")
|
||
if err != nil {
|
||
return errors.Wrap(err, "new webdav")
|
||
}
|
||
stImpl = storageImpl
|
||
|
||
} else if st.Implementation == storage1.ImplementationLocal {
|
||
ls := st.ToLocalSetting()
|
||
targetPath := ls.TvPath
|
||
if series.MediaType == media.MediaTypeMovie {
|
||
targetPath = ls.MoviePath
|
||
}
|
||
|
||
storageImpl, err := storage.NewLocalStorage(targetPath)
|
||
if err != nil {
|
||
return errors.Wrap(err, "new storage")
|
||
|
||
}
|
||
stImpl = storageImpl
|
||
|
||
}
|
||
if r.EpisodeID == 0 {
|
||
//season package download
|
||
if err := stImpl.Move(filepath.Join(s.db.GetDownloadDir(), torrent.Name()), r.TargetDir); err != nil {
|
||
return errors.Wrap(err, "move file")
|
||
|
||
}
|
||
|
||
} else {
|
||
if err := stImpl.Move(filepath.Join(s.db.GetDownloadDir(), torrent.Name()), filepath.Join(r.TargetDir, torrent.Name())); err != nil {
|
||
return errors.Wrap(err, "move file")
|
||
|
||
}
|
||
}
|
||
|
||
log.Infof("move downloaded files to target dir success, file: %v, target dir: %v", torrent.Name(), r.TargetDir)
|
||
return nil
|
||
}
|
||
|
||
func (s *Server) checkDownloadedSeriesFiles(m *ent.Media) error {
|
||
if m.MediaType != media.MediaTypeTv {
|
||
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
|
||
}
|
||
files, err := storageImpl.ReadDir(m.TargetDir)
|
||
if err != nil {
|
||
return errors.Wrapf(err, "read dir %s", m.TargetDir)
|
||
}
|
||
|
||
for _, in := range files {
|
||
if !in.IsDir() { //season dir, ignore file
|
||
continue
|
||
}
|
||
dir := filepath.Join(m.TargetDir, in.Name())
|
||
epFiles, err := storageImpl.ReadDir(dir)
|
||
if err != nil {
|
||
log.Errorf("read dir %s error: %v", dir, err)
|
||
continue
|
||
}
|
||
for _, ep := range epFiles {
|
||
log.Infof("found file: %v", ep.Name())
|
||
seNum, epNum, err := utils.FindSeasonEpisodeNum(ep.Name())
|
||
if err != nil {
|
||
log.Errorf("find season episode num error: %v", err)
|
||
continue
|
||
}
|
||
var dirname = filepath.Join(in.Name(), ep.Name())
|
||
log.Infof("found match, season num %d, episode num %d", seNum, epNum)
|
||
err = s.db.UpdateEpisodeFile(m.ID, seNum, epNum, dirname)
|
||
if err != nil {
|
||
log.Error("update episode: %v", err)
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
|
||
}
|
||
|
||
type Task struct {
|
||
//Processing bool
|
||
pkg.Torrent
|
||
}
|
||
|
||
func (s *Server) downloadTvSeries() {
|
||
log.Infof("begin check all tv series resources")
|
||
allSeries := s.db.GetMediaWatchlist(media.MediaTypeTv)
|
||
for _, series := range allSeries {
|
||
tvDetail := s.db.GetMediaDetails(series.ID)
|
||
for _, ep := range tvDetail.Episodes {
|
||
t, err := time.Parse("2006-01-02", ep.AirDate)
|
||
if err != nil {
|
||
log.Error("air date not known, skip: %v", ep.Title)
|
||
continue
|
||
}
|
||
if series.CreatedAt.Sub(t) > 24*time.Hour { //剧集在加入watchlist之前，不去下载
|
||
continue
|
||
}
|
||
if ep.Status != episode.StatusMissing { //已经下载的不去下载
|
||
continue
|
||
}
|
||
name, err := s.searchAndDownload(series.ID, ep.SeasonNumber, ep.EpisodeNumber)
|
||
if err != nil {
|
||
log.Infof("cannot find resource to download for %s: %v", ep.Title, err)
|
||
} else {
|
||
log.Infof("begin download torrent resource: %v", name)
|
||
}
|
||
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
func (s *Server) downloadMovie() {
|
||
log.Infof("begin check all movie resources")
|
||
allSeries := s.db.GetMediaWatchlist(media.MediaTypeMovie)
|
||
|
||
for _, series := range allSeries {
|
||
detail := s.db.GetMediaDetails(series.ID)
|
||
if len(detail.Episodes) == 0 {
|
||
log.Errorf("no related dummy episode: %v", detail.NameEn)
|
||
continue
|
||
}
|
||
ep := detail.Episodes[0]
|
||
if ep.Status == episode.StatusDownloaded {
|
||
continue
|
||
}
|
||
|
||
if err := s.downloadMovieSingleEpisode(ep); err != nil {
|
||
log.Errorf("download movie error: %v", err)
|
||
}
|
||
}
|
||
}
|
||
|
||
func (s *Server) downloadMovieSingleEpisode(ep *ent.Episode) error {
|
||
trc, err := s.getDownloadClient()
|
||
if err != nil {
|
||
return errors.Wrap(err, "connect transmission")
|
||
}
|
||
|
||
res, err := core.SearchMovie(s.db, ep.MediaID, true)
|
||
if err != nil {
|
||
|
||
return errors.Wrap(err, "search movie")
|
||
}
|
||
r1 := res[0]
|
||
log.Infof("begin download torrent resource: %v", r1.Name)
|
||
torrent, err := trc.Download(r1.Link, s.db.GetDownloadDir())
|
||
if err != nil {
|
||
return errors.Wrap(err, "downloading")
|
||
}
|
||
torrent.Start()
|
||
|
||
history, err := s.db.SaveHistoryRecord(ent.History{
|
||
MediaID: ep.MediaID,
|
||
EpisodeID: ep.ID,
|
||
SourceTitle: r1.Name,
|
||
TargetDir: "./",
|
||
Status: history.StatusRunning,
|
||
Size: r1.Size,
|
||
Saved: torrent.Save(),
|
||
})
|
||
if err != nil {
|
||
log.Errorf("save history error: %v", err)
|
||
}
|
||
|
||
s.tasks[history.ID] = &Task{Torrent: torrent}
|
||
|
||
s.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading)
|
||
return nil
|
||
}
|
||
|
||
func (s *Server) checkAllSeriesNewSeason() {
|
||
log.Infof("begin checking series all new season")
|
||
allSeries := s.db.GetMediaWatchlist(media.MediaTypeTv)
|
||
for _, series := range allSeries {
|
||
err := s.checkSeiesNewSeason(series)
|
||
if err != nil {
|
||
log.Errorf("check series new season error: series name %v, error: %v", series.NameEn, err)
|
||
}
|
||
}
|
||
}
|
||
|
||
func (s *Server) checkSeiesNewSeason(media *ent.Media) error{
|
||
d, err := s.MustTMDB().GetTvDetails(media.TmdbID, s.language)
|
||
if err != nil {
|
||
return errors.Wrap(err, "tmdb")
|
||
}
|
||
lastsSason := d.NumberOfSeasons
|
||
seasonDetail, err := s.MustTMDB().GetSeasonDetails(media.TmdbID, lastsSason, s.language)
|
||
if err != nil {
|
||
return errors.Wrap(err, "tmdb season")
|
||
}
|
||
|
||
for _, ep := range seasonDetail.Episodes {
|
||
epDb, err := s.db.GetEpisode(media.ID, ep.SeasonNumber, ep.EpisodeNumber)
|
||
if err != nil {
|
||
if ent.IsNotFound(err) {
|
||
log.Infof("add new episode: %+v", ep)
|
||
episode := &ent.Episode{
|
||
MediaID: media.ID,
|
||
SeasonNumber: ep.SeasonNumber,
|
||
EpisodeNumber: ep.EpisodeNumber,
|
||
Title: ep.Name,
|
||
Overview: ep.Overview,
|
||
AirDate: ep.AirDate,
|
||
Status: episode.StatusMissing,
|
||
}
|
||
s.db.SaveEposideDetail2(episode)
|
||
}
|
||
} else {//update episode
|
||
if ep.Name != epDb.Title || ep.Overview != epDb.Overview || ep.AirDate != epDb.AirDate {
|
||
log.Infof("update new episode: %+v", ep)
|
||
s.db.UpdateEpiode2(epDb.ID, ep.Name, ep.Overview, ep.AirDate)
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
} |