fix(server): add prop DisableOperationByOverUsedSeat on PolicyCheckPayload (#1829)

Author: soneda-yuya

server/gql/workspace.graphql

@@ -100,6 +100,7 @@ type CreateWorkspacePayload {
 type PolicyCheckPayload {
   workspaceId: ID!
   enableToCreatePrivateProject: Boolean!
+  disableOperationByOverUsedSeat: Boolean!
 }
 
 type UpdateWorkspacePayload {

server/internal/adapter/gql/generated.go

@@ -19,8 +19,9 @@ func TestWorkspacePolicyCheck_InputTypes(t *testing.T) {
 
 	// Test output structure
 	output := &gqlmodel.PolicyCheckPayload{
-		WorkspaceID:                  input.WorkspaceID,
-		EnableToCreatePrivateProject: true,
+		WorkspaceID:                    input.WorkspaceID,
+		EnableToCreatePrivateProject:   true,
+		DisableOperationByOverUsedSeat: false,
 	}
 
 	assert.Equal(t, input.WorkspaceID, output.WorkspaceID)
@@ -30,8 +31,9 @@ func TestWorkspacePolicyCheck_InputTypes(t *testing.T) {
 func TestWorkspacePolicyCheck_ResponseStructure(t *testing.T) {
 	// Test that the response structure is correctly defined
 	payload := &gqlmodel.PolicyCheckPayload{
-		WorkspaceID:                  gqlmodel.ID("01H4XCVR7QZJN0Z8V9XN9N9N9N"),
-		EnableToCreatePrivateProject: true,
+		WorkspaceID:                    gqlmodel.ID("01H4XCVR7QZJN0Z8V9XN9N9N9N"),
+		EnableToCreatePrivateProject:   true,
+		DisableOperationByOverUsedSeat: false,
 	}
 
 	assert.Equal(t, gqlmodel.ID("01H4XCVR7QZJN0Z8V9XN9N9N9N"), payload.WorkspaceID)

server/internal/adapter/gql/gqlmodel/models_gen.go

@@ -14,6 +14,7 @@ const (
 	PolicyCheckCustomDomainCount             PolicyCheckType = "visualizer_custom_domain_count"
 	PolicyCheckGeneralPrivateProjectCreation PolicyCheckType = "general_private_project_creation"
 	PolicyCheckGeneralPublicProjectCreation  PolicyCheckType = "general_public_project_creation"
+	PolicyCheckGeneralOperationAllowed       PolicyCheckType = "general_operation_allowed"
 )
 
 type PolicyCheckRequest struct {
@@ -48,3 +49,11 @@ func CreateGeneralPolicyCheckRequest(workspaceID accountdomain.WorkspaceID, visi
 		Value:       1,
 	}
 }
+
+func CreateGeneralOperationAllowedCheckRequest(workspaceID accountdomain.WorkspaceID) PolicyCheckRequest {
+	return PolicyCheckRequest{
+		WorkspaceID: workspaceID,
+		CheckType:   PolicyCheckGeneralOperationAllowed,
+		Value:       1,
+	}
+}

server/internal/adapter/gql/resolver_workspace_policy_test.go

@@ -30,15 +30,21 @@ func (i *Policy) GetWorkspacePolicy(ctx context.Context, wsid accountdomain.Work
 		return nil, err
 	}
 
-	res, err := i.policyChecker.CheckPolicy(ctx, gateway.CreateGeneralPolicyCheckRequest(ws.ID(), project.VisibilityPrivate))
+	createPrivateProject, err := i.policyChecker.CheckPolicy(ctx, gateway.CreateGeneralPolicyCheckRequest(ws.ID(), project.VisibilityPrivate))
 
 	if err != nil {
 		return nil, err
 	}
 
+	operationAllowed, err := i.policyChecker.CheckPolicy(ctx, gateway.CreateGeneralOperationAllowedCheckRequest(ws.ID()))
+	if err != nil {
+		return nil, err
+	}
+
 	return &policy.WorkspacePolicy{
-		WorkspaceID:                  wsid,
-		EnableToCreatePrivateProject: res.Allowed,
+		WorkspaceID:                    wsid,
+		EnableToCreatePrivateProject:   createPrivateProject.Allowed,
+		DisableOperationByOverUsedSeat: !operationAllowed.Allowed,
 	}, nil
 }
 

server/internal/usecase/gateway/policy_checker.go

@@ -12,8 +12,9 @@ type ID = workspace.PolicyID
 var ErrPolicyViolation = errors.New("policy violation")
 
 type WorkspacePolicy struct {
-	WorkspaceID                  workspace.ID
-	EnableToCreatePrivateProject bool
+	WorkspaceID                    workspace.ID
+	EnableToCreatePrivateProject   bool
+	DisableOperationByOverUsedSeat bool
 }
 
 type Policy struct {