✨ Notifications

pkg/server/accounts_api.go

@@ -20,7 +20,6 @@ func getUserinfo(c *fiber.Ctx) error {
 		Where(&models.Account{BaseModel: models.BaseModel{ID: user.ID}}).
 		Preload("Profile").
 		Preload("Contacts").
-		Preload("Factors").
 		Preload("Notifications", "read_at IS NULL").
 		First(&data).Error; err != nil {
 		return fiber.NewError(fiber.StatusInternalServerError, err.Error())

pkg/server/notifications_api.go

@@ -0,0 +1,59 @@
+package server
+
+import (
+	"code.smartsheep.studio/hydrogen/passport/pkg/database"
+	"code.smartsheep.studio/hydrogen/passport/pkg/models"
+	"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(&notifications).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)
+	}
+}

pkg/server/startup.go

@@ -60,6 +60,9 @@ func NewServer() {
 		api.Get("/avatar/:avatarId", getAvatar)
 		api.Put("/avatar", auth, setAvatar)
 
+		api.Get("/notifications", auth, getNotifications)
+		api.Put("/notifications/:notificationId/read", auth, markNotificationRead)
+
 		api.Get("/users/me", auth, getUserinfo)
 		api.Put("/users/me", auth, editUserinfo)
 		api.Get("/users/me/events", auth, getEvents)