package bridge import ( "github.com/42wim/matterbridge/bridge/api" "github.com/42wim/matterbridge/bridge/config" "github.com/42wim/matterbridge/bridge/discord" "github.com/42wim/matterbridge/bridge/gitter" "github.com/42wim/matterbridge/bridge/irc" "github.com/42wim/matterbridge/bridge/mattermost" "github.com/42wim/matterbridge/bridge/rocketchat" "github.com/42wim/matterbridge/bridge/slack" "github.com/42wim/matterbridge/bridge/telegram" "github.com/42wim/matterbridge/bridge/xmpp" log "github.com/Sirupsen/logrus" "strings" ) type Bridger interface { Send(msg config.Message) error Connect() error JoinChannel(channel string) error Disconnect() error } type Bridge struct { Config config.Protocol Bridger Name string Account string Protocol string ChannelsIn map[string]config.ChannelOptions ChannelsOut map[string]config.ChannelOptions } func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Bridge { b := new(Bridge) b.ChannelsIn = make(map[string]config.ChannelOptions) b.ChannelsOut = make(map[string]config.ChannelOptions) accInfo := strings.Split(bridge.Account, ".") protocol := accInfo[0] name := accInfo[1] b.Name = name b.Protocol = protocol b.Account = bridge.Account // override config from environment config.OverrideCfgFromEnv(cfg, protocol, name) switch protocol { case "mattermost": b.Config = cfg.Mattermost[name] b.Bridger = bmattermost.New(cfg.Mattermost[name], bridge.Account, c) case "irc": b.Config = cfg.IRC[name] b.Bridger = birc.New(cfg.IRC[name], bridge.Account, c) case "gitter": b.Config = cfg.Gitter[name] b.Bridger = bgitter.New(cfg.Gitter[name], bridge.Account, c) case "slack": b.Config = cfg.Slack[name] b.Bridger = bslack.New(cfg.Slack[name], bridge.Account, c) case "xmpp": b.Config = cfg.Xmpp[name] b.Bridger = bxmpp.New(cfg.Xmpp[name], bridge.Account, c) case "discord": b.Config = cfg.Discord[name] b.Bridger = bdiscord.New(cfg.Discord[name], bridge.Account, c) case "telegram": b.Config = cfg.Telegram[name] b.Bridger = btelegram.New(cfg.Telegram[name], bridge.Account, c) case "rocketchat": b.Config = cfg.Rocketchat[name] b.Bridger = brocketchat.New(cfg.Rocketchat[name], bridge.Account, c) case "api": b.Config = cfg.Api[name] b.Bridger = api.New(cfg.Api[name], bridge.Account, c) } return b } func (b *Bridge) JoinChannels() error { exists := make(map[string]bool) b.joinChannels(b.ChannelsIn, exists) b.joinChannels(b.ChannelsOut, exists) return nil } func (b *Bridge) joinChannels(cMap map[string]config.ChannelOptions, exists map[string]bool) error { mychannel := "" for channel, info := range cMap { if !exists[channel] { mychannel = channel log.Infof("%s: joining %s", b.Account, channel) if b.Protocol == "irc" && info.Key != "" { log.Debugf("using key %s for channel %s", info.Key, channel) mychannel = mychannel + " " + info.Key } b.JoinChannel(mychannel) exists[channel] = true } } return nil }