polaris/server/core/resources.go

package core

import (
	"fmt"
	"polaris/ent"
	"polaris/ent/episode"
	"polaris/ent/history"
	"polaris/log"
	"polaris/pkg/notifier/message"
	"polaris/pkg/torznab"
	"polaris/pkg/utils"

	"github.com/pkg/errors"
)

func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum, episodeNum int) (*string, error) {
	trc, dlc, err := c.GetDownloadClient()
	if err != nil {
		return nil, errors.Wrap(err, "connect transmission")
	}
	series := c.db.GetMediaDetails(seriesId)
	if series == nil {
		return nil, fmt.Errorf("no tv series of id %v", seriesId)
	}

	//check space available
	downloadDir := c.db.GetDownloadDir()
	size := utils.AvailableSpace(downloadDir)
	if size < uint64(r1.Size) {
		log.Errorf("space available %v, space needed %v", size, r1.Size)
		return nil, errors.New("no enough space")
	}

	var ep *ent.Episode
	if episodeNum > 0 {
		for _, e := range series.Episodes {
			if e.SeasonNumber == seasonNum && e.EpisodeNumber == episodeNum {
				ep = e
			}
		}
		if ep == nil {
			return nil, errors.Errorf("no episode of season %d episode %d", seasonNum, episodeNum)
		}
	} else { //season package download
		ep = &ent.Episode{}
	}
	torrent, err := trc.Download(r1.Link, downloadDir)
	if err != nil {
		return nil, errors.Wrap(err, "downloading")
	}
	torrent.Start()

	dir := fmt.Sprintf("%s/Season %02d/", series.TargetDir, seasonNum)

	history, err := c.db.SaveHistoryRecord(ent.History{
		MediaID:          seriesId,
		EpisodeID:        ep.ID,
		SourceTitle:      r1.Name,
		TargetDir:        dir,
		Status:           history.StatusRunning,
		Size:             r1.Size,
		Saved:            torrent.Save(),
		Link:             r1.Link,
		DownloadClientID: dlc.ID,
		IndexerID:        r1.IndexerId,
	})
	if err != nil {
		return nil, errors.Wrap(err, "save record")
	}
	if episodeNum > 0 {
		if ep.Status == episode.StatusMissing {
			c.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading)
		}
	} else {
		c.db.SetSeasonAllEpisodeStatus(seriesId, seasonNum, episode.StatusDownloading)
	}

	c.tasks[history.ID] = &Task{Torrent: torrent}
	c.sendMsg(fmt.Sprintf(message.BeginDownload, r1.Name))

	log.Infof("success add %s to download task", r1.Name)
	return &r1.Name, nil

}
func (c *Client) SearchAndDownload(seriesId, seasonNum, episodeNum int) (*string, error) {
	var episodes []int
	if episodeNum > 0 {
		episodes = append(episodes, episodeNum)
	}
	res, err := SearchTvSeries(c.db, &SearchParam{
		MediaId:         seriesId,
		SeasonNum:       seasonNum,
		Episodes:        episodes,
		CheckFileSize:   true,
		CheckResolution: true,
	})
	if err != nil {
		return nil, err
	}
	r1 := res[0]
	log.Infof("found resource to download: %+v", r1)
	return c.DownloadEpisodeTorrent(r1, seriesId, seasonNum, episodeNum)
}

func (c *Client) DownloadMovie(m *ent.Media, link, name string, size int, indexerID int) (*string, error) {
	trc, dlc, err := c.GetDownloadClient()
	if err != nil {
		return nil, errors.Wrap(err, "connect transmission")
	}

	torrent, err := trc.Download(link, c.db.GetDownloadDir())
	if err != nil {
		return nil, errors.Wrap(err, "downloading")
	}
	torrent.Start()

	if name == "" {
		name = m.OriginalName
	}
	go func() {
		ep, _ := c.db.GetMovieDummyEpisode(m.ID)
		history, err := c.db.SaveHistoryRecord(ent.History{
			MediaID:          m.ID,
			EpisodeID:        ep.ID,
			SourceTitle:      name,
			TargetDir:        m.TargetDir,
			Status:           history.StatusRunning,
			Size:             size,
			Saved:            torrent.Save(),
			Link:             link,
			DownloadClientID: dlc.ID,
			IndexerID:        indexerID,
		})
		if err != nil {
			log.Errorf("save history error: %v", err)
		}

		c.tasks[history.ID] = &Task{Torrent: torrent}

		if ep.Status == episode.StatusMissing {
			c.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading)
		}

	}()

	c.sendMsg(fmt.Sprintf(message.BeginDownload, name))
	log.Infof("success add %s to download task", name)
	return &name, nil

}