package databag import ( "bytes" "databag/internal/store" "encoding/json" "errors" "gorm.io/gorm" "net/http" ) var notify = make(chan *store.Notification, APPNotifyBuffer) var notifyExit = make(chan bool) //ExitNotifications stop forwarding notifications func ExitNotifications() { notifyExit <- true } //SendNotifications forward notifcations to contacts func SendNotifications() { // queue all saved notifications var notifications []store.Notification if err := store.DB.Find(¬ifications).Error; err != nil { ErrMsg(err) } for _, notification := range notifications { notify <- ¬ification } // send notifications until exit for { select { case notification := <-notify: node := getStrConfigValue(CNFDomain, "") if notification.Node == "" || notification.Node == node { sendLocalNotification(notification) } else { sendRemoteNotification(notification) } if err := store.DB.Delete(¬ification).Error; err != nil { ErrMsg(err) } case <-notifyExit: return } } } func sendLocalNotification(notification *store.Notification) { // pull reference account var card store.Card if err := store.DB.Preload("Account").Preload("CardSlot").Where("in_token = ?", notification.Token).First(&card).Error; err != nil { ErrMsg(err) return } if card.Account.Disabled { ErrMsg(errors.New("account is inactive")) return } if notification.Module == APPNotifyProfile { if err := NotifyProfileRevision(&card, notification.Revision); err != nil { ErrMsg(err) } } else if notification.Module == APPNotifyArticle { if err := NotifyArticleRevision(&card, notification.Revision); err != nil { ErrMsg(err) } } else if notification.Module == APPNotifyChannel { if err := NotifyChannelRevision(&card, notification.Revision); err != nil { ErrMsg(err) } } else if notification.Module == APPNotifyView { if err := NotifyViewRevision(&card, notification.Revision); err != nil { ErrMsg(err) } } else if notification.Module == APPPushNotify { SendPushEvent(card.Account, notification.Event) } else { LogMsg("unknown notification type") } } func sendRemoteNotification(notification *store.Notification) { var module string if notification.Module == APPNotifyProfile { module = "profile/revision" } else if notification.Module == APPNotifyArticle { module = "article/revision" } else if notification.Module == APPNotifyChannel { module = "channel/revision" } else if notification.Module == APPNotifyView { module = "view/revision" } else if notification.Module == APPPushNotify { module = "notification" } else { LogMsg("unknown notification type") return } if module == "notification" { body, err := json.Marshal(notification.Event) if err != nil { ErrMsg(err) return } url := "https://" + notification.Node + "/contact/" + module + "?contact=" + notification.GUID + "." + notification.Token req, err := http.NewRequest(http.MethodPost, url, bytes.NewBuffer(body)) if err != nil { ErrMsg(err) return } req.Header.Set("Content-Type", "application/json; charset=utf-8") client := &http.Client{} resp, err := client.Do(req) if err != nil { ErrMsg(err) } if resp.StatusCode != 200 { LogMsg("failed to notify contact") } } else { body, err := json.Marshal(notification.Revision) if err != nil { ErrMsg(err) return } url := "https://" + notification.Node + "/contact/" + module + "?contact=" + notification.GUID + "." + notification.Token req, err := http.NewRequest(http.MethodPut, url, bytes.NewBuffer(body)) if err != nil { ErrMsg(err) return } req.Header.Set("Content-Type", "application/json; charset=utf-8") client := &http.Client{} resp, err := client.Do(req) if err != nil { ErrMsg(err) } if resp == nil || resp.StatusCode != 200 { LogMsg("failed to notify contact") } } } //SetProfileNotification notifies all connected contacts of profile changes func SetProfileNotification(account *store.Account) { // select all connected cards var cards []store.Card if err := store.DB.Where("account_id = ? AND status = ?", account.GUID, APPCardConnected).Find(&cards).Error; err != nil { ErrMsg(err) return } // add new notification for each card err := store.DB.Transaction(func(tx *gorm.DB) error { for _, card := range cards { notification := &store.Notification{ Node: card.Node, Module: APPNotifyProfile, GUID: card.GUID, Token: card.OutToken, Revision: account.ProfileRevision, } if err := tx.Save(notification).Error; err != nil { return err } notify <- notification } return nil }) if err != nil { ErrMsg(err) } } //SetContactArticleNotification notifies all connected contacts of article changes // notify single card of article change: // for each card of groups in updated article data // for each card of group set or cleared from article (does not update data) func SetContactArticleNotification(account *store.Account, card *store.Card) { if card.Status != APPCardConnected { return } // add new notification for card notification := &store.Notification{ Node: card.Node, Module: APPNotifyArticle, GUID: card.GUID, Token: card.OutToken, Revision: account.ArticleRevision, } if res := store.DB.Save(notification).Error; res != nil { ErrMsg(res) } else { notify <- notification } } //SetContactViewNotification notifies all connected contacts of view change // notify single card of view change: // card set or cleared from a group // for each card in deleted group func SetContactViewNotification(account *store.Account, card *store.Card) { if card.Status != APPCardConnected { return } // add new notification for card notification := &store.Notification{ Node: card.Node, Module: APPNotifyView, GUID: card.GUID, Token: card.OutToken, Revision: card.ViewRevision, } if res := store.DB.Save(notification).Error; res != nil { ErrMsg(res) } else { notify <- notification } } //SetContactChannelNotification notifies all connected contacts of channel changes // notify single card of channel change: // for each card in updated channel data func SetContactChannelNotification(account *store.Account, card *store.Card) { if card.Status != APPCardConnected { return } // add new notification for card notification := &store.Notification{ Node: card.Node, Module: APPNotifyChannel, GUID: card.GUID, Token: card.OutToken, Revision: account.ChannelRevision, } if res := store.DB.Save(notification).Error; res != nil { ErrMsg(res) } else { notify <- notification } } //SetContactPushNotification notifies contacts of push event func SetContactPushNotification(card *store.Card, event string) { if card.Status != APPCardConnected { return } // add new notification for card notification := &store.Notification{ Node: card.Node, Module: APPPushNotify, GUID: card.GUID, Token: card.OutToken, Revision: 0, Event: event, }; if res := store.DB.Save(notification).Error; res != nil { ErrMsg(res) } else { notify <- notification } }