mirror of
https://github.com/grafana/grafana.git
synced 2025-02-20 11:48:34 -06:00
95 lines
2.8 KiB
Go
95 lines
2.8 KiB
Go
package apikeyimpl
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/services/apikey"
|
|
"github.com/grafana/grafana/pkg/services/quota"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
)
|
|
|
|
type Service struct {
|
|
store store
|
|
}
|
|
|
|
func ProvideService(db db.DB, cfg *setting.Cfg, quotaService quota.Service) (apikey.Service, error) {
|
|
s := &Service{
|
|
store: &sqlStore{db: db},
|
|
}
|
|
defaultLimits, err := readQuotaConfig(cfg)
|
|
if err != nil {
|
|
return s, err
|
|
}
|
|
|
|
if err := quotaService.RegisterQuotaReporter("a.NewUsageReporter{
|
|
TargetSrv: apikey.QuotaTargetSrv,
|
|
DefaultLimits: defaultLimits,
|
|
Reporter: s.Usage,
|
|
}); err != nil {
|
|
return s, err
|
|
}
|
|
|
|
return s, nil
|
|
}
|
|
|
|
func (s *Service) Usage(ctx context.Context, scopeParams *quota.ScopeParameters) (*quota.Map, error) {
|
|
return s.store.Count(ctx, scopeParams)
|
|
}
|
|
|
|
func (s *Service) GetAPIKeys(ctx context.Context, query *apikey.GetApiKeysQuery) ([]*apikey.APIKey, error) {
|
|
return s.store.GetAPIKeys(ctx, query)
|
|
}
|
|
func (s *Service) GetAllAPIKeys(ctx context.Context, orgID int64) ([]*apikey.APIKey, error) {
|
|
return s.store.GetAllAPIKeys(ctx, orgID)
|
|
}
|
|
func (s *Service) GetApiKeyById(ctx context.Context, query *apikey.GetByIDQuery) (*apikey.APIKey, error) {
|
|
return s.store.GetApiKeyById(ctx, query)
|
|
}
|
|
func (s *Service) GetApiKeyByName(ctx context.Context, query *apikey.GetByNameQuery) (*apikey.APIKey, error) {
|
|
return s.store.GetApiKeyByName(ctx, query)
|
|
}
|
|
func (s *Service) GetAPIKeyByHash(ctx context.Context, hash string) (*apikey.APIKey, error) {
|
|
return s.store.GetAPIKeyByHash(ctx, hash)
|
|
}
|
|
func (s *Service) DeleteApiKey(ctx context.Context, cmd *apikey.DeleteCommand) error {
|
|
return s.store.DeleteApiKey(ctx, cmd)
|
|
}
|
|
func (s *Service) AddAPIKey(ctx context.Context, cmd *apikey.AddCommand) (res *apikey.APIKey, err error) {
|
|
return s.store.AddAPIKey(ctx, cmd)
|
|
}
|
|
func (s *Service) UpdateAPIKeyLastUsedDate(ctx context.Context, tokenID int64) error {
|
|
return s.store.UpdateAPIKeyLastUsedDate(ctx, tokenID)
|
|
}
|
|
|
|
// IsDisabled returns true if the apikey service is disabled for the given org.
|
|
// This is the case if the org has no apikeys.
|
|
func (s *Service) IsDisabled(ctx context.Context, orgID int64) (bool, error) {
|
|
apikeys, err := s.store.CountAPIKeys(ctx, orgID)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return apikeys == 0, nil
|
|
}
|
|
|
|
func readQuotaConfig(cfg *setting.Cfg) (*quota.Map, error) {
|
|
limits := "a.Map{}
|
|
|
|
if cfg == nil {
|
|
return limits, nil
|
|
}
|
|
|
|
globalQuotaTag, err := quota.NewTag(apikey.QuotaTargetSrv, apikey.QuotaTarget, quota.GlobalScope)
|
|
if err != nil {
|
|
return limits, err
|
|
}
|
|
orgQuotaTag, err := quota.NewTag(apikey.QuotaTargetSrv, apikey.QuotaTarget, quota.OrgScope)
|
|
if err != nil {
|
|
return limits, err
|
|
}
|
|
|
|
limits.Set(globalQuotaTag, cfg.Quota.Global.ApiKey)
|
|
limits.Set(orgQuotaTag, cfg.Quota.Org.ApiKey)
|
|
return limits, nil
|
|
}
|