package gateway import ( "fmt" "github.com/42wim/matterbridge/bridge" "github.com/42wim/matterbridge/bridge/config" "github.com/42wim/matterbridge/gateway/samechannel" // "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{Message: make(chan config.Message), Gateways: make(map[string]*Gateway), Config: cfg} 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 { flog.Infof("Parsing gateway %s", gw.Name) for _, br := range gw.Bridges { m[br.Account] = br } } for _, br := range m { flog.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) 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 { // record all the message ID's of the different bridges var msgIDs []*BrMsgID if !gw.ignoreMessage(&msg) { msg.Timestamp = time.Now() gw.modifyMessage(&msg) gw.handleFiles(&msg) for _, br := range gw.Bridges { msgIDs = append(msgIDs, gw.handleMessage(msg, br)...) } // only add the message ID if it doesn't already exists if _, ok := gw.Messages.Get(msg.ID); !ok && msg.ID != "" { gw.Messages.Add(msg.ID, msgIDs) } } } } }