管理端全部功能跑通

2026-02-27 10:35:22 +08:00
parent 4d275c921d
commit 150799f41d
263 changed files with 22664 additions and 4053 deletions
--- a/EdgeAPI/internal/db/models/httpdns_app_secret_dao.go
+++ b/EdgeAPI/internal/db/models/httpdns_app_secret_dao.go
@@ -0,0 +1,125 @@
+package models
+
+import (
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/rands"
+	"github.com/iwind/TeaGo/types"
+	"time"
+)
+
+const (
+	HTTPDNSAppSecretStateEnabled  = 1
+	HTTPDNSAppSecretStateDisabled = 0
+)
+
+type HTTPDNSAppSecretDAO dbs.DAO
+
+func NewHTTPDNSAppSecretDAO() *HTTPDNSAppSecretDAO {
+	return dbs.NewDAO(&HTTPDNSAppSecretDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeHTTPDNSAppSecrets",
+			Model:  new(HTTPDNSAppSecret),
+			PkName: "id",
+		},
+	}).(*HTTPDNSAppSecretDAO)
+}
+
+var SharedHTTPDNSAppSecretDAO *HTTPDNSAppSecretDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedHTTPDNSAppSecretDAO = NewHTTPDNSAppSecretDAO()
+	})
+}
+
+func (this *HTTPDNSAppSecretDAO) InitAppSecret(tx *dbs.Tx, appDbId int64, signEnabled bool) (string, uint64, error) {
+	signSecret := "ss_" + rands.HexString(12)
+	now := uint64(time.Now().Unix())
+	var op = NewHTTPDNSAppSecretOperator()
+	op.AppId = appDbId
+	op.SignEnabled = signEnabled
+	op.SignSecret = signSecret
+	op.SignUpdatedAt = now
+	op.UpdatedAt = now
+	op.State = HTTPDNSAppSecretStateEnabled
+	err := this.Save(tx, op)
+	return signSecret, now, err
+}
+
+func (this *HTTPDNSAppSecretDAO) FindEnabledAppSecret(tx *dbs.Tx, appDbId int64) (*HTTPDNSAppSecret, error) {
+	one, err := this.Query(tx).
+		Attr("appId", appDbId).
+		State(HTTPDNSAppSecretStateEnabled).
+		Find()
+	if one == nil {
+		return nil, err
+	}
+	return one.(*HTTPDNSAppSecret), nil
+}
+
+func (this *HTTPDNSAppSecretDAO) UpdateSignEnabled(tx *dbs.Tx, appDbId int64, signEnabled bool) error {
+	_, err := this.Query(tx).
+		Attr("appId", appDbId).
+		State(HTTPDNSAppSecretStateEnabled).
+		Set("signEnabled", signEnabled).
+		Set("updatedAt", time.Now().Unix()).
+		Update()
+	return err
+}
+
+func (this *HTTPDNSAppSecretDAO) ResetSignSecret(tx *dbs.Tx, appDbId int64) (string, int64, error) {
+	signSecret := "ss_" + rands.HexString(12)
+	now := time.Now().Unix()
+	_, err := this.Query(tx).
+		Attr("appId", appDbId).
+		State(HTTPDNSAppSecretStateEnabled).
+		Set("signSecret", signSecret).
+		Set("signUpdatedAt", now).
+		Set("updatedAt", now).
+		Update()
+	if err != nil {
+		return "", 0, err
+	}
+	return signSecret, now, nil
+}
+
+func (this *HTTPDNSAppSecretDAO) FindSignEnabled(tx *dbs.Tx, appDbId int64) (bool, error) {
+	one, err := this.FindEnabledAppSecret(tx, appDbId)
+	if err != nil || one == nil {
+		return false, err
+	}
+	return one.SignEnabled, nil
+}
+
+func (this *HTTPDNSAppSecretDAO) FindSignSecretWithAppDbId(tx *dbs.Tx, appDbId int64) (string, error) {
+	return this.Query(tx).
+		Attr("appId", appDbId).
+		State(HTTPDNSAppSecretStateEnabled).
+		Result("signSecret").
+		FindStringCol("")
+}
+
+func (this *HTTPDNSAppSecretDAO) FindSignUpdatedAt(tx *dbs.Tx, appDbId int64) (int64, error) {
+	col, err := this.Query(tx).
+		Attr("appId", appDbId).
+		State(HTTPDNSAppSecretStateEnabled).
+		Result("signUpdatedAt").
+		FindCol(nil)
+	if err != nil {
+		return 0, err
+	}
+	return types.Int64(col), nil
+}
+
+func (this *HTTPDNSAppSecretDAO) DisableAppSecret(tx *dbs.Tx, appDbId int64) error {
+	_, err := this.Query(tx).
+		Attr("appId", appDbId).
+		State(HTTPDNSAppSecretStateEnabled).
+		Set("state", HTTPDNSAppSecretStateDisabled).
+		Set("updatedAt", time.Now().Unix()).
+		Update()
+	return err
+}