package docs

import (
	"fmt"
	"strings"
)

// BuildDocsMessages builds the messages for documentation generation.
// Exported for use by MCP server.
func BuildDocsMessages(language, code string) []map[string]string {
	style := docStyle(language)
	systemPrompt := fmt.Sprintf(
		"You are a documentation expert. Add %s documentation comments to the provided code. "+
			"Return ONLY the documented code with no explanations, markdown, or extra text. "+
			"Do NOT include markdown code fences. Document all public functions, methods, types, and constants.",
		style,
	)

	userPrompt := fmt.Sprintf("Add documentation comments to the following %s code:\n\n%s", language, code)

	return []map[string]string{
		{"role": "system", "content": systemPrompt},
		{"role": "user", "content": userPrompt},
	}
}

func docStyle(language string) string {
	switch strings.ToLower(language) {
	case "go":
		return "godoc"
	case "python":
		return "PEP 257 docstring"
	case "c", "c++":
		return "doxygen style"
	case "javascript", "typescript":
		return "JSDoc"
	case "rust":
		return "rustdoc"
	case "ruby":
		return "YARD"
	case "java":
		return "javadoc"
	case "shell":
		return "shell style comments"
	default:
		return "standard"
	}
}