Scert/backend/internal/app/app.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
	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
}

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)
	serverRepo := repository.NewSQLServerRepository(db.DB)
	taskRepo := repository.NewSQLTaskRepository(db.DB)
	auditRepo := repository.NewSQLAuditRepository(db.DB)
	sshKeys := sshkey.NewService(secretStore)
	accounts := certificate.NewAccountRepository(secretStore)
	certificates := certificate.NewService(cfg, repo, secretStore, dnsprovider.NewLegoFactory(), accounts)
	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.NewRenewalScheduler(repo, workflow, cfg.ACME.RenewalThresholdDays)
	server := httpapi.New(httpapi.Dependencies{
		Config: cfg, ConfigPath: configPath, Logger: logger, Repository: repo, Servers: serverRepo, SSHKeys: sshKeys,
		SetupDNS: dnsVerification, Firewall: firewall.NewManager(),
		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
		},
		CurrentCertificate: certificates,
		ResetStatePath:     admin.StatePath(cfg.Paths.Data),
		Audit:              auditRepo,
		Tasks:              taskRepo,
	})
	return &App{
		Config: cfg, Logger: logger, Database: db, Repository: repo,
		Tasks:       taskRepo,
		SecretStore: secretStore, SSHKeys: sshKeys, DNSVerification: dnsVerification, HTTP: server,
		Certificates: certificates,
		Workflow:     workflow,
		Renewal:      renewal,
	}, nil
}

func (a *App) Close() error {
	if a == nil {
		return nil
	}
	var first error
	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)
}