103 lines
2.9 KiB
Go
103 lines
2.9 KiB
Go
// Copyright 2024 The Forgejo Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package shared
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
quota_model "code.gitea.io/gitea/models/quota"
|
|
"code.gitea.io/gitea/routers/api/v1/utils"
|
|
"code.gitea.io/gitea/services/context"
|
|
"code.gitea.io/gitea/services/convert"
|
|
)
|
|
|
|
func GetQuota(ctx *context.APIContext, userID int64) {
|
|
used, err := quota_model.GetUsedForUser(ctx, userID)
|
|
if err != nil {
|
|
ctx.Error(http.StatusInternalServerError, "quota_model.GetUsedForUser", err)
|
|
return
|
|
}
|
|
|
|
groups, err := quota_model.GetGroupsForUser(ctx, userID)
|
|
if err != nil {
|
|
ctx.Error(http.StatusInternalServerError, "quota_model.GetGroupsForUser", err)
|
|
return
|
|
}
|
|
|
|
result := convert.ToQuotaInfo(used, groups, false)
|
|
ctx.JSON(http.StatusOK, &result)
|
|
}
|
|
|
|
func CheckQuota(ctx *context.APIContext, userID int64) {
|
|
subjectQuery := ctx.FormTrim("subject")
|
|
|
|
subject, err := quota_model.ParseLimitSubject(subjectQuery)
|
|
if err != nil {
|
|
ctx.Error(http.StatusUnprocessableEntity, "quota_model.ParseLimitSubject", err)
|
|
return
|
|
}
|
|
|
|
ok, err := quota_model.EvaluateForUser(ctx, userID, subject)
|
|
if err != nil {
|
|
ctx.Error(http.StatusInternalServerError, "quota_model.EvaluateForUser", err)
|
|
return
|
|
}
|
|
|
|
ctx.JSON(http.StatusOK, &ok)
|
|
}
|
|
|
|
func ListQuotaAttachments(ctx *context.APIContext, userID int64) {
|
|
opts := utils.GetListOptions(ctx)
|
|
count, attachments, err := quota_model.GetQuotaAttachmentsForUser(ctx, userID, opts)
|
|
if err != nil {
|
|
ctx.Error(http.StatusInternalServerError, "GetQuotaAttachmentsForUser", err)
|
|
return
|
|
}
|
|
|
|
result, err := convert.ToQuotaUsedAttachmentList(ctx, *attachments)
|
|
if err != nil {
|
|
ctx.Error(http.StatusInternalServerError, "convert.ToQuotaUsedAttachmentList", err)
|
|
}
|
|
|
|
ctx.SetLinkHeader(int(count), opts.PageSize)
|
|
ctx.SetTotalCountHeader(count)
|
|
ctx.JSON(http.StatusOK, result)
|
|
}
|
|
|
|
func ListQuotaPackages(ctx *context.APIContext, userID int64) {
|
|
opts := utils.GetListOptions(ctx)
|
|
count, packages, err := quota_model.GetQuotaPackagesForUser(ctx, userID, opts)
|
|
if err != nil {
|
|
ctx.Error(http.StatusInternalServerError, "GetQuotaPackagesForUser", err)
|
|
return
|
|
}
|
|
|
|
result, err := convert.ToQuotaUsedPackageList(ctx, *packages)
|
|
if err != nil {
|
|
ctx.Error(http.StatusInternalServerError, "convert.ToQuotaUsedPackageList", err)
|
|
}
|
|
|
|
ctx.SetLinkHeader(int(count), opts.PageSize)
|
|
ctx.SetTotalCountHeader(count)
|
|
ctx.JSON(http.StatusOK, result)
|
|
}
|
|
|
|
func ListQuotaArtifacts(ctx *context.APIContext, userID int64) {
|
|
opts := utils.GetListOptions(ctx)
|
|
count, artifacts, err := quota_model.GetQuotaArtifactsForUser(ctx, userID, opts)
|
|
if err != nil {
|
|
ctx.Error(http.StatusInternalServerError, "GetQuotaArtifactsForUser", err)
|
|
return
|
|
}
|
|
|
|
result, err := convert.ToQuotaUsedArtifactList(ctx, *artifacts)
|
|
if err != nil {
|
|
ctx.Error(http.StatusInternalServerError, "convert.ToQuotaUsedArtifactList", err)
|
|
}
|
|
|
|
ctx.SetLinkHeader(int(count), opts.PageSize)
|
|
ctx.SetTotalCountHeader(count)
|
|
ctx.JSON(http.StatusOK, result)
|
|
}
|