147 lines
3.7 KiB
Go
147 lines
3.7 KiB
Go
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())
|
|
|
|
// 兼容历史数据：如果已存在（可能是停用状态）则直接恢复并更新，避免 UNIQUE(appId) 冲突
|
|
old, err := this.Query(tx).
|
|
Attr("appId", appDbId).
|
|
Find()
|
|
if err != nil {
|
|
return "", 0, err
|
|
}
|
|
if old != nil {
|
|
oldSecret := old.(*HTTPDNSAppSecret)
|
|
_, err = this.Query(tx).
|
|
Pk(oldSecret.Id).
|
|
Set("signEnabled", signEnabled).
|
|
Set("signSecret", signSecret).
|
|
Set("signUpdatedAt", now).
|
|
Set("updatedAt", now).
|
|
Set("state", HTTPDNSAppSecretStateEnabled).
|
|
Update()
|
|
return signSecret, now, err
|
|
}
|
|
|
|
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
|
|
}
|