small pixel drawing of a pufferfish neoarkbot

main.go

package main

import (
	"flag"
	"io"
	"fmt"
	"os"
	"path/filepath"
	"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) {
			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
			}
			if e.Message == "stop" {
				fmt.Println("stopping")
				stream.Stop()
				return
			}
			if strings.HasPrefix(e.Message, "<a href=\"https://youtube") == true {
				fmt.Println("downloading...")
                videoID := "BaW_jenozKc"
                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()

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

				_, err = io.Copy(file, resp.Body)
				if err != nil {
                    fmt.Println(err)
					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)
			}
			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)
			}
		},
	})
}