99 lines
4.1 KiB
Go
99 lines
4.1 KiB
Go
package main
|
||
|
||
import (
|
||
"flag"
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
|
||
"git.pcux.cn/xuans/Smail/config"
|
||
"git.pcux.cn/xuans/Smail/utils/runtimepath"
|
||
)
|
||
|
||
// main 提供部署端显式配置迁移，不在网页或普通启动中自动改写配置。
|
||
func main() {
|
||
if err := run(os.Args[1:]); err != nil {
|
||
fmt.Fprintln(os.Stderr, err)
|
||
os.Exit(1)
|
||
}
|
||
}
|
||
|
||
func run(args []string) error {
|
||
flags := flag.NewFlagSet("smail-config", flag.ContinueOnError)
|
||
root := flags.String("root", "", "Smail 服务根目录，包含 config 子目录")
|
||
apply := flags.Bool("apply", false, "确认已停止后端，备份并执行迁移；默认只检查")
|
||
runtimeData := flags.Bool("runtime-data", false, "将运行数据迁到 data，保留当前五个配置文件")
|
||
historySource := flags.String("deployment-history", "", "停服复制完成后，从原部署导入配置历史并转换证书引用")
|
||
exportTLS := flags.String("export-project-tls", "", "将项目托管证书导出为本机 TLS 入口使用的合并 PEM 文件")
|
||
if err := flags.Parse(args); err != nil {
|
||
return err
|
||
}
|
||
if *root == "" || flags.NArg() != 0 {
|
||
return fmt.Errorf("请指定 --root 服务根目录；迁移前必须停止后端")
|
||
}
|
||
absolute, err := filepath.Abs(*root)
|
||
if err != nil {
|
||
return fmt.Errorf("服务根目录无效")
|
||
}
|
||
if *exportTLS != "" {
|
||
if *apply || *runtimeData || *historySource != "" || !filepath.IsAbs(*exportTLS) {
|
||
return fmt.Errorf("证书导出必须单独执行，目标必须是绝对路径")
|
||
}
|
||
if err := config.ExportProjectTLS(absolute, filepath.Clean(*exportTLS)); err != nil {
|
||
return fmt.Errorf("项目证书导出失败，原有入口证书保留：%w", err)
|
||
}
|
||
fmt.Println("项目证书同步检查完成。")
|
||
return nil
|
||
}
|
||
options := config.RuntimeValidationOptions{HasSuperAdmin: true, Production: true}
|
||
if *historySource != "" {
|
||
if !*apply || *runtimeData {
|
||
return fmt.Errorf("部署历史导入必须单独使用 --apply，且先停止源与目标后端")
|
||
}
|
||
if err := config.ImportDeploymentHistory(*historySource, absolute, options); err != nil {
|
||
return fmt.Errorf("配置历史导入失败：%w", err)
|
||
}
|
||
if err := config.NormalizeDeploymentReferences(absolute, options); err != nil {
|
||
return fmt.Errorf("部署引用转换失败：%w", err)
|
||
}
|
||
fmt.Println("配置历史已导入状态数据库，生效配置仍为 config 下的五个文件。")
|
||
return nil
|
||
}
|
||
if *runtimeData {
|
||
moves, err := runtimepath.Plan(absolute)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !*apply {
|
||
fmt.Printf("检查通过，共 %d 项待迁移；停止后端后加 --apply 执行。\n", len(moves))
|
||
return nil
|
||
}
|
||
if _, err := config.NewConfigLoader(absolute, config.ConfigLoaderOptions{Validation: options}).Load(); err != nil {
|
||
// 旧单文件部署先搬运行数据，再执行原有的分区配置迁移。
|
||
if _, legacyErr := config.PlanLegacyMigration(absolute, options); legacyErr != nil {
|
||
return fmt.Errorf("当前配置校验失败，未执行迁移")
|
||
}
|
||
}
|
||
if _, err := runtimepath.Migrate(absolute); err != nil {
|
||
return err
|
||
}
|
||
fmt.Printf("运行数据迁移完成，共 %d 项；当前五个配置文件未改写，废弃配置保留在 data/legacy-config。\n", len(moves))
|
||
return nil
|
||
}
|
||
plan, err := config.PlanLegacyMigration(absolute, options)
|
||
if err != nil {
|
||
// 不输出配置原文、未知字段或底层错误，避免凭据进入终端记录。
|
||
return fmt.Errorf("迁移检查未通过：未知字段 %d 项，需人工处理字段 %d 项；未执行迁移", len(plan.UnknownFields), len(plan.ManualFields))
|
||
}
|
||
if !*apply {
|
||
fmt.Println("旧版字段检查通过；尚未写入配置。执行迁移前请停止后端，使用 --apply 继续完整校验和备份迁移。")
|
||
return nil
|
||
}
|
||
result, err := config.MigrateLegacyConfig(absolute, options)
|
||
if err != nil {
|
||
return fmt.Errorf("配置迁移未完成，请保留原配置和备份，由部署管理员检查；不要手工删除分区文件")
|
||
}
|
||
fmt.Printf("配置迁移完成，代次：%d；原配置备份：%s。请使用最新版后端启动。\n", result.Generation, result.BackupPath)
|
||
return nil
|
||
}
|