3
0
mirror of https://github.com/42wim/matterbridge.git synced 2024-12-30 07:02:37 +01:00
matterbridge/bridge/bridge.go

119 lines
3.2 KiB
Go
Raw Normal View History

2016-07-11 21:23:33 +02:00
package bridge
import (
2017-02-18 23:10:22 +01:00
"github.com/42wim/matterbridge/bridge/api"
"github.com/42wim/matterbridge/bridge/config"
2016-09-19 20:53:26 +02:00
"github.com/42wim/matterbridge/bridge/discord"
2016-09-04 20:03:07 +02:00
"github.com/42wim/matterbridge/bridge/gitter"
"github.com/42wim/matterbridge/bridge/irc"
2017-02-20 00:49:27 +01:00
"github.com/42wim/matterbridge/bridge/matrix"
"github.com/42wim/matterbridge/bridge/mattermost"
2016-12-03 00:10:29 +01:00
"github.com/42wim/matterbridge/bridge/rocketchat"
2016-09-05 16:34:37 +02:00
"github.com/42wim/matterbridge/bridge/slack"
"github.com/42wim/matterbridge/bridge/sshchat"
2017-06-22 01:02:05 +02:00
"github.com/42wim/matterbridge/bridge/steam"
2016-11-15 23:15:57 +01:00
"github.com/42wim/matterbridge/bridge/telegram"
"github.com/42wim/matterbridge/bridge/xmpp"
2018-02-20 23:41:09 +01:00
log "github.com/sirupsen/logrus"
2016-07-11 21:23:33 +02:00
"strings"
)
2016-11-13 23:06:37 +01:00
type Bridger interface {
Send(msg config.Message) (string, error)
Connect() error
JoinChannel(channel config.ChannelInfo) error
Disconnect() error
2016-07-11 21:23:33 +02:00
}
2016-11-13 23:06:37 +01:00
type Bridge struct {
Config config.Protocol
Bridger
2017-03-28 23:56:58 +02:00
Name string
Account string
Protocol string
Channels map[string]config.ChannelInfo
Joined map[string]bool
2016-11-13 23:06:37 +01:00
}
var flog *log.Entry
func init() {
flog = log.WithFields(log.Fields{"prefix": "bridge"})
}
2016-11-13 23:06:37 +01:00
func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Bridge {
b := new(Bridge)
2017-03-28 23:56:58 +02:00
b.Channels = make(map[string]config.ChannelInfo)
accInfo := strings.Split(bridge.Account, ".")
protocol := accInfo[0]
name := accInfo[1]
2016-11-13 23:06:37 +01:00
b.Name = name
b.Protocol = protocol
b.Account = bridge.Account
b.Joined = make(map[string]bool)
bridgeConfig := &config.BridgeConfig{General: &cfg.General, Account: bridge.Account, Remote: c}
2016-11-13 23:06:37 +01:00
// override config from environment
config.OverrideCfgFromEnv(cfg, protocol, name)
switch protocol {
case "mattermost":
bridgeConfig.Config = cfg.Mattermost[name]
b.Bridger = bmattermost.New(bridgeConfig)
2016-08-16 00:08:38 +02:00
case "irc":
bridgeConfig.Config = cfg.IRC[name]
b.Bridger = birc.New(bridgeConfig)
2016-09-04 20:03:07 +02:00
case "gitter":
bridgeConfig.Config = cfg.Gitter[name]
b.Bridger = bgitter.New(bridgeConfig)
2016-09-05 16:34:37 +02:00
case "slack":
bridgeConfig.Config = cfg.Slack[name]
b.Bridger = bslack.New(bridgeConfig)
case "xmpp":
bridgeConfig.Config = cfg.Xmpp[name]
b.Bridger = bxmpp.New(bridgeConfig)
2016-09-19 20:53:26 +02:00
case "discord":
bridgeConfig.Config = cfg.Discord[name]
b.Bridger = bdiscord.New(bridgeConfig)
2016-11-15 23:15:57 +01:00
case "telegram":
bridgeConfig.Config = cfg.Telegram[name]
b.Bridger = btelegram.New(bridgeConfig)
2016-12-03 00:10:29 +01:00
case "rocketchat":
bridgeConfig.Config = cfg.Rocketchat[name]
b.Bridger = brocketchat.New(bridgeConfig)
2017-02-20 00:49:27 +01:00
case "matrix":
bridgeConfig.Config = cfg.Matrix[name]
b.Bridger = bmatrix.New(bridgeConfig)
2017-06-22 01:02:05 +02:00
case "steam":
bridgeConfig.Config = cfg.Steam[name]
b.Bridger = bsteam.New(bridgeConfig)
case "sshchat":
bridgeConfig.Config = cfg.Sshchat[name]
b.Bridger = bsshchat.New(bridgeConfig)
2017-02-18 23:10:22 +01:00
case "api":
bridgeConfig.Config = cfg.Api[name]
b.Bridger = api.New(bridgeConfig)
2016-08-16 00:08:38 +02:00
}
2017-12-26 19:13:27 +01:00
b.Config = bridgeConfig.Config
2016-11-13 23:06:37 +01:00
return b
2016-08-16 00:08:38 +02:00
}
func (b *Bridge) JoinChannels() error {
err := b.joinChannels(b.Channels, b.Joined)
2017-07-14 00:35:01 +02:00
return err
}
2017-03-28 23:56:58 +02:00
func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error {
for ID, channel := range channels {
if !exists[ID] {
flog.Infof("%s: joining %s (ID: %s)", b.Account, channel.Name, ID)
err := b.JoinChannel(channel)
if err != nil {
return err
}
2017-03-28 23:56:58 +02:00
exists[ID] = true
}
}
return nil
}