databag/net/server/internal/api_setChannelNotification.go

package databag

import (
  "databag/internal/store"
  "errors"
  "github.com/gorilla/mux"
  "net/http"
)

//SetChannelNotifications enables or disables notifcation
func SetChannelNotification(w http.ResponseWriter, r *http.Request) {

  // get referenced channel
  params := mux.Vars(r)
  channelID := params["channelID"]

  // get enabled state
  var flag bool
  if err := ParseRequest(r, w, &flag); err != nil {
    ErrResponse(w, http.StatusBadRequest, err)
    return
  }

  tokenType := ParamTokenType(r)
  if tokenType == APPTokenAgent {
    account, code, err := ParamAgentToken(r, false)
    if err != nil {
      ErrResponse(w, code, err)
      return
    }

    // update host notification status
    if err = store.DB.Model(&store.Channel{}).Where("account_id = ? AND id = ?", account.ID, channelID).Update("host_push", flag).Error; err != nil {
      ErrResponse(w, http.StatusInternalServerError, err)
      return
    }
  } else if tokenType == APPTokenContact {
    card, code, err := ParamContactToken(r, true)
    if err != nil {
      ErrResponse(w, code, err)
      return
    }

    // update member notification status
    if err := store.DB.Model(&store.Member{}).Where("channel_id = ? AND card_id = ?", channelID, card.ID).Update("push_enabled", flag).Error; err != nil {
      ErrResponse(w, http.StatusInternalServerError, err)
    }
  } else {
    ErrResponse(w, http.StatusUnauthorized, errors.New("invalid access token"))
    return
  }
}