✨ Punishment API

pkg/internal/web/api/auth_api.go

@@ -39,6 +39,8 @@ func doAuthenticate(c *fiber.Ctx) error {
 		return fiber.NewError(fiber.StatusBadRequest, fmt.Sprintf("account was not found: %v", err.Error()))
 	} else if user.SuspendedAt != nil {
 		return fiber.NewError(fiber.StatusForbidden, "account was suspended")
+	} else if err := services.CheckLoginAbility(user); err != nil {
+		return err
 	}
 
 	ticket, err := services.NewTicket(user, c.IP(), c.Get(fiber.HeaderUserAgent))

pkg/internal/web/api/index.go

@@ -54,6 +54,16 @@ func MapControllers(app *fiber.App, baseURL string) {
 			}
 		}
 
+		punishments := api.Group("/punishments").Name("Punishments API")
+		{
+			punishments.Get("/", listUserPunishment)
+			punishments.Get("/given", listMadePunishment)
+			punishments.Get("/:id", getPunishment)
+			punishments.Post("/", createPunishment)
+			punishments.Put("/:id", editPunishment)
+			punishments.Delete("/:id", deletePunishment)
+		}
+
 		api.Get("/users", getUserInBatch)
 		api.Get("/users/lookup", lookupAccount)
 		api.Get("/users/search", searchAccount)

pkg/internal/web/api/punishments_api.go

@@ -0,0 +1,180 @@
+package api
+
+import (
+	"time"
+
+	"git.solsynth.dev/hypernet/passport/pkg/authkit/models"
+	"git.solsynth.dev/hypernet/passport/pkg/internal/database"
+	"git.solsynth.dev/hypernet/passport/pkg/internal/services"
+	"git.solsynth.dev/hypernet/passport/pkg/internal/web/exts"
+	"github.com/gofiber/fiber/v2"
+)
+
+func getPunishment(c *fiber.Ctx) error {
+	id, _ := c.ParamsInt("id")
+	data, err := services.GetPunishment(uint(id), true)
+	if err != nil {
+		return err
+	}
+	return c.JSON(data)
+}
+
+func listUserPunishment(c *fiber.Ctx) error {
+	if err := exts.EnsureAuthenticated(c); err != nil {
+		return err
+	}
+	user := c.Locals("user").(models.Account)
+
+	data, err := services.ListPunishments(user)
+	if err != nil {
+		return err
+	}
+	return c.JSON(data)
+}
+
+func listMadePunishment(c *fiber.Ctx) error {
+	if err := exts.EnsureAuthenticated(c); err != nil {
+		return err
+	}
+	moderator := c.Locals("user").(models.Account)
+
+	take := c.QueryInt("take", 0)
+	offset := c.QueryInt("offset", 0)
+
+	if c.QueryBool("all", false) {
+		if err := exts.EnsureGrantedPerm(c, "OverridePunishments", true); err != nil {
+			return err
+		}
+		count, err := services.CountAllPunishments()
+		data, err := services.ListAllPunishments(take, offset)
+		if err != nil {
+			return err
+		}
+		return c.JSON(fiber.Map{
+			"count": count,
+			"data":  data,
+		})
+	}
+
+	count, err := services.CountMadePunishments(moderator)
+	data, err := services.ListMadePunishments(moderator, take, offset)
+	if err != nil {
+		return err
+	}
+	return c.JSON(fiber.Map{
+		"count": count,
+		"data":  data,
+	})
+}
+
+func createPunishment(c *fiber.Ctx) error {
+	if err := exts.EnsureGrantedPerm(c, "CreatePunishments", true); err != nil {
+		return err
+	}
+	user := c.Locals("user").(models.Account)
+
+	var data struct {
+		Reason    string         `json:"reason" validate:"required"`
+		Type      int            `json:"type"`
+		ExpiredAt *time.Time     `json:"expired_at"`
+		PermNodes map[string]any `json:"perm_nodes"`
+		AccountID uint           `json:"account_id"`
+	}
+
+	if err := exts.BindAndValidate(c, &data); err != nil {
+		return err
+	}
+
+	var account models.Account
+	if err := database.C.Where("id = ?", data.AccountID).First(&account).Error; err != nil {
+		return fiber.NewError(fiber.StatusNotFound, err.Error())
+	}
+
+	punishment := models.Punishment{
+		Reason:    data.Reason,
+		Type:      data.Type,
+		PermNodes: data.PermNodes,
+		ExpiredAt: data.ExpiredAt,
+		Account:   account,
+		AccountID: account.ID,
+	}
+
+	if punishment, err := services.NewPunishment(punishment, user); err != nil {
+		return err
+	} else {
+		return c.JSON(punishment)
+	}
+}
+
+func editPunishment(c *fiber.Ctx) error {
+	if err := exts.EnsureAuthenticated(c); err != nil {
+		return err
+	}
+	user := c.Locals("user").(models.Account)
+
+	id, _ := c.ParamsInt("id", 0)
+
+	var data struct {
+		Reason    string         `json:"reason" validate:"required"`
+		Type      int            `json:"type"`
+		ExpiredAt *time.Time     `json:"expired_at"`
+		PermNodes map[string]any `json:"perm_nodes"`
+	}
+
+	if err := exts.BindAndValidate(c, &data); err != nil {
+		return err
+	}
+
+	var err error
+	var punishment models.Punishment
+	if c.QueryBool("override", false) {
+		if err = exts.EnsureGrantedPerm(c, "OverridePunishments", true); err != nil {
+			return err
+		}
+		punishment, err = services.GetPunishment(uint(id))
+	} else {
+		punishment, err = services.GetMadePunishment(uint(id), user)
+	}
+	if err != nil {
+		return fiber.NewError(fiber.StatusNotFound, err.Error())
+	}
+
+	punishment.Reason = data.Reason
+	punishment.Type = data.Type
+	punishment.ExpiredAt = data.ExpiredAt
+	punishment.PermNodes = data.PermNodes
+
+	if punishment, err := services.EditPunishment(punishment); err != nil {
+		return err
+	} else {
+		return c.JSON(punishment)
+	}
+}
+
+func deletePunishment(c *fiber.Ctx) error {
+	if err := exts.EnsureAuthenticated(c); err != nil {
+		return err
+	}
+	user := c.Locals("user").(models.Account)
+
+	id := c.QueryInt("id")
+
+	var err error
+	var punishment models.Punishment
+	if c.QueryBool("override", false) {
+		if err = exts.EnsureGrantedPerm(c, "OverridePunishments", true); err != nil {
+			return err
+		}
+		punishment, err = services.GetPunishment(uint(id))
+	} else {
+		punishment, err = services.GetMadePunishment(uint(id), user)
+	}
+	if err != nil {
+		return fiber.NewError(fiber.StatusNotFound, err.Error())
+	}
+
+	if err := services.DeletePunishment(punishment); err != nil {
+		return err
+	}
+	return c.SendStatus(fiber.StatusOK)
+}