NyxErinys/crunchy-cli
1
0
Fork 0
mirror of https://github.com/crunchy-labs/crunchy-cli.git synced 2026-01-21 12:12:00 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added 'worst' and 'best' resolution filter to GetFormat(...)

Browse source
This commit is contained in:
bytedream 2022-02-13 13:15:21 +01:00
parent 1c3de9ea93
commit c557486089
1 changed files with 24 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

24
episode.go
View file

@ -4,6 +4,8 @@ import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"regexp" "regexp"
"strconv"
"strings"
"time" "time"
) )
@ -147,11 +149,33 @@ func (e *Episode) GetFormat(resolution string, subtitle LOCALE, hardsub bool) (*
if err != nil { if err != nil {
return nil, err return nil, err
} }
var res *Format
for _, format := range formats { for _, format := range formats {
if resolution == "worst" || resolution == "best" {
curSplitRes := strings.SplitN(format.Video.Resolution, "x", 1)
curResX, _ := strconv.Atoi(curSplitRes[0])
curResY, _ := strconv.Atoi(curSplitRes[1])
resSplitRes := strings.SplitN(res.Video.Resolution, "x", 1)
resResX, _ := strconv.Atoi(resSplitRes[0])
resResY, _ := strconv.Atoi(resSplitRes[1])
if resolution == "worst" && curResX+curResY < resResX+resResY {
res = format
} else if resolution == "best" && curResX+curResY > resResX+resResY {
res = format
}
}
if format.Video.Resolution == resolution { if format.Video.Resolution == resolution {
return format, nil return format, nil
} }
} }
if res != nil {
return res, nil
}
return nil, fmt.Errorf("no matching resolution found") return nil, fmt.Errorf("no matching resolution found")
} }