mirror of
https://github.com/andatoshiki/shikigrid.git
synced 2026-06-05 19:56:27 +00:00
102 lines
2.4 KiB
Go
102 lines
2.4 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/evilsocket/islazy/log"
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/andatoshiki/shikigrid/mesh"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"sort"
|
|
)
|
|
|
|
// 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)
|
|
}
|
|
|
|
// 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)
|
|
}
|
|
|
|
// 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)
|
|
}
|
|
|
|
// 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,
|
|
})
|
|
}
|
|
|
|
// GET /api/v1/mesh/data
|
|
func (api *API) PeerGetMeshData(w http.ResponseWriter, r *http.Request) {
|
|
JSON(w, http.StatusOK, api.Peer.Data())
|
|
}
|
|
|
|
// POST /api/v1/mesh/data
|
|
func (api *API) PeerSetMeshData(w http.ResponseWriter, r *http.Request) {
|
|
var newData map[string]interface{}
|
|
|
|
body, err := ioutil.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,
|
|
})
|
|
}
|