Greg Gauthier
6bd72aad25
Simplify the subRe regular expression by using a non-capturing group for the section labels instead of alternation, and switch the lookahead to a non-capturing group for consistency. Add comments to document the purpose of stepRe and subRe for better maintainability.
87 lines
2.2 KiB
Go
87 lines
2.2 KiB
Go
package recipe
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
var (
|
|
// stepRe finds every "### Step N: Title" heading
|
|
stepRe = regexp.MustCompile(`(?m)^### Step (\d+): (.+)$`)
|
|
|
|
// subRe finds the three labelled sections inside each step.
|
|
// We use a simple non-capturing group + word-boundary approach instead of lookahead.
|
|
subRe = regexp.MustCompile(`(?m)^(\*\*(?:Objective|Instructions|Expected output):\*\*)\s*(.+?)(?:\n\n|\n###|\z)`)
|
|
)
|
|
|
|
// Load reads a recipe from disk and fully parses it.
|
|
func Load(path string, params map[string]any) (*Recipe, error) {
|
|
b, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// split frontmatter
|
|
parts := bytes.SplitN(b, []byte("---"), 3)
|
|
if len(parts) < 3 {
|
|
return nil, fmt.Errorf("missing YAML frontmatter")
|
|
}
|
|
|
|
var r Recipe
|
|
if err := yaml.Unmarshal(parts[1], &r); err != nil {
|
|
return nil, fmt.Errorf("yaml parse: %w", err)
|
|
}
|
|
|
|
// simple template render on the whole body (so {{.package_path}} works everywhere)
|
|
tpl, err := template.New("recipe").Parse(string(parts[2]))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var rendered bytes.Buffer
|
|
if err := tpl.Execute(&rendered, params); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
body := rendered.String()
|
|
|
|
// extract steps
|
|
matches := stepRe.FindAllStringSubmatch(body, -1)
|
|
for i, m := range matches {
|
|
stepNum := i + 1
|
|
title := m[2]
|
|
|
|
// crude but effective sub-section extraction
|
|
start := strings.Index(body, m[0])
|
|
end := len(body)
|
|
if i+1 < len(matches) {
|
|
end = strings.Index(body[start:], matches[i+1][0]) + start
|
|
}
|
|
section := body[start:end]
|
|
|
|
step := Step{Number: stepNum, Title: title}
|
|
for _, sub := range subRe.FindAllStringSubmatch(section, -1) {
|
|
switch sub[1] {
|
|
case "**Objective:**":
|
|
step.Objective = strings.TrimSpace(sub[2])
|
|
case "**Instructions:**":
|
|
step.Instructions = strings.TrimSpace(sub[2])
|
|
case "**Expected output:**":
|
|
step.Expected = strings.TrimSpace(sub[2])
|
|
}
|
|
}
|
|
r.Steps = append(r.Steps, step)
|
|
}
|
|
|
|
// final summary is everything after the last step
|
|
lastStepEnd := strings.LastIndex(body, matches[len(matches)-1][0])
|
|
r.FinalSummaryPrompt = strings.TrimSpace(body[lastStepEnd+len(matches[len(matches)-1][0]):])
|
|
|
|
return &r, nil
|
|
}
|