feat(zone): add normalized zone domain schema

internal/model/openflare_zone.go

@@ -0,0 +1,115 @@
+// Copyright 2026 Arctel.net
+// SPDX-License-Identifier: Apache-2.0
+
+package model
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+
+	"github.com/Rain-kl/Wavelet/internal/db"
+	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
+)
+
+const (
+	tableOfZones       = "of_zones"
+	tableOfZoneDomains = "of_zone_domains"
+)
+
+var errZoneDomainBoundToAnotherRoute = errors.New("zone domain is already bound to another proxy route")
+
+// Zone OpenFlare 注册根域实体。
+type Zone struct {
+	ID        uint      `json:"id" gorm:"primaryKey;autoIncrement"`
+	Domain    string    `json:"domain" gorm:"uniqueIndex:idx_of_zones_domain;size:255;not null"`
+	Remark    string    `json:"remark" gorm:"size:255;not null;default:''"`
+	CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"`
+	UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"`
+}
+
+// TableName 表名。
+func (Zone) TableName() string {
+	return tableOfZones
+}
+
+// ZoneDomain OpenFlare Zone 下的明确域名实体。
+type ZoneDomain struct {
+	ID           uint      `json:"id" gorm:"primaryKey;autoIncrement"`
+	ZoneID       uint      `json:"zone_id" gorm:"not null;index:idx_of_zone_domains_zone_id"`
+	ProxyRouteID *uint     `json:"proxy_route_id" gorm:"index:idx_of_zone_domains_proxy_route_id"`
+	Domain       string    `json:"domain" gorm:"uniqueIndex:idx_of_zone_domains_domain;size:255;not null"`
+	CertID       *uint     `json:"cert_id" gorm:"index:idx_of_zone_domains_cert_id"`
+	Remark       string    `json:"remark" gorm:"size:255;not null;default:''"`
+	CreatedAt    time.Time `json:"created_at" gorm:"autoCreateTime"`
+	UpdatedAt    time.Time `json:"updated_at" gorm:"autoUpdateTime"`
+}
+
+// TableName 表名。
+func (ZoneDomain) TableName() string {
+	return tableOfZoneDomains
+}
+
+// ListZoneDomainsByRouteID returns the domains bound to a proxy route.
+func ListZoneDomainsByRouteID(ctx context.Context, routeID uint) ([]ZoneDomain, error) {
+	var domains []ZoneDomain
+	if err := db.DB(ctx).Where("proxy_route_id = ?", routeID).Order("id asc").Find(&domains).Error; err != nil {
+		return nil, err
+	}
+	return domains, nil
+}
+
+// ReplaceZoneDomainRouteBindings replaces every ZoneDomain binding for a proxy route.
+func ReplaceZoneDomainRouteBindings(ctx context.Context, routeID uint, domainIDs []uint) error {
+	conn := db.DB(ctx)
+	if conn == nil {
+		return errors.New("database is not initialized")
+	}
+
+	return conn.Transaction(func(tx *gorm.DB) error {
+		var requested []ZoneDomain
+		if len(domainIDs) > 0 {
+			if err := tx.Clauses(clause.Locking{Strength: "UPDATE"}).
+				Where("id IN ?", domainIDs).
+				Find(&requested).Error; err != nil {
+				return err
+			}
+			if len(requested) != len(uniqueZoneDomainIDs(domainIDs)) {
+				return fmt.Errorf("one or more zone domains do not exist")
+			}
+			for _, domain := range requested {
+				if domain.ProxyRouteID != nil && *domain.ProxyRouteID != routeID {
+					return errZoneDomainBoundToAnotherRoute
+				}
+			}
+		}
+
+		current := tx.Model(&ZoneDomain{}).Where("proxy_route_id = ?", routeID)
+		if len(domainIDs) > 0 {
+			current = current.Where("id NOT IN ?", domainIDs)
+		}
+		if err := current.Update("proxy_route_id", nil).Error; err != nil {
+			return err
+		}
+
+		if len(domainIDs) == 0 {
+			return nil
+		}
+		return tx.Model(&ZoneDomain{}).Where("id IN ?", domainIDs).Update("proxy_route_id", routeID).Error
+	})
+}
+
+func uniqueZoneDomainIDs(domainIDs []uint) []uint {
+	seen := make(map[uint]struct{}, len(domainIDs))
+	ids := make([]uint, 0, len(domainIDs))
+	for _, id := range domainIDs {
+		if _, ok := seen[id]; ok {
+			continue
+		}
+		seen[id] = struct{}{}
+		ids = append(ids, id)
+	}
+	return ids
+}