ytdl-web/web/formats.go

35 lines
717 B
Go
Raw Normal View History

package web
import (
"github.com/samber/lo"
2023-05-23 18:44:05 -04:00
"go.fifitido.net/ytdl-web/ytdl/metadata"
)
type Video struct {
2023-05-23 18:44:05 -04:00
Meta *metadata.Metadata
Formats []metadata.Format
}
2023-05-23 18:44:05 -04:00
func GetVideos(meta *metadata.Metadata) []Video {
if meta.Type == "playlist" {
2023-05-23 18:44:05 -04:00
return lo.Map(meta.Entries, func(video metadata.Metadata, _ int) Video {
return GetVideos(&video)[0]
})
}
2023-05-23 18:44:05 -04:00
formats := lo.Filter(meta.Formats, func(item metadata.Format, _ int) bool {
return item.ACodec != "none" && item.VCodec != "none" && item.Protocol != "m3u8_native"
})
for i, j := 0, len(formats)-1; i < j; i, j = i+1, j-1 {
formats[i], formats[j] = formats[j], formats[i]
}
return []Video{
{
Meta: meta,
Formats: formats,
},
}
}