Add compliance registry snapshots

Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>

pkg/coredata/compliance_registry.go

@@ -40,6 +40,8 @@ type (
 		LastReviewDate         *time.Time               `db:"last_review_date"`
 		DueDate                *time.Time               `db:"due_date"`
 		Status                 ComplianceRegistryStatus `db:"status"`
+		SnapshotID             *gid.GID                 `db:"snapshot_id"`
+		SourceID               *gid.GID                 `db:"source_id"`
 		CreatedAt              time.Time                `db:"created_at"`
 		UpdatedAt              time.Time                `db:"updated_at"`
 	}
@@ -74,6 +76,8 @@ func (cr *ComplianceRegistry) LoadByID(
 SELECT
 	id,
 	organization_id,
+	snapshot_id,
+	source_id,
 	reference_id,
 	area,
 	source,
@@ -119,6 +123,7 @@ func (crs *ComplianceRegistries) CountByOrganizationID(
 	conn pg.Conn,
 	scope Scoper,
 	organizationID gid.GID,
+	filter *ComplianceRegistryFilter,
 ) (int, error) {
 	q := `
 SELECT
@@ -128,12 +133,14 @@ FROM
 WHERE
 	%s
 	AND organization_id = @organization_id
+	AND %s
 `
 
-	q = fmt.Sprintf(q, scope.SQLFragment())
+	q = fmt.Sprintf(q, scope.SQLFragment(), filter.SQLFragment())
 
 	args := pgx.StrictNamedArgs{"organization_id": organizationID}
 	maps.Copy(args, scope.SQLArguments())
+	maps.Copy(args, filter.SQLArguments())
 
 	row := conn.QueryRow(ctx, q, args)
 
@@ -152,6 +159,7 @@ func (crs *ComplianceRegistries) LoadByOrganizationID(
 	scope Scoper,
 	organizationID gid.GID,
 	cursor *page.Cursor[ComplianceRegistryOrderField],
+	filter *ComplianceRegistryFilter,
 ) error {
 	q := `
 SELECT
@@ -167,6 +175,8 @@ SELECT
 	last_review_date,
 	due_date,
 	status,
+	snapshot_id,
+	source_id,
 	created_at,
 	updated_at
 FROM
@@ -175,12 +185,14 @@ WHERE
 	%s
 	AND organization_id = @organization_id
 	AND %s
+	AND %s
 `
 
-	q = fmt.Sprintf(q, scope.SQLFragment(), cursor.SQLFragment())
+	q = fmt.Sprintf(q, scope.SQLFragment(), filter.SQLFragment(), cursor.SQLFragment())
 
 	args := pgx.StrictNamedArgs{"organization_id": organizationID}
 	maps.Copy(args, scope.SQLArguments())
+	maps.Copy(args, filter.SQLArguments())
 	maps.Copy(args, cursor.SQLArguments())
 
 	rows, err := conn.Query(ctx, q, args)
@@ -218,6 +230,8 @@ INSERT INTO compliance_registries (
 	last_review_date,
 	due_date,
 	status,
+	snapshot_id,
+	source_id,
 	created_at,
 	updated_at
 ) VALUES (
@@ -234,6 +248,8 @@ INSERT INTO compliance_registries (
 	@last_review_date,
 	@due_date,
 	@status,
+	@snapshot_id,
+	@source_id,
 	@created_at,
 	@updated_at
 )
@@ -253,6 +269,8 @@ INSERT INTO compliance_registries (
 		"last_review_date":          cr.LastReviewDate,
 		"due_date":                  cr.DueDate,
 		"status":                    cr.Status,
+		"snapshot_id":               cr.SnapshotID,
+		"source_id":                 cr.SourceID,
 		"created_at":                cr.CreatedAt,
 		"updated_at":                cr.UpdatedAt,
 	}
@@ -286,6 +304,7 @@ UPDATE compliance_registries SET
 WHERE
 	%s
 	AND id = @id
+	AND snapshot_id IS NULL
 `
 
 	q = fmt.Sprintf(q, scope.SQLFragment())
@@ -324,6 +343,7 @@ DELETE FROM compliance_registries
 WHERE
 	%s
 	AND id = @id
+	AND snapshot_id IS NULL
 `
 
 	q = fmt.Sprintf(q, scope.SQLFragment())
@@ -338,3 +358,64 @@ WHERE
 
 	return nil
 }
+
+func (crs ComplianceRegistries) Snapshot(ctx context.Context, conn pg.Conn, scope Scoper, organizationID, snapshotID gid.GID) error {
+	query := `
+INSERT INTO compliance_registries (
+	id,
+	tenant_id,
+	snapshot_id,
+	source_id,
+	organization_id,
+	reference_id,
+	area,
+	source,
+	requirement,
+	actions_to_be_implemented,
+	regulator,
+	owner_id,
+	last_review_date,
+	due_date,
+	status,
+	created_at,
+	updated_at
+)
+SELECT
+	generate_gid(decode_base64_unpadded(@tenant_id), @compliance_registry_entity_type),
+	@tenant_id,
+	@snapshot_id,
+	r.id,
+	r.organization_id,
+	r.reference_id,
+	r.area,
+	r.source,
+	r.requirement,
+	r.actions_to_be_implemented,
+	r.regulator,
+	r.owner_id,
+	r.last_review_date,
+	r.due_date,
+	r.status,
+	r.created_at,
+	r.updated_at
+FROM compliance_registries r
+WHERE %s AND r.organization_id = @organization_id AND r.snapshot_id IS NULL
+	`
+
+	query = fmt.Sprintf(query, scope.SQLFragment())
+
+	args := pgx.StrictNamedArgs{
+		"tenant_id":                       scope.GetTenantID(),
+		"snapshot_id":                     snapshotID,
+		"organization_id":                 organizationID,
+		"compliance_registry_entity_type": ComplianceRegistryEntityType,
+	}
+	maps.Copy(args, scope.SQLArguments())
+
+	_, err := conn.Exec(ctx, query, args)
+	if err != nil {
+		return fmt.Errorf("cannot insert compliance registry snapshots: %w", err)
+	}
+
+	return nil
+}

pkg/coredata/compliance_registry_filter.go

@@ -0,0 +1,54 @@
+// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
+//
+// Permission to use, copy, modify, and/or distribute this software for any
+// purpose with or without fee is hereby granted, provided that the above
+// copyright notice and this permission notice appear in all copies.
+//
+// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+// PERFORMANCE OF THIS SOFTWARE.
+
+package coredata
+
+import (
+	"github.com/getprobo/probo/pkg/gid"
+	"github.com/jackc/pgx/v5"
+)
+
+type (
+	ComplianceRegistryFilter struct {
+		snapshotID **gid.GID
+	}
+)
+
+func NewComplianceRegistryFilter(snapshotID **gid.GID) *ComplianceRegistryFilter {
+	return &ComplianceRegistryFilter{
+		snapshotID: snapshotID,
+	}
+}
+
+func (f *ComplianceRegistryFilter) SQLArguments() pgx.NamedArgs {
+	args := pgx.NamedArgs{}
+
+	if f.snapshotID != nil && *f.snapshotID != nil {
+		args["filter_snapshot_id"] = **f.snapshotID
+	}
+
+	return args
+}
+
+func (f *ComplianceRegistryFilter) SQLFragment() string {
+	if f.snapshotID == nil {
+		return "TRUE"
+	}
+
+	if *f.snapshotID == nil {
+		return "snapshot_id IS NULL"
+	} else {
+		return "snapshot_id = @filter_snapshot_id"
+	}
+}

pkg/coredata/datum_filter.go

@@ -25,13 +25,7 @@ type (
 	}
 )
 
-func NewDatumFilter() *DatumFilter {
-	return &DatumFilter{
-		snapshotID: nil,
-	}
-}
-
-func NewDatumFilterBySnapshotID(snapshotID **gid.GID) *DatumFilter {
+func NewDatumFilter(snapshotID **gid.GID) *DatumFilter {
 	return &DatumFilter{
 		snapshotID: snapshotID,
 	}

pkg/coredata/migrations/20250828T201559Z.sql

@@ -0,0 +1,11 @@
+ALTER TABLE compliance_registries ADD COLUMN snapshot_id TEXT;
+ALTER TABLE compliance_registries ADD COLUMN source_id TEXT;
+
+ALTER TABLE compliance_registries ADD CONSTRAINT compliance_registries_snapshot_id_fkey
+    FOREIGN KEY (snapshot_id)
+    REFERENCES snapshots(id)
+    ON UPDATE CASCADE
+    ON DELETE CASCADE;
+
+ALTER TABLE compliance_registries ADD CONSTRAINT compliance_registries_source_id_snapshot_id_key
+    UNIQUE (source_id, snapshot_id);

pkg/coredata/nonconformity_registry_filter.go

@@ -25,18 +25,12 @@ type (
 	}
 )
 
-func NewNonconformityRegistryFilter() *NonconformityRegistryFilter {
+func NewNonconformityRegistryFilter(snapshotID **gid.GID) *NonconformityRegistryFilter {
 	return &NonconformityRegistryFilter{
 		snapshotID: nil,
 	}
 }
 
-func NewNonconformityRegistryFilterBySnapshotID(snapshotID **gid.GID) *NonconformityRegistryFilter {
-	return &NonconformityRegistryFilter{
-		snapshotID: snapshotID,
-	}
-}
-
 func (f *NonconformityRegistryFilter) SQLArguments() pgx.NamedArgs {
 	args := pgx.NamedArgs{}
 

pkg/coredata/snapshottable.go

@@ -32,6 +32,8 @@ func GetSnapshottable(snapshotType SnapshotsType) (Snapshottable, error) {
 		return Data{}, nil
 	case SnapshotsTypeNonConformityRegistries:
 		return NonconformityRegistries{}, nil
+	case SnapshotsTypeComplianceRegistries:
+		return ComplianceRegistries{}, nil
 	default:
 		return nil, fmt.Errorf("unsupported snapshot type: %s", snapshotType)
 	}

pkg/coredata/vendor_filter.go

@@ -26,20 +26,10 @@ type (
 	}
 )
 
-func NewVendorFilter() *VendorFilter {
-	return &VendorFilter{}
-}
-
-func NewVendorTrustCenterFilter() *VendorFilter {
-	showOnTrustCenter := true
+func NewVendorFilter(snapshotID **gid.GID, showOnTrustCenter *bool) *VendorFilter {
 	return &VendorFilter{
-		showOnTrustCenter: &showOnTrustCenter,
-	}
-}
-
-func NewVendorFilterBySnapshotID(snapshotID **gid.GID) *VendorFilter {
-	return &VendorFilter{
-		snapshotID: snapshotID,
+		snapshotID:        snapshotID,
+		showOnTrustCenter: showOnTrustCenter,
 	}
 }