e28d2f5d89
Adds an EventGetChannelMembers event, which gets send every x time to all bridges. Bridges should respond on this event with a Message containing ChannelMembers in the EventGetChannelMembers key in the Extra field. handleEventGetChannelMembers will handle this Message and sets the contained ChannelMembers to the Bridge struct.
122 lines
2.6 KiB
Go
122 lines
2.6 KiB
Go
package bridge
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/42wim/matterbridge/bridge/config"
|
|
"github.com/sirupsen/logrus"
|
|
"sync"
|
|
)
|
|
|
|
type Bridger interface {
|
|
Send(msg config.Message) (string, error)
|
|
Connect() error
|
|
JoinChannel(channel config.ChannelInfo) error
|
|
Disconnect() error
|
|
}
|
|
|
|
type Bridge struct {
|
|
Bridger
|
|
Name string
|
|
Account string
|
|
Protocol string
|
|
Channels map[string]config.ChannelInfo
|
|
Joined map[string]bool
|
|
ChannelMembers *config.ChannelMembers
|
|
Log *logrus.Entry
|
|
Config config.Config
|
|
General *config.Protocol
|
|
*sync.RWMutex
|
|
}
|
|
|
|
type Config struct {
|
|
// General *config.Protocol
|
|
Remote chan config.Message
|
|
Log *logrus.Entry
|
|
*Bridge
|
|
}
|
|
|
|
// Factory is the factory function to create a bridge
|
|
type Factory func(*Config) Bridger
|
|
|
|
func New(bridge *config.Bridge) *Bridge {
|
|
b := &Bridge{
|
|
Channels: make(map[string]config.ChannelInfo),
|
|
RWMutex: new(sync.RWMutex),
|
|
Joined: make(map[string]bool),
|
|
}
|
|
accInfo := strings.Split(bridge.Account, ".")
|
|
protocol := accInfo[0]
|
|
name := accInfo[1]
|
|
b.Name = name
|
|
b.Protocol = protocol
|
|
b.Account = bridge.Account
|
|
return b
|
|
}
|
|
|
|
func (b *Bridge) JoinChannels() error {
|
|
err := b.joinChannels(b.Channels, b.Joined)
|
|
return err
|
|
}
|
|
|
|
// SetChannelMembers sets the newMembers to the bridge ChannelMembers
|
|
func (b *Bridge) SetChannelMembers(newMembers *config.ChannelMembers) {
|
|
b.Lock()
|
|
b.ChannelMembers = newMembers
|
|
b.Unlock()
|
|
}
|
|
|
|
func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error {
|
|
for ID, channel := range channels {
|
|
if !exists[ID] {
|
|
b.Log.Infof("%s: joining %s (ID: %s)", b.Account, channel.Name, ID)
|
|
err := b.JoinChannel(channel)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
exists[ID] = true
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (b *Bridge) GetBool(key string) bool {
|
|
val, ok := b.Config.GetBool(b.Account + "." + key)
|
|
if !ok {
|
|
val, _ = b.Config.GetBool("general." + key)
|
|
}
|
|
return val
|
|
}
|
|
|
|
func (b *Bridge) GetInt(key string) int {
|
|
val, ok := b.Config.GetInt(b.Account + "." + key)
|
|
if !ok {
|
|
val, _ = b.Config.GetInt("general." + key)
|
|
}
|
|
return val
|
|
}
|
|
|
|
func (b *Bridge) GetString(key string) string {
|
|
val, ok := b.Config.GetString(b.Account + "." + key)
|
|
if !ok {
|
|
val, _ = b.Config.GetString("general." + key)
|
|
}
|
|
return val
|
|
}
|
|
|
|
func (b *Bridge) GetStringSlice(key string) []string {
|
|
val, ok := b.Config.GetStringSlice(b.Account + "." + key)
|
|
if !ok {
|
|
val, _ = b.Config.GetStringSlice("general." + key)
|
|
}
|
|
return val
|
|
}
|
|
|
|
func (b *Bridge) GetStringSlice2D(key string) [][]string {
|
|
val, ok := b.Config.GetStringSlice2D(b.Account + "." + key)
|
|
if !ok {
|
|
val, _ = b.Config.GetStringSlice2D("general." + key)
|
|
}
|
|
return val
|
|
}
|