jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-06-07 10:37:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

code refactor and add season pack size limit

Browse Source
This commit is contained in:
Simon Ding
2024-08-11 17:40:01 +08:00
parent 93e8e78591
commit f110f257d4
5 changed files with 115 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
server/resources.go
View File

@@ -13,7 +13,13 @@ import (
func (s *Server) searchAndDownloadSeasonPackage(seriesId, seasonNum int) (*string, error) {
res, err := core.SearchTvSeries(s.db, seriesId, seasonNum, nil, true, true)
res, err := core.SearchTvSeries(s.db, &core.SearchParam{
MediaId: seriesId,
SeasonNum: seasonNum,
Episodes: nil,
CheckResolution: true,
CheckFileSize: true,
})
if err != nil {
return nil, err
}
@@ -46,13 +52,21 @@ func (s *Server) SearchAvailableTorrents(c *gin.Context) (interface{}, error) {
if in.Episode == 0 {
//search season package
log.Infof("search series season package S%02d", in.Season)
res, err = core.SearchTvSeries(s.db, in.ID, in.Season, nil, false, false)
res, err = core.SearchTvSeries(s.db, &core.SearchParam{
MediaId: in.ID,
SeasonNum: in.Season,
Episodes: nil,
})
if err != nil {
return nil, errors.Wrap(err, "search season package")
}
} else {
log.Infof("search series episode S%02dE%02d", in.Season, in.Episode)
res, err = core.SearchTvSeries(s.db, in.ID, in.Season, []int{in.Episode}, false, false)
res, err = core.SearchTvSeries(s.db, &core.SearchParam{
MediaId: in.ID,
SeasonNum: in.Season,
Episodes: []int{in.Episode},
})
if err != nil {
if err.Error() == "no resource found" {
return []string{}, nil
@@ -63,7 +77,9 @@ func (s *Server) SearchAvailableTorrents(c *gin.Context) (interface{}, error) {
}
} else {
log.Info("search movie %d", in.ID)
res, err = core.SearchMovie(s.db, in.ID, false, false)
res, err = core.SearchMovie(s.db, &core.SearchParam{
MediaId: in.ID,
})
if err != nil {
if err.Error() == "no resource found" {
return []string{}, nil