package processing

import (
	"strings"

	"github.com/descrybe/descrybe-v2/apps/api/internal/aiprompts"
	"github.com/descrybe/descrybe-v2/apps/api/internal/company"
)

func resolveProductPromptTemplates(in ProductInput) (systemTpl, userTpl string) {
	systemTpl = strings.TrimSpace(in.EnhanceSystemTemplate)
	userTpl = strings.TrimSpace(in.EnhanceUserTemplate)
	// Per-category prompt wins for the user message (company system keeps JSON schema / brand).
	if cat := strings.TrimSpace(in.CategoryEnhancePrompt); cat != "" {
		userTpl = cat
	}
	def, ok := aiprompts.DefaultFor(aiprompts.KeyProductEnhance)
	if !ok {
		return systemTpl, userTpl
	}
	if systemTpl == "" {
		systemTpl = def.SystemTemplate
	}
	if userTpl == "" {
		userTpl = def.UserTemplate
	}
	return systemTpl, userTpl
}

// RenderProductEnhancePrompts fills company/built-in templates with product variables.
func RenderProductEnhancePrompts(systemTpl, userTpl, category, name, description, gtin, brandPrompt, language string, attrs map[string]any) (system, user string) {
	attrsJSON := ""
	compact := CompactAttrs(attrs, MaxAttrKeys)
	if len(compact) > 0 {
		attrsJSON = sanitizeJSON(compact)
	}
	vars := aiprompts.Vars{
		"name":        SanitizeText(truncateRunes(name, 200)),
		"description": SanitizeText(truncateRunes(description, MaxProductDescRunes)),
		"category":    SanitizeText(category),
		"attrs":       attrsJSON,
		"gtin":        SanitizeText(gtin),
		"brand_voice": CompactBrandPrompt(brandPrompt),
		"language":    company.LanguageLabel(language),
	}
	system = strings.TrimSpace(aiprompts.Render(systemTpl, vars))
	user = strings.TrimSpace(aiprompts.Render(userTpl, vars))
	if user == "" {
		// Safety net if a custom user template renders empty.
		user = ProductEnhanceUser(category, name, description, attrs)
	}
	return system, user
}