polaris/server/activity.go

package server

import (
	"fmt"
	"polaris/ent"
	"polaris/ent/episode"
	"polaris/ent/history"
	"polaris/log"
	"polaris/pkg/utils"
	"strconv"

	"github.com/gin-gonic/gin"
	"github.com/pkg/errors"
)

type Activity struct {
	*ent.History
	Progress int `json:"progress"`
}

func (s *Server) GetAllActivities(c *gin.Context) (interface{}, error) {
	q := c.Query("status")
	his := s.db.GetHistories()
	var activities = make([]Activity, 0, len(his))
	for _, h := range his {
		if q == "active" && (h.Status != history.StatusRunning && h.Status != history.StatusUploading) {
			continue //active downloads
		} else if q == "archive" && (h.Status == history.StatusRunning || h.Status == history.StatusUploading) {
			continue //archived downloads
		}

		a := Activity{
			History: h,
		}
		for id, task := range s.tasks {
			if h.ID == id && task.Exists() {
				a.Progress = task.Progress()
			}
		}
		activities = append(activities, a)
	}

	return activities, nil
}

func (s *Server) RemoveActivity(c *gin.Context) (interface{}, error) {
	ids := c.Param("id")
	id, err := strconv.Atoi(ids)
	if err != nil {
		return nil, errors.Wrap(err, "convert")
	}
	his := s.db.GetHistory(id)
	if his == nil {
		log.Errorf("no record of id: %d", id)
		return nil, nil
	}
	torrent := s.tasks[his.ID]
	if torrent != nil {
		if err := torrent.Remove(); err != nil {
			return nil, errors.Wrap(err, "remove torrent")
		}
		delete(s.tasks, his.ID)
	}
	if his.EpisodeID != 0 {
		s.db.SetEpisodeStatus(his.EpisodeID, episode.StatusMissing)

	} else {
		seasonNum, err := utils.SeasonId(his.TargetDir)
		if err != nil {
			log.Errorf("no season id: %v", his.TargetDir)
			seasonNum = -1
		}
		s.db.SetSeasonAllEpisodeStatus(his.MediaID, seasonNum, episode.StatusMissing)

	}

	err = s.db.DeleteHistory(id)
	if err != nil {
		return nil, errors.Wrap(err, "db")
	}
	log.Infof("history record successful deleted: %v", his.SourceTitle)
	return nil, nil
}
func (s *Server) GetMediaDownloadHistory(c *gin.Context) (interface{}, error) {
	var ids = c.Param("id")
	id, err := strconv.Atoi(ids)
	if err != nil {
		return nil, fmt.Errorf("id is not correct: %v", ids)
	}
	his, err := s.db.GetDownloadHistory(id)
	if err != nil {
		return nil, errors.Wrap(err, "db")
	}
	return his, nil
}

type TorrentInfo struct {
	Name      string  `json:"name"`
	ID        int64     `json:"id"`
	SeedRatio float32 `json:"seed_ratio"`
	Progress  int     `json:"progress"`
}

func (s *Server) GetAllTorrents(c *gin.Context) (interface{}, error) {
	trc, _, err := s.getDownloadClient()
	if err != nil {
		return nil, errors.Wrap(err, "connect transmission")
	}
	all, err := trc.GetAll()
	if err != nil {
		return nil, errors.Wrap(err, "get all")
	}
	var infos []TorrentInfo
	for _, t := range all {
		if !t.Exists() {
			continue
		}
		infos = append(infos, TorrentInfo{
			Name: t.Name(),
			ID: t.ID,
			Progress: t.Progress(),
		})
	}
	return infos, nil
}