Logic for formatting the files output correctly

rithikjain · rithikjain · commit 21100d44b218 · 2020-03-29T21:51:16.000+05:30
diff --git a/api/handler/gists.go b/api/handler/gists.go
@@ -0,0 +1,42 @@
+package handler
+
+import (
+	"encoding/json"
+	"github.com/rithikjain/GistsBackend/api/view"
+	"github.com/rithikjain/GistsBackend/pkg/gists"
+	"net/http"
+)
+
+func test(s gists.Service) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		g, err := s.ViewAllFiles(1)
+		if err != nil {
+			view.Wrap(err, w)
+		}
+
+		var files []gists.File
+		var allGists []gists.Gist
+		mGists := *g
+
+		// Logic for formatting the files in right format
+		for i := 0; i < len(mGists); i++ {
+			allGists = append(allGists, mGists[i])
+			for _, file := range mGists[i].Files {
+				file.GistID = allGists[i].ID
+				file.GistUrl = allGists[i].Url
+				file.IsPublic = allGists[i].IsPublic
+				file.UpdatedAt = allGists[i].UpdatedAt
+				files = append(files, file)
+			}
+		}
+
+		w.Header().Add("Content-Type", "application/json; charset=utf-8")
+		_ = json.NewEncoder(w).Encode(map[string]interface{}{
+			"files": files,
+		})
+	})
+}
+
+func MakGistsHandler(r *http.ServeMux, svc gists.Service) {
+	r.Handle("/api/test", test(svc))
+}
diff --git a/api/view/errors.go b/api/view/errors.go
@@ -0,0 +1,53 @@
+package view
+
+import (
+	"encoding/json"
+	"errors"
+	"github.com/rithikjain/GistsBackend/pkg"
+	"net/http"
+)
+
+type ErrView struct {
+	Message string `json:"message"`
+	Status  int    `json:"status"`
+}
+
+//noinspection ALL
+var (
+	ErrMethodNotAllowed = errors.New("Error: Method is not allowed")
+	ErrInvalidToken     = errors.New("Error: Invalid Authorization token")
+	ErrUserExists       = errors.New("User already exists")
+)
+
+var ErrHTTPStatusMap = map[string]int{
+	pkg.ErrNotFound.Error():     http.StatusNotFound,
+	pkg.ErrInvalidSlug.Error():  http.StatusBadRequest,
+	pkg.ErrExists.Error():       http.StatusConflict,
+	pkg.ErrNoContent.Error():    http.StatusNotFound,
+	pkg.ErrDatabase.Error():     http.StatusInternalServerError,
+	pkg.ErrUnauthorized.Error(): http.StatusUnauthorized,
+	pkg.ErrForbidden.Error():    http.StatusForbidden,
+	pkg.ErrEmail.Error():        http.StatusBadRequest,
+	pkg.ErrPassword.Error():     http.StatusBadRequest,
+	ErrMethodNotAllowed.Error(): http.StatusMethodNotAllowed,
+	ErrInvalidToken.Error():     http.StatusBadRequest,
+	ErrUserExists.Error():       http.StatusConflict,
+}
+
+func Wrap(err error, w http.ResponseWriter) {
+	msg := err.Error()
+	code := ErrHTTPStatusMap[msg]
+
+	if code == 0 {
+		code = http.StatusInternalServerError
+	}
+
+	w.WriteHeader(code)
+	w.Header().Add("Content-Type", "application/json; charset=utf-8")
+
+	errView := ErrView{
+		Message: msg,
+		Status:  code,
+	}
+	_ = json.NewEncoder(w).Encode(errView)
+}
diff --git a/main.go b/main.go
@@ -5,6 +5,8 @@ import (
 	"github.com/jinzhu/gorm"
 	_ "github.com/jinzhu/gorm/dialects/postgres"
 	"github.com/joho/godotenv"
+	"github.com/rithikjain/GistsBackend/api/handler"
+	"github.com/rithikjain/GistsBackend/pkg/gists"
 	"log"
 	"net/http"
 	"os"
@@ -54,8 +56,13 @@ func main() {
 	fmt.Println("Connected to DB...")
 	db.LogMode(true)
 
+	gistsSvc := gists.NewService(db)
+
 	// Setting up the router
 	r := http.NewServeMux()
+
+	handler.MakGistsHandler(r, gistsSvc)
+
 	r.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		w.WriteHeader(http.StatusOK)
 		_, _ = w.Write([]byte("Hello There"))
diff --git a/pkg/gists/entities.go b/pkg/gists/entities.go
@@ -1,15 +1,19 @@
 package gists
 
 type Gist struct {
-	Url       string            `json:"url"`
-	ID        string            `json:"id"`
-	IsPublic  bool              `json:"public"`
-	UpdatedAt string            `json:"updated_at"`
-	Files     []map[string]File `json:"files"`
+	Url       string          `json:"url"`
+	ID        string          `json:"id"`
+	IsPublic  bool            `json:"public"`
+	UpdatedAt string          `json:"updated_at"`
+	Files     map[string]File `json:"files"`
 }
 
 type File struct {
-	Filename string `json:"filename"`
-	Language string `json:"language"`
-	RawUrl   string `json:"raw_url"`
+	GistID    string `json:"gist_id"`
+	GistUrl   string `json:"gist_url"`
+	IsPublic  bool   `json:"public"`
+	UpdatedAt string `json:"updated_at"`
+	Filename  string `json:"filename"`
+	Language  string `json:"language"`
+	RawUrl    string `json:"raw_url"`
 }
