package server import "C" import ( "code.smartsheep.studio/hydrogen/passport/pkg/database" "code.smartsheep.studio/hydrogen/passport/pkg/models" "code.smartsheep.studio/hydrogen/passport/pkg/services" "github.com/gofiber/fiber/v2" "github.com/samber/lo" "time" ) func getNotifications(c *fiber.Ctx) error { user := c.Locals("principal").(models.Account) take := c.QueryInt("take", 0) offset := c.QueryInt("offset", 0) var count int64 var notifications []models.Notification if err := database.C. Where(&models.Notification{RecipientID: user.ID}). Model(&models.Notification{}). Count(&count).Error; err != nil { return fiber.NewError(fiber.StatusInternalServerError, err.Error()) } if err := database.C. Where(&models.Notification{RecipientID: user.ID}). Limit(take). Offset(offset). Order("read_at desc"). 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 { user := c.Locals("principal").(models.Account) id, _ := c.ParamsInt("notificationId", 0) var data models.Notification if err := database.C.Where(&models.Notification{ BaseModel: models.BaseModel{ID: uint(id)}, RecipientID: user.ID, }).First(&data).Error; err != nil { return fiber.NewError(fiber.StatusNotFound, err.Error()) } data.ReadAt = lo.ToPtr(time.Now()) if err := database.C.Save(&data).Error; err != nil { return fiber.NewError(fiber.StatusInternalServerError, err.Error()) } else { return c.SendStatus(fiber.StatusOK) } } func addNotifySubscriber(c *fiber.Ctx) error { user := c.Locals("principal").(models.Account) var data struct { Provider string `json:"provider" validate:"required"` DeviceID string `json:"device_id" validate:"required"` } if err := BindAndValidate(c, &data); err != nil { return err } var count int64 if err := database.C.Where(&models.NotificationSubscriber{ DeviceID: data.DeviceID, 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, c.Get(fiber.HeaderUserAgent), ) if err != nil { return fiber.NewError(fiber.StatusBadRequest, err.Error()) } return c.JSON(subscriber) }