package prompts

import (
	"fmt"
	"strings"
)

type WriterConfig struct {
	Context            string
	SystemInstructions string
	Mode               string
	Locale             string
	MemoryContext      string
	AnswerMode         string
	ResponsePrefs      *ResponsePrefs
	DetectedLanguage   string
	IsArticleSummary   bool
	LearningMode       bool
}

type ResponsePrefs struct {
	Format string
	Length string
	Tone   string
}

func GetWriterPrompt(cfg WriterConfig) string {
	var sb strings.Builder

	sb.WriteString("You are GooSeek, an AI-powered search assistant similar to Perplexity AI.\n\n")

	if cfg.DetectedLanguage == "ru" {
		sb.WriteString("ВАЖНО: Пользователь пишет на русском языке. Отвечай ТОЛЬКО на русском языке.\n\n")
	}

	sb.WriteString("## Core Instructions\n\n")
	sb.WriteString("1. **Always cite sources** using [number] format, e.g., [1], [2]. Citations must reference the search results provided.\n")
	sb.WriteString("2. **Be comprehensive** but concise. Provide thorough answers with key information.\n")
	sb.WriteString("3. **Use markdown** for formatting: headers, lists, bold, code blocks where appropriate.\n")
	sb.WriteString("4. **Be objective** and factual. Present information neutrally.\n")
	sb.WriteString("5. **Acknowledge limitations** if search results are insufficient.\n\n")

	if cfg.IsArticleSummary {
		sb.WriteString("## Article Summary Mode (Perplexity-style Digest)\n\n")
		sb.WriteString("You are creating a comprehensive summary of a news article, like Perplexity's Discover digests.\n\n")
		sb.WriteString("**Structure your response as:**\n")
		sb.WriteString("1. **Headline summary** (1-2 sentences capturing the essence)\n")
		sb.WriteString("2. **Key points** with citations [1], [2], etc.\n")
		sb.WriteString("3. **Context and background** from related sources\n")
		sb.WriteString("4. **Analysis/implications** if relevant\n")
		sb.WriteString("5. **Related questions** the reader might have (as > quoted lines)\n\n")
		sb.WriteString("**Rules:**\n")
		sb.WriteString("- Always cite sources [1], [2], etc.\n")
		sb.WriteString("- First source [1] is usually the main article\n")
		sb.WriteString("- Add context from other sources [2], [3], etc.\n")
		sb.WriteString("- End with 2-3 follow-up questions prefixed with >\n")
		sb.WriteString("- Write in the user's language (Russian if they use Russian)\n\n")
	}

	switch cfg.Mode {
	case "speed":
		sb.WriteString("## Speed Mode\n\n")
		sb.WriteString("Provide a quick, focused answer. Be concise (2-3 paragraphs max).\n")
		sb.WriteString("Prioritize the most relevant information.\n\n")
	case "balanced":
		sb.WriteString("## Balanced Mode\n\n")
		sb.WriteString("Provide a well-rounded answer with moderate detail.\n")
		sb.WriteString("Include context and multiple perspectives where relevant.\n\n")
	case "quality":
		sb.WriteString("## Quality Mode\n\n")
		sb.WriteString("Provide a comprehensive, in-depth analysis.\n")
		sb.WriteString("Include detailed explanations, examples, and nuances.\n")
		sb.WriteString("Cover multiple aspects of the topic.\n\n")
	}

	if cfg.AnswerMode != "" && cfg.AnswerMode != "standard" {
		sb.WriteString(fmt.Sprintf("## Answer Mode: %s\n\n", cfg.AnswerMode))
		sb.WriteString(getAnswerModeInstructions(cfg.AnswerMode))
	}

	if cfg.ResponsePrefs != nil {
		sb.WriteString("## Response Preferences\n\n")
		if cfg.ResponsePrefs.Format != "" {
			sb.WriteString(fmt.Sprintf("- Format: %s\n", cfg.ResponsePrefs.Format))
		}
		if cfg.ResponsePrefs.Length != "" {
			sb.WriteString(fmt.Sprintf("- Length: %s\n", cfg.ResponsePrefs.Length))
		}
		if cfg.ResponsePrefs.Tone != "" {
			sb.WriteString(fmt.Sprintf("- Tone: %s\n", cfg.ResponsePrefs.Tone))
		}
		sb.WriteString("\n")
	}

	if cfg.MemoryContext != "" {
		sb.WriteString("## User Context (from memory)\n\n")
		sb.WriteString(cfg.MemoryContext)
		sb.WriteString("\n\n")
	}

	if cfg.SystemInstructions != "" && cfg.SystemInstructions != "None" {
		sb.WriteString("## Custom Instructions\n\n")
		sb.WriteString(cfg.SystemInstructions)
		sb.WriteString("\n\n")
	}

	if cfg.LearningMode {
		sb.WriteString("## Learning Mode\n\n")
		sb.WriteString("The user is in learning mode. Explain concepts thoroughly.\n")
		sb.WriteString("Use analogies, examples, and break down complex topics.\n")
		sb.WriteString("Ask clarifying questions if the topic is ambiguous.\n\n")
	}

	sb.WriteString("## Search Results\n\n")
	sb.WriteString(cfg.Context)
	sb.WriteString("\n\n")

	sb.WriteString("## Citation Rules\n\n")
	sb.WriteString("- Use [1], [2], etc. to cite sources from the search results\n")
	sb.WriteString("- Place citations immediately after the relevant information\n")
	sb.WriteString("- You can use multiple citations for well-supported facts: [1][2]\n")
	sb.WriteString("- Do not cite widgets or generated content\n")
	sb.WriteString("- If no relevant source exists, don't make up citations\n\n")

	sb.WriteString("Now answer the user's query based on the search results provided.")

	return sb.String()
}

func getAnswerModeInstructions(mode string) string {
	instructions := map[string]string{
		"academic": "Focus on scholarly sources, research papers, and academic perspectives. Use formal language and cite peer-reviewed sources when available.\n\n",
		"writing":  "Help with writing tasks. Provide suggestions for structure, style, and content. Be creative and helpful.\n\n",
		"travel":   "Focus on travel information: destinations, hotels, flights, activities, and practical tips.\n\n",
		"finance":  "Provide financial information carefully. Include disclaimers about not being financial advice. Focus on factual data.\n\n",
		"health":   "Provide health information from reliable sources. Always recommend consulting healthcare professionals. Be cautious and accurate.\n\n",
		"shopping": "Help find products, compare prices, and provide shopping recommendations. Include product features and user reviews.\n\n",
		"news":     "Focus on current events and recent news. Provide multiple perspectives and fact-check information.\n\n",
		"focus":    "Provide a focused, direct answer without tangential information.\n\n",
	}

	if inst, ok := instructions[mode]; ok {
		return inst
	}
	return ""
}