lolly/internal/app/import.go

package app

import (
	"fmt"
	"os"
	"path/filepath"

	"rua.plus/lolly/internal/config"
	"rua.plus/lolly/internal/converter/nginx"
	"rua.plus/lolly/internal/version"

	"gopkg.in/yaml.v3"
)

// Run decides behavior based on flags: generate config, import nginx config, print version, or start the app.
func Run(cfgPath string, genConfig bool, outputPath string, importPath string, showVersion bool) int {
	if genConfig && importPath != "" {
		fmt.Fprintln(os.Stderr, "error: --generate-config and --import are mutually exclusive")
		return 1
	}
	if outputPath != "" && !genConfig && importPath == "" {
		fmt.Fprintln(os.Stderr, "error: -o requires either --generate-config or --import")
		return 1
	}

	if genConfig {
		return generateConfig(outputPath)
	}

	if importPath != "" {
		if err := importNginxConfig(importPath, outputPath); err != nil {
			fmt.Fprintln(os.Stderr, "error:", err)
			return 1
		}
		return 0
	}

	if showVersion {
		printVersion()
		return 0
	}

	app := NewApp(cfgPath)
	return app.Run()
}

func generateConfig(outputPath string) int {
	cfg := config.DefaultConfig()
	yamlData, err := config.GenerateConfigYAML(cfg)
	if err != nil {
		fmt.Fprintf(os.Stderr, "生成配置失败: %v\n", err)
		return 1
	}

	if outputPath == "" {
		fmt.Print(string(yamlData))
	} else {
		if err := os.WriteFile(outputPath, yamlData, 0o644); err != nil {
			fmt.Fprintf(os.Stderr, "写入文件失败: %v\n", err)
			return 1
		}
		fmt.Printf("配置已写入: %s\n", outputPath)
	}
	return 0
}

func importNginxConfig(path, outputPath string) error {
	nginxCfg, err := nginx.ParseFile(path)
	if err != nil {
		return fmt.Errorf("解析 nginx 配置失败: %w", err)
	}

	result, err := nginx.Convert(nginxCfg)
	if err != nil {
		return fmt.Errorf("转换配置失败: %w", err)
	}

	for _, w := range result.Warnings {
		fmt.Fprintf(os.Stderr, "warning: %s:line %d: %s\n", w.File, w.Line, w.Message)
	}

	if err := config.Validate(result.Config); err != nil {
		return fmt.Errorf("转换后配置验证失败: %w", err)
	}

	yamlData, err := yaml.Marshal(result.Config)
	if err != nil {
		return fmt.Errorf("序列化 YAML 失败: %w", err)
	}

	if outputPath == "" {
		os.Stdout.Write(yamlData)
	} else {
		if err := os.MkdirAll(filepath.Dir(outputPath), 0o755); err != nil {
			return fmt.Errorf("创建输出目录失败: %w", err)
		}
		if err := os.WriteFile(outputPath, yamlData, 0o644); err != nil {
			return fmt.Errorf("写入文件失败: %w", err)
		}
		fmt.Printf("配置已写入: %s\n", outputPath)
	}

	return nil
}

func printVersion() {
	fmt.Printf("lolly version %s\n", version.Version)
	fmt.Printf("  Git: %s (%s)\n", version.GitCommit, version.GitBranch)
	fmt.Printf("  Built: %s\n", version.BuildTime)
	fmt.Printf("  Go: %s\n", version.GoVersion)
	fmt.Printf("  Platform: %s\n", version.BuildPlatform)
}