Add new models

Signed-off-by: Émile Ré <emile@getprobo.com>

pkg/coredata/cookie_consent_record.go

@@ -0,0 +1,205 @@
+// Copyright (c) 2025-2026 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 (
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"maps"
+	"time"
+
+	"github.com/jackc/pgx/v5"
+	"go.gearno.de/kit/pg"
+	"go.probo.inc/probo/pkg/gid"
+	"go.probo.inc/probo/pkg/page"
+)
+
+type (
+	ConsentRecord struct {
+		ID             gid.GID             `db:"id"`
+		CookieBannerID gid.GID             `db:"cookie_banner_id"`
+		VisitorID      string              `db:"visitor_id"`
+		IPAddress      *string             `db:"ip_address"`
+		UserAgent      *string             `db:"user_agent"`
+		ConsentData    json.RawMessage     `db:"consent_data"`
+		Action         CookieConsentAction `db:"action"`
+		CreatedAt      time.Time           `db:"created_at"`
+	}
+
+	ConsentRecords []*ConsentRecord
+)
+
+func (r *ConsentRecord) CursorKey(field CookieConsentRecordOrderField) page.CursorKey {
+	switch field {
+	case CookieConsentRecordOrderFieldCreatedAt:
+		return page.NewCursorKey(r.ID, r.CreatedAt)
+	}
+
+	panic(fmt.Sprintf("unsupported order by: %s", field))
+}
+
+func (r *ConsentRecord) AuthorizationAttributes(ctx context.Context, conn pg.Querier) (map[string]string, error) {
+	q := `
+SELECT cb.organization_id
+FROM cookie_consent_records cr
+JOIN cookie_banners cb ON cr.cookie_banner_id = cb.id
+WHERE cr.id = $1
+LIMIT 1;
+`
+
+	var organizationID gid.GID
+	if err := conn.QueryRow(ctx, q, r.ID).Scan(&organizationID); err != nil {
+		if errors.Is(err, pgx.ErrNoRows) {
+			return nil, ErrResourceNotFound
+		}
+		return nil, fmt.Errorf("cannot query consent record authorization attributes: %w", err)
+	}
+
+	return map[string]string{"organization_id": organizationID.String()}, nil
+}
+
+func (r *ConsentRecords) LoadByCookieBannerID(
+	ctx context.Context,
+	conn pg.Querier,
+	scope Scoper,
+	cookieBannerID gid.GID,
+	cursor *page.Cursor[CookieConsentRecordOrderField],
+	filter *CookieConsentRecordFilter,
+) error {
+	q := `
+SELECT
+	id,
+	cookie_banner_id,
+	visitor_id,
+	ip_address,
+	user_agent,
+	consent_data,
+	action,
+	created_at
+FROM
+	cookie_consent_records
+WHERE
+	%s
+	AND cookie_banner_id = @cookie_banner_id
+	AND %s
+	AND %s
+`
+
+	q = fmt.Sprintf(q, scope.SQLFragment(), filter.SQLFragment(), cursor.SQLFragment())
+
+	args := pgx.StrictNamedArgs{"cookie_banner_id": cookieBannerID}
+	maps.Copy(args, scope.SQLArguments())
+	maps.Copy(args, filter.SQLArguments())
+	maps.Copy(args, cursor.SQLArguments())
+
+	rows, err := conn.Query(ctx, q, args)
+	if err != nil {
+		return fmt.Errorf("cannot query consent records: %w", err)
+	}
+
+	records, err := pgx.CollectRows(rows, pgx.RowToAddrOfStructByName[ConsentRecord])
+	if err != nil {
+		return fmt.Errorf("cannot collect consent records: %w", err)
+	}
+
+	*r = records
+
+	return nil
+}
+
+func (r *ConsentRecords) CountByCookieBannerID(
+	ctx context.Context,
+	conn pg.Querier,
+	scope Scoper,
+	cookieBannerID gid.GID,
+	filter *CookieConsentRecordFilter,
+) (int, error) {
+	q := `
+SELECT
+	COUNT(id)
+FROM
+	cookie_consent_records
+WHERE
+	%s
+	AND cookie_banner_id = @cookie_banner_id
+	AND %s
+`
+
+	q = fmt.Sprintf(q, scope.SQLFragment(), filter.SQLFragment())
+
+	args := pgx.StrictNamedArgs{"cookie_banner_id": cookieBannerID}
+	maps.Copy(args, scope.SQLArguments())
+	maps.Copy(args, filter.SQLArguments())
+
+	row := conn.QueryRow(ctx, q, args)
+
+	var count int
+	if err := row.Scan(&count); err != nil {
+		return 0, fmt.Errorf("cannot scan count: %w", err)
+	}
+
+	return count, nil
+}
+
+func (r *ConsentRecord) Insert(
+	ctx context.Context,
+	tx pg.Tx,
+	scope Scoper,
+) error {
+	q := `
+INSERT INTO cookie_consent_records (
+	id,
+	tenant_id,
+	cookie_banner_id,
+	visitor_id,
+	ip_address,
+	user_agent,
+	consent_data,
+	action,
+	created_at
+) VALUES (
+	@id,
+	@tenant_id,
+	@cookie_banner_id,
+	@visitor_id,
+	@ip_address,
+	@user_agent,
+	@consent_data,
+	@action,
+	@created_at
+)
+`
+
+	args := pgx.StrictNamedArgs{
+		"id":               r.ID,
+		"tenant_id":        scope.GetTenantID(),
+		"cookie_banner_id": r.CookieBannerID,
+		"visitor_id":       r.VisitorID,
+		"ip_address":       r.IPAddress,
+		"user_agent":       r.UserAgent,
+		"consent_data":     r.ConsentData,
+		"action":           r.Action,
+		"created_at":       r.CreatedAt,
+	}
+
+	_, err := tx.Exec(ctx, q, args)
+	if err != nil {
+		return fmt.Errorf("cannot insert consent record: %w", err)
+	}
+
+	return nil
+}