small pixel drawing of a pufferfish neoarkbot

main.go

package main

import (
	"flag"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"regexp"
	"sort"
	"strings"

	"github.com/kkdai/youtube/v2"
	"layeh.com/gumble/gumble"
	"layeh.com/gumble/gumbleffmpeg"
	"layeh.com/gumble/gumbleutil"
	_ "layeh.com/gumble/opus"
)

func main() {
	files := make(map[string]string)
	var stream *gumbleffmpeg.Stream

	flag.Usage = func() {
		fmt.Fprintf(os.Stderr, "Usage of %s: [flags] [audio files...]\n", os.Args[0])
		flag.PrintDefaults()
	}

	gumbleutil.Main(gumbleutil.AutoBitrate, gumbleutil.Listener{
		Connect: func(e *gumble.ConnectEvent) {
			// hardcode access token cuz idgaf rn
			e.Client.Config.Tokens = []string{"welcome"}
			e.Client.Send(e.Client.Config.Tokens)

			for _, file := range flag.Args() {
				key := filepath.Base(file)
				key = strings.TrimSuffix(key, filepath.Ext(key))
				files[key] = file
			}

			fmt.Printf("audio player loaded! (%d files)\n", len(files))
		},

		TextMessage: func(e *gumble.TextMessageEvent) {
			if e.Sender == nil {
				return
			}
			// old commands started with #
			if strings.HasPrefix(e.Message, "#") {
				fmt.Println("old detected")
				// shittily trim the string
				e.Message = e.Message[1:]
			}
			if e.Message == "stop" {
				fmt.Println("stopping")
				stream.Stop()
				return
			}
			if e.Message == "list" {
				// send list to sender
				fmt.Println("listing")
				var keys []string
				for k, _ := range files {
					if strings.HasPrefix(k, "s-") {
						continue
					}
					keys = append(keys, k)
				}
				sort.Strings(keys)
				list := strings.Join(keys, "\n")
				e.Sender.Send(list)
			}
			var file string
			matched, err := regexp.MatchString(`https:\/\/(www.)?youtube\.com\/watch\?v=[A-z_0-9]{11}`, e.Message)
			if err != nil {
				fmt.Printf("regex match blew up: %s", err)
				return
			}
			if matched {
				m := regexp.MustCompile(`https:\/\/(www.)?youtube\.com\/watch\?v=[A-z_0-9]{11}`)
				videoUrl := m.FindString(e.Message)
				fmt.Println("downloading...")
				parts := strings.Split(videoUrl, "?v=")
				fmt.Printf("%v", parts)
				if len(parts) != 2 {
					fmt.Println("youtube url is weird")
					return
				}
				fmt.Println("trash")
				fmt.Println(parts[1])
				//                videoID := parts[1]
				videoID := parts[1]
				client := youtube.Client{}
				video, err := client.GetVideo(videoID)
				if err != nil {
					fmt.Println(err)
					return
				}

				resp, err := client.GetStream(video, &video.Formats[0])
				if err != nil {
					fmt.Println(err)
					return
				}
				defer resp.Body.Close()

				wfile, err := os.Create("video.mp4")
				if err != nil {
					fmt.Println(err)
					return
				}
				defer wfile.Close()

				_, err = io.Copy(wfile, resp.Body)
				if err != nil {
					fmt.Println(err)
					return
				}
				file = "video.mp4"
			} else {
				var ok bool
				file, ok = files[e.Message]
				if !ok {
					return
				}
			}
			if stream != nil && stream.State() == gumbleffmpeg.StatePlaying {
				fmt.Println("something else is playing already")
			}
			stream = gumbleffmpeg.New(e.Client, gumbleffmpeg.SourceFile(file))
			if err := stream.Play(); err != nil {
				fmt.Printf("%s\n", err)
			} else {
				fmt.Printf("playing %s\n", file)
			}
		},
	})
}