package halo

import (
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"net/url"
	"os"
	"strings"
)

func checkErr(err error) {
	if err != nil {
		log.Fatal(err)
	}
}

func EventsForMatch(baseurl, title, matchid string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/matches/%s/events", baseurl, title, matchid))
	checkErr(err)
	q := url.Query()
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func MatchesForPlayer(baseurl, title, player, modes string, start, count int) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/players/%s/matches", baseurl, title, player))
	checkErr(err)
	q := url.Query()

	if modes != "" {
		q.Set("modes", modes)
	}
	if start != 0 {
		q.Set("start", string(start))
	}
	if count != 0 {
		q.Set("count", string(count))
	}
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func PlayerLeaderboard(baseurl, title, seasonid, playlistid string, count int) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/player-leaderboards/csr/%s/%s", baseurl, title, seasonid, playlistid))
	checkErr(err)
	q := url.Query()

	if count != 0 {
		q.Set("count", string(count))
	}
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func CarnageReportArena(baseurl, title, matchid string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/arena/matches/%s", baseurl, title, matchid))
	checkErr(err)
	q := url.Query()
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func CarnageReportCampaign(baseurl, title, matchid string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/campaign/matches/%s", baseurl, title, matchid))
	checkErr(err)
	q := url.Query()
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func CarnageReportCustom(baseurl, title, matchid string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/custom/matches/%s", baseurl, title, matchid))
	checkErr(err)
	q := url.Query()
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func CarnageReportWarzone(baseurl, title, matchid string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/warzone/matches/%s", baseurl, title, matchid))
	checkErr(err)
	q := url.Query()
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func ServiceRecordArena(baseurl, title, players, seasonid string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/servicerecords/arena", baseurl, title))
	checkErr(err)
	q := url.Query()
	q.Set("players", players)
	if seasonid != "" {
		q.Set("seasonId", seasonid)
	}
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func ServiceRecordCampaign(baseurl, title, players string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/servicerecords/campaign", baseurl, title))
	checkErr(err)
	q := url.Query()
	q.Set("players", players)
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func ServiceRecordCustom(baseurl, title, players string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/servicerecords/custom", baseurl, title))
	checkErr(err)
	q := url.Query()
	q.Set("players", players)
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func ServiceRecordWarzone(baseurl, title, players string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/stats/%s/servicerecords/warzone", baseurl, title))
	checkErr(err)
	q := url.Query()
	q.Set("players", players)
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

// Begin profile section
func EmblemImage(baseurl, title, player string, size int) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/profile/%s/profiles/%s", baseurl, title, player))
	checkErr(err)
	q := url.Query()
	if (size == 95) || (size == 128) || (size == 190) || (size == 256) || (size == 512) {
		q.Set("size", string(size))
	}
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func SpartanImage(baseurl, title, player string, size int, crop string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/profile/%s/profiles/%s/spartan", baseurl, title, player))
	checkErr(err)
	q := url.Query()
	if (size == 95) || (size == 128) || (size == 190) || (size == 256) || (size == 512) {
		q.Set("size", string(size))
	}
	if (strings.ToLower(crop) == "full") || (strings.ToLower(crop) == "portrait") {
		q.Set("crop", crop)
	}
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

// Begin metadata section
func CampaignMissions(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "campaign-missions", "")
}

func Commendations(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "commendations", "")
}

func CsrDesignations(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "csr-designations", "")
}

func Enemies(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "enemies", "")
}

func FlexibleStats(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "flexible-stats", "")
}

func GameBaseVariants(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "game-base-variants", "")
}

func GameVariants(baseurl, title, id string) []byte {
	return metadataRequest(baseurl, title, "game-variants", id)
}

func Impulses(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "impulses", "")
}

func MapVariants(baseurl, title, id string) []byte {
	return metadataRequest(baseurl, title, "map-variants", id)
}

func Maps(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "maps", "")
}

func Medals(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "medals", "")
}

func Playlists(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "playlists", "")
}

func RequisitionPacks(baseurl, title, id string) []byte {
	return metadataRequest(baseurl, title, "requisition-packs", id)
}

func Requisitions(baseurl, title, id string) []byte {
	return metadataRequest(baseurl, title, "requisitions", id)
}

func Seasons(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "seasons", "")
}

func Skulls(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "skulls", "")
}

func SpartanRanks(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "spartan-ranks", "")
}

func TeamColors(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "team-colors", "")
}

func Vehicles(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "vehicles", "")
}

func Weapons(baseurl, title string) []byte {
	return metadataRequest(baseurl, title, "weapons", "")
}

func metadataRequest(baseurl, title, datatype, id string) []byte {
	url, err := url.Parse(fmt.Sprintf("%s/metadata/%s/metadata/%s/%s", baseurl, title, datatype, id))
	checkErr(err)
	q := url.Query()
	url.RawQuery = q.Encode()
	response := sendRequest(url.String())
	return response
}

func sendRequest(url string) []byte {
	request, err := http.NewRequest("GET", url, nil)
	checkErr(err)
	request.Header.Set("Ocp-Apim-Subscription-Key", getAPIKey())

	response, err := http.DefaultClient.Do(request)
	checkErr(err)
	defer response.Body.Close()

	// Return the URL of the image for SpartanImage and EmblemImage
	if url != response.Request.URL.String() {
		return []byte(response.Request.URL.String())
	}

	// If its not SpartanImage or EmblemImage return the body
	contents, err := ioutil.ReadAll(response.Body)
	checkErr(err)

	return contents
}

func getAPIKey() string {
	apikey := os.Getenv("HALO_API_KEY")
	if len(apikey) != 32 {
		fmt.Println("Invalid API Key")
	}
	return apikey
}