package main

import (
	"fmt"
	"html/template"
	"log"
	"net/http"
	"strconv"
	"strings"
	"time"
)

// LanguageOption represents a language option for search
type LanguageOption struct {
	Code string
	Name string
}

var languageOptions = []LanguageOption{
	{Code: "", Name: "Any Language"},
	{Code: "lang_en", Name: "English"},
	{Code: "lang_af", Name: "Afrikaans"},
	{Code: "lang_ar", Name: "العربية (Arabic)"},
	{Code: "lang_hy", Name: "Հայերեն (Armenian)"},
	{Code: "lang_be", Name: "Беларуская (Belarusian)"},
	{Code: "lang_bg", Name: "български (Bulgarian)"},
	{Code: "lang_ca", Name: "Català (Catalan)"},
	{Code: "lang_zh-CN", Name: "中文 (简体) (Chinese Simplified)"},
	{Code: "lang_zh-TW", Name: "中文 (繁體) (Chinese Traditional)"},
	{Code: "lang_hr", Name: "Hrvatski (Croatian)"},
	{Code: "lang_cs", Name: "Čeština (Czech)"},
	{Code: "lang_da", Name: "Dansk (Danish)"},
	{Code: "lang_nl", Name: "Nederlands (Dutch)"},
	{Code: "lang_eo", Name: "Esperanto"},
	{Code: "lang_et", Name: "Eesti (Estonian)"},
	{Code: "lang_tl", Name: "Filipino (Tagalog)"},
	{Code: "lang_fi", Name: "Suomi (Finnish)"},
	{Code: "lang_fr", Name: "Français (French)"},
	{Code: "lang_de", Name: "Deutsch (German)"},
	{Code: "lang_el", Name: "Ελληνικά (Greek)"},
	{Code: "lang_iw", Name: "עברית (Hebrew)"},
	{Code: "lang_hi", Name: "हिन्दी (Hindi)"},
	{Code: "lang_hu", Name: "magyar (Hungarian)"},
	{Code: "lang_is", Name: "íslenska (Icelandic)"},
	{Code: "lang_id", Name: "Bahasa Indonesia (Indonesian)"},
	{Code: "lang_it", Name: "italiano (Italian)"},
	{Code: "lang_ja", Name: "日本語 (Japanese)"},
	{Code: "lang_ko", Name: "한국어 (Korean)"},
	{Code: "lang_lv", Name: "latviešu (Latvian)"},
	{Code: "lang_lt", Name: "lietuvių (Lithuanian)"},
	{Code: "lang_no", Name: "norsk (Norwegian)"},
	{Code: "lang_fa", Name: "فارسی (Persian)"},
	{Code: "lang_pl", Name: "polski (Polish)"},
	{Code: "lang_pt", Name: "português (Portuguese)"},
	{Code: "lang_ro", Name: "română (Romanian)"},
	{Code: "lang_ru", Name: "русский (Russian)"},
	{Code: "lang_sr", Name: "српски (Serbian)"},
	{Code: "lang_sk", Name: "slovenčina (Slovak)"},
	{Code: "lang_sl", Name: "slovenščina (Slovenian)"},
	{Code: "lang_es", Name: "español (Spanish)"},
	{Code: "lang_sw", Name: "Kiswahili (Swahili)"},
	{Code: "lang_sv", Name: "svenska (Swedish)"},
	{Code: "lang_th", Name: "ไทย (Thai)"},
	{Code: "lang_tr", Name: "Türkçe (Turkish)"},
	{Code: "lang_uk", Name: "українська (Ukrainian)"},
	{Code: "lang_vi", Name: "Tiếng Việt (Vietnamese)"},
}

var funcs = template.FuncMap{
	"title":   func(s string) string { return strings.Title(s) },
	"url_for": func(filename string) string { return "/" + filename },
}

var templates = template.Must(template.New("").Funcs(funcs).ParseFiles("templates/results.html"))

func main() {
	http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("static"))))
	http.HandleFunc("/", handleSearch)
	http.HandleFunc("/search", handleSearch)
	fmt.Println("Server is listening on http://localhost:5000")
	log.Fatal(http.ListenAndServe(":5000", nil))
}

func handleSearch(w http.ResponseWriter, r *http.Request) {
	var query, safe, lang string

	if r.Method == "GET" {
		query = r.URL.Query().Get("q")
		safe = r.URL.Query().Get("safe")
		lang = r.URL.Query().Get("lang")
	} else if r.Method == "POST" {
		query = r.FormValue("q")
		safe = r.FormValue("safe")
		lang = r.FormValue("lang")
	}

	if query == "" {
		http.ServeFile(w, r, "static/search.html")
		return
	}

	start := time.Now() // This line is correctly placed
	results, err := PerformTextSearch(query, safe, lang)
	if err != nil {
		http.Error(w, "Failed to fetch search results", http.StatusInternalServerError)
		return
	}
	elapsed := time.Since(start) // Correctly captures the elapsed time after search
	fetched := fmt.Sprintf("Fetched the results in %.2f seconds", elapsed.Seconds())

	data := struct {
		Results         []SearchResult
		Query           string
		Fetched         string
		ElapsedTime     string
		LanguageOptions []LanguageOption
		CurrentLang     string
	}{
		Results:         results,
		Query:           query,
		Fetched:         fetched,
		ElapsedTime:     strconv.FormatFloat(time.Since(start).Seconds(), 'f', 2, 64),
		LanguageOptions: languageOptions,
		CurrentLang:     lang,
	}

	err = templates.ExecuteTemplate(w, "results.html", data)
	if err != nil {
		http.Error(w, "Failed to render template", http.StatusInternalServerError)
		return
	}
}