package searcher

import (
	"fmt"
	"io"

	"gitea.stevedudenhoeffer.com/steve/go-extractor"
	"gitea.stevedudenhoeffer.com/steve/go-extractor/sites/duckduckgo"
	gollm "gitea.stevedudenhoeffer.com/steve/go-llm"
)

func deferClose(closer io.Closer) {
	if closer != nil {
		_ = closer.Close()
	}
}

type searchResult struct {
	Answer  string   `json:"answer"`
	Sources []string `json:"sources"`
}

func fnSearch(ctx *gollm.Context, args struct {
	Query    string `description:"The search query to perform on duckduckgo"`
	Question string `description:"The question(s) you are trying to answer when you read the search results. e.g: "`
}) (string, error) {
	browser, ok := ctx.Value("browser").(extractor.Browser)
	if !ok {
		return "", fmt.Errorf("browser not found")
	}

	cfg := duckduckgo.Config{
		SafeSearch: duckduckgo.SafeSearchOff,
		Region:     "us-en",
	}

	page, err := cfg.OpenSearch(ctx, browser, args.Query)
	defer deferClose(page)

	if err != nil {
		return "", fmt.Errorf("failed to search: %w", err)
	}

	return "", nil
}