Merge pull request #20 from ConductorOne/jirwin/skip-entitlements-grants

Add support to skip entitlements and grants on a resource type

Author: jirwin

pkg/bsql/config.go

@@ -19,12 +19,13 @@ type DatabaseConfig struct {
 }
 
 type ResourceType struct {
-	Name               string                `yaml:"name" json:"name"`
-	List               *ListQuery            `yaml:"list,omitempty" json:"list,omitempty"`
-	Entitlements       *EntitlementsQuery    `yaml:"entitlements,omitempty" json:"entitlements,omitempty"`
-	StaticEntitlements []*EntitlementMapping `yaml:"static_entitlements,omitempty" json:"static_entitlements,omitempty"`
-	Grants             []*GrantsQuery        `yaml:"grants,omitempty" json:"grants,omitempty"`
-	Description        string                `yaml:"description,omitempty" json:"description,omitempty"`
+	Name                      string                `yaml:"name" json:"name"`
+	List                      *ListQuery            `yaml:"list,omitempty" json:"list,omitempty"`
+	Entitlements              *EntitlementsQuery    `yaml:"entitlements,omitempty" json:"entitlements,omitempty"`
+	StaticEntitlements        []*EntitlementMapping `yaml:"static_entitlements,omitempty" json:"static_entitlements,omitempty"`
+	Grants                    []*GrantsQuery        `yaml:"grants,omitempty" json:"grants,omitempty"`
+	Description               string                `yaml:"description,omitempty" json:"description,omitempty"`
+	SkipEntitlementsAndGrants bool                  `yaml:"skip_entitlements_and_grants,omitempty" json:"skip_entitlements_and_grants,omitempty"`
 }
 
 type ListQuery struct {

pkg/bsql/resource_types.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 
 	v2 "github.com/conductorone/baton-sdk/pb/c1/connector/v2"
+	"github.com/conductorone/baton-sdk/pkg/annotations"
 )
 
 func (c Config) extractTraits(rtID string) ([]v2.ResourceType_Trait, error) {
@@ -54,12 +55,19 @@ func (c Config) GetResourceTypes(ctx context.Context) ([]*v2.ResourceType, error
 			return nil, err
 		}
 
-		resourceTypes = append(resourceTypes, &v2.ResourceType{
+		resourceType := &v2.ResourceType{
 			Id:          rtID,
 			DisplayName: rt.Name,
 			Description: rt.Description,
 			Traits:      traits,
-		})
+		}
+
+		if rt.SkipEntitlementsAndGrants {
+			annos := annotations.Annotations(resourceType.Annotations)
+			annos.Update(&v2.SkipEntitlementsAndGrants{})
+			resourceType.Annotations = annos
+		}
+		resourceTypes = append(resourceTypes, resourceType)
 	}
 	return resourceTypes, nil
 }
@@ -75,10 +83,18 @@ func (c Config) GetResourceType(ctx context.Context, rtID string) (*v2.ResourceT
 		return nil, fmt.Errorf("resource type %s not found in config", rtID)
 	}
 
-	return &v2.ResourceType{
+	ret := &v2.ResourceType{
 		Id:          rtID,
 		DisplayName: rt.Name,
 		Description: rt.Description,
 		Traits:      traits,
-	}, nil
+	}
+
+	if rt.SkipEntitlementsAndGrants {
+		annos := annotations.Annotations(ret.Annotations)
+		annos.Update(&v2.SkipEntitlementsAndGrants{})
+		ret.Annotations = annos
+	}
+
+	return ret, nil
 }