mumble-discord-bridge/handlers.go

package main

import (
	"fmt"
	"log"
	"strings"
	"time"

	"github.com/bwmarrin/discordgo"
)

func ready(s *discordgo.Session, event *discordgo.Ready) {
	log.Println("READY event registered")
}

func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {

	// Ignore all messages created by the bot itself
	// This isn't required in this specific example but it's a good practice.
	log.Println("Checking message")
	if m.Author.ID == s.State.User.ID {
		return
	}
	prefix := "!" + BridgeConf.Command
	if strings.HasPrefix(m.Content, prefix+" link") {

		// 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.
		for _, vs := range g.VoiceStates {
			if vs.UserID == m.Author.ID {
				log.Printf("Trying to join GID %v and VID %v\n", g.ID, vs.ChannelID)
				die := make(chan bool)
				Bridge.ActiveConn = die
				go startBridge(s, g.ID, vs.ChannelID, BridgeConf.Config, BridgeConf.MumbleAddr, BridgeConf.MumbleInsecure, die)
				return
			}
		}
	}

	if strings.HasPrefix(m.Content, prefix+" unlink") {

		// 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.
		for _, vs := range g.VoiceStates {
			if vs.UserID == m.Author.ID {
				log.Printf("Trying to leave GID %v and VID %v\n", g.ID, vs.ChannelID)
				Bridge.ActiveConn <- true
				Bridge.ActiveConn = nil
				MumbleReset()
				DiscordReset()
				return
			}
		}
	}

	if strings.HasPrefix(m.Content, prefix+" refresh") {

		// 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.
		for _, vs := range g.VoiceStates {
			if vs.UserID == m.Author.ID {
				log.Printf("Trying to refresh GID %v and VID %v\n", g.ID, vs.ChannelID)
				Bridge.ActiveConn <- true
				MumbleReset()
				DiscordReset()
				time.Sleep(5 * time.Second)
				Bridge.ActiveConn = make(chan bool)
				go startBridge(s, g.ID, vs.ChannelID, BridgeConf.Config, BridgeConf.MumbleAddr, BridgeConf.MumbleInsecure, Bridge.ActiveConn)
				return
			}
		}
	}

	if strings.HasPrefix(m.Content, prefix+" auto") {
		if BridgeConf.Auto == false {
			BridgeConf.Auto = true
			Bridge.AutoChan = make(chan bool)
			go AutoBridge(s)
		} else {
			Bridge.AutoChan <- true
			BridgeConf.Auto = false
		}
	}
}

func guildCreate(s *discordgo.Session, event *discordgo.GuildCreate) {

	if event.Guild.Unavailable {
		return
	}

	for _, channel := range event.Guild.Channels {
		if channel.ID == event.Guild.ID {
			log.Println("Mumble-Discord bridge is active in new guild")
			return
		}
	}
}

func voiceUpdate(s *discordgo.Session, event *discordgo.VoiceStateUpdate) {
	if event.GuildID == BridgeConf.GID {
		if event.ChannelID == BridgeConf.CID {
			//check to see if this is actually a new user
			g, err := s.State.Guild(event.GuildID)
			if err != nil {
				log.Println("Could not find guild while checking VoiceStateUpdate")
				return
			}
			for _, vs := range g.VoiceStates {
				if vs.UserID == event.UserID {
					//user is already in channel (and is probably just muting/etc), ignore
					return
				}
			}
			log.Println("user joined watched discord channel")
			u, err := s.User(event.UserID)
			if err != nil {
				log.Printf("error looking up user for uid %v", event.UserID)
			} else if Bridge.Connected {
				Bridge.CurrentChannel.Send(fmt.Sprintf("%v has joined Discord channel\n", u.Username), false)
			}
			Bridge.DiscordUserCount = Bridge.DiscordUserCount + 1
		}
		if event.ChannelID == "" {
			//leave event, trigger recount of active users
			//TODO when next version of discordgo comes out, switch to PreviousState
			g, err := s.State.Guild(event.GuildID)
			if err != nil {
				// Could not find guild.
				return
			}

			// Look for current voice states in watched channel
			count := 0
			for _, vs := range g.VoiceStates {
				if vs.ChannelID == BridgeConf.CID {
					count = count + 1
				}
			}
			if Bridge.DiscordUserCount > count {
				log.Println("user left watched discord channel")
				Bridge.DiscordUserCount = count
			}
		}

	}
	return
}