129 lines
3.3 KiB
Go
129 lines
3.3 KiB
Go
package api
|
|
|
|
import (
|
|
"git.solsynth.dev/hydrogen/passport/pkg/internal/database"
|
|
"git.solsynth.dev/hydrogen/passport/pkg/internal/models"
|
|
"git.solsynth.dev/hydrogen/passport/pkg/internal/server/exts"
|
|
"git.solsynth.dev/hydrogen/passport/pkg/internal/services"
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
func getNotifications(c *fiber.Ctx) error {
|
|
take := c.QueryInt("take", 0)
|
|
offset := c.QueryInt("offset", 0)
|
|
|
|
if err := exts.EnsureAuthenticated(c); err != nil {
|
|
return err
|
|
}
|
|
user := c.Locals("user").(models.Account)
|
|
|
|
tx := database.C.Where(&models.Notification{AccountID: user.ID}).Model(&models.Notification{})
|
|
|
|
var count int64
|
|
var notifications []models.Notification
|
|
if err := tx.
|
|
Count(&count).Error; err != nil {
|
|
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
|
}
|
|
|
|
if err := tx.
|
|
Limit(take).
|
|
Offset(offset).
|
|
Find(¬ifications).Error; err != nil {
|
|
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
|
}
|
|
|
|
return c.JSON(fiber.Map{
|
|
"count": count,
|
|
"data": notifications,
|
|
})
|
|
}
|
|
|
|
func markNotificationRead(c *fiber.Ctx) error {
|
|
if err := exts.EnsureAuthenticated(c); err != nil {
|
|
return err
|
|
}
|
|
user := c.Locals("user").(models.Account)
|
|
id, _ := c.ParamsInt("notificationId", 0)
|
|
|
|
if err := exts.EnsureAuthenticated(c); err != nil {
|
|
return err
|
|
}
|
|
|
|
var notify models.Notification
|
|
if err := database.C.Where(&models.Notification{
|
|
BaseModel: models.BaseModel{ID: uint(id)},
|
|
AccountID: user.ID,
|
|
}).First(¬ify).Error; err != nil {
|
|
return fiber.NewError(fiber.StatusNotFound, err.Error())
|
|
}
|
|
|
|
if err := database.C.Delete(¬ify).Error; err != nil {
|
|
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
|
} else {
|
|
return c.SendStatus(fiber.StatusOK)
|
|
}
|
|
}
|
|
|
|
func markNotificationReadBatch(c *fiber.Ctx) error {
|
|
if err := exts.EnsureAuthenticated(c); err != nil {
|
|
return err
|
|
}
|
|
user := c.Locals("user").(models.Account)
|
|
|
|
var data struct {
|
|
MessageIDs []uint `json:"messages"`
|
|
}
|
|
|
|
if err := exts.BindAndValidate(c, &data); err != nil {
|
|
return fiber.NewError(fiber.StatusBadRequest, err.Error())
|
|
}
|
|
|
|
if err := database.C.Model(&models.Notification{}).
|
|
Where("account_id = ? AND id IN ?", user.ID, data.MessageIDs).
|
|
Delete(&models.Notification{}).Error; err != nil {
|
|
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
|
} else {
|
|
return c.SendStatus(fiber.StatusOK)
|
|
}
|
|
}
|
|
|
|
func addNotifySubscriber(c *fiber.Ctx) error {
|
|
if err := exts.EnsureAuthenticated(c); err != nil {
|
|
return err
|
|
}
|
|
user := c.Locals("user").(models.Account)
|
|
|
|
var data struct {
|
|
Provider string `json:"provider" validate:"required"`
|
|
DeviceToken string `json:"device_token" validate:"required"`
|
|
DeviceID string `json:"device_id" validate:"required"`
|
|
}
|
|
|
|
if err := exts.BindAndValidate(c, &data); err != nil {
|
|
return err
|
|
}
|
|
|
|
var count int64
|
|
if err := database.C.Where(&models.NotificationSubscriber{
|
|
DeviceID: data.DeviceID,
|
|
DeviceToken: data.DeviceToken,
|
|
AccountID: user.ID,
|
|
}).Model(&models.NotificationSubscriber{}).Count(&count).Error; err != nil || count > 0 {
|
|
return c.SendStatus(fiber.StatusOK)
|
|
}
|
|
|
|
subscriber, err := services.AddNotifySubscriber(
|
|
user,
|
|
data.Provider,
|
|
data.DeviceID,
|
|
data.DeviceToken,
|
|
c.Get(fiber.HeaderUserAgent),
|
|
)
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusBadRequest, err.Error())
|
|
}
|
|
|
|
return c.JSON(subscriber)
|
|
}
|