package gateway import ( "fmt" "github.com/42wim/matterbridge/bridge" "github.com/42wim/matterbridge/bridge/config" "github.com/42wim/matterbridge/gateway/samechannel" log "github.com/Sirupsen/logrus" // "github.com/davecgh/go-spew/spew" "time" ) type Router struct { Gateways map[string]*Gateway Message chan config.Message *config.Config } func NewRouter(cfg *config.Config) (*Router, error) { r := &Router{} r.Config = cfg r.Message = make(chan config.Message) r.Gateways = make(map[string]*Gateway) sgw := samechannelgateway.New(cfg) gwconfigs := sgw.GetConfig() for _, entry := range append(gwconfigs, cfg.Gateway...) { if !entry.Enable { continue } if entry.Name == "" { return nil, fmt.Errorf("%s", "Gateway without name found") } if _, ok := r.Gateways[entry.Name]; ok { return nil, fmt.Errorf("Gateway with name %s already exists", entry.Name) } r.Gateways[entry.Name] = New(entry, r) } return r, nil } func (r *Router) Start() error { m := make(map[string]*bridge.Bridge) for _, gw := range r.Gateways { for _, br := range gw.Bridges { m[br.Account] = br } } for _, br := range m { log.Infof("Starting bridge: %s ", br.Account) err := br.Connect() if err != nil { return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) } err = br.JoinChannels() if err != nil { return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) } } go r.handleReceive() return nil } func (r *Router) getBridge(account string) *bridge.Bridge { for _, gw := range r.Gateways { if br, ok := gw.Bridges[account]; ok { return br } } return nil } func (r *Router) getGatewayName(channelID string) string { for _, gw := range r.Gateways { if _, ok := gw.Channels[channelID]; ok { return gw.Name } } return "" } func (r *Router) handleReceive() { for msg := range r.Message { if msg.Event == config.EVENT_FAILURE { Loop: for _, gw := range r.Gateways { for _, br := range gw.Bridges { if msg.Account == br.Account { go gw.reconnectBridge(br) break Loop } } } } if msg.Event == config.EVENT_REJOIN_CHANNELS { for _, gw := range r.Gateways { for _, br := range gw.Bridges { if msg.Account == br.Account { br.Joined = make(map[string]bool) br.JoinChannels() } } } } for _, gw := range r.Gateways { if gw.Name == r.getGatewayName(getChannelID(msg)) { if !gw.ignoreMessage(&msg) { msg.Timestamp = time.Now() gw.modifyMessage(&msg) for _, br := range gw.Bridges { gw.handleMessage(msg, br) } } } } } }