From 475bed5e19ea867b143cf3a0602fa78b3a32adc9 Mon Sep 17 00:00:00 2001 From: Wim Date: Wed, 26 Oct 2016 01:01:36 +0200 Subject: [PATCH] Add support for discord channel ID. See #57 --- bridge/discord/discord.go | 30 ++++++++++++++++++++++-------- 1 file changed, 22 insertions(+), 8 deletions(-) diff --git a/bridge/discord/discord.go b/bridge/discord/discord.go index 7cf7e89f..903c80c7 100644 --- a/bridge/discord/discord.go +++ b/bridge/discord/discord.go @@ -4,16 +4,18 @@ import ( "github.com/42wim/matterbridge/bridge/config" log "github.com/Sirupsen/logrus" "github.com/bwmarrin/discordgo" + "strings" ) type bdiscord struct { - c *discordgo.Session - Config *config.Protocol - Remote chan config.Message - protocol string - origin string - Channels []*discordgo.Channel - Nick string + c *discordgo.Session + Config *config.Protocol + Remote chan config.Message + protocol string + origin string + Channels []*discordgo.Channel + Nick string + UseChannelID bool } var flog *log.Entry @@ -75,6 +77,10 @@ func (b *bdiscord) FullOrigin() string { } func (b *bdiscord) JoinChannel(channel string) error { + idcheck := strings.Split(channel, "ID:") + if len(idcheck) > 1 { + b.UseChannelID = true + } return nil } @@ -115,11 +121,19 @@ func (b *bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat return } flog.Debugf("Sending message from %s on %s to gateway", m.Author.Username, b.FullOrigin()) - b.Remote <- config.Message{Username: m.Author.Username, Text: m.Content, Channel: b.getChannelName(m.ChannelID), + channelName := b.getChannelName(m.ChannelID) + if b.UseChannelID { + channelName = "ID:" + m.ChannelID + } + b.Remote <- config.Message{Username: m.Author.Username, Text: m.Content, Channel: channelName, Origin: b.origin, Protocol: b.protocol, FullOrigin: b.FullOrigin()} } func (b *bdiscord) getChannelID(name string) string { + idcheck := strings.Split(name, "ID:") + if len(idcheck) > 1 { + return idcheck[1] + } for _, channel := range b.Channels { if channel.Name == name { return channel.ID