summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go')
-rw-r--r--vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go169
1 files changed, 81 insertions, 88 deletions
diff --git a/vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go b/vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
index ebd5456d..ee27a65d 100644
--- a/vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
+++ b/vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
@@ -21,15 +21,17 @@ type MMhook struct {
}
type MMapi struct {
- mc *matterclient.MMClient
- mmMap map[string]string
+ mc *matterclient.MMClient
+ mmMap map[string]string
+ mmIgnoreNicks []string
}
type MMirc struct {
- i *irc.Connection
- ircNick string
- ircMap map[string]string
- names map[string][]string
+ i *irc.Connection
+ ircNick string
+ ircMap map[string]string
+ names map[string][]string
+ ircIgnoreNicks []string
}
type MMMessage struct {
@@ -53,6 +55,8 @@ type FancyLog struct {
var flog FancyLog
+const Legacy = "legacy"
+
func initFLog() {
flog.irc = log.WithFields(log.Fields{"module": "irc"})
flog.mm = log.WithFields(log.Fields{"module": "mattermost"})
@@ -66,7 +70,9 @@ func NewBridge(name string, config *Config, kind string) *Bridge {
b.ircNick = b.Config.IRC.Nick
b.ircMap = make(map[string]string)
b.MMirc.names = make(map[string][]string)
- if kind == "legacy" {
+ b.ircIgnoreNicks = strings.Fields(b.Config.IRC.IgnoreNicks)
+ b.mmIgnoreNicks = strings.Fields(b.Config.Mattermost.IgnoreNicks)
+ if kind == Legacy {
if len(b.Config.Token) > 0 {
for _, val := range b.Config.Token {
b.ircMap[val.IRCChannel] = val.MMChannel
@@ -87,10 +93,14 @@ func NewBridge(name string, config *Config, kind string) *Bridge {
}
b.mc = matterclient.New(b.Config.Mattermost.Login, b.Config.Mattermost.Password,
b.Config.Mattermost.Team, b.Config.Mattermost.Server)
+ b.mc.SkipTLSVerify = b.Config.Mattermost.SkipTLSVerify
+ b.mc.NoTLS = b.Config.Mattermost.NoTLS
+ flog.mm.Infof("Trying login %s (team: %s) on %s", b.Config.Mattermost.Login, b.Config.Mattermost.Team, b.Config.Mattermost.Server)
err := b.mc.Login()
if err != nil {
- flog.mm.Fatal("can not connect", err)
+ flog.mm.Fatal("Can not connect", err)
}
+ flog.mm.Info("Login ok")
b.mc.JoinChannel(b.Config.Mattermost.Channel)
if len(b.Config.Channel) > 0 {
for _, val := range b.Config.Channel {
@@ -99,7 +109,9 @@ func NewBridge(name string, config *Config, kind string) *Bridge {
}
go b.mc.WsReceiver()
}
+ flog.irc.Info("Trying IRC connection")
b.i = b.createIRC(name)
+ flog.irc.Info("Connection succeeded")
go b.handleMatter()
return b
}
@@ -111,37 +123,51 @@ func (b *Bridge) createIRC(name string) *irc.Connection {
if b.Config.IRC.Password != "" {
i.Password = b.Config.IRC.Password
}
- i.AddCallback("*", b.handleOther)
+ i.AddCallback(ircm.RPL_WELCOME, b.handleNewConnection)
i.Connect(b.Config.IRC.Server + ":" + strconv.Itoa(b.Config.IRC.Port))
return i
}
func (b *Bridge) handleNewConnection(event *irc.Event) {
+ flog.irc.Info("Registering callbacks")
+ i := b.i
b.ircNick = event.Arguments[0]
+ i.AddCallback("PRIVMSG", b.handlePrivMsg)
+ i.AddCallback("CTCP_ACTION", b.handlePrivMsg)
+ i.AddCallback(ircm.RPL_ENDOFNAMES, b.endNames)
+ i.AddCallback(ircm.RPL_NAMREPLY, b.storeNames)
+ i.AddCallback(ircm.RPL_TOPICWHOTIME, b.handleTopicWhoTime)
+ i.AddCallback(ircm.NOTICE, b.handleNotice)
+ i.AddCallback(ircm.RPL_MYINFO, func(e *irc.Event) { flog.irc.Infof("%s: %s", e.Code, strings.Join(e.Arguments[1:], " ")) })
+ i.AddCallback("PING", func(e *irc.Event) {
+ i.SendRaw("PONG :" + e.Message())
+ flog.irc.Debugf("PING/PONG")
+ })
+ if b.Config.Mattermost.ShowJoinPart {
+ i.AddCallback("JOIN", b.handleJoinPart)
+ i.AddCallback("PART", b.handleJoinPart)
+ }
+ i.AddCallback("*", b.handleOther)
b.setupChannels()
}
func (b *Bridge) setupChannels() {
i := b.i
- flog.irc.Info("Joining ", b.Config.IRC.Channel, " as ", b.ircNick)
- i.Join(b.Config.IRC.Channel)
- if b.kind == "legacy" {
+ if b.Config.IRC.Channel != "" {
+ flog.irc.Infof("Joining %s as %s", b.Config.IRC.Channel, b.ircNick)
+ i.Join(b.Config.IRC.Channel)
+ }
+ if b.kind == Legacy {
for _, val := range b.Config.Token {
- flog.irc.Info("Joining ", val.IRCChannel, " as ", b.ircNick)
+ flog.irc.Infof("Joining %s as %s", val.IRCChannel, b.ircNick)
i.Join(val.IRCChannel)
}
} else {
for _, val := range b.Config.Channel {
- flog.irc.Info("Joining ", val.IRC, " as ", b.ircNick)
+ flog.irc.Infof("Joining %s as %s", val.IRC, b.ircNick)
i.Join(val.IRC)
}
}
- i.AddCallback("PRIVMSG", b.handlePrivMsg)
- i.AddCallback("CTCP_ACTION", b.handlePrivMsg)
- if b.Config.Mattermost.ShowJoinPart {
- i.AddCallback("JOIN", b.handleJoinPart)
- i.AddCallback("PART", b.handleJoinPart)
- }
}
func (b *Bridge) handleIrcBotCommand(event *irc.Event) bool {
@@ -177,6 +203,9 @@ func (b *Bridge) ircNickFormat(nick string) string {
}
func (b *Bridge) handlePrivMsg(event *irc.Event) {
+ if b.ignoreMessage(event.Nick, event.Message(), "irc") {
+ return
+ }
if b.handleIrcBotCommand(event) {
return
}
@@ -238,81 +267,21 @@ func (b *Bridge) endNames(event *irc.Event) {
b.MMirc.names[channel] = nil
}
-func (b *Bridge) handleTopicWhoTime(event *irc.Event) bool {
+func (b *Bridge) handleTopicWhoTime(event *irc.Event) {
parts := strings.Split(event.Arguments[2], "!")
- t_i, err := strconv.ParseInt(event.Arguments[3], 10, 64)
+ t, err := strconv.ParseInt(event.Arguments[3], 10, 64)
if err != nil {
flog.irc.Errorf("Invalid time stamp: %s", event.Arguments[3])
- return false
}
user := parts[0]
if len(parts) > 1 {
user += " [" + parts[1] + "]"
}
- flog.irc.Infof("%s: Topic set by %s [%s]", event.Code, user, time.Unix(t_i, 0))
- return true
+ flog.irc.Infof("%s: Topic set by %s [%s]", event.Code, user, time.Unix(t, 0))
}
func (b *Bridge) handleOther(event *irc.Event) {
flog.irc.Debugf("%#v", event)
- switch event.Code {
- case ircm.RPL_WELCOME:
- b.handleNewConnection(event)
- case ircm.RPL_ENDOFNAMES:
- b.endNames(event)
- case ircm.RPL_NAMREPLY:
- b.storeNames(event)
- case ircm.RPL_ISUPPORT:
- fallthrough
- case ircm.RPL_LUSEROP:
- fallthrough
- case ircm.RPL_LUSERUNKNOWN:
- fallthrough
- case ircm.RPL_LUSERCHANNELS:
- fallthrough
- case ircm.RPL_MYINFO:
- flog.irc.Infof("%s: %s", event.Code, strings.Join(event.Arguments[1:], " "))
- case ircm.RPL_YOURHOST:
- fallthrough
- case ircm.RPL_CREATED:
- fallthrough
- case ircm.RPL_STATSDLINE:
- fallthrough
- case ircm.RPL_LUSERCLIENT:
- fallthrough
- case ircm.RPL_LUSERME:
- fallthrough
- case ircm.RPL_LOCALUSERS:
- fallthrough
- case ircm.RPL_GLOBALUSERS:
- fallthrough
- case ircm.RPL_MOTD:
- flog.irc.Infof("%s: %s", event.Code, event.Message())
- // flog.irc.Info(event.Message())
- case ircm.RPL_TOPIC:
- flog.irc.Infof("%s: Topic for %s: %s", event.Code, event.Arguments[1], event.Message())
- case ircm.RPL_TOPICWHOTIME:
- if !b.handleTopicWhoTime(event) {
- break
- }
- case ircm.MODE:
- flog.irc.Infof("%s: %s %s", event.Code, event.Arguments[1], event.Arguments[0])
- case ircm.JOIN:
- fallthrough
- case ircm.PING:
- fallthrough
- case ircm.PONG:
- flog.irc.Infof("%s: %s", event.Code, event.Message())
- case ircm.RPL_ENDOFMOTD:
- case ircm.RPL_MOTDSTART:
- case ircm.ERR_NICKNAMEINUSE:
- flog.irc.Warn(event.Message())
- case ircm.NOTICE:
- b.handleNotice(event)
- default:
- flog.irc.Infof("UNKNOWN EVENT: %#v", event)
- return
- }
}
func (b *Bridge) Send(nick string, message string, channel string) error {
@@ -327,7 +296,7 @@ func (b *Bridge) SendType(nick string, message string, channel string, mtype str
message = nick + " " + message
}
}
- if b.kind == "legacy" {
+ if b.kind == Legacy {
matterMessage := matterhook.OMessage{IconURL: b.Config.Mattermost.IconURL}
matterMessage.Channel = channel
matterMessage.UserName = nick
@@ -371,24 +340,34 @@ func (b *Bridge) handleMatterClient(mchan chan *MMMessage) {
}
func (b *Bridge) handleMatter() {
+ flog.mm.Infof("Choosing Mattermost connection type %s", b.kind)
mchan := make(chan *MMMessage)
- if b.kind == "legacy" {
+ if b.kind == Legacy {
go b.handleMatterHook(mchan)
} else {
go b.handleMatterClient(mchan)
}
+ flog.mm.Info("Start listening for Mattermost messages")
for message := range mchan {
var username string
+ if b.ignoreMessage(message.Username, message.Text, "mattermost") {
+ continue
+ }
username = message.Username + ": "
if b.Config.IRC.RemoteNickFormat != "" {
username = strings.Replace(b.Config.IRC.RemoteNickFormat, "{NICK}", message.Username, -1)
} else if b.Config.IRC.UseSlackCircumfix {
username = "<" + message.Username + "> "
}
- cmd := strings.Fields(message.Text)[0]
+ cmds := strings.Fields(message.Text)
+ // empty message
+ if len(cmds) == 0 {
+ continue
+ }
+ cmd := cmds[0]
switch cmd {
case "!users":
- flog.mm.Info("received !users from ", message.Username)
+ flog.mm.Info("Received !users from ", message.Username)
b.i.SendRaw("NAMES " + b.getIRCChannel(message.Channel))
continue
case "!gif":
@@ -425,7 +404,7 @@ func (b *Bridge) getMMChannel(ircChannel string) string {
}
func (b *Bridge) getIRCChannel(channel string) string {
- if b.kind == "legacy" {
+ if b.kind == Legacy {
ircchannel := b.Config.IRC.Channel
_, ok := b.Config.Token[channel]
if ok {
@@ -439,3 +418,17 @@ func (b *Bridge) getIRCChannel(channel string) string {
}
return ircchannel
}
+
+func (b *Bridge) ignoreMessage(nick string, message string, protocol string) bool {
+ var ignoreNicks = b.mmIgnoreNicks
+ if protocol == "irc" {
+ ignoreNicks = b.ircIgnoreNicks
+ }
+ // should we discard messages ?
+ for _, entry := range ignoreNicks {
+ if nick == entry {
+ return true
+ }
+ }
+ return false
+}