package pages

import (
	"net/http"
	"strconv"

	"codeberg.org/rimgo/rimgo/render"
	"codeberg.org/rimgo/rimgo/utils"
)

func HandleSearch(w http.ResponseWriter, r *http.Request) error {
	utils.SetHeaders(w)
	w.Header().Set("X-Frame-Options", "DENY")
	w.Header().Set("Cache-Control", "public,max-age=604800")
	w.Header().Set("Content-Security-Policy", "default-src 'none'; frame-ancestors 'none'; base-uri 'none'; form-action 'self'; style-src 'unsafe-inline' 'self'; media-src 'self'; img-src 'self'; manifest-src 'self'; block-all-mixed-content")

	query := r.URL.Query().Get("q")

	if utils.ImgurRe.MatchString(query) {
		w.Header().Set("Location", utils.ImgurRe.ReplaceAllString(query, ""))
		w.WriteHeader(302)
		return nil
	}

	page := r.URL.Query().Get("page")
	if page == "" {
		page = "0"
	}

	pageNumber, err := strconv.Atoi(page)
	if err != nil {
		pageNumber = 0
	}

	results, err := ApiClient.Search(query, page)
	if err != nil {
		return err
	}

	return render.Render(w, "search", map[string]any{
		"query":    query,
		"results":  results,
		"page":     pageNumber,
		"nextPage": pageNumber + 1,
		"prevPage": pageNumber - 1,
	})
}