117 lines
2.3 KiB
Go
117 lines
2.3 KiB
Go
package arn
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/aerogo/aero"
|
|
"github.com/aerogo/api"
|
|
jsoniter "github.com/json-iterator/go"
|
|
)
|
|
|
|
// Force interface implementations
|
|
var (
|
|
_ api.Editable = (*PushSubscriptions)(nil)
|
|
_ api.Filter = (*PushSubscriptions)(nil)
|
|
)
|
|
|
|
// Actions
|
|
func init() {
|
|
API.RegisterActions("PushSubscriptions", []*api.Action{
|
|
// Add subscription
|
|
{
|
|
Name: "add",
|
|
Route: "/add",
|
|
Run: func(obj interface{}, ctx aero.Context) error {
|
|
subscriptions := obj.(*PushSubscriptions)
|
|
|
|
// Parse body
|
|
body, err := ctx.Request().Body().Bytes()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var subscription *PushSubscription
|
|
err = jsoniter.Unmarshal(body, &subscription)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Add subscription
|
|
err = subscriptions.Add(subscription)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
subscriptions.Save()
|
|
|
|
return nil
|
|
},
|
|
},
|
|
|
|
// Remove subscription
|
|
{
|
|
Name: "remove",
|
|
Route: "/remove",
|
|
Run: func(obj interface{}, ctx aero.Context) error {
|
|
subscriptions := obj.(*PushSubscriptions)
|
|
|
|
// Parse body
|
|
body, err := ctx.Request().Body().Bytes()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var subscription *PushSubscription
|
|
err = jsoniter.Unmarshal(body, &subscription)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Remove subscription
|
|
if !subscriptions.Remove(subscription.ID()) {
|
|
return errors.New("PushSubscription does not exist")
|
|
}
|
|
|
|
subscriptions.Save()
|
|
|
|
return nil
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
// Filter removes privacy critical fields from the settings object.
|
|
func (list *PushSubscriptions) Filter() {
|
|
for _, item := range list.Items {
|
|
item.P256DH = ""
|
|
item.Auth = ""
|
|
item.Endpoint = ""
|
|
}
|
|
}
|
|
|
|
// ShouldFilter tells whether data needs to be filtered in the given context.
|
|
func (list *PushSubscriptions) ShouldFilter(ctx aero.Context) bool {
|
|
ctxUser := GetUserFromContext(ctx)
|
|
|
|
if ctxUser != nil && ctxUser.Role == "admin" {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// Authorize returns an error if the given API request is not authorized.
|
|
func (list *PushSubscriptions) Authorize(ctx aero.Context, action string) error {
|
|
return AuthorizeIfLoggedInAndOwnData(ctx, "id")
|
|
}
|
|
|
|
// Save saves the push subscriptions in the database.
|
|
func (list *PushSubscriptions) Save() {
|
|
DB.Set("PushSubscriptions", list.UserID, list)
|
|
}
|