package formatter

import (
	"fmt"
	"strings"

	"git.db.org.ai/dborg/internal/models"
	"git.db.org.ai/dborg/internal/utils"
)

func FormatShortlinksResults(response *models.ShortlinksSearchResponse, asJSON bool) error {
	if asJSON {
		return utils.PrintJSON(response)
	}

	if response.Results == nil {
		PrintWarning("No results found")
		return nil
	}

	resultsMap, ok := response.Results.(map[string]interface{})
	if !ok {
		return utils.PrintJSON(response)
	}

	hits, ok := resultsMap["hits"].([]interface{})
	if !ok || len(hits) == 0 {
		PrintWarning("No results found")
		return nil
	}

	numHits, _ := resultsMap["num_hits"].(float64)
	elapsed, _ := resultsMap["elapsed_time_micros"].(float64)

	fmt.Printf("%s %s %s\n",
		Cyan("Found"),
		Bold(Yellow(fmt.Sprintf("%d", int(numHits)))),
		Cyan(fmt.Sprintf("credentials (%.2fms)", elapsed/1000)))
	fmt.Println()

	for i, hit := range hits {
		hitMap, ok := hit.(map[string]interface{})
		if !ok {
			continue
		}

		if i > 0 {
			fmt.Println(Dim(strings.Repeat("─", 80)))
		}

		username, _ := hitMap["username"].(string)
		password, _ := hitMap["password"].(string)
		url, _ := hitMap["url"].(string)
		filename, _ := hitMap["filename"].(string)

		fmt.Printf("%s %s\n", Bold(Green("●")), Bold(username))

		if password != "" {
			fmt.Printf("  %s %s\n", Dim("Password:"), Yellow(password))
		}

		if url != "" {
			fmt.Printf("  %s %s\n", Dim("URL:"), Blue(url))
		}

		if filename != "" {
			fmt.Printf("  %s %s\n", Dim("Source:"), Magenta(filename))
		}

		fmt.Println()
	}

	if response.Credits.Unlimited {
		fmt.Printf("%s: %s\n", Dim("Credits"), Green("Unlimited"))
	} else {
		fmt.Printf("%s: %s\n", Dim("Credits Remaining"), FormatCredits(int64(response.Credits.Remaining)))
	}

	return nil
}