make RelationshipCheck client functions (#578)

Author: rocktavious

.changes/unreleased/Feature-20250731-110119.yaml

@@ -0,0 +1,3 @@
+kind: Feature
+body: Add new client functions for `CreateCheckRelationship` and `UpdateCheckRelationship`
+time: 2025-07-31T11:01:19.557368-04:00

check.go

@@ -38,6 +38,7 @@ var CheckCreateConstructors = map[CheckType]CheckInputConstructor{
 	CheckTypeTagDefined:          func() any { return &CheckTagDefinedCreateInput{} },
 	CheckTypeToolUsage:           func() any { return &CheckToolUsageCreateInput{} },
 	CheckTypePackageVersion:      func() any { return &CheckPackageVersionCreateInput{} },
+	CheckTypeRelationship:        func() any { return &CheckRelationshipCreateInput{} },
 }
 
 var CheckUpdateConstructors = map[CheckType]CheckInputConstructor{
@@ -61,6 +62,7 @@ var CheckUpdateConstructors = map[CheckType]CheckInputConstructor{
 	CheckTypeTagDefined:          func() any { return &CheckTagDefinedUpdateInput{} },
 	CheckTypeToolUsage:           func() any { return &CheckToolUsageUpdateInput{} },
 	CheckTypePackageVersion:      func() any { return &CheckPackageVersionUpdateInput{} },
+	CheckTypeRelationship:        func() any { return &CheckRelationshipUpdateInput{} },
 }
 
 func UnmarshalCheckCreateInput(checkType CheckType, data []byte) (any, error) {
@@ -164,6 +166,8 @@ func (client *Client) CreateCheck(input any) (*Check, error) {
 		return client.CreateCheckToolUsage(*v)
 	case *CheckPackageVersionCreateInput:
 		return client.CreateCheckPackageVersion(*v)
+	case *CheckRelationshipCreateInput:
+		return client.CreateCheckRelationship(*v)
 	}
 	return nil, fmt.Errorf("unknown input type %T", input)
 }
@@ -252,6 +256,8 @@ func (client *Client) UpdateCheck(input any) (*Check, error) {
 		return client.UpdateCheckToolUsage(*v)
 	case *CheckPackageVersionUpdateInput:
 		return client.UpdateCheckPackageVersion(*v)
+	case *CheckRelationshipUpdateInput:
+		return client.UpdateCheckRelationship(*v)
 	}
 	return nil, fmt.Errorf("unknown input type %T", input)
 }

check_relationship.go

@@ -0,0 +1,28 @@
+package opslevel
+
+type RelationshipCheckFragment struct {
+	RelationshipCountPredicate *Predicate                 `graphql:"relationshipCountPredicate"` // The condition that should be satisfied by the number of RelatedTo relationships
+	RelationshipDefinition     RelationshipDefinitionType `graphql:"relationshipDefinition"`     // The relationship definition that the check is based on.
+}
+
+func (client *Client) CreateCheckRelationship(input CheckRelationshipCreateInput) (*Check, error) {
+	var m struct {
+		Payload CheckResponsePayload `graphql:"checkRelationshipCreate(input: $input)"`
+	}
+	v := PayloadVariables{
+		"input": input,
+	}
+	err := client.Mutate(&m, v, WithName("CheckRelationshipCreate"))
+	return &m.Payload.Check, HandleErrors(err, m.Payload.Errors)
+}
+
+func (client *Client) UpdateCheckRelationship(input CheckRelationshipUpdateInput) (*Check, error) {
+	var m struct {
+		Payload CheckResponsePayload `graphql:"checkRelationshipUpdate(input: $input)"`
+	}
+	v := PayloadVariables{
+		"input": input,
+	}
+	err := client.Mutate(&m, v, WithName("CheckRelationshipUpdate"))
+	return &m.Payload.Check, HandleErrors(err, m.Payload.Errors)
+}