Manager now exposes a Resource Grant manifest and the Agnet control-plane response carries runtime state, agent instances, and permission_manifest so frontend runs can submit bounded resource grants without plaintext credentials. Constraint: Manager remains the user console while NewAPI stays independent and OpenBao is referenced through secret_ref only. Rejected: platform-side high-risk approval | client approval is the product boundary; Agnet only validates approval evidence. Confidence: medium Scope-risk: moderate Directive: Do not mix child Agnet runtime model selection with NewAPI billing or expose OpenBao as a public route. Tested: git diff --check; jq empty locale JSON; go vet ./controller ./model ./router; go test -count=1 ./controller ./model ./router Not-tested: frontend typecheck/build because local node_modules tooling is absent and user requested builds happen on the VM. Co-authored-by: OmX <omx@oh-my-codex.dev>
828 lines
25 KiB
Go
828 lines
25 KiB
Go
package controller
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/heicode/manager/common"
|
|
"github.com/heicode/manager/model"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var allowedResourceTypes = map[string]bool{
|
|
"git": true,
|
|
"sk": true,
|
|
"project_document": true,
|
|
"cloud_account": true,
|
|
"cloud_resource": true,
|
|
}
|
|
|
|
var allowedResourceStatuses = map[string]bool{
|
|
"active": true,
|
|
"disabled": true,
|
|
"revoked": true,
|
|
}
|
|
|
|
var secretLikeKeys = map[string]bool{
|
|
"access_key": true,
|
|
"access_key_id": true,
|
|
"api_key": true,
|
|
"client_secret": true,
|
|
"database_password": true,
|
|
"db_password": true,
|
|
"newapi_key": true,
|
|
"password": true,
|
|
"private_key": true,
|
|
"refresh_token": true,
|
|
"secret": true,
|
|
"secret_key": true,
|
|
"ssh_key": true,
|
|
"token": true,
|
|
}
|
|
|
|
type resourcePayload struct {
|
|
TenantId string `json:"tenant_id"`
|
|
ProjectId string `json:"project_id"`
|
|
BindingScope string `json:"binding_scope"`
|
|
Name string `json:"name"`
|
|
ResourceType string `json:"resource_type"`
|
|
Provider string `json:"provider"`
|
|
ExternalId string `json:"external_id"`
|
|
SecretRef string `json:"secret_ref"`
|
|
Metadata map[string]any `json:"metadata"`
|
|
PermissionScope map[string]any `json:"permission_scope"`
|
|
Constraints map[string]any `json:"constraints"`
|
|
Status string `json:"status"`
|
|
}
|
|
|
|
type resourceSecretPayload struct {
|
|
Data map[string]any `json:"data"`
|
|
}
|
|
|
|
type resourceResponse struct {
|
|
Id int `json:"id"`
|
|
UserId int `json:"user_id"`
|
|
TenantId string `json:"tenant_id"`
|
|
ProjectId string `json:"project_id"`
|
|
BindingScope string `json:"binding_scope"`
|
|
Name string `json:"name"`
|
|
ResourceType string `json:"resource_type"`
|
|
Provider string `json:"provider"`
|
|
ExternalId string `json:"external_id"`
|
|
SecretRef string `json:"secret_ref"`
|
|
Metadata map[string]any `json:"metadata"`
|
|
PermissionScope map[string]any `json:"permission_scope"`
|
|
Constraints map[string]any `json:"constraints"`
|
|
Status string `json:"status"`
|
|
CreatedAt int64 `json:"created_at"`
|
|
UpdatedAt int64 `json:"updated_at"`
|
|
}
|
|
|
|
type resourceGrantPayload struct {
|
|
TenantId string `json:"tenant_id"`
|
|
ProjectId string `json:"project_id"`
|
|
BindingScope string `json:"binding_scope"`
|
|
ResourceId int `json:"resource_id"`
|
|
Role string `json:"role"`
|
|
AgnetId string `json:"agnet_id"`
|
|
PermissionScope map[string]any `json:"permission_scope"`
|
|
Constraints map[string]any `json:"constraints"`
|
|
Status string `json:"status"`
|
|
}
|
|
|
|
type resourceGrantResponse struct {
|
|
Id int `json:"id"`
|
|
UserId int `json:"user_id"`
|
|
TenantId string `json:"tenant_id"`
|
|
ProjectId string `json:"project_id"`
|
|
BindingScope string `json:"binding_scope"`
|
|
ResourceId int `json:"resource_id"`
|
|
Role string `json:"role"`
|
|
AgnetId string `json:"agnet_id"`
|
|
PermissionScope map[string]any `json:"permission_scope"`
|
|
Constraints map[string]any `json:"constraints"`
|
|
Status string `json:"status"`
|
|
RevokedAt int64 `json:"revoked_at"`
|
|
Resource *resourceResponse `json:"resource,omitempty"`
|
|
CreatedAt int64 `json:"created_at"`
|
|
UpdatedAt int64 `json:"updated_at"`
|
|
}
|
|
|
|
type resourceGrantManifestGrant struct {
|
|
GrantId int `json:"grant_id"`
|
|
ResourceId int `json:"resource_id"`
|
|
ResourceType string `json:"resource_type"`
|
|
ResourceRef string `json:"resource_ref"`
|
|
AllowedActions []any `json:"allowed_actions"`
|
|
PermissionScope map[string]any `json:"permission_scope"`
|
|
Constraints map[string]any `json:"constraints"`
|
|
SecretRef string `json:"secret_ref,omitempty"`
|
|
Status string `json:"status"`
|
|
}
|
|
|
|
type resourceGrantManifestResponse struct {
|
|
UserId int `json:"user_id"`
|
|
BindingScope string `json:"binding_scope"`
|
|
AgentRole string `json:"agent_role"`
|
|
TargetAgentRef string `json:"target_agent_ref"`
|
|
ResourceGrants []resourceGrantManifestGrant `json:"resource_grants"`
|
|
}
|
|
|
|
func normalizeResourcePayload(p resourcePayload) (resourcePayload, error) {
|
|
p.TenantId = strings.TrimSpace(p.TenantId)
|
|
p.ProjectId = strings.TrimSpace(p.ProjectId)
|
|
p.BindingScope = strings.TrimSpace(p.BindingScope)
|
|
p.Name = strings.TrimSpace(p.Name)
|
|
p.ResourceType = strings.ToLower(strings.TrimSpace(p.ResourceType))
|
|
p.Provider = strings.TrimSpace(p.Provider)
|
|
p.ExternalId = strings.TrimSpace(p.ExternalId)
|
|
p.SecretRef = strings.TrimSpace(p.SecretRef)
|
|
p.Status = strings.ToLower(strings.TrimSpace(p.Status))
|
|
|
|
if p.BindingScope == "" {
|
|
p.BindingScope = inferResourceBindingScope(p)
|
|
}
|
|
if p.Name == "" {
|
|
return p, errors.New("name required")
|
|
}
|
|
if !allowedResourceTypes[p.ResourceType] {
|
|
return p, fmt.Errorf("resource_type must be one of %s", strings.Join(sortedResourceTypes(), ", "))
|
|
}
|
|
if p.Provider == "" {
|
|
p.Provider = "custom"
|
|
}
|
|
if p.Status == "" {
|
|
p.Status = "active"
|
|
}
|
|
if !allowedResourceStatuses[p.Status] {
|
|
return p, errors.New("status must be active, disabled, or revoked")
|
|
}
|
|
if p.Metadata == nil {
|
|
p.Metadata = map[string]any{}
|
|
}
|
|
if p.PermissionScope == nil {
|
|
p.PermissionScope = map[string]any{}
|
|
}
|
|
if p.Constraints == nil {
|
|
p.Constraints = map[string]any{}
|
|
}
|
|
if containsPlaintextSecret(p.Metadata) || containsPlaintextSecret(p.PermissionScope) || containsPlaintextSecret(p.Constraints) {
|
|
return p, errors.New("plaintext secrets are not allowed; store credentials in Secret Store and provide secret_ref only")
|
|
}
|
|
return p, nil
|
|
}
|
|
|
|
func normalizeResourceGrantPayload(p resourceGrantPayload) (resourceGrantPayload, error) {
|
|
p.TenantId = strings.TrimSpace(p.TenantId)
|
|
p.ProjectId = strings.TrimSpace(p.ProjectId)
|
|
p.BindingScope = strings.TrimSpace(p.BindingScope)
|
|
p.Role = strings.TrimSpace(p.Role)
|
|
p.AgnetId = strings.TrimSpace(p.AgnetId)
|
|
p.Status = strings.ToLower(strings.TrimSpace(p.Status))
|
|
|
|
if p.ResourceId <= 0 {
|
|
return p, errors.New("resource_id required")
|
|
}
|
|
if p.Role == "" {
|
|
return p, errors.New("role required")
|
|
}
|
|
if p.AgnetId == "" {
|
|
return p, errors.New("agnet_id required")
|
|
}
|
|
if p.Status == "" {
|
|
p.Status = "active"
|
|
}
|
|
if !allowedResourceStatuses[p.Status] {
|
|
return p, errors.New("status must be active, disabled, or revoked")
|
|
}
|
|
if p.PermissionScope == nil {
|
|
p.PermissionScope = map[string]any{}
|
|
}
|
|
if p.Constraints == nil {
|
|
p.Constraints = map[string]any{}
|
|
}
|
|
if containsPlaintextSecret(p.PermissionScope) || containsPlaintextSecret(p.Constraints) {
|
|
return p, errors.New("plaintext secrets are not allowed; store credentials in Secret Store and provide secret_ref only")
|
|
}
|
|
return p, nil
|
|
}
|
|
|
|
func inferResourceBindingScope(p resourcePayload) string {
|
|
switch {
|
|
case p.ExternalId != "":
|
|
return p.ExternalId
|
|
case p.ProjectId != "":
|
|
return p.ProjectId
|
|
case p.TenantId != "":
|
|
return p.TenantId
|
|
case p.Name != "":
|
|
return p.Name
|
|
default:
|
|
return "resource"
|
|
}
|
|
}
|
|
|
|
func sortedResourceTypes() []string {
|
|
types := make([]string, 0, len(allowedResourceTypes))
|
|
for resourceType := range allowedResourceTypes {
|
|
types = append(types, resourceType)
|
|
}
|
|
sort.Strings(types)
|
|
return types
|
|
}
|
|
|
|
func containsPlaintextSecret(value any) bool {
|
|
switch v := value.(type) {
|
|
case map[string]any:
|
|
for key, child := range v {
|
|
if isSecretLikeKey(key) {
|
|
return true
|
|
}
|
|
if containsPlaintextSecret(child) {
|
|
return true
|
|
}
|
|
}
|
|
case []any:
|
|
for _, child := range v {
|
|
if containsPlaintextSecret(child) {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func isSecretLikeKey(key string) bool {
|
|
normalized := strings.ToLower(strings.TrimSpace(key))
|
|
normalized = strings.ReplaceAll(normalized, "-", "_")
|
|
normalized = strings.ReplaceAll(normalized, " ", "_")
|
|
if secretLikeKeys[normalized] {
|
|
return true
|
|
}
|
|
return strings.HasSuffix(normalized, "_token") || strings.HasSuffix(normalized, "_secret") || strings.HasSuffix(normalized, "_password") || strings.HasSuffix(normalized, "_private_key")
|
|
}
|
|
|
|
func marshalResourceJSON(value map[string]any) (string, error) {
|
|
if value == nil {
|
|
value = map[string]any{}
|
|
}
|
|
data, err := common.Marshal(value)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(data), nil
|
|
}
|
|
|
|
func unmarshalResourceJSON(raw string) map[string]any {
|
|
if strings.TrimSpace(raw) == "" {
|
|
return map[string]any{}
|
|
}
|
|
var value map[string]any
|
|
if err := common.UnmarshalJsonStr(raw, &value); err != nil || value == nil {
|
|
return map[string]any{}
|
|
}
|
|
return value
|
|
}
|
|
|
|
func resourceToResponse(resource model.ResourceBinding) resourceResponse {
|
|
return resourceResponse{
|
|
Id: resource.Id,
|
|
UserId: resource.UserId,
|
|
TenantId: resource.TenantId,
|
|
ProjectId: resource.ProjectId,
|
|
BindingScope: resource.BindingScope,
|
|
Name: resource.Name,
|
|
ResourceType: resource.ResourceType,
|
|
Provider: resource.Provider,
|
|
ExternalId: resource.ExternalId,
|
|
SecretRef: resource.SecretRef,
|
|
Metadata: unmarshalResourceJSON(resource.Metadata),
|
|
PermissionScope: unmarshalResourceJSON(resource.PermissionScope),
|
|
Constraints: unmarshalResourceJSON(resource.Constraints),
|
|
Status: resource.Status,
|
|
CreatedAt: resource.CreatedAt,
|
|
UpdatedAt: resource.UpdatedAt,
|
|
}
|
|
}
|
|
|
|
func resourceGrantToResponse(grant model.ResourceGrant, resource *model.ResourceBinding) resourceGrantResponse {
|
|
resp := resourceGrantResponse{
|
|
Id: grant.Id,
|
|
UserId: grant.UserId,
|
|
TenantId: grant.TenantId,
|
|
ProjectId: grant.ProjectId,
|
|
BindingScope: grant.BindingScope,
|
|
ResourceId: grant.ResourceId,
|
|
Role: grant.Role,
|
|
AgnetId: grant.AgnetId,
|
|
PermissionScope: unmarshalResourceJSON(grant.PermissionScope),
|
|
Constraints: unmarshalResourceJSON(grant.Constraints),
|
|
Status: grant.Status,
|
|
RevokedAt: grant.RevokedAt,
|
|
CreatedAt: grant.CreatedAt,
|
|
UpdatedAt: grant.UpdatedAt,
|
|
}
|
|
if resource != nil {
|
|
resourceResp := resourceToResponse(*resource)
|
|
resp.Resource = &resourceResp
|
|
}
|
|
return resp
|
|
}
|
|
|
|
func resourceGrantToManifestGrant(grant model.ResourceGrant, resource model.ResourceBinding) resourceGrantManifestGrant {
|
|
permissionScope := unmarshalResourceJSON(grant.PermissionScope)
|
|
return resourceGrantManifestGrant{
|
|
GrantId: grant.Id,
|
|
ResourceId: grant.ResourceId,
|
|
ResourceType: resource.ResourceType,
|
|
ResourceRef: resourceManifestRef(resource),
|
|
AllowedActions: manifestAllowedActions(permissionScope),
|
|
PermissionScope: permissionScope,
|
|
Constraints: unmarshalResourceJSON(grant.Constraints),
|
|
SecretRef: resource.SecretRef,
|
|
Status: grant.Status,
|
|
}
|
|
}
|
|
|
|
func resourceManifestRef(resource model.ResourceBinding) string {
|
|
for _, value := range []string{resource.ExternalId, resource.BindingScope, resource.Name} {
|
|
if strings.TrimSpace(value) != "" {
|
|
return value
|
|
}
|
|
}
|
|
return fmt.Sprintf("resource:%d", resource.Id)
|
|
}
|
|
|
|
func manifestAllowedActions(permissionScope map[string]any) []any {
|
|
actions, ok := permissionScope["actions"]
|
|
if !ok {
|
|
return []any{}
|
|
}
|
|
switch v := actions.(type) {
|
|
case []any:
|
|
return v
|
|
case []string:
|
|
out := make([]any, 0, len(v))
|
|
for _, action := range v {
|
|
out = append(out, action)
|
|
}
|
|
return out
|
|
case string:
|
|
if strings.TrimSpace(v) == "" {
|
|
return []any{}
|
|
}
|
|
return []any{v}
|
|
default:
|
|
return []any{}
|
|
}
|
|
}
|
|
|
|
func ListResources(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
query := model.DB.Where("user_id = ?", userId)
|
|
if tenantId := strings.TrimSpace(c.Query("tenant_id")); tenantId != "" {
|
|
query = query.Where("tenant_id = ?", tenantId)
|
|
}
|
|
if projectId := strings.TrimSpace(c.Query("project_id")); projectId != "" {
|
|
query = query.Where("project_id = ?", projectId)
|
|
}
|
|
if bindingScope := strings.TrimSpace(c.Query("binding_scope")); bindingScope != "" {
|
|
query = query.Where("binding_scope = ?", bindingScope)
|
|
}
|
|
if resourceType := strings.TrimSpace(c.Query("resource_type")); resourceType != "" {
|
|
query = query.Where("resource_type = ?", strings.ToLower(resourceType))
|
|
}
|
|
var resources []model.ResourceBinding
|
|
if err := query.Order("id desc").Find(&resources).Error; err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
items := make([]resourceResponse, 0, len(resources))
|
|
for _, resource := range resources {
|
|
items = append(items, resourceToResponse(resource))
|
|
}
|
|
common.ApiSuccess(c, gin.H{"items": items})
|
|
}
|
|
|
|
func CreateResource(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
var payload resourcePayload
|
|
if err := c.ShouldBindJSON(&payload); err != nil {
|
|
c.JSON(http.StatusOK, gin.H{"success": false, "message": "invalid params"})
|
|
return
|
|
}
|
|
payload, err := normalizeResourcePayload(payload)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
metadata, permissionScope, constraints, err := marshalResourcePayloadJSON(payload)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
resource := model.ResourceBinding{
|
|
UserId: userId,
|
|
TenantId: payload.TenantId,
|
|
ProjectId: payload.ProjectId,
|
|
BindingScope: payload.BindingScope,
|
|
Name: payload.Name,
|
|
ResourceType: payload.ResourceType,
|
|
Provider: payload.Provider,
|
|
ExternalId: payload.ExternalId,
|
|
SecretRef: payload.SecretRef,
|
|
Metadata: metadata,
|
|
PermissionScope: permissionScope,
|
|
Constraints: constraints,
|
|
Status: payload.Status,
|
|
}
|
|
if err := model.DB.Create(&resource).Error; err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
common.ApiSuccess(c, resourceToResponse(resource))
|
|
}
|
|
|
|
func UpdateResource(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
var resource model.ResourceBinding
|
|
if err := model.DB.Where("id = ? AND user_id = ?", c.Param("id"), userId).First(&resource).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
common.ApiErrorMsg(c, "resource not found")
|
|
return
|
|
}
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
var payload resourcePayload
|
|
if err := c.ShouldBindJSON(&payload); err != nil {
|
|
c.JSON(http.StatusOK, gin.H{"success": false, "message": "invalid params"})
|
|
return
|
|
}
|
|
payload, err := normalizeResourcePayload(payload)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
metadata, permissionScope, constraints, err := marshalResourcePayloadJSON(payload)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
resource.TenantId = payload.TenantId
|
|
resource.ProjectId = payload.ProjectId
|
|
resource.BindingScope = payload.BindingScope
|
|
resource.Name = payload.Name
|
|
resource.ResourceType = payload.ResourceType
|
|
resource.Provider = payload.Provider
|
|
resource.ExternalId = payload.ExternalId
|
|
resource.SecretRef = payload.SecretRef
|
|
resource.Metadata = metadata
|
|
resource.PermissionScope = permissionScope
|
|
resource.Constraints = constraints
|
|
resource.Status = payload.Status
|
|
if err := model.DB.Save(&resource).Error; err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
common.ApiSuccess(c, resourceToResponse(resource))
|
|
}
|
|
|
|
func DeleteResource(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
res := model.DB.Where("id = ? AND user_id = ?", c.Param("id"), userId).Delete(&model.ResourceBinding{})
|
|
if res.Error != nil {
|
|
common.ApiError(c, res.Error)
|
|
return
|
|
}
|
|
if res.RowsAffected == 0 {
|
|
common.ApiErrorMsg(c, "resource not found")
|
|
return
|
|
}
|
|
common.ApiSuccess(c, gin.H{"deleted": true})
|
|
}
|
|
|
|
func UpsertResourceSecret(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
var resource model.ResourceBinding
|
|
if err := model.DB.Where("id = ? AND user_id = ?", c.Param("id"), userId).First(&resource).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
common.ApiErrorMsg(c, "resource not found")
|
|
return
|
|
}
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
var payload resourceSecretPayload
|
|
if err := c.ShouldBindJSON(&payload); err != nil {
|
|
c.JSON(http.StatusOK, gin.H{"success": false, "message": "invalid params"})
|
|
return
|
|
}
|
|
if len(payload.Data) == 0 {
|
|
common.ApiErrorMsg(c, "secret data required")
|
|
return
|
|
}
|
|
client, err := newSecretStoreClientFromEnv()
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
secretPath := resourceSecretPath(resource)
|
|
if err := client.putKV2(secretPath, payload.Data); err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
resource.SecretRef = fmt.Sprintf("vault://%s/%s", client.mount, secretPath)
|
|
if err := model.DB.Save(&resource).Error; err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
common.ApiSuccess(c, gin.H{
|
|
"resource": resourceToResponse(resource),
|
|
"secret_ref": resource.SecretRef,
|
|
})
|
|
}
|
|
|
|
func resourceSecretPath(resource model.ResourceBinding) string {
|
|
scope := resource.BindingScope
|
|
if strings.TrimSpace(scope) == "" {
|
|
scope = resource.ExternalId
|
|
}
|
|
if strings.TrimSpace(scope) == "" {
|
|
scope = resource.Name
|
|
}
|
|
return strings.Join([]string{
|
|
"users",
|
|
fmt.Sprintf("%d", resource.UserId),
|
|
"bindings",
|
|
sanitizeSecretPathSegment(scope),
|
|
"resources",
|
|
fmt.Sprintf("%d", resource.Id),
|
|
}, "/")
|
|
}
|
|
|
|
func sanitizeSecretPathSegment(value string) string {
|
|
value = strings.TrimSpace(value)
|
|
if value == "" {
|
|
return "_"
|
|
}
|
|
var b strings.Builder
|
|
for _, r := range value {
|
|
switch {
|
|
case r >= 'a' && r <= 'z':
|
|
b.WriteRune(r)
|
|
case r >= 'A' && r <= 'Z':
|
|
b.WriteRune(r)
|
|
case r >= '0' && r <= '9':
|
|
b.WriteRune(r)
|
|
case r == '_' || r == '-' || r == '.':
|
|
b.WriteRune(r)
|
|
default:
|
|
b.WriteRune('-')
|
|
}
|
|
}
|
|
return b.String()
|
|
}
|
|
|
|
func marshalResourcePayloadJSON(payload resourcePayload) (string, string, string, error) {
|
|
metadata, err := marshalResourceJSON(payload.Metadata)
|
|
if err != nil {
|
|
return "", "", "", err
|
|
}
|
|
permissionScope, err := marshalResourceJSON(payload.PermissionScope)
|
|
if err != nil {
|
|
return "", "", "", err
|
|
}
|
|
constraints, err := marshalResourceJSON(payload.Constraints)
|
|
if err != nil {
|
|
return "", "", "", err
|
|
}
|
|
return metadata, permissionScope, constraints, nil
|
|
}
|
|
|
|
func ListResourceGrants(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
query := model.DB.Where("user_id = ?", userId)
|
|
if tenantId := strings.TrimSpace(c.Query("tenant_id")); tenantId != "" {
|
|
query = query.Where("tenant_id = ?", tenantId)
|
|
}
|
|
if projectId := strings.TrimSpace(c.Query("project_id")); projectId != "" {
|
|
query = query.Where("project_id = ?", projectId)
|
|
}
|
|
if bindingScope := strings.TrimSpace(c.Query("binding_scope")); bindingScope != "" {
|
|
query = query.Where("binding_scope = ?", bindingScope)
|
|
}
|
|
if agnetId := strings.TrimSpace(c.Query("agnet_id")); agnetId != "" {
|
|
query = query.Where("agnet_id = ?", agnetId)
|
|
}
|
|
var grants []model.ResourceGrant
|
|
if err := query.Order("id desc").Find(&grants).Error; err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
items := make([]resourceGrantResponse, 0, len(grants))
|
|
for _, grant := range grants {
|
|
items = append(items, resourceGrantToResponse(grant, nil))
|
|
}
|
|
common.ApiSuccess(c, gin.H{"items": items})
|
|
}
|
|
|
|
func GenerateResourceGrantManifest(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
bindingScope := strings.TrimSpace(c.Query("binding_scope"))
|
|
agentId := strings.TrimSpace(c.Query("agnet_id"))
|
|
role := strings.TrimSpace(c.Query("role"))
|
|
|
|
query := model.DB.Where("user_id = ? AND status = ?", userId, "active")
|
|
if bindingScope != "" {
|
|
query = query.Where("binding_scope = ?", bindingScope)
|
|
}
|
|
if agentId != "" {
|
|
query = query.Where("agnet_id = ?", agentId)
|
|
}
|
|
if role != "" {
|
|
query = query.Where("role = ?", role)
|
|
}
|
|
|
|
var grants []model.ResourceGrant
|
|
if err := query.Order("id asc").Find(&grants).Error; err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
|
|
items := make([]resourceGrantManifestGrant, 0, len(grants))
|
|
for _, grant := range grants {
|
|
var resource model.ResourceBinding
|
|
if err := model.DB.Where("id = ? AND user_id = ? AND status = ?", grant.ResourceId, userId, "active").First(&resource).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
continue
|
|
}
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
items = append(items, resourceGrantToManifestGrant(grant, resource))
|
|
}
|
|
|
|
common.ApiSuccess(c, resourceGrantManifestResponse{
|
|
UserId: userId,
|
|
BindingScope: bindingScope,
|
|
AgentRole: role,
|
|
TargetAgentRef: agentId,
|
|
ResourceGrants: items,
|
|
})
|
|
}
|
|
|
|
func CreateResourceGrant(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
var payload resourceGrantPayload
|
|
if err := c.ShouldBindJSON(&payload); err != nil {
|
|
c.JSON(http.StatusOK, gin.H{"success": false, "message": "invalid params"})
|
|
return
|
|
}
|
|
payload, err := normalizeResourceGrantPayload(payload)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
resource, err := findGrantResource(userId, payload.ResourceId, payload.BindingScope)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
payload = inheritResourceGrantScope(payload, resource)
|
|
permissionScope, constraints, err := marshalResourceGrantPayloadJSON(payload)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
grant := model.ResourceGrant{
|
|
UserId: userId,
|
|
TenantId: payload.TenantId,
|
|
ProjectId: payload.ProjectId,
|
|
BindingScope: payload.BindingScope,
|
|
ResourceId: payload.ResourceId,
|
|
Role: payload.Role,
|
|
AgnetId: payload.AgnetId,
|
|
PermissionScope: permissionScope,
|
|
Constraints: constraints,
|
|
Status: payload.Status,
|
|
}
|
|
if err := model.DB.Create(&grant).Error; err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
common.ApiSuccess(c, resourceGrantToResponse(grant, &resource))
|
|
}
|
|
|
|
func UpdateResourceGrant(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
var grant model.ResourceGrant
|
|
if err := model.DB.Where("id = ? AND user_id = ?", c.Param("id"), userId).First(&grant).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
common.ApiErrorMsg(c, "resource grant not found")
|
|
return
|
|
}
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
var payload resourceGrantPayload
|
|
if err := c.ShouldBindJSON(&payload); err != nil {
|
|
c.JSON(http.StatusOK, gin.H{"success": false, "message": "invalid params"})
|
|
return
|
|
}
|
|
payload, err := normalizeResourceGrantPayload(payload)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
resource, err := findGrantResource(userId, payload.ResourceId, payload.BindingScope)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
payload = inheritResourceGrantScope(payload, resource)
|
|
permissionScope, constraints, err := marshalResourceGrantPayloadJSON(payload)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
grant.TenantId = payload.TenantId
|
|
grant.ProjectId = payload.ProjectId
|
|
grant.BindingScope = payload.BindingScope
|
|
grant.ResourceId = payload.ResourceId
|
|
grant.Role = payload.Role
|
|
grant.AgnetId = payload.AgnetId
|
|
grant.PermissionScope = permissionScope
|
|
grant.Constraints = constraints
|
|
grant.Status = payload.Status
|
|
if payload.Status == "revoked" && grant.RevokedAt == 0 {
|
|
grant.RevokedAt = common.GetTimestamp()
|
|
}
|
|
if payload.Status != "revoked" {
|
|
grant.RevokedAt = 0
|
|
}
|
|
if err := model.DB.Save(&grant).Error; err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
common.ApiSuccess(c, resourceGrantToResponse(grant, &resource))
|
|
}
|
|
|
|
func DeleteResourceGrant(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
res := model.DB.Where("id = ? AND user_id = ?", c.Param("id"), userId).Delete(&model.ResourceGrant{})
|
|
if res.Error != nil {
|
|
common.ApiError(c, res.Error)
|
|
return
|
|
}
|
|
if res.RowsAffected == 0 {
|
|
common.ApiErrorMsg(c, "resource grant not found")
|
|
return
|
|
}
|
|
common.ApiSuccess(c, gin.H{"deleted": true})
|
|
}
|
|
|
|
func marshalResourceGrantPayloadJSON(payload resourceGrantPayload) (string, string, error) {
|
|
permissionScope, err := marshalResourceJSON(payload.PermissionScope)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
constraints, err := marshalResourceJSON(payload.Constraints)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
return permissionScope, constraints, nil
|
|
}
|
|
|
|
func inheritResourceGrantScope(payload resourceGrantPayload, resource model.ResourceBinding) resourceGrantPayload {
|
|
if payload.BindingScope == "" {
|
|
payload.BindingScope = resource.BindingScope
|
|
}
|
|
if payload.TenantId == "" {
|
|
payload.TenantId = resource.TenantId
|
|
}
|
|
if payload.ProjectId == "" {
|
|
payload.ProjectId = resource.ProjectId
|
|
}
|
|
return payload
|
|
}
|
|
|
|
func findGrantResource(userId int, resourceId int, bindingScope string) (model.ResourceBinding, error) {
|
|
var resource model.ResourceBinding
|
|
if err := model.DB.Where("id = ? AND user_id = ?", resourceId, userId).First(&resource).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return resource, errors.New("resource not found")
|
|
}
|
|
return resource, err
|
|
}
|
|
if bindingScope != "" && resource.BindingScope != "" && resource.BindingScope != bindingScope {
|
|
return resource, errors.New("resource binding_scope does not match grant binding_scope")
|
|
}
|
|
return resource, nil
|
|
}
|