From 99c6bd379a1ea465bcd8c769190e83a5b8d2398f Mon Sep 17 00:00:00 2001 From: William P Date: Thu, 22 Jan 2026 23:12:19 -0500 Subject: [PATCH] start discord-side of things --- app/go.mod | 2 ++ app/go.sum | 10 ++++++ app/main.go | 92 ++++++++++++++++++++++++++++++++++++++++++++++++++--- 3 files changed, 99 insertions(+), 5 deletions(-) diff --git a/app/go.mod b/app/go.mod index 3c59237..5fbb7f8 100644 --- a/app/go.mod +++ b/app/go.mod @@ -4,7 +4,9 @@ go 1.25.2 require ( github.com/ProtonMail/go-crypto v1.3.0 // indirect + github.com/bwmarrin/discordgo v0.29.0 // indirect github.com/cloudflare/circl v1.6.1 // indirect + github.com/gorilla/websocket v1.4.2 // indirect github.com/lrstanley/go-ytdlp v1.2.7 // indirect github.com/ulikunitz/xz v0.5.13 // indirect golang.org/x/crypto v0.41.0 // indirect diff --git a/app/go.sum b/app/go.sum index 7f2816e..8bfee9d 100644 --- a/app/go.sum +++ b/app/go.sum @@ -1,12 +1,22 @@ github.com/ProtonMail/go-crypto v1.3.0 h1:ILq8+Sf5If5DCpHQp4PbZdS1J7HDFRXz/+xKBiRGFrw= github.com/ProtonMail/go-crypto v1.3.0/go.mod h1:9whxjD8Rbs29b4XWbB8irEcE8KHMqaR2e7GWU1R+/PE= +github.com/bwmarrin/discordgo v0.29.0 h1:FmWeXFaKUwrcL3Cx65c20bTRW+vOb6k8AnaP+EgjDno= +github.com/bwmarrin/discordgo v0.29.0/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY= github.com/cloudflare/circl v1.6.1 h1:zqIqSPIndyBh1bjLVVDHMPpVKqp8Su/V+6MeDzzQBQ0= github.com/cloudflare/circl v1.6.1/go.mod h1:uddAzsPgqdMAYatqJ0lsjX1oECcQLIlRpzZh3pJrofs= +github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc= +github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= github.com/lrstanley/go-ytdlp v1.2.7 h1:YNDvKkd0OCJSZLZePZvJwcirBCfL8Yw3eCwrTCE5w7Q= github.com/lrstanley/go-ytdlp v1.2.7/go.mod h1:38IL64XM6gULrWtKTiR0+TTNCVbxesNSbTyaFG2CGTI= github.com/ulikunitz/xz v0.5.13 h1:ar98gWrjf4H1ev05fYP/o29PDZw9DrI3niHtnEqyuXA= github.com/ulikunitz/xz v0.5.13/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14= +golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4= golang.org/x/crypto v0.41.0 h1:WKYxWedPGCTVVl5+WHSSrOBT0O8lx32+zxmHxijgXp4= golang.org/x/crypto v0.41.0/go.mod h1:pO5AFd7FA68rFak7rOAGVuygIISepHftHnr8dr6+sUc= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.35.0 h1:vz1N37gP5bs89s7He8XuIYXpyY0+QlsKmzipCbUtyxI= golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= diff --git a/app/main.go b/app/main.go index 54f8724..02c05cb 100644 --- a/app/main.go +++ b/app/main.go @@ -1,18 +1,100 @@ package main import ( - "fmt" + "log" "os" + "os/signal" + "syscall" + + "github.com/bwmarrin/discordgo" ) func main() { - out_dir := os.Getenv("OUT_DIR") - var url string = "https://www.youtube.com/watch?v=e_isb3YDUDA" + out_dir := os.Getenv("OUT_PATH") + bot_token := os.Getenv("DISCORD_TOKEN") if out_dir == "" { panic("No output dir specified") } - downloadVideo(out_dir, url) -} \ No newline at end of file + s, err := discordgo.New("Bot " + bot_token) + if err != nil { + log.Fatalf("Invalid bot parameters: %v", err) + } + + err = s.Open() + if err != nil { + log.Fatalf("Error opening connection: %v", err) + } + + var defaultMemberPermissions int64 = discordgo.PermissionAllText + var interactionPrivateChannel = discordgo.InteractionContextPrivateChannel + + var commands = []*discordgo.ApplicationCommand{ + { + Name: "download", + Description: "Download video and save it to 'youtube-vids", + DefaultMemberPermissions: &defaultMemberPermissions, + Contexts: &[]discordgo.InteractionContextType{interactionPrivateChannel}, + Options: []*discordgo.ApplicationCommandOption{ + { + Name: "url", + Description: "URL", + Type: discordgo.ApplicationCommandOptionString, + Required: true, + }, + }, + }, + } + + var commandHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){ + "download": func(s *discordgo.Session, i *discordgo.InteractionCreate) { + options := i.ApplicationCommandData().Options + optionMap := make(map[string]*discordgo.ApplicationCommandInteractionDataOption, len(options)) + for _, opt := range options { + optionMap[opt.Name] = opt + } + response := "" + if option, ok := optionMap["url"]; ok { + response = "It works! Your URL is: " + option.StringValue() + } else { + response = "It works!" + } + s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{ + Type: discordgo.InteractionResponseChannelMessageWithSource, + Data: &discordgo.InteractionResponseData{ + Content: response, + Flags: discordgo.MessageFlagsEphemeral, + }, + }) + }, + } + + s.AddHandler(func(s *discordgo.Session, i *discordgo.InteractionCreate) { + if h, ok := commandHandlers[i.ApplicationCommandData().Name]; ok { + h(s, i) + } + }) + + log.Println("Adding commands") + registeredCommands := make([]*discordgo.ApplicationCommand, len(commands)) + for i, v := range commands { + cmd, err := s.ApplicationCommandCreate(s.State.User.ID, "", v) + if err != nil { + log.Panicf("Cannot create '%v' command: %v", v.Name, err) + } + registeredCommands[i] = cmd + } + + log.Println("Bot is now running. Press CTRL+C to exit") + sc := make(chan os.Signal, 1) + signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt) + <-sc + + s.Close() + + //var url string = "https://www.youtube.com/watch?v=WpBWSFF03eI" + + //downloadVideo(out_dir, url, DownloadOptions{EmbedThumbnail: true, IncludeSubtitles: true}) +}