package main

import (
	"encoding/json"
	"errors"
	"log"
	"net/http"
)

var database Database

type UpdateRequest struct {
	Name    string `json:"name"`
	Healthy bool   `json:"healthy"`
}

func health(w http.ResponseWriter, req *http.Request) {
	if req.Method == "GET" {
		getHealth(w, req)
	} else if req.Method == "POST" {
		updateHealth(w, req)
	} else {
		w.WriteHeader(http.StatusBadRequest)
	}
}

func getHealth(w http.ResponseWriter, req *http.Request) {
	serviceId := req.URL.Query().Get("service")

	if serviceId == "" {
		w.WriteHeader(http.StatusBadRequest)
		w.Write([]byte("Service parameter not supplied"))
		return
	}

	service := database.Get(serviceId)

	if service == nil {
		w.WriteHeader(http.StatusNotFound)
		w.Write([]byte("Service not found with id " + serviceId))
		return
	}

	response, err := json.Marshal(service)
	if err != nil {
		log.Fatalf("Error marshalling JSON response: %v", err)
	}

	w.Header().Set("Content-Type", "application/json")
	w.Write(response)
}
func updateHealth(w http.ResponseWriter, req *http.Request) error {
	var updateRequest UpdateRequest
	if req.Body == nil {
		http.Error(w, "Request has no body", http.StatusBadRequest)
		return errors.New("request has no body")
	}
	err := json.NewDecoder(req.Body).Decode(&updateRequest)
	if err != nil {
		http.Error(w, "Error while parsing request", http.StatusBadRequest)
		log.Default().Println(err)
		return err
	}

	err = database.Update(Service{Name: updateRequest.Name, Healthy: updateRequest.Healthy})
	if err != nil {
		http.Error(w, "Error while updating service", 500)
		return err
	}

	w.WriteHeader(200)
	return nil
}

func status(w http.ResponseWriter, req *http.Request) {
	result, err := database.GetAllStatus()

	if err != nil {
		http.Error(w, "Error while fetching status", 500)
		return
	}

	response, err := json.Marshal(result)
	if err != nil {
		log.Fatalf("Error marshalling JSON response: %v", err)
	}

	w.Header().Set("Content-Type", "application/json")
	w.Header().Set("Access-Control-Allow-Origin", "*") // URL of the Kea control agent API endpoint
	w.Write(response)
}

func webServer(db Database) {
	database = db

	http.HandleFunc("/health", health)
	http.HandleFunc("/status", status)

	http.ListenAndServe(":8090", nil)
}