mirror of
https://github.com/lyx0/nourybot-matrix.git
synced 2024-11-13 19:49:54 +01:00
78 lines
2 KiB
Go
78 lines
2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/google/uuid"
|
||
|
|
"github.com/wader/goutubedl"
|
||
|
|
"maunium.net/go/mautrix/event"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
//CATBOX_ENDPOINT = "https://litterbox.catbox.moe/resources/internals/api.php"
|
||
|
|
//GOFILE_ENDPOINT = "https://store1.gofile.io/uploadFile"
|
||
|
|
//KAPPA_ENDPOINT = "https://kappa.lol/api/upload"
|
||
|
|
YAF_ENDPOINT = "https://i.yaf.li/upload"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (app *application) NewDownload(destination string, evt *event.Event, link string) {
|
||
|
|
uuid := uuid.NewString()
|
||
|
|
|
||
|
|
app.Log.Info().Msgf("Link: %s", link)
|
||
|
|
switch destination {
|
||
|
|
case "yaf":
|
||
|
|
app.YafDownload(evt, link, uuid)
|
||
|
|
// case "catbox":
|
||
|
|
// app.CatboxDownload(target, link, identifier, msg)
|
||
|
|
//case "kappa":
|
||
|
|
// app.KappaDownload(target, link, identifier, msg)
|
||
|
|
//case "gofile":
|
||
|
|
// app.GofileDownload(target, link, identifier, msg)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (app *application) YafDownload(evt *event.Event, link, uuid string) {
|
||
|
|
goutubedl.Path = "yt-dlp"
|
||
|
|
var rExt string
|
||
|
|
|
||
|
|
app.SendText(evt, "Downloading...")
|
||
|
|
result, err := goutubedl.New(context.Background(), link, goutubedl.Options{})
|
||
|
|
if err != nil {
|
||
|
|
app.Log.Error().Err(err).Msg("Failed to download")
|
||
|
|
app.SendText(evt, fmt.Sprintf("Something went wrong FeelsBadMan: %q", err))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// For some reason youtube links return webm as result.Info.Ext but
|
||
|
|
// are in reality mp4.
|
||
|
|
if strings.HasPrefix(link, "https://www.youtube.com/") {
|
||
|
|
rExt = "mp4"
|
||
|
|
} else {
|
||
|
|
rExt = result.Info.Ext
|
||
|
|
}
|
||
|
|
|
||
|
|
downloadResult, err := result.Download(context.TODO(), "best")
|
||
|
|
if err != nil {
|
||
|
|
app.Log.Error().Err(err).Msg("Failed to download")
|
||
|
|
app.SendText(evt, fmt.Sprintf("Something went wrong FeelsBadMan: %q", err))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer downloadResult.Close()
|
||
|
|
|
||
|
|
fileName := fmt.Sprintf("%s.%s", uuid, rExt)
|
||
|
|
f, err := os.Create(fileName)
|
||
|
|
if err != nil {
|
||
|
|
app.Log.Error().Err(err).Msg("Failed to download")
|
||
|
|
app.SendText(evt, fmt.Sprintf("Something went wrong FeelsBadMan: %q", err))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer f.Close()
|
||
|
|
io.Copy(f, downloadResult)
|
||
|
|
|
||
|
|
app.NewUpload("yaf", evt, fileName)
|
||
|
|
}
|