move entry unmarshal to reposotyr.GetJSON

1 files changed, 2 insertions, 10 deletions

diff --git a/internal/server/handler/handler.go b/internal/server/handler/handler.go
index 2630433..b834dd0 100644
--- a/internal/server/handler/handler.go
+++ b/internal/server/handler/handler.go
@@ -53,20 +53,12 @@ func (h Handler) List(w http.ResponseWriter, r *http.Request) error {
 }
 
 func (h Handler) Get(w http.ResponseWriter, r *http.Request) error {
-	id := r.URL.Query().Get("id")
-
-	ent, err := repository.Instance(h.conf).Get(id)
-	if err != nil {
-		return err
-	}
-
-	// TODO: Move to repository.go
-	bytes, err := ent.JSONMarshal()
+	json, err := repository.Instance(h.conf).GetJSON(r.URL.Query().Get("id"))
 	if err != nil {
 		return err
 	}
 
-	fmt.Fprint(w, string(bytes))
+	fmt.Fprint(w, json)
 	return nil
 }