Add new models

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

pkg/coredata/cookie_banner_state.go

@@ -0,0 +1,75 @@
+// 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 (
+	"database/sql/driver"
+	"fmt"
+)
+
+type CookieBannerState string
+
+const (
+	CookieBannerStateDraft     CookieBannerState = "DRAFT"
+	CookieBannerStatePublished CookieBannerState = "PUBLISHED"
+	CookieBannerStateDisabled  CookieBannerState = "DISABLED"
+)
+
+func CookieBannerStates() []CookieBannerState {
+	return []CookieBannerState{
+		CookieBannerStateDraft,
+		CookieBannerStatePublished,
+		CookieBannerStateDisabled,
+	}
+}
+
+func (s CookieBannerState) String() string {
+	return string(s)
+}
+
+func (s *CookieBannerState) Scan(value any) error {
+	var v string
+	switch val := value.(type) {
+	case string:
+		v = val
+	case []byte:
+		v = string(val)
+	default:
+		return fmt.Errorf("unsupported type for CookieBannerState: %T", value)
+	}
+
+	switch CookieBannerState(v) {
+	case CookieBannerStateDraft:
+		*s = CookieBannerStateDraft
+	case CookieBannerStatePublished:
+		*s = CookieBannerStatePublished
+	case CookieBannerStateDisabled:
+		*s = CookieBannerStateDisabled
+	default:
+		return fmt.Errorf("invalid CookieBannerState value: %q", v)
+	}
+	return nil
+}
+
+func (s CookieBannerState) Value() (driver.Value, error) {
+	switch s {
+	case CookieBannerStateDraft,
+		CookieBannerStatePublished,
+		CookieBannerStateDisabled:
+		return string(s), nil
+	default:
+		return nil, fmt.Errorf("invalid CookieBannerState: %s", s)
+	}
+}