ok-2

internal/templating/engine.go

@@ -0,0 +1,104 @@
+// internal/templating/engine.go
+package templating
+
+import (
+	"bytes"
+	"fmt"
+	"os"
+	"path/filepath"
+	"text/template"
+
+	"gopkg.in/yaml.v3"
+)
+
+// LoadTemplate 从文件加载 YAML 模板
+func LoadTemplate(path string) (*Template, error) {
+	data, err := os.ReadFile(path)
+	if err != nil {
+		return nil, fmt.Errorf("无法读取模板文件 %s: %w", path, err)
+	}
+	var tmpl Template
+	if err := yaml.Unmarshal(data, &tmpl); err != nil {
+		return nil, fmt.Errorf("YAML 解析失败: %w", err)
+	}
+	return &tmpl, nil
+}
+
+// Render 渲染模板为具体操作
+func (t *Template) Render(ctx Context) (map[string][]RenderedStep, error) {
+	result := make(map[string][]RenderedStep)
+
+	for stageName, steps := range t.Stages {
+		var renderedSteps []RenderedStep
+		for _, step := range steps {
+			// 处理 condition
+			if step.Condition != "" {
+				condTmpl, err := template.New("condition").Parse(step.Condition)
+				if err != nil {
+					return nil, fmt.Errorf("条件模板语法错误: %w", err)
+				}
+				var buf bytes.Buffer
+				if err := condTmpl.Execute(&buf, ctx); err != nil {
+					return nil, fmt.Errorf("执行条件模板失败: %w", err)
+				}
+				if buf.String() == "" {
+					continue // 条件不满足，跳过
+				}
+			}
+
+			// 渲染 content
+			contentTmpl, err := template.New("content").Parse(step.Content)
+			if err != nil {
+				return nil, fmt.Errorf("内容模板语法错误: %w", err)
+			}
+			var buf bytes.Buffer
+			if err := contentTmpl.Execute(&buf, ctx); err != nil {
+				return nil, fmt.Errorf("执行内容模板失败: %w", err)
+			}
+
+			renderedSteps = append(renderedSteps, RenderedStep{
+				Type:    step.Type,
+				Path:    step.Path,
+				Content: buf.String(),
+			})
+		}
+		result[stageName] = renderedSteps
+	}
+	return result, nil
+}
+
+// RenderedStep 是渲染后的步骤
+type RenderedStep struct {
+	Type    string
+	Path    string
+	Content string
+}
+
+// WriteFiles 将 file 类型步骤写入磁盘
+func WriteFiles(steps []RenderedStep, rootDir string) error {
+	for _, s := range steps {
+		if s.Type != "file" {
+			continue
+		}
+		fullPath := s.Path
+		if !filepath.IsAbs(s.Path) {
+			fullPath = filepath.Join(rootDir, s.Path)
+		}
+		if err := os.MkdirAll(filepath.Dir(fullPath), 0755); err != nil {
+			return err
+		}
+		if err := os.WriteFile(fullPath, []byte(s.Content), 0644); err != nil {
+			return fmt.Errorf("写入文件 %s 失败: %w", fullPath, err)
+		}
+	}
+	return nil
+}
+
+// PrintScripts 打印 script 内容（安全起见，先不自动执行）
+func PrintScripts(steps []RenderedStep) {
+	for _, s := range steps {
+		if s.Type == "script" {
+			fmt.Printf("# --- 脚本开始 ---\n%s\n# --- 脚本结束 ---\n", s.Content)
+		}
+	}
+}