重构项目

pkg/core/alias.go

@@ -2,6 +2,7 @@ package core
 
 import (
 	"encoding/json"
+	"fmt"
 
 	"gopkg.d7z.net/gitea-pages/pkg/utils"
 )
@@ -13,10 +14,10 @@ type Alias struct {
 }
 
 type DomainAlias struct {
-	config utils.Config
+	config utils.KVConfig
 }
 
-func NewDomainAlias(config utils.Config) *DomainAlias {
+func NewDomainAlias(config utils.KVConfig) *DomainAlias {
 	return &DomainAlias{config: config}
 }
 
@@ -32,17 +33,34 @@ func (a *DomainAlias) Query(domain string) (*Alias, error) {
 	return rel, nil
 }
 
-func (a *DomainAlias) Bind(domain, owner, repo, branch string) error {
-	save := &Alias{
+func (a *DomainAlias) Bind(domains []string, owner, repo, branch string) error {
+	oldDomains := make([]string, 0)
+	rKey := fmt.Sprintf("domain/r-alias/%s/%s/%s", owner, repo, branch)
+	if oldStr, err := a.config.Get(rKey); err == nil {
+		_ = json.Unmarshal([]byte(oldStr), &oldDomains)
+	}
+	for _, oldDomain := range oldDomains {
+		if err := a.Unbind(oldDomain); err != nil {
+			return err
+		}
+	}
+	if domains == nil || len(domains) == 0 {
+		return nil
+	}
+	aliasMeta := &Alias{
 		Owner:  owner,
 		Repo:   repo,
 		Branch: branch,
 	}
-	saveB, err := json.Marshal(save)
-	if err != nil {
-		return err
+	aliasMetaRaw, _ := json.Marshal(aliasMeta)
+	domainsRaw, _ := json.Marshal(domains)
+	_ = a.config.Put(rKey, string(domainsRaw), utils.TtlKeep)
+	for _, domain := range domains {
+		if err := a.config.Put("domain/alias/"+domain, string(aliasMetaRaw), utils.TtlKeep); err != nil {
+			return err
+		}
 	}
-	return a.config.Put("domain/alias/"+domain, string(saveB), utils.TtlKeep)
+	return nil
 }
 
 func (a *DomainAlias) Unbind(domain string) error {