mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
63cd906cfc
- Merge all files inside openflare-server to the repository root directory. - Relocate agent, relay, and flared subprojects from internal/ to internal/apps/. - Combine docker-compose files and update build context paths to root. - Update GitHub workflows and Dockerfiles to refer to new directories and package names. - Rewrite Go package imports across all files. - Resolve database renew test race condition and clean up docs.
83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package model
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/db"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// AcmeAccount OpenFlare ACME 账号实体。
|
|
type AcmeAccount struct {
|
|
ID uint `json:"id" gorm:"primaryKey;autoIncrement"`
|
|
Email string `json:"email" gorm:"size:255"`
|
|
URL string `json:"url" gorm:"size:255"`
|
|
PrivateKey string `json:"-" gorm:"type:text;not null"`
|
|
CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"`
|
|
UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"`
|
|
}
|
|
|
|
// TableName 表名。
|
|
func (AcmeAccount) TableName() string {
|
|
return "of_acme_accounts"
|
|
}
|
|
|
|
// GetAcmeAccountByID 按 ID 查询 ACME 账号。
|
|
func GetAcmeAccountByID(ctx context.Context, id uint) (*AcmeAccount, error) {
|
|
conn := db.DB(ctx)
|
|
if conn == nil {
|
|
return nil, errors.New(errDatabaseNotInitialized)
|
|
}
|
|
var account AcmeAccount
|
|
if err := conn.First(&account, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &account, nil
|
|
}
|
|
|
|
// CreateAcmeAccountRecord 创建 ACME 账号。
|
|
func CreateAcmeAccountRecord(ctx context.Context, account *AcmeAccount) error {
|
|
conn := db.DB(ctx)
|
|
if conn == nil {
|
|
return errors.New(errDatabaseNotInitialized)
|
|
}
|
|
return conn.Create(account).Error
|
|
}
|
|
|
|
// SaveAcmeAccount 保存 ACME 账号。
|
|
func SaveAcmeAccount(ctx context.Context, account *AcmeAccount) error {
|
|
conn := db.DB(ctx)
|
|
if conn == nil {
|
|
return errors.New(errDatabaseNotInitialized)
|
|
}
|
|
return conn.Save(account).Error
|
|
}
|
|
|
|
// GetDefaultAcmeAccount 获取默认 ACME 账号，不存在时创建占位记录。
|
|
func GetDefaultAcmeAccount(ctx context.Context) (*AcmeAccount, error) {
|
|
conn := db.DB(ctx)
|
|
if conn == nil {
|
|
return nil, errors.New(errDatabaseNotInitialized)
|
|
}
|
|
var account AcmeAccount
|
|
err := conn.Order("id asc").First(&account).Error
|
|
if err == nil {
|
|
return &account, nil
|
|
}
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, err
|
|
}
|
|
account = AcmeAccount{
|
|
Email: "admin@openflare.dev",
|
|
}
|
|
if err = conn.Create(&account).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &account, nil
|
|
}
|