fix: wizard now loads and uses custom template from config

- Pass config template path to RunWizard and template editor
- Load custom template in wizard step 4 instead of always using default
- Only override config template if user actually edits it in wizard
- Remove template from isComplete check since it can come from config

Author: ezynda3

cmd/iteratr/build.go

@@ -185,7 +185,7 @@ func runBuild(cmd *cobra.Command, args []string) error {
 		}
 		defer cleanup()
 
-		result, err := wizard.RunWizard(wizardStore)
+		result, err := wizard.RunWizard(wizardStore, buildFlags.template)
 		if err != nil {
 			return fmt.Errorf("wizard failed: %w", err)
 		}

internal/tui/wizard/template_editor.go

@@ -36,10 +36,12 @@ type TemplateEditorStep struct {
 	width    int            // Available width
 	height   int            // Available height
 	tmpFile  string         // Path to temp file for editing
+	edited   bool           // True if user edited the template via external editor
 }
 
 // NewTemplateEditorStep creates a new template editor step.
-func NewTemplateEditorStep() *TemplateEditorStep {
+// templatePath is the custom template path from config (empty string means use default).
+func NewTemplateEditorStep(templatePath string) *TemplateEditorStep {
 	// Create viewport
 	vp := viewport.New(
 		viewport.WithWidth(60),
@@ -50,8 +52,12 @@ func NewTemplateEditorStep() *TemplateEditorStep {
 	vp.MouseWheelEnabled = true
 	vp.MouseWheelDelta = 3
 
-	// Get default template content
-	content := template.DefaultTemplate
+	// Load template content - use custom path if provided, otherwise use default
+	content, err := template.GetTemplate(templatePath)
+	if err != nil {
+		// Fall back to default if custom template can't be loaded
+		content = template.DefaultTemplate
+	}
 
 	// Set highlighted content
 	vp.SetContent(highlightTemplate(content))
@@ -124,6 +130,7 @@ func (t *TemplateEditorStep) Update(msg tea.Msg) tea.Cmd {
 	case TemplateEditedMsg:
 		// Editor returned with new content
 		t.content = msg.Content
+		t.edited = true // Mark as edited so wizard knows to use this content
 		t.viewport.SetContent(highlightTemplate(t.content))
 		t.viewport.GotoTop()
 		// Clean up temp file
@@ -217,6 +224,12 @@ func (t *TemplateEditorStep) Content() string {
 	return t.content
 }
 
+// WasEdited returns true if the user edited the template via external editor.
+// Used to determine whether to override config template settings.
+func (t *TemplateEditorStep) WasEdited() bool {
+	return t.edited
+}
+
 // TemplateEditedMsg is sent when the external editor returns with new content.
 type TemplateEditedMsg struct {
 	Content string

internal/tui/wizard/wizard.go

@@ -37,6 +37,8 @@ type WizardModel struct {
 
 	// Session store for session operations
 	sessionStore *session.Store
+	// Template path from config (empty = use default)
+	templatePath string
 
 	// Step components
 	sessionSelectorStep *SessionSelectorStep
@@ -52,13 +54,15 @@ type WizardModel struct {
 
 // RunWizard is the entry point for the build wizard.
 // It creates a standalone BubbleTea program, runs it, and returns the result.
+// templatePath is the custom template path from config (empty string means use default).
 // Returns nil result and error if user cancels or an error occurs.
-func RunWizard(sessionStore *session.Store) (*WizardResult, error) {
+func RunWizard(sessionStore *session.Store, templatePath string) (*WizardResult, error) {
 	// Create initial model
 	m := &WizardModel{
 		step:         0,
 		cancelled:    false,
 		sessionStore: sessionStore,
+		templatePath: templatePath,
 	}
 
 	// Create BubbleTea program
@@ -219,7 +223,10 @@ func (m *WizardModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 
 	case ConfigCompleteMsg:
 		// Config complete in step 4
-		m.result.Template = m.templateEditorStep.Content()
+		// Only set template if user edited it; otherwise let config template be used
+		if m.templateEditorStep.WasEdited() {
+			m.result.Template = m.templateEditorStep.Content()
+		}
 		m.result.SessionName = m.configStep.SessionName()
 		m.result.Iterations = m.configStep.Iterations()
 		return m, tea.Quit
@@ -365,7 +372,10 @@ func (m *WizardModel) goNext() (tea.Model, tea.Cmd) {
 	case 4:
 		// Config step - finish wizard
 		if m.configStep != nil && m.configStep.IsValid() {
-			m.result.Template = m.templateEditorStep.Content()
+			// Only set template if user edited it; otherwise let config template be used
+			if m.templateEditorStep.WasEdited() {
+				m.result.Template = m.templateEditorStep.Content()
+			}
 			m.result.SessionName = m.configStep.SessionName()
 			m.result.Iterations = m.configStep.Iterations()
 			return m, tea.Quit
@@ -443,7 +453,7 @@ func (m *WizardModel) initCurrentStep() {
 		}
 	case 3:
 		if m.templateEditorStep == nil {
-			m.templateEditorStep = NewTemplateEditorStep()
+			m.templateEditorStep = NewTemplateEditorStep(m.templatePath)
 		}
 	case 4:
 		if m.configStep == nil {
@@ -781,9 +791,8 @@ func (m *WizardModel) isStepValid() bool {
 
 // isComplete checks if all required steps have valid data.
 func (m *WizardModel) isComplete() bool {
-	// TODO: Validate each step
+	// Template is optional here - can come from config if not edited in wizard
 	return m.result.SpecPath != "" &&
 		m.result.Model != "" &&
-		m.result.Template != "" &&
 		m.result.SessionName != ""
 }