package api

import (
	"encoding/json"
	"io"
	"net/http"
	"sort"

	"github.com/evilsocket/islazy/log"
	"github.com/go-chi/chi/v5"
	"github.com/andatoshiki/shikigrid/mesh"
)

// PeerGetPeers GET /api/v1/mesh/peers
func (api *API) PeerGetPeers(w http.ResponseWriter, r *http.Request) {
	peers := make([]*mesh.Peer, 0)
	mesh.Peers.Range(func(key, value interface{}) bool {
		peers = append(peers, value.(*mesh.Peer))
		return true
	})

	// closer first
	sort.Slice(peers, func(i, j int) bool {
		return peers[i].RSSI > peers[j].RSSI
	})

	JSON(w, http.StatusOK, peers)
}

// PeerGetMemory GET /api/v1/mesh/memory
func (api *API) PeerGetMemory(w http.ResponseWriter, r *http.Request) {
	peers := api.Mesh.Memory()
	// higher number of encounters first
	sort.Slice(peers, func(i, j int) bool {
		return peers[i].Encounters > peers[j].Encounters
	})
	JSON(w, http.StatusOK, peers)
}

// PeerGetMemoryOf GET /api/v1/mesh/memory/<fingerprint>
func (api *API) PeerGetMemoryOf(w http.ResponseWriter, r *http.Request) {
	fingerprint := chi.URLParam(r, "fingerprint")
	peer := api.Mesh.MemoryOf(fingerprint)
	if peer == nil {
		ERROR(w, http.StatusNotFound, ErrEmpty)
		return
	}
	JSON(w, http.StatusOK, peer)
}

// PeerSetSignaling GET /api/v1/mesh/<status>
func (api *API) PeerSetSignaling(w http.ResponseWriter, r *http.Request) {
	status := chi.URLParam(r, "status")

	if status == "enabled" || status == "true" {
		api.Peer.Advertise(true)
	} else if status == "disabled" || status == "false" {
		api.Peer.Advertise(false)
	} else {
		ERROR(w, http.StatusNotFound, ErrEmpty)
		return
	}

	JSON(w, http.StatusOK, map[string]interface{}{
		"success": true,
	})
}

// PeerGetMeshData GET /api/v1/mesh/data
func (api *API) PeerGetMeshData(w http.ResponseWriter, r *http.Request) {
	JSON(w, http.StatusOK, api.Peer.Data())
}

// PeerSetMeshData POST /api/v1/mesh/data
func (api *API) PeerSetMeshData(w http.ResponseWriter, r *http.Request) {
	var newData map[string]interface{}

	if api.Peer.ForceDisabled == true {
		api.Peer.Advertise(false)
		JSON(w, http.StatusOK, map[string]interface{}{
			"success": true, // this should be changed later when shikigotchi can handle shikigrid being force advertise disabled
		})
		return
	}

	body, err := io.ReadAll(r.Body)
	if err != nil {
		ERROR(w, http.StatusUnprocessableEntity, err)
		return
	}

	log.Debug("%s", body)

	if err = json.Unmarshal(body, &newData); err != nil {
		ERROR(w, http.StatusUnprocessableEntity, err)
		return
	}

	// this makes sure that the shikigrid server receives advertisements
	api.Client.SetData(map[string]interface{}{
		"advertisement": newData,
	})

	// update mesh advertisement data
	api.Peer.SetData(newData)

	JSON(w, http.StatusOK, map[string]interface{}{
		"success": true,
	})
}