From 57bbc5a1ac27e2c4702a937a126181b03254c29a Mon Sep 17 00:00:00 2001 From: "Arik Jones (aider)" Date: Thu, 19 Sep 2024 16:35:46 -0500 Subject: [PATCH] refactor: update writeMultipleFiles to create single output file --- cmd/web.go | 31 +++++++++++++------------------ 1 file changed, 13 insertions(+), 18 deletions(-) diff --git a/cmd/web.go b/cmd/web.go index f3e93e5..be27c03 100644 --- a/cmd/web.go +++ b/cmd/web.go @@ -113,25 +113,20 @@ func writeSingleFile(content map[string]string) error { return nil } -func writeMultipleFiles(content map[string]string) error { - for url, c := range content { - filename, err := getFilenameFromContent(c, url) - if err != nil { - return fmt.Errorf("error generating filename for %s: %v", url, err) - } - file, err := os.Create(filename) - if err != nil { - return fmt.Errorf("error creating output file %s: %v", filename, err) - } - - _, err = file.WriteString(fmt.Sprintf("# Content from %s\n\n%s", url, c)) - file.Close() - if err != nil { - return fmt.Errorf("error writing content to file %s: %v", filename, err) - } - - fmt.Printf("Content from %s has been saved to %s\n", url, filename) +func writeMultipleFiles(content map[string]string, outputFilename string) error { + file, err := os.Create(outputFilename) + if err != nil { + return fmt.Errorf("error creating output file %s: %v", outputFilename, err) } + defer file.Close() + + for path, c := range content { + _, err = file.WriteString(fmt.Sprintf("# File: %s\n\n%s\n\n", path, c)) + if err != nil { + return fmt.Errorf("error writing content to file %s: %v", outputFilename, err) + } + } + return nil }