forked from jshiffer/matterbridge
Move common code to matterbridge-plus bridge package
This commit is contained in:
Wim
parent
4b1ebaf7d5
commit
6b18257185
55
config.go
55
config.go
@ -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
|
|
||||||
}
|
|
||||||
250
matterbridge.go
250
matterbridge.go
@ -1,257 +1,11 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"crypto/tls"
|
|
||||||
"flag"
|
"flag"
|
||||||
"github.com/42wim/matterbridge/matterhook"
|
"github.com/42wim/matterbridge-plus/bridge"
|
||||||
log "github.com/Sirupsen/logrus"
|
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() {
|
func init() {
|
||||||
log.SetFormatter(&log.TextFormatter{FullTimestamp: true})
|
log.SetFormatter(&log.TextFormatter{FullTimestamp: true})
|
||||||
}
|
}
|
||||||
@ -264,6 +18,6 @@ func main() {
|
|||||||
log.Info("enabling debug")
|
log.Info("enabling debug")
|
||||||
log.SetLevel(log.DebugLevel)
|
log.SetLevel(log.DebugLevel)
|
||||||
}
|
}
|
||||||
NewBridge("matterbot", NewConfig(*flagConfig))
|
bridge.NewBridge("matterbot", bridge.NewConfig(*flagConfig), "legacy")
|
||||||
select {}
|
select {}
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user