rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add consent record operations and rename ConsentRecord to CookieConsentRecord

Browse Source 
Signed-off-by: Émile Ré <emile@getprobo.com>
This commit is contained in:
Émile Ré
2026-04-10 17:31:59 +04:00
parent 327652163f
commit c594b10e9f
3 changed files with 125 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pkg/coredata/cookie_consent_record.go
View File

@@ -29,7 +29,7 @@ import (
)
type (
ConsentRecord struct {
CookieConsentRecord struct {
ID gid.GID `db:"id"`
CookieBannerID gid.GID `db:"cookie_banner_id"`
VisitorID string `db:"visitor_id"`
@@ -40,10 +40,10 @@ type (
CreatedAt time.Time `db:"created_at"`
}
ConsentRecords []*ConsentRecord
CookieConsentRecords []*CookieConsentRecord
)
func (r *ConsentRecord) CursorKey(field CookieConsentRecordOrderField) page.CursorKey {
func (r *CookieConsentRecord) CursorKey(field CookieConsentRecordOrderField) page.CursorKey {
switch field {
case CookieConsentRecordOrderFieldCreatedAt:
return page.NewCursorKey(r.ID, r.CreatedAt)
@@ -52,7 +52,7 @@ func (r *ConsentRecord) CursorKey(field CookieConsentRecordOrderField) page.Curs
panic(fmt.Sprintf("unsupported order by: %s", field))
}
func (r *ConsentRecord) AuthorizationAttributes(ctx context.Context, conn pg.Querier) (map[string]string, error) {
func (r *CookieConsentRecord) AuthorizationAttributes(ctx context.Context, conn pg.Querier) (map[string]string, error) {
q := `
SELECT cb.organization_id
FROM cookie_consent_records cr
@@ -72,7 +72,7 @@ LIMIT 1;
return map[string]string{"organization_id": organizationID.String()}, nil
}
func (r *ConsentRecords) LoadByCookieBannerID(
func (r *CookieConsentRecords) LoadByCookieBannerID(
ctx context.Context,
conn pg.Querier,
scope Scoper,
@@ -111,7 +111,7 @@ WHERE
return fmt.Errorf("cannot query consent records: %w", err)
}
records, err := pgx.CollectRows(rows, pgx.RowToAddrOfStructByName[ConsentRecord])
records, err := pgx.CollectRows(rows, pgx.RowToAddrOfStructByName[CookieConsentRecord])
if err != nil {
return fmt.Errorf("cannot collect consent records: %w", err)
}
@@ -121,7 +121,7 @@ WHERE
return nil
}
func (r *ConsentRecords) CountByCookieBannerID(
func (r *CookieConsentRecords) CountByCookieBannerID(
ctx context.Context,
conn pg.Querier,
scope Scoper,
@@ -155,7 +155,7 @@ WHERE
return count, nil
}
func (r *ConsentRecord) Insert(
func (r *CookieConsentRecord) Insert(
ctx context.Context,
tx pg.Tx,
scope Scoper,

6
pkg/coredata/entity_type_reg.go
View File

@@ -100,7 +100,7 @@ const (
AccessEntryDecisionHistoryEntityType uint16 = 74
CookieBannerEntityType uint16 = 75
CookieCategoryEntityType uint16 = 76
ConsentRecordEntityType uint16 = 77
CookieConsentRecordEntityType uint16 = 77
)
func NewEntityFromID(id gid.GID) (any, bool) {
@@ -251,8 +251,8 @@ func NewEntityFromID(id gid.GID) (any, bool) {
return &CookieBanner{ID: id}, true
case CookieCategoryEntityType:
return &CookieCategory{ID: id}, true
case ConsentRecordEntityType:
return &ConsentRecord{ID: id}, true
case CookieConsentRecordEntityType:
return &CookieConsentRecord{ID: id}, true
default:
return nil, false
}