248 lines
9.4 KiB
Go
248 lines
9.4 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"git.pcux.cn/xuans/Scert/backend/internal/admin"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/certificate"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/config"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/database"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/deployment"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/dnsprovider"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/firewall"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/httpapi"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/logging"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/notification"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/repository"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/secretstore"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/setup"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/sshkey"
|
|
)
|
|
|
|
type App struct {
|
|
Config config.Config
|
|
Logger *logging.Logger
|
|
Database *database.Database
|
|
Repository *repository.SQLDomainRepository
|
|
Servers *repository.SQLServerRepository
|
|
Users *repository.SQLUserRepository
|
|
Tasks *repository.SQLTaskRepository
|
|
Deployment *deployment.Service
|
|
SecretStore *secretstore.Store
|
|
SSHKeys *sshkey.Service
|
|
DNSVerification *setup.DNSVerificationManager
|
|
Certificates *certificate.Service
|
|
Workflow *CertificateWorkflow
|
|
Renewal *certificate.RenewalScheduler
|
|
HTTP *httpapi.Server
|
|
maintenanceStop context.CancelFunc
|
|
maintenanceDone chan struct{}
|
|
}
|
|
|
|
const (
|
|
auditLogRetentionDays = 30
|
|
auditCleanupInterval = 24 * time.Hour
|
|
)
|
|
|
|
func New(ctx context.Context, cfg config.Config) (*App, error) {
|
|
return NewWithConfigPath(ctx, cfg, filepath.Join(cfg.Paths.Config, "scert.json"))
|
|
}
|
|
|
|
func NewWithConfigPath(ctx context.Context, cfg config.Config, configPath string) (*App, error) {
|
|
if err := cfg.Validate(); err != nil {
|
|
return nil, err
|
|
}
|
|
for _, path := range []string{cfg.Paths.Install, cfg.Paths.Certificate, cfg.Paths.Config, cfg.Paths.Data, cfg.Paths.Log, cfg.Paths.Backup} {
|
|
if err := makeDirectory(path); err != nil {
|
|
return nil, fmt.Errorf("创建运行目录失败：%w", err)
|
|
}
|
|
}
|
|
logger, err := logging.New(cfg.Paths.Log, slog.LevelInfo)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("初始化日志失败：%w", err)
|
|
}
|
|
db, err := database.Open(ctx, cfg.Database)
|
|
if err != nil {
|
|
_ = logger.Close()
|
|
return nil, err
|
|
}
|
|
if err := database.Migrate(ctx, db.DB, cfg.Database.Driver); err != nil {
|
|
_ = db.Close()
|
|
_ = logger.Close()
|
|
return nil, err
|
|
}
|
|
secretStore, err := secretstore.Open(cfg.Paths.Config)
|
|
if err != nil {
|
|
_ = db.Close()
|
|
_ = logger.Close()
|
|
return nil, fmt.Errorf("初始化敏感信息存储失败：%w", err)
|
|
}
|
|
dnsTester := dnsprovider.NewTester(dnsprovider.NewLegoFactory())
|
|
dnsVerification := setup.NewDNSVerificationManager(dnsTester, secretStore)
|
|
repo := repository.NewSQLDomainRepository(db.DB, cfg.Database.Driver)
|
|
serverRepo := repository.NewSQLServerRepository(db.DB, cfg.Database.Driver)
|
|
userRepo := repository.NewSQLUserRepository(db.DB, cfg.Database.Driver)
|
|
if cfg.SetupReady && cfg.Admin.Username != "" && cfg.Admin.PasswordHash != "" {
|
|
if err := userRepo.EnsureSuperAdmin(ctx, repository.User{ID: "primary-super-admin", Username: cfg.Admin.Username, Email: cfg.Admin.Email, PasswordHash: cfg.Admin.PasswordHash}); err != nil {
|
|
_ = db.Close()
|
|
_ = logger.Close()
|
|
return nil, fmt.Errorf("初始化超级管理员用户失败：%w", err)
|
|
}
|
|
}
|
|
taskRepo := repository.NewSQLTaskRepository(db.DB, cfg.Database.Driver)
|
|
if recovered, err := taskRepo.RecoverInterruptedTasks(ctx, time.Now().UTC()); err != nil {
|
|
_ = db.Close()
|
|
_ = logger.Close()
|
|
return nil, fmt.Errorf("\u6062\u590d\u670d\u52a1\u91cd\u542f\u524d\u672a\u5b8c\u6210\u4efb\u52a1\u5931\u8d25\uff1a%w", err)
|
|
} else if recovered > 0 {
|
|
logger.Warn("\u670d\u52a1\u91cd\u542f\u524d\u672a\u5b8c\u6210\u4efb\u52a1\u5df2\u6807\u8bb0\u4e3a\u5931\u8d25", "count", recovered)
|
|
}
|
|
auditRepo := repository.NewSQLAuditRepository(db.DB, cfg.Database.Driver)
|
|
if deleted, err := auditRepo.PruneExpiredAuditLogs(ctx, time.Now().UTC().AddDate(0, 0, -auditLogRetentionDays)); err != nil {
|
|
logger.Warn("启动时清理过期审计日志失败", "error", err)
|
|
} else if deleted > 0 {
|
|
logger.Info("启动时已清理过期审计日志", "count", deleted)
|
|
}
|
|
sshKeys := sshkey.NewService(secretStore)
|
|
accounts := certificate.NewAccountRepository(secretStore)
|
|
certificates := certificate.NewService(cfg, repo, secretStore, dnsprovider.NewLegoFactory(), accounts)
|
|
aliyunAuthChecker := dnsprovider.NewAliyunAuthorizationChecker()
|
|
certificates.SetDNSAuthorizationChecker(aliyunAuthChecker)
|
|
deploymentService := deployment.NewService(deployment.SSHFactory{Secrets: secretStore})
|
|
notifier := notification.New(cfg.Notification, secretStore)
|
|
remoteValidator := certificate.RemoteValidator{}
|
|
workflow := &CertificateWorkflow{
|
|
Certificates: certificates,
|
|
Domains: repo,
|
|
Servers: serverRepo,
|
|
Deployment: deploymentService,
|
|
RemoteVerifier: func(ctx context.Context, endpoint string, expectedDomains []string) (certificate.ValidationResult, error) {
|
|
return remoteValidator.ValidateHTTPS(ctx, endpoint, expectedDomains)
|
|
},
|
|
Notifier: notifier,
|
|
NotifyError: func(err error) { logger.Error("发送邮件通知失败", "error", err) },
|
|
CurrentCertificate: certificates.Current,
|
|
}
|
|
renewal := certificate.NewRenewalSchedulerWithPolicy(repo, workflow, certificate.RenewalPolicy{
|
|
CheckInterval: time.Duration(cfg.ACME.RenewalCheckIntervalDays) * 24 * time.Hour,
|
|
NearExpiryThreshold: time.Duration(cfg.ACME.RenewalNearExpiryDays) * 24 * time.Hour,
|
|
RenewalThreshold: time.Duration(cfg.ACME.RenewalThresholdDays) * 24 * time.Hour,
|
|
UrgentThreshold: time.Duration(cfg.ACME.RenewalUrgentThresholdDays) * 24 * time.Hour,
|
|
MaxRetries: cfg.ACME.RenewalMaxRetries,
|
|
RetryInterval: time.Duration(cfg.ACME.RenewalRetryIntervalMins) * time.Minute,
|
|
})
|
|
server := httpapi.New(httpapi.Dependencies{
|
|
Config: cfg, ConfigPath: configPath, Logger: logger, Repository: repo, Servers: serverRepo, Users: userRepo, SSHKeys: sshKeys,
|
|
SetupDNS: dnsVerification, Secrets: secretStore, Firewall: firewall.NewManager(), AliyunChecker: aliyunAuthChecker,
|
|
IssueCertificate: workflow.IssueForHTTP,
|
|
IssueSetupCertificate: func(ctx context.Context, request certificate.SetupCertificateRequest) (certificate.SavedResource, error) {
|
|
return certificates.IssueForSetup(ctx, request)
|
|
},
|
|
RollbackSetupCertificate: func(ctx context.Context, saved certificate.SavedResource) error {
|
|
return certificate.RollbackSavedResource(ctx, saved)
|
|
},
|
|
ImportCertificate: func(ctx context.Context, request certificate.ImportRequest) (certificate.IssueResult, deployment.Report, error) {
|
|
result, err := workflow.Import(ctx, request)
|
|
return result.Certificate, result.Deployment, err
|
|
},
|
|
DeployCertificate: func(ctx context.Context, domainID string) (certificate.IssueResult, deployment.Report, error) {
|
|
result, err := workflow.DeployCurrent(ctx, domainID)
|
|
return result.Certificate, result.Deployment, err
|
|
},
|
|
RestoreCertificateVersion: func(ctx context.Context, domainID, version string) (certificate.IssueResult, error) {
|
|
return certificates.RestoreVersion(ctx, domainID, version)
|
|
},
|
|
RunRenewal: func(ctx context.Context) (certificate.RenewalReport, error) {
|
|
return renewal.RunOnce(ctx)
|
|
},
|
|
UpdateRenewalPolicy: renewal.UpdatePolicy,
|
|
CurrentCertificate: certificates,
|
|
ResetStatePath: admin.StatePath(cfg.Paths.Data),
|
|
Audit: auditRepo,
|
|
Tasks: taskRepo,
|
|
ReadinessCheck: db.DB.PingContext,
|
|
})
|
|
application := &App{
|
|
Config: cfg, Logger: logger, Database: db, Repository: repo,
|
|
Servers: serverRepo,
|
|
Users: userRepo,
|
|
Tasks: taskRepo,
|
|
Deployment: deploymentService,
|
|
SecretStore: secretStore, SSHKeys: sshKeys, DNSVerification: dnsVerification, HTTP: server,
|
|
Certificates: certificates,
|
|
Workflow: workflow,
|
|
Renewal: renewal,
|
|
}
|
|
application.startAuditCleanup(auditRepo)
|
|
return application, nil
|
|
}
|
|
|
|
func (a *App) startAuditCleanup(repo *repository.SQLAuditRepository) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
done := make(chan struct{})
|
|
a.maintenanceStop = cancel
|
|
a.maintenanceDone = done
|
|
go func() {
|
|
defer close(done)
|
|
ticker := time.NewTicker(auditCleanupInterval)
|
|
defer ticker.Stop()
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-ticker.C:
|
|
deleted, err := repo.PruneExpiredAuditLogs(ctx, time.Now().UTC().AddDate(0, 0, -auditLogRetentionDays))
|
|
if err != nil {
|
|
a.Logger.Warn("定时清理过期审计日志失败", "error", err)
|
|
continue
|
|
}
|
|
if deleted > 0 {
|
|
a.Logger.Info("已定时清理过期审计日志", "count", deleted)
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (a *App) Close() error {
|
|
if a == nil {
|
|
return nil
|
|
}
|
|
var first error
|
|
if a.Renewal != nil {
|
|
a.Renewal.Stop()
|
|
}
|
|
if a.maintenanceStop != nil {
|
|
a.maintenanceStop()
|
|
}
|
|
if a.maintenanceDone != nil {
|
|
<-a.maintenanceDone
|
|
}
|
|
if a.HTTP != nil {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
if err := a.HTTP.Shutdown(ctx); err != nil && first == nil {
|
|
first = err
|
|
}
|
|
cancel()
|
|
}
|
|
if a.Database != nil {
|
|
if err := a.Database.Close(); err != nil && first == nil {
|
|
first = err
|
|
}
|
|
}
|
|
if a.Logger != nil {
|
|
if err := a.Logger.Close(); err != nil && first == nil {
|
|
first = err
|
|
}
|
|
}
|
|
return first
|
|
}
|
|
|
|
func makeDirectory(path string) error {
|
|
return osMkdirAll(path)
|
|
}
|