package components

import (
	"fmt"
	"maps"

	"dk/internal/auth"
	"dk/internal/csrf"
	"dk/internal/middleware"
	"dk/internal/router"
	"dk/internal/template"
)

// GenerateTopNav generates the top navigation HTML based on authentication status
func GenerateTopNav(ctx router.Ctx) string {
	if middleware.IsAuthenticated(ctx) {
		csrfField := csrf.HiddenField(ctx, auth.Manager)
		return fmt.Sprintf(`<form action="/logout" method="post" class="logout">
				%s
				<button class="img-button" type="submit"><img src="/assets/images/button_logout.gif" alt="Log Out" title="Log Out"></button>
			</form>
			<a href="/help"><img src="/assets/images/button_help.gif" alt="Help" title="Help"></a>`, csrfField)
	} else {
		return `<a href="/login"><img src="/assets/images/button_login.gif" alt="Log In" title="Log In"></a>
			<a href="/register"><img src="/assets/images/button_register.gif" alt="Register" title="Register"></a>
			<a href="/help"><img src="/assets/images/button_help.gif" alt="Help" title="Help"></a>`
	}
}

// PageData holds common page template data
type PageData struct {
	Title      string
	Content    string
	TopNav     string
	LeftSide   string
	RightSide  string
	TotalTime  string
	NumQueries string
	Version    string
	Build      string
}

// RenderPage renders a page using the layout template with common data and additional custom data
func RenderPage(ctx router.Ctx, pageData PageData, additionalData map[string]any) error {
	if template.Cache == nil || auth.Manager == nil {
		return fmt.Errorf("template.Cache or auth.Manager not initialized")
	}

	layoutTmpl, err := template.Cache.Load("layout.html")
	if err != nil {
		return fmt.Errorf("failed to load layout template: %w", err)
	}

	// Build the base template data with common fields
	data := map[string]any{
		"title":      pageData.Title,
		"content":    pageData.Content,
		"topnav":     GenerateTopNav(ctx),
		"leftside":   pageData.LeftSide,
		"rightside":  pageData.RightSide,
		"totaltime":  middleware.GetRequestTime(ctx),
		"numqueries": pageData.NumQueries,
		"version":    pageData.Version,
		"build":      pageData.Build,
	}

	// Merge in additional data (overwrites common data if keys conflict)
	maps.Copy(data, additionalData)

	// Set defaults for empty fields
	if data["leftside"] == "" {
		data["leftside"] = LeftAside(ctx)
	}
	if data["rightside"] == "" {
		data["rightside"] = RightAside(ctx)
	}
	if data["numqueries"] == "" {
		data["numqueries"] = "0"
	}
	if data["version"] == "" {
		data["version"] = "1.0.0"
	}
	if data["build"] == "" {
		data["build"] = "dev"
	}

	layoutTmpl.WriteTo(ctx, data)
	return nil
}

// NewPageData creates a new PageData with sensible defaults
func NewPageData(title, content string) PageData {
	return PageData{
		Title:      title,
		Content:    content,
		LeftSide:   "",
		RightSide:  "",
		NumQueries: "0",
		Version:    "1.0.0",
		Build:      "dev",
	}
}