summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
diff options
context:
space:
mode:
authorWim <wim@42.be>2017-01-28 22:45:32 +0100
committerWim <wim@42.be>2017-01-28 22:45:32 +0100
commit79e6c9fa6cfca4bfe8913fd891fd9b12bd5bb505 (patch)
treebbcbbcf10127a0f5d030345b4beabbc9df7713d2 /vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
parent1426ddec5f2250024c80765017b899e847bf2996 (diff)
downloadmatterbridge-msglm-79e6c9fa6cfca4bfe8913fd891fd9b12bd5bb505.tar.gz
matterbridge-msglm-79e6c9fa6cfca4bfe8913fd891fd9b12bd5bb505.tar.bz2
matterbridge-msglm-79e6c9fa6cfca4bfe8913fd891fd9b12bd5bb505.zip
Update vendor
Diffstat (limited to 'vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go')
-rw-r--r--vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go434
1 files changed, 0 insertions, 434 deletions
diff --git a/vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go b/vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
deleted file mode 100644
index ee27a65d..00000000
--- a/vendor/github.com/42wim/matterbridge-plus/bridge/bridge.go
+++ /dev/null
@@ -1,434 +0,0 @@
-package bridge
-
-import (
- "crypto/tls"
- "github.com/42wim/matterbridge-plus/matterclient"
- "github.com/42wim/matterbridge/matterhook"
- log "github.com/Sirupsen/logrus"
- "github.com/peterhellberg/giphy"
- ircm "github.com/sorcix/irc"
- "github.com/thoj/go-ircevent"
- "regexp"
- "sort"
- "strconv"
- "strings"
- "time"
-)
-
-//type Bridge struct {
-type MMhook struct {
- mh *matterhook.Client
-}
-
-type MMapi struct {
- 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
- ircIgnoreNicks []string
-}
-
-type MMMessage struct {
- Text string
- Channel string
- Username string
-}
-
-type Bridge struct {
- MMhook
- MMapi
- MMirc
- *Config
- kind string
-}
-
-type FancyLog struct {
- irc *log.Entry
- mm *log.Entry
-}
-
-var flog FancyLog
-
-const Legacy = "legacy"
-
-func initFLog() {
- flog.irc = log.WithFields(log.Fields{"module": "irc"})
- flog.mm = log.WithFields(log.Fields{"module": "mattermost"})
-}
-
-func NewBridge(name string, config *Config, kind string) *Bridge {
- initFLog()
- b := &Bridge{}
- b.Config = config
- b.kind = kind
- b.ircNick = b.Config.IRC.Nick
- b.ircMap = make(map[string]string)
- b.MMirc.names = make(map[string][]string)
- 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
- }
- }
-
- b.mh = matterhook.New(b.Config.Mattermost.URL,
- matterhook.Config{Port: b.Config.Mattermost.Port, Token: b.Config.Mattermost.Token,
- InsecureSkipVerify: b.Config.Mattermost.SkipTLSVerify,
- BindAddress: b.Config.Mattermost.BindAddress})
- } else {
- b.mmMap = make(map[string]string)
- if len(b.Config.Channel) > 0 {
- for _, val := range b.Config.Channel {
- b.ircMap[val.IRC] = val.Mattermost
- b.mmMap[val.Mattermost] = val.IRC
- }
- }
- 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.Info("Login ok")
- b.mc.JoinChannel(b.Config.Mattermost.Channel)
- if len(b.Config.Channel) > 0 {
- for _, val := range b.Config.Channel {
- b.mc.JoinChannel(val.Mattermost)
- }
- }
- 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
-}
-
-func (b *Bridge) createIRC(name string) *irc.Connection {
- i := irc.IRC(b.Config.IRC.Nick, b.Config.IRC.Nick)
- i.UseTLS = b.Config.IRC.UseTLS
- i.TLSConfig = &tls.Config{InsecureSkipVerify: b.Config.IRC.SkipTLSVerify}
- if b.Config.IRC.Password != "" {
- i.Password = b.Config.IRC.Password
- }
- 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
- 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.Infof("Joining %s as %s", val.IRCChannel, b.ircNick)
- i.Join(val.IRCChannel)
- }
- } else {
- for _, val := range b.Config.Channel {
- flog.irc.Infof("Joining %s as %s", val.IRC, b.ircNick)
- i.Join(val.IRC)
- }
- }
-}
-
-func (b *Bridge) handleIrcBotCommand(event *irc.Event) bool {
- parts := strings.Fields(event.Message())
- exp, _ := regexp.Compile("[:,]+$")
- channel := event.Arguments[0]
- command := ""
- if len(parts) == 2 {
- command = parts[1]
- }
- if exp.ReplaceAllString(parts[0], "") == b.ircNick {
- switch command {
- case "users":
- usernames := b.mc.UsernamesInChannel(b.getMMChannel(channel))
- sort.Strings(usernames)
- b.i.Privmsg(channel, "Users on Mattermost: "+strings.Join(usernames, ", "))
- default:
- b.i.Privmsg(channel, "Valid commands are: [users, help]")
- }
- return true
- }
- return false
-}
-
-func (b *Bridge) ircNickFormat(nick string) string {
- if nick == b.ircNick {
- return nick
- }
- if b.Config.Mattermost.RemoteNickFormat == nil {
- return "irc-" + nick
- }
- return strings.Replace(*b.Config.Mattermost.RemoteNickFormat, "{NICK}", nick, -1)
-}
-
-func (b *Bridge) handlePrivMsg(event *irc.Event) {
- if b.ignoreMessage(event.Nick, event.Message(), "irc") {
- return
- }
- if b.handleIrcBotCommand(event) {
- return
- }
- msg := ""
- if event.Code == "CTCP_ACTION" {
- msg = event.Nick + " "
- }
- msg += event.Message()
- b.Send(b.ircNickFormat(event.Nick), msg, b.getMMChannel(event.Arguments[0]))
-}
-
-func (b *Bridge) handleJoinPart(event *irc.Event) {
- b.Send(b.ircNick, b.ircNickFormat(event.Nick)+" "+strings.ToLower(event.Code)+"s "+event.Message(), b.getMMChannel(event.Arguments[0]))
-}
-
-func (b *Bridge) handleNotice(event *irc.Event) {
- if strings.Contains(event.Message(), "This nickname is registered") {
- b.i.Privmsg(b.Config.IRC.NickServNick, "IDENTIFY "+b.Config.IRC.NickServPassword)
- }
-}
-
-func (b *Bridge) nicksPerRow() int {
- if b.Config.Mattermost.NicksPerRow < 1 {
- return 4
- }
- return b.Config.Mattermost.NicksPerRow
-}
-
-func (b *Bridge) formatnicks(nicks []string, continued bool) string {
- switch b.Config.Mattermost.NickFormatter {
- case "table":
- return tableformatter(nicks, b.nicksPerRow(), continued)
- default:
- return plainformatter(nicks, b.nicksPerRow())
- }
-}
-
-func (b *Bridge) storeNames(event *irc.Event) {
- channel := event.Arguments[2]
- b.MMirc.names[channel] = append(
- b.MMirc.names[channel],
- strings.Split(strings.TrimSpace(event.Message()), " ")...)
-}
-
-func (b *Bridge) endNames(event *irc.Event) {
- channel := event.Arguments[1]
- sort.Strings(b.MMirc.names[channel])
- maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow()
- continued := false
- for len(b.MMirc.names[channel]) > maxNamesPerPost {
- b.Send(
- b.ircNick,
- b.formatnicks(b.MMirc.names[channel][0:maxNamesPerPost], continued),
- b.getMMChannel(channel))
- b.MMirc.names[channel] = b.MMirc.names[channel][maxNamesPerPost:]
- continued = true
- }
- b.Send(b.ircNick, b.formatnicks(b.MMirc.names[channel], continued), b.getMMChannel(channel))
- b.MMirc.names[channel] = nil
-}
-
-func (b *Bridge) handleTopicWhoTime(event *irc.Event) {
- parts := strings.Split(event.Arguments[2], "!")
- t, err := strconv.ParseInt(event.Arguments[3], 10, 64)
- if err != nil {
- flog.irc.Errorf("Invalid time stamp: %s", event.Arguments[3])
- }
- 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, 0))
-}
-
-func (b *Bridge) handleOther(event *irc.Event) {
- flog.irc.Debugf("%#v", event)
-}
-
-func (b *Bridge) Send(nick string, message string, channel string) error {
- return b.SendType(nick, message, channel, "")
-}
-
-func (b *Bridge) SendType(nick string, message string, channel string, mtype string) error {
- if b.Config.Mattermost.PrefixMessagesWithNick {
- if IsMarkup(message) {
- message = nick + "\n\n" + message
- } else {
- message = nick + " " + message
- }
- }
- if b.kind == Legacy {
- matterMessage := matterhook.OMessage{IconURL: b.Config.Mattermost.IconURL}
- matterMessage.Channel = channel
- matterMessage.UserName = nick
- matterMessage.Type = mtype
- matterMessage.Text = message
- err := b.mh.Send(matterMessage)
- if err != nil {
- flog.mm.Info(err)
- return err
- }
- return nil
- }
- flog.mm.Debug("->mattermost channel: ", channel, " ", message)
- b.mc.PostMessage(channel, message)
- return nil
-}
-
-func (b *Bridge) handleMatterHook(mchan chan *MMMessage) {
- for {
- message := b.mh.Receive()
- m := &MMMessage{}
- m.Username = message.UserName
- m.Text = message.Text
- m.Channel = message.Token
- mchan <- m
- }
-}
-
-func (b *Bridge) handleMatterClient(mchan chan *MMMessage) {
- for message := range b.mc.MessageChan {
- // do not post our own messages back to irc
- if message.Raw.Action == "posted" && b.mc.User.Username != message.Username {
- m := &MMMessage{}
- m.Username = message.Username
- m.Channel = message.Channel
- m.Text = message.Text
- flog.mm.Debugf("<-mattermost channel: %s %#v %#v", message.Channel, message.Post, message.Raw)
- mchan <- m
- }
- }
-}
-
-func (b *Bridge) handleMatter() {
- flog.mm.Infof("Choosing Mattermost connection type %s", b.kind)
- mchan := make(chan *MMMessage)
- 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 + "> "
- }
- 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)
- b.i.SendRaw("NAMES " + b.getIRCChannel(message.Channel))
- continue
- case "!gif":
- message.Text = b.giphyRandom(strings.Fields(strings.Replace(message.Text, "!gif ", "", 1)))
- b.Send(b.ircNick, message.Text, b.getIRCChannel(message.Channel))
- continue
- }
- texts := strings.Split(message.Text, "\n")
- for _, text := range texts {
- flog.mm.Debug("Sending message from " + message.Username + " to " + message.Channel)
- b.i.Privmsg(b.getIRCChannel(message.Channel), username+text)
- }
- }
-}
-
-func (b *Bridge) giphyRandom(query []string) string {
- g := giphy.DefaultClient
- if b.Config.General.GiphyAPIKey != "" {
- g.APIKey = b.Config.General.GiphyAPIKey
- }
- res, err := g.Random(query)
- if err != nil {
- return "error"
- }
- return res.Data.FixedHeightDownsampledURL
-}
-
-func (b *Bridge) getMMChannel(ircChannel string) string {
- mmchannel, ok := b.ircMap[ircChannel]
- if !ok {
- mmchannel = b.Config.Mattermost.Channel
- }
- return mmchannel
-}
-
-func (b *Bridge) getIRCChannel(channel string) string {
- if b.kind == Legacy {
- ircchannel := b.Config.IRC.Channel
- _, ok := b.Config.Token[channel]
- if ok {
- ircchannel = b.Config.Token[channel].IRCChannel
- }
- return ircchannel
- }
- ircchannel, ok := b.mmMap[channel]
- if !ok {
- ircchannel = b.Config.IRC.Channel
- }
- 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
-}