feat: add a option to select a acme user when obtaining a certificate.

This commit is contained in:
Jacky 2024-04-30 11:16:30 +08:00
parent e77d37bbaa
commit f0dcd67004
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D
8 changed files with 186 additions and 2 deletions

View file

@ -36,6 +36,7 @@ func newAcmeUser(db *gorm.DB, opts ...gen.DOOption) acmeUser {
_acmeUser.Email = field.NewString(tableName, "email")
_acmeUser.CADir = field.NewString(tableName, "ca_dir")
_acmeUser.Registration = field.NewField(tableName, "registration")
_acmeUser.Key = field.NewField(tableName, "key")
_acmeUser.fillFieldMap()
@ -54,6 +55,7 @@ type acmeUser struct {
Email field.String
CADir field.String
Registration field.Field
Key field.Field
fieldMap map[string]field.Expr
}
@ -78,6 +80,7 @@ func (a *acmeUser) updateTableName(table string) *acmeUser {
a.Email = field.NewString(table, "email")
a.CADir = field.NewString(table, "ca_dir")
a.Registration = field.NewField(table, "registration")
a.Key = field.NewField(table, "key")
a.fillFieldMap()
@ -94,7 +97,7 @@ func (a *acmeUser) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
}
func (a *acmeUser) fillFieldMap() {
a.fieldMap = make(map[string]field.Expr, 8)
a.fieldMap = make(map[string]field.Expr, 9)
a.fieldMap["id"] = a.ID
a.fieldMap["created_at"] = a.CreatedAt
a.fieldMap["updated_at"] = a.UpdatedAt
@ -103,6 +106,7 @@ func (a *acmeUser) fillFieldMap() {
a.fieldMap["email"] = a.Email
a.fieldMap["ca_dir"] = a.CADir
a.fieldMap["registration"] = a.Registration
a.fieldMap["key"] = a.Key
}
func (a acmeUser) clone(db *gorm.DB) acmeUser {

View file

@ -40,6 +40,7 @@ func newCert(db *gorm.DB, opts ...gen.DOOption) cert {
_cert.AutoCert = field.NewInt(tableName, "auto_cert")
_cert.ChallengeMethod = field.NewString(tableName, "challenge_method")
_cert.DnsCredentialID = field.NewInt(tableName, "dns_credential_id")
_cert.ACMEUserID = field.NewInt(tableName, "acme_user_id")
_cert.KeyType = field.NewString(tableName, "key_type")
_cert.Log = field.NewString(tableName, "log")
_cert.DnsCredential = certBelongsToDnsCredential{
@ -48,6 +49,12 @@ func newCert(db *gorm.DB, opts ...gen.DOOption) cert {
RelationField: field.NewRelation("DnsCredential", "model.DnsCredential"),
}
_cert.ACMEUser = certBelongsToACMEUser{
db: db.Session(&gorm.Session{}),
RelationField: field.NewRelation("ACMEUser", "model.AcmeUser"),
}
_cert.fillFieldMap()
return _cert
@ -69,10 +76,13 @@ type cert struct {
AutoCert field.Int
ChallengeMethod field.String
DnsCredentialID field.Int
ACMEUserID field.Int
KeyType field.String
Log field.String
DnsCredential certBelongsToDnsCredential
ACMEUser certBelongsToACMEUser
fieldMap map[string]field.Expr
}
@ -100,6 +110,7 @@ func (c *cert) updateTableName(table string) *cert {
c.AutoCert = field.NewInt(table, "auto_cert")
c.ChallengeMethod = field.NewString(table, "challenge_method")
c.DnsCredentialID = field.NewInt(table, "dns_credential_id")
c.ACMEUserID = field.NewInt(table, "acme_user_id")
c.KeyType = field.NewString(table, "key_type")
c.Log = field.NewString(table, "log")
@ -118,7 +129,7 @@ func (c *cert) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
}
func (c *cert) fillFieldMap() {
c.fieldMap = make(map[string]field.Expr, 15)
c.fieldMap = make(map[string]field.Expr, 17)
c.fieldMap["id"] = c.ID
c.fieldMap["created_at"] = c.CreatedAt
c.fieldMap["updated_at"] = c.UpdatedAt
@ -131,6 +142,7 @@ func (c *cert) fillFieldMap() {
c.fieldMap["auto_cert"] = c.AutoCert
c.fieldMap["challenge_method"] = c.ChallengeMethod
c.fieldMap["dns_credential_id"] = c.DnsCredentialID
c.fieldMap["acme_user_id"] = c.ACMEUserID
c.fieldMap["key_type"] = c.KeyType
c.fieldMap["log"] = c.Log
@ -217,6 +229,77 @@ func (a certBelongsToDnsCredentialTx) Count() int64 {
return a.tx.Count()
}
type certBelongsToACMEUser struct {
db *gorm.DB
field.RelationField
}
func (a certBelongsToACMEUser) Where(conds ...field.Expr) *certBelongsToACMEUser {
if len(conds) == 0 {
return &a
}
exprs := make([]clause.Expression, 0, len(conds))
for _, cond := range conds {
exprs = append(exprs, cond.BeCond().(clause.Expression))
}
a.db = a.db.Clauses(clause.Where{Exprs: exprs})
return &a
}
func (a certBelongsToACMEUser) WithContext(ctx context.Context) *certBelongsToACMEUser {
a.db = a.db.WithContext(ctx)
return &a
}
func (a certBelongsToACMEUser) Session(session *gorm.Session) *certBelongsToACMEUser {
a.db = a.db.Session(session)
return &a
}
func (a certBelongsToACMEUser) Model(m *model.Cert) *certBelongsToACMEUserTx {
return &certBelongsToACMEUserTx{a.db.Model(m).Association(a.Name())}
}
type certBelongsToACMEUserTx struct{ tx *gorm.Association }
func (a certBelongsToACMEUserTx) Find() (result *model.AcmeUser, err error) {
return result, a.tx.Find(&result)
}
func (a certBelongsToACMEUserTx) Append(values ...*model.AcmeUser) (err error) {
targetValues := make([]interface{}, len(values))
for i, v := range values {
targetValues[i] = v
}
return a.tx.Append(targetValues...)
}
func (a certBelongsToACMEUserTx) Replace(values ...*model.AcmeUser) (err error) {
targetValues := make([]interface{}, len(values))
for i, v := range values {
targetValues[i] = v
}
return a.tx.Replace(targetValues...)
}
func (a certBelongsToACMEUserTx) Delete(values ...*model.AcmeUser) (err error) {
targetValues := make([]interface{}, len(values))
for i, v := range values {
targetValues[i] = v
}
return a.tx.Delete(targetValues...)
}
func (a certBelongsToACMEUserTx) Clear() error {
return a.tx.Clear()
}
func (a certBelongsToACMEUserTx) Count() int64 {
return a.tx.Count()
}
type certDo struct{ gen.DO }
// FirstByID Where("id=@id")