Add control categories

Signed-off-by: gearnode <bryan@frimin.fr>
getprobo · Feb 13, 2025 · 72d4b09 · 72d4b09
1 parent bd7fb97
commit 72d4b09
Show file tree

Hide file tree

Showing 6 changed files with 67 additions and 0 deletions.
diff --git a/pkg/api/console/v1/schema.graphql b/pkg/api/console/v1/schema.graphql
@@ -157,6 +157,7 @@ type ControlEdge {
 
 type Control implements Node {
   id: ID!
+  category: String!
   name: String!
   description: String!
   state: ControlState!

diff --git a/pkg/api/console/v1/schema/schema.go b/pkg/api/console/v1/schema/schema.go
diff --git a/pkg/api/console/v1/types/control.go b/pkg/api/console/v1/types/control.go
@@ -42,6 +42,7 @@ func NewControlEdge(c *coredata.Control) *ControlEdge {
 func NewControl(c *coredata.Control) *Control {
 	return &Control{
 		ID:          c.ID,
+		Category:    c.Category,
 		Name:        c.Name,
 		Description: c.Description,
 		State:       c.State,

diff --git a/pkg/api/console/v1/types/types.go b/pkg/api/console/v1/types/types.go
diff --git a/pkg/probo/coredata/control.go b/pkg/probo/coredata/control.go
@@ -31,6 +31,7 @@ type (
 	Control struct {
 		ID          gid.GID
 		FrameworkID gid.GID
+		Category    string
 		Name        string
 		Description string
 		State       ControlState
@@ -50,6 +51,7 @@ func (c *Control) scan(r pgx.Row) error {
 	return r.Scan(
 		&c.ID,
 		&c.FrameworkID,
+		&c.Category,
 		&c.Name,
 		&c.Description,
 		&c.State,
@@ -82,6 +84,7 @@ WITH control_states AS (
 SELECT
     id,
     framework_id,
+    category,
     name,
     description,
     cs.to_state AS state,
@@ -125,6 +128,7 @@ INSERT INTO
     controls (
         id,
         control_id,
+		category,
         name,
         description,
         content_ref,
@@ -179,6 +183,7 @@ WITH control_states AS (
 SELECT
     id,
     framework_id,
+	category,
     name,
     description,
     cs.to_state AS state,

diff --git a/pkg/probo/coredata/migrations/20250212T174900Z.sql b/pkg/probo/coredata/migrations/20250212T174900Z.sql
@@ -0,0 +1,5 @@
+ALTER TABLE controls ADD COLUMN category TEXT;
+
+UPDATE controls SET category = 'security';
+
+ALTER TABLE controls ALTER COLUMN category SET NOT NULL;