EngiBot/helpers.go

package main

import (
    "log"
    "fmt"
    "net/http"
    "errors"

    "github.com/wader/goutubedl"
    "golang.org/x/exp/slices"
)

var FileExtensions = []string {"webm","mp4","mkv"}

func getRedirectURL(input string) (string, error) {
    request, err := http.NewRequest("GET", input, nil)
    if err != nil {
        return "", err
    }

    client := &http.Client{}
    resp, err := client.Do(request)
    if err != nil {
        return "", err
    }

    log.Printf("Found new URL: %s", resp.Request.URL.String())
    return resp.Request.URL.String(), nil
}

func getLargestFormat(input goutubedl.Result, sizeLimit float64) (format goutubedl.Format, err error) {
    for i := len(input.Formats())-1; i >= 0; i-- {
        var size float64
        if input.Formats()[i].FilesizeApprox > 0 {
            size = input.Formats()[i].FilesizeApprox
        } else {
            size = input.Formats()[i].Filesize
        }

        if size <= sizeLimit && input.Formats()[i].VCodec != "h265" && input.Formats()[i].ACodec != "none" && slices.Contains(FileExtensions, input.Formats()[i].Ext) { // Discord can't embed HEVC
            return input.Formats()[i], nil
        }
    }
    return goutubedl.Format{}, errors.New(fmt.Sprintf("Could not find a video format that is under the size limit of %f", sizeLimit))
}

// Cringe YouTube
func getLargestDashFormat(input goutubedl.Result, sizeLimit float64) (videoFormat goutubedl.Format, audioFormat goutubedl.Format, err error) {
    // Get best Opus audio format
    var bestAudio goutubedl.Format
    for i := len(input.Formats())-1; i >= 0; i-- {
        if input.Formats()[i].ACodec == "opus" {
            bestAudio = input.Formats()[i]
            break
        }
    }

    // If there aren't any Opus formats (somehow), use whatever else we can find
    if len(bestAudio.FormatID) == 0 {
        for i := len(input.Formats())-1; i >= 0; i-- {
            if input.Formats()[i].ABR > bestAudio.ABR {
                bestAudio = input.Formats()[i]
            }
        }
    }


    // Get largest VP9 video that will fit in the limit with the audio
    for i := len(input.Formats())-1; i >= 0; i-- {
        if input.Formats()[i].VCodec == "vp9" {
            var size float64
            if input.Formats()[i].FilesizeApprox > 0 {
                size = input.Formats()[i].FilesizeApprox + bestAudio.FilesizeApprox
            } else {
                size = input.Formats()[i].Filesize + bestAudio.Filesize
            }
            if size <= sizeLimit {
                return input.Formats()[i], bestAudio, nil
            }
        }
    }

    // If VP9 isn't available, just use whatever fits
    for i := len(input.Formats())-1; i >= 0; i-- {
        var size float64
        if input.Formats()[i].FilesizeApprox > 0 {
            size = input.Formats()[i].FilesizeApprox + bestAudio.FilesizeApprox
        } else {
            size = input.Formats()[i].Filesize + bestAudio.Filesize
        }
        if size <= sizeLimit {
            return input.Formats()[i], bestAudio, nil
        }
    }

    return goutubedl.Format{}, goutubedl.Format{}, errors.New(fmt.Sprintf("Could not find a video format that is under the size limit of %f", sizeLimit))
}

func getLargestYTFormat(input goutubedl.Result, sizeLimit float64) (format goutubedl.Format, err error) {
    for i := len(input.Formats())-1; i >= 0; i-- {
        if input.Formats()[i].FPS > 0 && input.Formats()[i].ASR > 0 && slices.Contains(FileExtensions, input.Formats()[i].Ext) {
            var size float64
            if input.Formats()[i].FilesizeApprox > 0 {
                size = input.Formats()[i].FilesizeApprox
            } else {
                size = input.Formats()[i].Filesize
            }
            if size <= sizeLimit {
                return input.Formats()[i], nil
            }
        }
    }

    return goutubedl.Format{}, errors.New(fmt.Sprintf("Could not find a video format that is under the size limit of %f", sizeLimit))
}