logan/EngiBot
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Compare commits

base: logan:a1330b3a3380ae4ab6fb47735f9f986345d0a0b1
Branches Tags
logan:master
...
compare: logan:1a2b019b4bb7d9a6a660ac88e7a4da5220aea56e
Branches Tags
logan:master

3 commits
a1330b3a33 ... 1a2b019b4b

Author SHA1 Message Date
Logan Gartner
1a2b019b4b
Removed redundant check 2022-09-16 18:32:40 -06:00
Logan Gartner
f4705a0b7f
Added file extension check to YT download 2022-09-16 18:32:22 -06:00
Logan Gartner
d3d7cc9a9c
Don't forget to remove the . 2022-09-16 18:29:47 -06:00
1 changed files with 3 additions and 3 deletions
Show stats Expand all files Collapse all files

6
helpers.go
View file

@ -10,7 +10,7 @@ import (
"golang.org/x/exp/slices" "golang.org/x/exp/slices"
) )
var FileExtensions = []string {".webm",".mp4",".mkv"} var FileExtensions = []string {"webm","mp4","mkv"}
func getRedirectURL(input string) (string, error) { func getRedirectURL(input string) (string, error) {
request, err := http.NewRequest("GET", input, nil) request, err := http.NewRequest("GET", input, nil)
@ -97,14 +97,14 @@ func getLargestDashFormat(input goutubedl.Result, sizeLimit float64) (videoForma
func getLargestYTFormat(input goutubedl.Result, sizeLimit float64) (format goutubedl.Format, err error) { func getLargestYTFormat(input goutubedl.Result, sizeLimit float64) (format goutubedl.Format, err error) {
for i := len(input.Formats())-1; i >= 0; i-- { for i := len(input.Formats())-1; i >= 0; i-- {
if input.Formats()[i].VBR > 0 && input.Formats()[i].ASR > 0 { if input.Formats()[i].VBR > 0 && input.Formats()[i].ASR > 0 && slices.Contains(FileExtensions, input.Formats()[i].Ext) {
var size float64 var size float64
if input.Formats()[i].FilesizeApprox > 0 { if input.Formats()[i].FilesizeApprox > 0 {
size = input.Formats()[i].FilesizeApprox size = input.Formats()[i].FilesizeApprox
} else { } else {
size = input.Formats()[i].Filesize size = input.Formats()[i].Filesize
} }
if size <= sizeLimit && input.Formats()[i].FormatID != "17" { if size <= sizeLimit {
return input.Formats()[i], nil return input.Formats()[i], nil
} }
} }