mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
063160aae2
* pass url parameters through context.Context * fix url param names without colon prefix * change context params to vars * replace url vars in tests using new api * rename vars to params * add some comments * rename seturlvars to seturlparams
70 lines
2.0 KiB
Go
70 lines
2.0 KiB
Go
package api
|
|
|
|
import (
|
|
"github.com/grafana/grafana/pkg/api/response"
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
macaron "gopkg.in/macaron.v1"
|
|
)
|
|
|
|
func GetOrgQuotas(c *models.ReqContext) response.Response {
|
|
if !setting.Quota.Enabled {
|
|
return response.Error(404, "Quotas not enabled", nil)
|
|
}
|
|
query := models.GetOrgQuotasQuery{OrgId: c.ParamsInt64(":orgId")}
|
|
|
|
if err := bus.Dispatch(&query); err != nil {
|
|
return response.Error(500, "Failed to get org quotas", err)
|
|
}
|
|
|
|
return response.JSON(200, query.Result)
|
|
}
|
|
|
|
func UpdateOrgQuota(c *models.ReqContext, cmd models.UpdateOrgQuotaCmd) response.Response {
|
|
if !setting.Quota.Enabled {
|
|
return response.Error(404, "Quotas not enabled", nil)
|
|
}
|
|
cmd.OrgId = c.ParamsInt64(":orgId")
|
|
cmd.Target = macaron.Params(c.Req)[":target"]
|
|
|
|
if _, ok := setting.Quota.Org.ToMap()[cmd.Target]; !ok {
|
|
return response.Error(404, "Invalid quota target", nil)
|
|
}
|
|
|
|
if err := bus.Dispatch(&cmd); err != nil {
|
|
return response.Error(500, "Failed to update org quotas", err)
|
|
}
|
|
return response.Success("Organization quota updated")
|
|
}
|
|
|
|
func GetUserQuotas(c *models.ReqContext) response.Response {
|
|
if !setting.Quota.Enabled {
|
|
return response.Error(404, "Quotas not enabled", nil)
|
|
}
|
|
query := models.GetUserQuotasQuery{UserId: c.ParamsInt64(":id")}
|
|
|
|
if err := bus.Dispatch(&query); err != nil {
|
|
return response.Error(500, "Failed to get org quotas", err)
|
|
}
|
|
|
|
return response.JSON(200, query.Result)
|
|
}
|
|
|
|
func UpdateUserQuota(c *models.ReqContext, cmd models.UpdateUserQuotaCmd) response.Response {
|
|
if !setting.Quota.Enabled {
|
|
return response.Error(404, "Quotas not enabled", nil)
|
|
}
|
|
cmd.UserId = c.ParamsInt64(":id")
|
|
cmd.Target = macaron.Params(c.Req)[":target"]
|
|
|
|
if _, ok := setting.Quota.User.ToMap()[cmd.Target]; !ok {
|
|
return response.Error(404, "Invalid quota target", nil)
|
|
}
|
|
|
|
if err := bus.Dispatch(&cmd); err != nil {
|
|
return response.Error(500, "Failed to update org quotas", err)
|
|
}
|
|
return response.Success("Organization quota updated")
|
|
}
|