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

Compare commits

base: logan:9b5b980e1d495c3631b8f8b15fa1ae7bc4c8575f
Branches Tags
logan:master
..
compare: logan:7f2c9fa130ea6065078f83a4e30b9dc80cfb98cb
Branches Tags
logan:master

No commits in common. "9b5b980e1d495c3631b8f8b15fa1ae7bc4c8575f" and "7f2c9fa130ea6065078f83a4e30b9dc80cfb98cb" have entirely different histories.
9b5b980e1d ... 7f2c9fa130

2 changed files with 5 additions and 84 deletions
Show stats Expand all files Collapse all files

44
helpers.go
View file

@ -42,7 +42,7 @@ func getLargestFormat(input goutubedl.Result, sizeLimit float64) (format goutube
// Cringe YouTube
func getLargestDashFormat(input goutubedl.Result, sizeLimit float64) (videoFormat goutubedl.Format, audioFormat goutubedl.Format, err error) {
// Get best Opus audio format
// Get best audio format
var bestAudio goutubedl.Format
for i := len(input.Formats())-1; i >= 0; i-- {
if input.Formats()[i].ACodec == "opus" {
@ -51,17 +51,7 @@ func getLargestDashFormat(input goutubedl.Result, sizeLimit float64) (videoForma
}
}
// 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
// Get largest 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
@ -76,36 +66,6 @@ func getLargestDashFormat(input goutubedl.Result, sizeLimit float64) (videoForma
}
}
// 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].VBR > 0 && input.Formats()[i].ASR > 0 {
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))
}

45
messageCreate.go
View file

@ -5,14 +5,14 @@ import (
"log"
"regexp"
"context"
"bytes"
"github.com/bwmarrin/discordgo"
"github.com/wader/goutubedl"
"mvdan.cc/xurls/v2" // Peak lazy
)
// This function will be called every time a new message is created on any channel that the authenticated bot has access to.
// This function will be called (due to AddHandler above) every time a new
// message is created on any channel that the authenticated bot has access to.
func messageCreate(session *discordgo.Session, message *discordgo.MessageCreate) {
// Ignore all messages created by the bot itself
if message.Author.ID == session.State.User.ID {
@ -106,17 +106,10 @@ func messageCreate(session *discordgo.Session, message *discordgo.MessageCreate)
log.Println(err)
continue
} else {
/*
videoChoice, audioChoice, err := getLargestDashFormat(result, 8*1024*1024)
log.Printf("Choice: %s+%s | Size: %fM\n", videoChoice.FormatID, audioChoice.FormatID, (videoChoice.Filesize+audioChoice.Filesize)/1024/1024)
if err == nil {
downloadResult, err := result.Download(context.Background(), fmt.Sprintf("%s+%s", videoChoice.FormatID, audioChoice.FormatID))
*/
choice, err := getLargestYTFormat(result, 8*1024*1024)
log.Printf("Choice: %s | Size: %fM\n", choice.FormatID, choice.FilesizeApprox/1024/1024)
if err == nil {
downloadResult, err := result.Download(context.Background(), choice.FormatID)
if err != nil {
log.Println(err)
//session.ChannelMessageSend(message.ChannelID, err.Error())
@ -124,8 +117,7 @@ func messageCreate(session *discordgo.Session, message *discordgo.MessageCreate)
}
response.Files = append(response.Files, &discordgo.File {
//Name: fmt.Sprintf("%s.%s", result.Info.ID, videoChoice.Ext),
Name: fmt.Sprintf("%s.%s", result.Info.ID, choice.Ext),
Name: fmt.Sprintf("%s.%s", result.Info.ID, videoChoice.Ext),
ContentType: "text/plain", // This is of course not true, but Discord doesn't give a shit
Reader: downloadResult,
})
@ -136,37 +128,6 @@ func messageCreate(session *discordgo.Session, message *discordgo.MessageCreate)
}
}
respond = true
} else if output, _ := regexp.MatchString("(http.*instagram\\.com/p/.*)|(http.*instagram\\.com/tv/.*)|(http.*instagram\\.com/reel/.*)", url); output {
log.Println("Instagram detected.")
response.Content = response.Content + fmt.Sprintf("<%s>", regexp.MustCompile("http.*instagram\\.com").ReplaceAllString(url, "https://bibliogram.art"))
result, err := goutubedl.New(context.Background(), url, goutubedl.Options{})
if err != nil {
log.Println(err)
continue
} else {
downloadResult, err := result.Download(context.Background(), result.Formats()[0].FormatID) // Insta only has one possible format to download :(
if err != nil {
log.Println(err)
continue
}
buf := new(bytes.Buffer)
buf.ReadFrom(downloadResult)
downloadResult.Close()
if buf.Len() <= 8*1024*1024 {
response.Files = append(response.Files, &discordgo.File {
Name: fmt.Sprintf("%s.%s", result.Info.ID, result.Formats()[0].Ext),
ContentType: "text/plain", // This is of course not true, but Discord doesn't give a shit
Reader: buf,
})
}
}
respond = true
}
}