65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package view
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"local/whodunit/config"
|
|
"local/whodunit/storage"
|
|
"local/whodunit/storage/entity"
|
|
"log"
|
|
"net/http"
|
|
)
|
|
|
|
func Html(g storage.Graph) error {
|
|
port := config.New().Port
|
|
log.Println("listening on", port)
|
|
err := http.ListenAndServe(fmt.Sprintf(":%d", port), foo(g))
|
|
return err
|
|
}
|
|
|
|
func foo(g storage.Graph) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
var err error
|
|
switch r.URL.Path {
|
|
case "/who":
|
|
err = who(g, w, r)
|
|
default:
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
}
|
|
})
|
|
}
|
|
|
|
func who(g storage.Graph, w http.ResponseWriter, r *http.Request) error {
|
|
ids := r.URL.Query()["id"]
|
|
_, verbose := r.URL.Query()["v"]
|
|
results := make(map[string]entity.One)
|
|
for i := 0; i < len(ids); i++ {
|
|
id := ids[i]
|
|
ones, err := g.List(r.Context(), id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(ones) != 1 {
|
|
ones = append(ones, entity.One{})
|
|
}
|
|
one := ones[0]
|
|
results[id] = one
|
|
if verbose {
|
|
ones, err := g.List(r.Context(), one.Peers()...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for i, one := range ones {
|
|
one.Connections = nil
|
|
results[id].Connections[i] = one
|
|
}
|
|
}
|
|
}
|
|
log.Println("results:", results)
|
|
return json.NewEncoder(w).Encode(results)
|
|
}
|