HyperNet/Paperclip
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

✨ Pay for upload

Browse Source
This commit is contained in:
LittleSheep
2025-01-29 19:12:54 +08:00
parent 96f429e1d9
commit 80739eab52
5 changed files with 100 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/internal/server/api/up_direct_api.go
View File

@ -32,8 +32,8 @@ func createAttachmentDirectly(c *fiber.Ctx) error {
return err
}
if !user.HasPermNode("CreateAttachments", file.Size) {
return fiber.NewError(fiber.StatusForbidden, "you are not permitted to create attachments like this large")
if !user.HasPermNode("CreateAttachments", true) {
return fiber.NewError(fiber.StatusForbidden, "you are not permitted to create attachments")
} else if pool.Config.Data().MaxFileSize != nil && file.Size > *pool.Config.Data().MaxFileSize {
return fiber.NewError(fiber.StatusBadRequest, fmt.Sprintf("attachment pool %s doesn't allow file larger than %d", pool.Alias, *pool.Config.Data().MaxFileSize))
}
@ -62,6 +62,13 @@ func createAttachmentDirectly(c *fiber.Ctx) error {
return fiber.NewError(fiber.StatusBadRequest, err.Error())
}
// If pool has no belongs to, it means it is shared pool, apply shared attachment discount
withDiscount := pool.AccountID == nil
if err := services.PlaceOrder(user.ID, file.Size, withDiscount); err != nil {
tx.Rollback()
return fiber.NewError(fiber.StatusPaymentRequired, err.Error())
}
tx.Commit()
metadata.Pool = &pool