fix(xml): Support tree-only mode in XML format

Add FilePaths field to TemplateData to provide file paths independently
of file contents. This allows XML format to build the directory structure
correctly in tree-only mode when Files array is empty.

- Add FilePaths to TemplateData struct
- Populate FilePaths in PrepareTemplateData (always, not just tree-only)
- Update XML format to use FilePaths for directory structure
- Update tests to include and verify FilePaths field

Author: epilande

internal/generator/format.go

@@ -14,6 +14,7 @@ type FileData struct {
 type TemplateData struct {
 	Structure string
 	Files     []FileData
+	FilePaths []string
 }
 
 // Format defines the interface for different output formats

internal/generator/formats/formats_test.go

@@ -153,13 +153,15 @@ func createTestTemplateData() generator.TemplateData {
 				Language: "go",
 			},
 		},
+		FilePaths: []string{"main.go"},
 	}
 }
 
 func createTreeOnlyTemplateData() generator.TemplateData {
 	return generator.TemplateData{
 		Structure: "test-project/\n├── src/\n│   └── main.go\n└── README.md\n",
 		Files:     []generator.FileData{},
+		FilePaths: []string{"src/main.go", "README.md"},
 	}
 }
 
@@ -224,8 +226,19 @@ func TestXMLFormatTreeOnly(t *testing.T) {
 	if !strings.Contains(content, "<project>") {
 		t.Errorf("Expected content to contain '<project>'")
 	}
+	// Directory structure should be built from FilePaths
+	if !strings.Contains(content, "<directory name=\"src\">") {
+		t.Errorf("Expected content to contain '<directory name=\"src\">' from FilePaths")
+	}
+	if !strings.Contains(content, "<file name=\"main.go\">") {
+		t.Errorf("Expected content to contain '<file name=\"main.go\">' in filesystem section")
+	}
+	if !strings.Contains(content, "<file name=\"README.md\">") {
+		t.Errorf("Expected content to contain '<file name=\"README.md\">' in filesystem section")
+	}
+	// Files section should be empty (no file contents)
 	if strings.Contains(content, "<file path=") {
-		t.Errorf("Expected content to NOT contain file elements in tree-only mode")
+		t.Errorf("Expected content to NOT contain file content elements in tree-only mode")
 	}
 	if tokens <= 0 {
 		t.Errorf("Expected tokens to be positive, got %d", tokens)

internal/generator/formats/xml.go

@@ -61,9 +61,8 @@ func (f *XMLFormat) Render(data generator.TemplateData) (string, int, error) {
 		files:   []string{},
 	}
 
-	// Get all file paths from the files data
-	for _, file := range data.Files {
-		addFileToTree(root, file.Path)
+	for _, path := range data.FilePaths {
+		addFileToTree(root, path)
 	}
 
 	// Convert our internal tree to the XML structure

internal/generator/generator.go

@@ -202,6 +202,11 @@ func (g *Generator) PrepareTemplateData() (TemplateData, error) {
 
 	g.SelectedFiles = expandedSelection
 
+	filePaths := make([]string, 0, len(expandedSelection))
+	for path := range expandedSelection {
+		filePaths = append(filePaths, path)
+	}
+
 	rootNode := g.buildTree()
 	var structureBuilder strings.Builder
 	baseRootName := filepath.Base(g.RootPath)
@@ -239,5 +244,6 @@ func (g *Generator) PrepareTemplateData() (TemplateData, error) {
 	return TemplateData{
 		Structure: structureBuilder.String(),
 		Files:     filesData,
+		FilePaths: filePaths,
 	}, nil
 }

internal/generator/generator_test.go

@@ -151,6 +151,11 @@ func TestPrepareTemplateData(t *testing.T) {
 			t.Errorf("Expected language for %q to be %q, got %q", tf.path, "go", file.Language)
 		}
 	}
+
+	// Verify FilePaths is also populated
+	if len(data.FilePaths) != 3 {
+		t.Errorf("Expected FilePaths to have 3 entries, got %d", len(data.FilePaths))
+	}
 }
 
 func TestGenerateString(t *testing.T) {
@@ -282,6 +287,10 @@ func TestPrepareTemplateDataTreeOnly(t *testing.T) {
 	if len(data.Files) != 0 {
 		t.Errorf("Expected Files to be empty in tree-only mode, got %d files", len(data.Files))
 	}
+	// FilePaths should still be populated for XML format support
+	if len(data.FilePaths) != 3 {
+		t.Errorf("Expected FilePaths to have 3 entries in tree-only mode, got %d", len(data.FilePaths))
+	}
 }
 
 type mockFormat struct {