39 lines
769 B
Go
39 lines
769 B
Go
package reddit
|
|
|
|
import (
|
|
"fmt"
|
|
"govd/models"
|
|
"govd/util"
|
|
"govd/util/parser"
|
|
"regexp"
|
|
)
|
|
|
|
const (
|
|
hlsURLFormat = "https://v.redd.it/%s/HLSPlaylist.m3u8"
|
|
)
|
|
|
|
var videoURLPattern = regexp.MustCompile(`https?://v\.redd\.it/([^/]+)`)
|
|
|
|
func GetHLSFormats(videoURL string, thumbnail string, duration int64) ([]*models.MediaFormat, error) {
|
|
matches := videoURLPattern.FindStringSubmatch(videoURL)
|
|
if len(matches) < 2 {
|
|
return nil, nil
|
|
}
|
|
|
|
videoID := matches[1]
|
|
hlsURL := fmt.Sprintf(hlsURLFormat, videoID)
|
|
|
|
formats, err := parser.ParseM3U8FromURL(hlsURL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, format := range formats {
|
|
format.Duration = duration
|
|
if thumbnail != "" {
|
|
format.Thumbnail = []string{util.FixURL(thumbnail)}
|
|
}
|
|
}
|
|
|
|
return formats, nil
|
|
}
|