优化部分代码

pkg/core/alias.go

@@ -0,0 +1,50 @@
+package core
+
+import (
+	"encoding/json"
+
+	"code.d7z.net/d7z-project/gitea-pages/pkg/utils"
+)
+
+type Alias struct {
+	Owner  string `json:"owner"`
+	Repo   string `json:"repo"`
+	Branch string `json:"branch"`
+}
+
+type DomainAlias struct {
+	config utils.Config
+}
+
+func NewDomainAlias(config utils.Config) *DomainAlias {
+	return &DomainAlias{config: config}
+}
+
+func (a *DomainAlias) Query(domain string) (*Alias, error) {
+	get, err := a.config.Get("alias/" + domain)
+	if err != nil {
+		return nil, err
+	}
+	rel := &Alias{}
+	if err = json.Unmarshal([]byte(get), rel); err != nil {
+		return nil, err
+	}
+	return rel, nil
+}
+
+func (a *DomainAlias) Bind(domain, owner, repo, branch string) error {
+	save := &Alias{
+		Owner:  owner,
+		Repo:   repo,
+		Branch: branch,
+	}
+	saveB, err := json.Marshal(save)
+	if err != nil {
+		return err
+	}
+	return a.config.Put("domain/"+domain, string(saveB), utils.TtlKeep)
+}
+
+func (a *DomainAlias) Unbind(domain string) error {
+	return a.config.Delete("domain/" + domain)
+}