JSON API new spec with deref

view/html.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"local/whodunit/config"
 	"local/whodunit/storage"
+	"local/whodunit/storage/entity"
 	"log"
 	"net/http"
 )
@@ -33,16 +34,30 @@ func foo(g storage.Graph) http.Handler {
 }
 
 func who(g storage.Graph, w http.ResponseWriter, r *http.Request) error {
-	results := make(map[string]storage.One)
-	for _, id := range r.URL.Query()["id"] {
+	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, storage.One{})
+			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
+			}
 		}
-		results[id] = ones[0]
 	}
 	log.Println("results:", results)
 	return json.NewEncoder(w).Encode(results)

view/html_test.go

@@ -6,39 +6,25 @@ import (
 	"fmt"
 	"local/whodunit/config"
 	"local/whodunit/storage"
+	"local/whodunit/storage/entity"
 	"net/http"
 	"os"
 	"testing"
 	"time"
+
+	"github.com/google/uuid"
 )
 
 func TestHtml(t *testing.T) {
-	if len(os.Getenv("INTEGRATION")) > 0 {
+	if len(os.Getenv("INTEGRATION")) == 0 {
 		t.Logf("skipping because $INTEGRATION unset")
 		return
 	}
 
 	os.Args = os.Args[:1]
-	g := storage.NewGraph()
-	ones := []storage.One{
-		storage.One{
-			ID:   "A",
-			Know: []storage.One{storage.One{}},
-		},
-		storage.One{
-			ID:   "B",
-			Know: []storage.One{storage.One{}},
-		},
-	}
-	ones[0].Know[0] = ones[1]
-	ones[0].Know[0].Know = nil
-	ones[0].Know[0].Relation = ":)"
-	ones[1].Know[0] = ones[0]
-	ones[1].Know[0].Know = nil
-	ones[1].Know[0].Relation = ":("
 
-	g.Insert(context.TODO(), ones[0])
-	g.Insert(context.TODO(), ones[1])
+	g := storage.NewGraph()
+	ones := fillDB(t, g)
 
 	go func() {
 		if err := Html(g); err != nil {
@@ -46,7 +32,7 @@ func TestHtml(t *testing.T) {
 		}
 	}()
 	time.Sleep(time.Millisecond * 250)
-	resp, err := http.Get(fmt.Sprintf("http://localhost:%d/who?id=A&id=B", config.New().Port))
+	resp, err := http.Get(fmt.Sprintf("http://localhost:%d/who?id=%s&v", config.New().Port, ones[len(ones)-1].Name))
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -61,3 +47,34 @@ func TestHtml(t *testing.T) {
 	}
 	t.Logf("\n%s\n", b)
 }
+
+func fillDB(t *testing.T, g storage.Graph) []entity.One {
+	ones := make([]entity.One, 5)
+	for i := range ones {
+		ones[i] = randomOne()
+		if i > 0 {
+			ones[i].Connections = []entity.One{entity.One{
+				Name:         ones[i-1].Name,
+				Relationship: ":D",
+			}}
+		}
+	}
+	for i := range ones {
+		if err := g.Insert(context.TODO(), ones[i]); err != nil {
+			t.Fatal(err)
+		}
+	}
+	return ones
+}
+
+func randomOne() entity.One {
+	return entity.One{
+		Name:        "name-" + uuid.New().String()[:5],
+		Type:        "type-" + uuid.New().String()[:5],
+		Title:       "titl-" + uuid.New().String()[:5],
+		Image:       "imge-" + uuid.New().String()[:5],
+		Text:        "text-" + uuid.New().String()[:5],
+		Modified:    time.Now().UnixNano(),
+		Connections: []entity.One{},
+	}
+}