From 6b18257185b1830bd2eff83fae30bdd2055f78b0 Mon Sep 17 00:00:00 2001 From: Wim Date: Wed, 23 Mar 2016 00:25:04 +0100 Subject: [PATCH] Move common code to matterbridge-plus bridge package --- config.go | 55 ----------- matterbridge.go | 250 +----------------------------------------------- 2 files changed, 2 insertions(+), 303 deletions(-) delete mode 100644 config.go diff --git a/config.go b/config.go deleted file mode 100644 index 1c16d771..00000000 --- a/config.go +++ /dev/null @@ -1,55 +0,0 @@ -package main - -import ( - log "github.com/Sirupsen/logrus" - "gopkg.in/gcfg.v1" - "io/ioutil" -) - -type Config struct { - IRC struct { - UseTLS bool - SkipTLSVerify bool - Server string - Port int - Nick string - Password string - Channel string - UseSlackCircumfix bool - NickServNick string - NickServPassword string - } - Mattermost struct { - URL string - Port int - ShowJoinPart bool - Token string - IconURL string - SkipTLSVerify bool - BindAddress string - Channel string - PrefixMessagesWithNick bool - NicksPerRow int - NickFormatter string - } - Token map[string]*struct { - IRCChannel string - MMChannel string - } - General struct { - GiphyAPIKey string - } -} - -func NewConfig(cfgfile string) *Config { - var cfg Config - content, err := ioutil.ReadFile(cfgfile) - if err != nil { - log.Fatal(err) - } - err = gcfg.ReadStringInto(&cfg, string(content)) - if err != nil { - log.Fatal("Failed to parse "+cfgfile+":", err) - } - return &cfg -} diff --git a/matterbridge.go b/matterbridge.go index 55713acf..5e0ed2d6 100644 --- a/matterbridge.go +++ b/matterbridge.go @@ -1,257 +1,11 @@ package main import ( - "crypto/tls" "flag" - "github.com/42wim/matterbridge/matterhook" + "github.com/42wim/matterbridge-plus/bridge" log "github.com/Sirupsen/logrus" - "github.com/peterhellberg/giphy" - "github.com/thoj/go-ircevent" - "strconv" - "strings" ) -type Bridge struct { - i *irc.Connection - m *matterhook.Client - cmap map[string]string - ircNick string - ircNickPass string - *Config -} - -func NewBridge(name string, config *Config) *Bridge { - b := &Bridge{} - b.Config = config - b.cmap = make(map[string]string) - b.ircNick = b.Config.IRC.Nick - if len(b.Config.Token) > 0 { - for _, val := range b.Config.Token { - b.cmap[val.IRCChannel] = val.MMChannel - } - } - b.m = 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}) - b.i = b.createIRC(name) - 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("*", b.handleOther) - i.Connect(b.Config.IRC.Server + ":" + strconv.Itoa(b.Config.IRC.Port)) - return i -} - -func (b *Bridge) handleNewConnection(event *irc.Event) { - b.ircNick = event.Arguments[0] - b.setupChannels() -} - -func (b *Bridge) setupChannels() { - i := b.i - log.Info("Joining ", b.Config.IRC.Channel, " as ", b.ircNick) - i.Join(b.Config.IRC.Channel) - for _, val := range b.Config.Token { - log.Info("Joining ", val.IRCChannel, " as ", b.ircNick) - i.Join(val.IRCChannel) - } - 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) handlePrivMsg(event *irc.Event) { - msg := "" - if event.Code == "CTCP_ACTION" { - msg = event.Nick + " " - } - msg += event.Message() - b.Send("irc-"+event.Nick, msg, b.getMMChannel(event.Arguments[0])) -} - -func (b *Bridge) handleJoinPart(event *irc.Event) { - b.Send(b.ircNick, "irc-"+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 tableformatter(nicks_s string, nicksPerRow int) string { - nicks := strings.Split(nicks_s, " ") - result := "|IRC users" - if nicksPerRow < 1 { - nicksPerRow = 4 - } - for i := 0; i < 2; i++ { - for j := 1; j <= nicksPerRow && j <= len(nicks); j++ { - if i == 0 { - result += "|" - } else { - result += ":-|" - } - } - result += "\r\n|" - } - result += nicks[0] + "|" - for i := 1; i < len(nicks); i++ { - if i%nicksPerRow == 0 { - result += "\r\n|" + nicks[i] + "|" - } else { - result += nicks[i] + "|" - } - } - return result -} - -func plainformatter(nicks string, nicksPerRow int) string { - return nicks + " currently on IRC" -} - -func (b *Bridge) formatnicks(nicks string) string { - switch b.Config.Mattermost.NickFormatter { - case "table": - return tableformatter(nicks, b.Config.Mattermost.NicksPerRow) - default: - return plainformatter(nicks, b.Config.Mattermost.NicksPerRow) - } -} - -func (b *Bridge) handleOther(event *irc.Event) { - switch event.Code { - case "001": - b.handleNewConnection(event) - case "353": - log.Debug("handleOther ", b.getMMChannel(event.Arguments[0])) - b.Send(b.ircNick, b.formatnicks(event.Message()), b.getMMChannel(event.Arguments[0])) - case "NOTICE": - b.handleNotice(event) - default: - log.Debugf("UNKNOWN EVENT: %+v", event) - return - } - log.Debugf("%+v", event) -} - -func (b *Bridge) Send(nick string, message string, channel string) error { - return b.SendType(nick, message, channel, "") -} - -func IsMarkup(message string) bool { - switch message[0] { - case '|': - fallthrough - case '#': - fallthrough - case '_': - fallthrough - case '*': - fallthrough - case '~': - fallthrough - case '-': - fallthrough - case ':': - fallthrough - case '>': - fallthrough - case '=': - return true - } - return false -} - -func (b *Bridge) SendType(nick string, message string, channel string, mtype string) error { - matterMessage := matterhook.OMessage{IconURL: b.Config.Mattermost.IconURL} - matterMessage.Channel = channel - matterMessage.UserName = nick - matterMessage.Type = mtype - if b.Config.Mattermost.PrefixMessagesWithNick { - if IsMarkup(message) { - matterMessage.Text = nick + ":\n\n" + message - } else { - matterMessage.Text = nick + ": " + message - } - } else { - matterMessage.Text = message - } - err := b.m.Send(matterMessage) - if err != nil { - log.Info(err) - return err - } - return nil -} - -func (b *Bridge) handleMatter() { - var username string - for { - message := b.m.Receive() - username = message.UserName + ": " - if b.Config.IRC.UseSlackCircumfix { - username = "<" + message.UserName + "> " - } - cmd := strings.Fields(message.Text)[0] - switch cmd { - case "!users": - log.Info("received !users from ", message.UserName) - b.i.SendRaw("NAMES " + b.getIRCChannel(message.Token)) - return - case "!gif": - message.Text = b.giphyRandom(strings.Fields(strings.Replace(message.Text, "!gif ", "", 1))) - b.Send(b.ircNick, message.Text, b.getIRCChannel(message.Token)) - return - } - texts := strings.Split(message.Text, "\n") - for _, text := range texts { - b.i.Privmsg(b.getIRCChannel(message.Token), 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.cmap[ircChannel] - if !ok { - mmchannel = b.Config.Mattermost.Channel - } - return mmchannel -} - -func (b *Bridge) getIRCChannel(token string) string { - ircchannel := b.Config.IRC.Channel - _, ok := b.Config.Token[token] - if ok { - ircchannel = b.Config.Token[token].IRCChannel - } - return ircchannel -} - func init() { log.SetFormatter(&log.TextFormatter{FullTimestamp: true}) } @@ -264,6 +18,6 @@ func main() { log.Info("enabling debug") log.SetLevel(log.DebugLevel) } - NewBridge("matterbot", NewConfig(*flagConfig)) + bridge.NewBridge("matterbot", bridge.NewConfig(*flagConfig), "legacy") select {} }