2025-04-01 10:38:02 +09:00

27 lines
705 B
Go

package clients
import (
"github.com/grafana/grafana/pkg/services/org"
"github.com/grafana/grafana/pkg/setting"
)
// roleExtractor should return the org role, optional isGrafanaAdmin or an error
type roleExtractor func() (org.RoleType, *bool, error)
// getRoles only handles one org role for now, could be subject to change
func getRoles(cfg *setting.Cfg, extract roleExtractor) (map[int64]org.RoleType, *bool, error) {
role, isGrafanaAdmin, err := extract()
orgRoles := make(map[int64]org.RoleType, 0)
if err != nil {
return orgRoles, nil, err
}
if role == "" || !role.IsValid() {
return orgRoles, nil, nil
}
orgRoles[cfg.DefaultOrgID()] = role
return orgRoles, isGrafanaAdmin, nil
}