Refactored, added comprehensive testing.

prompt/prompt.go

@@ -0,0 +1,55 @@
+package prompt
+
+import (
+	"fmt"
+	"strings"
+
+	"git.wisehodl.dev/jay/aicli/input"
+)
+
+const defaultPrompt = "Analyze the following:"
+
+// ConstructQuery formats prompts and files into a complete query string.
+func ConstructQuery(prompts []string, files []input.FileData) string {
+	promptStr := formatPrompts(prompts)
+	filesStr := formatFiles(files)
+	return combineContent(promptStr, filesStr)
+}
+
+// formatPrompts joins prompt strings with newlines.
+func formatPrompts(prompts []string) string {
+	if len(prompts) == 0 {
+		return ""
+	}
+	return strings.Join(prompts, "\n")
+}
+
+// formatFiles wraps each file in a template with path and content.
+func formatFiles(files []input.FileData) string {
+	if len(files) == 0 {
+		return ""
+	}
+
+	var parts []string
+	for _, f := range files {
+		parts = append(parts, fmt.Sprintf("File: %s\n\n```\n%s\n```", f.Path, f.Content))
+	}
+	return strings.Join(parts, "\n\n")
+}
+
+// combineContent merges formatted prompts and files with appropriate separators.
+func combineContent(promptStr, filesStr string) string {
+	if promptStr == "" && filesStr == "" {
+		return ""
+	}
+
+	if promptStr == "" && filesStr != "" {
+		return defaultPrompt + "\n\n" + filesStr
+	}
+
+	if promptStr != "" && filesStr == "" {
+		return promptStr
+	}
+
+	return promptStr + "\n\n" + filesStr
+}