Add song player handler

changelog.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## 4.1.0 - 2019-07-28
+### Added
+- Song player handler
+
+## 4.0.2 - 2019-07-28
+### Changed
+- Info about 7th subday
+
 ## 4.0.1 - 2019-07-22
 ### Changed
 - Twitch chat library version from v1 to v2

go.mod

@@ -5,7 +5,7 @@ require (
 	github.com/gempir/go-twitch-irc/v2 v2.2.0
 	github.com/pkg/errors v0.8.1
 	github.com/stretchr/testify v1.3.0 // indirect
-	go.uber.org/atomic v1.3.2 // indirect
+	go.uber.org/atomic v1.3.2
 	go.uber.org/dig v1.7.0 // indirect
 	go.uber.org/fx v1.9.0
 	go.uber.org/goleak v0.10.0 // indirect

modules/discord/discord.go

@@ -1,12 +1,17 @@
 package discord
 
 import (
+	"encoding/binary"
 	"fmt"
+	"io"
 	"log"
+	"os"
 
 	"galched-bot/modules/settings"
 	"galched-bot/modules/subday"
 
+	"go.uber.org/atomic"
+
 	"github.com/bwmarrin/discordgo"
 	"github.com/pkg/errors"
 )
@@ -31,6 +36,16 @@ func New(s *settings.Settings, subday *subday.Subday) (*Discord, error) {
 		processor.AddHandler(subdayHandler)
 	}
 
+	polka, err := loadSong(s.PolkaPath)
+	if err != nil {
+		return nil, errors.Wrap(err, "cannot read polka song")
+	}
+	processor.AddHandler(&polkaHandler{
+		polka:        polka,
+		voiceChannel: s.DiscordVoiceChannel,
+		lock:         atomic.NewBool(false),
+	})
+
 	log.Printf("discord: added %d message handlers", len(processor.handlers))
 	if len(processor.handlers) > 0 {
 		for i := range processor.handlers {
@@ -45,6 +60,48 @@ func New(s *settings.Settings, subday *subday.Subday) (*Discord, error) {
 	}, nil
 }
 
+func loadSong(path string) ([][]byte, error) {
+	file, err := os.Open(path)
+	if err != nil {
+		return nil, errors.Wrap(err, "error opening dca file")
+	}
+
+	var (
+		opuslen int16
+		buffer  = make([][]byte, 0)
+	)
+
+	for {
+		// Read opus frame length from dca file.
+		err = binary.Read(file, binary.LittleEndian, &opuslen)
+
+		// If this is the end of the file, just return.
+		if err == io.EOF || err == io.ErrUnexpectedEOF {
+			err := file.Close()
+			if err != nil {
+				return nil, err
+			}
+			return buffer, nil
+		}
+
+		if err != nil {
+			return nil, errors.Wrap(err, "error reading from dca file")
+		}
+
+		// Read encoded pcm from dca file.
+		InBuf := make([]byte, opuslen)
+		err = binary.Read(file, binary.LittleEndian, &InBuf)
+
+		// Should not be any end of file errors
+		if err != nil {
+			return nil, errors.Wrap(err, "error  reading from dca file")
+		}
+
+		// Append encoded pcm data to the buffer.
+		buffer = append(buffer, InBuf)
+	}
+}
+
 func LogMessage(m *discordgo.MessageCreate) {
 	log.Printf("discord: msg [%s]: %s", m.Author.Username, m.Content)
 }

modules/discord/polkahandler.go

@@ -0,0 +1,92 @@
+package discord
+
+import (
+	"log"
+	"time"
+
+	"github.com/bwmarrin/discordgo"
+	"go.uber.org/atomic"
+)
+
+type polkaHandler struct {
+	polka        [][]byte
+	voiceChannel string
+	lock         *atomic.Bool
+}
+
+func (h *polkaHandler) Signature() string {
+	return "!song"
+}
+
+func (h *polkaHandler) Description() string {
+	return "сыграть гимн галчед (только для избранных)"
+}
+
+func (h *polkaHandler) IsValid(msg string) bool {
+	return msg == "!song"
+}
+
+func (h *polkaHandler) Handle(s *discordgo.Session, m *discordgo.MessageCreate) {
+	if m.Author.Username != "YoMedved" && m.Author.Username != "Rummy_Quamox" && m.Author.Username != "Lidiya_owl" {
+		log.Printf("discord: unathorized polka message from %s", m.Author.Username)
+		return
+	}
+
+	// Find the channel that the message came from.
+	c, err := s.State.Channel(m.ChannelID)
+	if err != nil {
+		// Could not find channel.
+		return
+	}
+
+	// Find the guild for that channel.
+	g, err := s.State.Guild(c.GuildID)
+	if err != nil {
+		// Could not find guild.
+		return
+	}
+
+	// Look for the message sender in that guild's current voice states.
+	LogMessage(m)
+	if h.lock.CAS(false, true) {
+		defer h.lock.Store(false)
+		err = h.playSound(s, g.ID, h.voiceChannel)
+		if err != nil {
+			log.Println("discord: error playing sound:", err)
+		}
+		time.Sleep(10 * time.Second)
+		return
+	}
+}
+
+// playSound plays the current buffer to the provided channel.
+func (h *polkaHandler) playSound(s *discordgo.Session, guildID, channelID string) (err error) {
+
+	// Join the provided voice channel.
+	vc, err := s.ChannelVoiceJoin(guildID, channelID, false, true)
+	if err != nil {
+		return err
+	}
+
+	// Sleep for a specified amount of time before playing the sound
+	time.Sleep(250 * time.Millisecond)
+
+	// Start speaking.
+	vc.Speaking(true)
+
+	// Send the buffer data.
+	for _, buff := range h.polka {
+		vc.OpusSend <- buff
+	}
+
+	// Stop speaking
+	vc.Speaking(false)
+
+	// Sleep for a specified amount of time before ending.
+	time.Sleep(250 * time.Millisecond)
+
+	// Disconnect from the provided voice channel.
+	vc.Disconnect()
+
+	return nil
+}

modules/discord/subdayhandlers.go

@@ -153,7 +153,8 @@ func (h *SubdayHistoryHandler) Handle(s *discordgo.Session, m *discordgo.Message
 		"**22.12.18**: _True Crime: Streets of LA_ -> _Serious Sam 3_ -> _Kholat_\n" +
 		"**26.01.19**: _Disney’s Aladdin_ -> _~~Gothic~~_ -> _Scrapland_ -> _Donut County_\n" +
 		"**24.02.19**: _Tetris 99_ -> _~~Bully~~_ -> _~~GTA: Vice City~~_\n" +
-		"**02.06.19**: _Spec Ops: The Line_ -> _Escape from Tarkov_\n"
+		"**02.06.19**: _Spec Ops: The Line_ -> _Escape from Tarkov_\n" +
+		"**28.07.19**: _Crypt of the Necrodancer_ -> _My Friend Pedro_ -> _Ape Out_\n"
 	SendMessage(s, m, message)
 }
 

modules/settings/settings.go

@@ -6,7 +6,7 @@ import (
 )
 
 const (
-	version          = "4.0.1"
+	version          = "4.1.0"
 	twitchUser       = "galchedbot"
 	twitchIRCRoom    = "galched"
 	discordTokenPath = "./tokens/.discordtoken"
@@ -22,13 +22,15 @@ const (
 
 type (
 	Settings struct {
-		Version        string
+		Version             string
-		DiscordToken   string
+		DiscordToken        string
-		TwitchUser     string
+		TwitchUser          string
-		TwitchIRCRoom  string
+		TwitchIRCRoom       string
-		TwitchToken    string
+		TwitchToken         string
-		SubdayDataPath string
+		SubdayDataPath      string
-		PermittedRoles []string
+		PermittedRoles      []string
+		PolkaPath           string
+		DiscordVoiceChannel string
 	}
 )
 
@@ -43,12 +45,14 @@ func New() (*Settings, error) {
 	}
 
 	return &Settings{
-		Version:        version,
+		Version:             version,
-		DiscordToken:   string(discordToken),
+		DiscordToken:        string(discordToken),
-		TwitchToken:    string(twitchToken),
+		TwitchToken:         string(twitchToken),
-		TwitchUser:     twitchUser,
+		TwitchUser:          twitchUser,
-		TwitchIRCRoom:  twitchIRCRoom,
+		TwitchIRCRoom:       twitchIRCRoom,
-		SubdayDataPath: subdayDataPath,
+		SubdayDataPath:      subdayDataPath,
-		PermittedRoles: []string{subRole1, subRole2, galchedRole, smorcRole},
+		PolkaPath:           "songs/polka.dca",
+		DiscordVoiceChannel: "301793085522706432",
+		PermittedRoles:      []string{subRole1, subRole2, galchedRole, smorcRole},
 	}, nil
 }