137 lines
3.1 KiB
Go
137 lines
3.1 KiB
Go
package view
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"io/ioutil"
|
|
"local/dndex/storage"
|
|
"local/dndex/storage/entity"
|
|
"local/dndex/storage/operator"
|
|
"net/http"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/buger/jsonparser"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
)
|
|
|
|
func who(g storage.Graph, w http.ResponseWriter, r *http.Request) error {
|
|
namespace := strings.TrimLeft(r.URL.Path, path.Dir(r.URL.Path))
|
|
if len(namespace) == 0 {
|
|
http.NotFound(w, r)
|
|
return nil
|
|
}
|
|
namespace = strings.Replace(namespace, "/", ".", -1)
|
|
|
|
switch r.Method {
|
|
case http.MethodGet:
|
|
return whoGet(namespace, g, w, r)
|
|
case http.MethodPut:
|
|
return whoPut(namespace, g, w, r)
|
|
case http.MethodPost:
|
|
return whoPost(namespace, g, w, r)
|
|
default:
|
|
http.NotFound(w, r)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func whoGet(namespace string, g storage.Graph, w http.ResponseWriter, r *http.Request) error {
|
|
id := r.URL.Query().Get("id")
|
|
if id == "" {
|
|
http.Error(w, `{"error":"no ?id provided"}`, http.StatusBadRequest)
|
|
return nil
|
|
}
|
|
_, light := r.URL.Query()["light"]
|
|
|
|
ones, err := g.List(r.Context(), namespace, id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(ones) == 0 {
|
|
http.NotFound(w, r)
|
|
return nil
|
|
}
|
|
if len(ones) > 1 {
|
|
return errors.New("more than one result found matching " + id)
|
|
}
|
|
one := ones[0]
|
|
|
|
if !light && len(one.Connections) > 0 {
|
|
ones, err := g.List(r.Context(), namespace, one.Peers()...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, another := range ones {
|
|
another.Relationship = one.Connections[another.Name].Relationship
|
|
one.Connections[another.Name] = another
|
|
}
|
|
}
|
|
|
|
enc := json.NewEncoder(w)
|
|
enc.SetIndent("", " ")
|
|
return enc.Encode(one)
|
|
}
|
|
|
|
func whoPut(namespace string, g storage.Graph, w http.ResponseWriter, r *http.Request) error {
|
|
id := r.URL.Query().Get("id")
|
|
if id == "" {
|
|
http.Error(w, `{"error":"no ?id provided"}`, http.StatusBadRequest)
|
|
return nil
|
|
}
|
|
|
|
body, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
operation := entity.One{}
|
|
if err := json.Unmarshal(body, &operation); err != nil {
|
|
return err
|
|
}
|
|
if operation.Name != "" {
|
|
http.Error(w, `{"error":"cannot specify name in request body"}`, http.StatusBadRequest)
|
|
return nil
|
|
}
|
|
if operation.Modified != 0 {
|
|
http.Error(w, `{"error":"cannot specify modified in request body"}`, http.StatusBadRequest)
|
|
return nil
|
|
}
|
|
b, err := bson.Marshal(operation)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
op := bson.M{}
|
|
if err := bson.Unmarshal(b, &op); err != nil {
|
|
return err
|
|
}
|
|
for k := range op {
|
|
if _, _, _, err := jsonparser.Get(body, k); err != nil {
|
|
delete(op, k)
|
|
}
|
|
}
|
|
if err := g.Update(r.Context(), namespace, entity.One{Name: id}, operator.SetMany{Value: op}); err != nil {
|
|
return err
|
|
}
|
|
|
|
return whoGet(namespace, g, w, r)
|
|
}
|
|
|
|
func whoPost(namespace string, g storage.Graph, w http.ResponseWriter, r *http.Request) error {
|
|
id := r.URL.Query().Get("id")
|
|
if id == "" {
|
|
http.Error(w, `{"error":"no ?id provided"}`, http.StatusBadRequest)
|
|
return nil
|
|
}
|
|
|
|
one := entity.One{}
|
|
if err := json.NewDecoder(r.Body).Decode(&one); err != nil {
|
|
return err
|
|
}
|
|
one.Name = id
|
|
if err := g.Insert(r.Context(), namespace, one); err != nil {
|
|
return err
|
|
}
|
|
|
|
return whoGet(namespace, g, w, r)
|
|
}
|