109 lines
2.1 KiB
Go
109 lines
2.1 KiB
Go
package ninegag
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"govd/enums"
|
|
"govd/models"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func FindBestPhoto(
|
|
images map[string]*Media,
|
|
) (*Media, error) {
|
|
var bestPhoto *Media
|
|
var maxWidth int
|
|
|
|
for _, photo := range images {
|
|
if !strings.HasSuffix(photo.URL, ".jpg") {
|
|
continue
|
|
}
|
|
if photo.Width > maxWidth {
|
|
maxWidth = photo.Width
|
|
bestPhoto = photo
|
|
}
|
|
}
|
|
|
|
if bestPhoto == nil {
|
|
return nil, errors.New("no photo found in post")
|
|
}
|
|
|
|
return bestPhoto, nil
|
|
}
|
|
|
|
func ParseVideoFormats(
|
|
images map[string]*Media,
|
|
) ([]*models.MediaFormat, error) {
|
|
var video *Media
|
|
var thumbnailURL string
|
|
|
|
for _, media := range images {
|
|
if media.Duration > 0 {
|
|
video = media
|
|
}
|
|
if strings.HasSuffix(media.URL, ".jpg") {
|
|
thumbnailURL = media.URL
|
|
}
|
|
}
|
|
if video == nil {
|
|
return nil, errors.New("no video found in post")
|
|
}
|
|
|
|
codecMapping := map[string]struct {
|
|
Field string
|
|
Codec enums.MediaCodec
|
|
}{
|
|
"url": {"URL", enums.MediaCodecAVC},
|
|
"h265Url": {"H265URL", enums.MediaCodecHEVC},
|
|
"vp8Url": {"Vp8URL", enums.MediaCodecVP8},
|
|
"vp9Url": {"Vp9URL", enums.MediaCodecVP9},
|
|
"av1Url": {"Av1URL", enums.MediaCodecAV1},
|
|
}
|
|
|
|
formats := make([]*models.MediaFormat, 0, len(codecMapping))
|
|
|
|
for _, mapping := range codecMapping {
|
|
url := getField(video, mapping.Field)
|
|
if url == "" {
|
|
continue
|
|
}
|
|
|
|
format := &models.MediaFormat{
|
|
FormatID: fmt.Sprintf("video_%s", mapping.Codec),
|
|
Type: enums.MediaTypeVideo,
|
|
VideoCodec: mapping.Codec,
|
|
AudioCodec: enums.MediaCodecAAC,
|
|
URL: []string{url},
|
|
Width: int64(video.Width),
|
|
Height: int64(video.Height),
|
|
Duration: int64(video.Duration),
|
|
}
|
|
|
|
if thumbnailURL != "" {
|
|
format.Thumbnail = []string{thumbnailURL}
|
|
}
|
|
|
|
formats = append(formats, format)
|
|
}
|
|
|
|
return formats, nil
|
|
}
|
|
|
|
func getField(media *Media, fieldName string) string {
|
|
switch fieldName {
|
|
case "URL":
|
|
return media.URL
|
|
case "H265URL":
|
|
return media.H265URL
|
|
case "Vp8URL":
|
|
return media.Vp8URL
|
|
case "Vp9URL":
|
|
return media.Vp9URL
|
|
case "Av1URL":
|
|
return media.Av1URL
|
|
default:
|
|
return ""
|
|
}
|
|
}
|