package main

import (
	"io/ioutil"
	"log"
	"net/http"
	"net/url"
	"regexp"
	"strings"

	"gopkg.in/yaml.v2"
)

var bangs = loadBangs()

func main() {
	http.HandleFunc("/eval", func(w http.ResponseWriter, r *http.Request) {
		searchQuery := r.URL.Query().Get("query")
		searchEngine := r.URL.Query().Get("engine")

		if searchQuery == "" || searchEngine == "" {
			http.Error(w, "400 Bad Request", http.StatusBadRequest)
			return
		}

		searchQuery, searchEngine = evaluateBang(searchQuery, searchEngine)

		searchURL := buildSearchURL(searchEngine, searchQuery)
		http.Redirect(w, r, searchURL, http.StatusSeeOther)
	})

	log.Fatalf("Failed to start http server: %v", http.ListenAndServe(":8081", nil))
}

func loadBangs() (bangs map[string]string) {
	data, err := ioutil.ReadFile("bangs.yml")
	if err != nil {
		log.Fatalf("Failed to read bangs: %v", err)
	}

	err = yaml.Unmarshal(data, &bangs)
	if err != nil {
		log.Fatalf("Failed to decode bangs: %v", err)
	}

	return
}

func buildSearchURL(template string, query string) (searchUrl string) {
	searchUrl = strings.Replace(template, "%s", url.QueryEscape(query), 1)
	searchUrl = strings.Replace(searchUrl, "%S", query, 1)
	return searchUrl
}

func parseBang(searchQuery string) (bang string, query string) {
	regexStartBang, _ := regexp.Compile("\\!(.*) (.*)")
	regexEndBang, _ := regexp.Compile("(.*) \\!(.*)")

	startBang := regexStartBang.FindSubmatch([]byte(searchQuery))
	endBang := regexEndBang.FindSubmatch([]byte(searchQuery))

	if len(endBang) == 3 {
		bang = strings.ToLower(string(endBang[2]))
		query = string(endBang[1])
		return
	} else if len(startBang) == 3 {
		bang = strings.ToLower(string(startBang[1]))
		query = string(startBang[2])
		return
	} else {
		return "", searchQuery
	}
}

func evaluateBang(searchQuery string, searchEngine string) (query string, engine string) {
	bang, query := parseBang(searchQuery)
	engine = bangs[bang]

	if bang == "" || engine == "" {
		return searchQuery, searchEngine
	} else {
		return query, engine
	}
}