Swagger for import-exoprt

view/json.go

@@ -27,6 +27,10 @@ func jsonHandler(g storage.Graph) http.Handler {
 		foo    func(g storage.Graph, w http.ResponseWriter, r *http.Request) error
 		noauth bool
 	}{
+		{
+			path: "/port",
+			foo:  port,
+		},
 		{
 			path: "/who",
 			foo:  who,

view/port.go

@@ -0,0 +1,78 @@
+package view
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"local/dndex/storage"
+	"local/dndex/storage/entity"
+	"net/http"
+
+	"github.com/buger/jsonparser"
+)
+
+func port(g storage.Graph, w http.ResponseWriter, r *http.Request) error {
+	switch r.Method {
+	case http.MethodGet:
+		return portGet(g, w, r)
+	case http.MethodPost:
+		return portPost(g, w, r)
+	default:
+		http.NotFound(w, r)
+		return nil
+	}
+}
+
+func portGet(g storage.Graph, w http.ResponseWriter, r *http.Request) error {
+	namespace, err := getNamespace(r)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return nil
+	}
+
+	ones, err := g.List(r.Context(), namespace)
+	if err != nil {
+		return err
+	}
+
+	return json.NewEncoder(w).Encode(map[string]interface{}{namespace: ones})
+}
+
+func portPost(g storage.Graph, w http.ResponseWriter, r *http.Request) error {
+	namespace, err := getNamespace(r)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return nil
+	}
+
+	b, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		return err
+	}
+
+	var errIn error
+	n, err := jsonparser.ArrayEach(b, func(b []byte, _ jsonparser.ValueType, _ int, err error) {
+		if err != nil {
+			errIn = err
+			return
+		}
+
+		o := entity.One{}
+		if err := json.Unmarshal(b, &o); err != nil {
+			errIn = err
+			return
+		}
+
+		if err := g.Insert(r.Context(), namespace, o); err != nil {
+			errIn = err
+			return
+		}
+	}, namespace)
+	if err != nil {
+		return err
+	}
+	if errIn != nil {
+		return errIn
+	}
+
+	return json.NewEncoder(w).Encode(map[string]int{namespace: n})
+}

view/port_test.go

@@ -0,0 +1,112 @@
+package view
+
+import (
+	"bytes"
+	"context"
+	"io/ioutil"
+	"local/dndex/storage"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"strings"
+	"testing"
+)
+
+func TestPort(t *testing.T) {
+	os.Args = os.Args[:1]
+	f, err := ioutil.TempFile(os.TempDir(), "pattern*")
+	if err != nil {
+		t.Fatal(err)
+	}
+	f.Close()
+	defer os.Remove(f.Name())
+	os.Setenv("DBURI", f.Name())
+	os.Setenv("AUTH", "false")
+
+	g := storage.NewGraph()
+	reset := func() {
+		if err := g.Delete(context.TODO(), "col", map[string]string{}); err != nil {
+			t.Fatal(err)
+		}
+		fillDB(t, g)
+	}
+	handler := jsonHandler(g)
+
+	t.Run("port: wrong path", func(t *testing.T) {
+		reset()
+		r := httptest.NewRequest(http.MethodGet, "/port/", nil)
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusNotFound {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+	})
+
+	t.Run("port: wrong method", func(t *testing.T) {
+		reset()
+		r := httptest.NewRequest(http.MethodPut, "/port", nil)
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusNotFound {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+	})
+
+	t.Run("port: post: no namespace", func(t *testing.T) {
+		reset()
+		r := httptest.NewRequest(http.MethodPost, "/port", strings.NewReader(``))
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusBadRequest {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+	})
+
+	t.Run("port: get: no namespace", func(t *testing.T) {
+		reset()
+		r := httptest.NewRequest(http.MethodGet, "/port", nil)
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusBadRequest {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+	})
+
+	t.Run("port: ex => im", func(t *testing.T) {
+		reset()
+		r := httptest.NewRequest(http.MethodGet, "/port?namespace=col", nil)
+		w := httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusOK {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+
+		b, err := ioutil.ReadAll(w.Body)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		if err := g.Delete(context.TODO(), "col", map[string]string{}); err != nil {
+			t.Fatal(err)
+		}
+
+		if ones, err := g.List(context.TODO(), "col"); err != nil {
+			t.Fatal(err)
+		} else if len(ones) != 0 {
+			t.Fatal(len(ones))
+		}
+
+		r = httptest.NewRequest(http.MethodPost, "/port?namespace=col", bytes.NewReader(b))
+		w = httptest.NewRecorder()
+		handler.ServeHTTP(w, r)
+		if w.Code != http.StatusOK {
+			t.Fatalf("%d: %s", w.Code, w.Body.Bytes())
+		}
+
+		if ones, err := g.List(context.TODO(), "col"); err != nil {
+			t.Fatal(err)
+		} else if len(ones) < 10 {
+			t.Fatal(len(ones))
+		}
+	})
+}