refactor: backend

api/resource.go

@@ -1,130 +1,40 @@
 package api
 
-import (
-	"encoding/json"
-	"io/ioutil"
-	"memos/api/e"
-	"memos/store"
-	"net/http"
-	"strings"
+type Resource struct {
+	Id        int   `jsonapi:"primary,resource"`
+	CreatedTs int64 `jsonapi:"attr,createdTs"`
+	UpdatedTs int64 `jsonapi:"attr,updatedTs"`
 
-	"github.com/gorilla/mux"
-)
+	Filename string `jsonapi:"attr,filename"`
+	Blob     []byte `jsonapi:"attr,blob"`
+	Type     string `jsonapi:"attr,type"`
+	Size     int64  `jsonapi:"attr,size"`
 
-func handleGetMyResources(w http.ResponseWriter, r *http.Request) {
-	userId, _ := GetUserIdInSession(r)
-
-	resources, err := store.GetResourcesByUserId(userId)
-
-	if err != nil {
-		e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
-		return
-	}
-
-	json.NewEncoder(w).Encode(Response{
-		Succeed: true,
-		Message: "",
-		Data:    resources,
-	})
+	CreatorId int `jsonapi:"attr,creatorId"`
 }
 
-func handleUploadResource(w http.ResponseWriter, r *http.Request) {
-	userId, _ := GetUserIdInSession(r)
+type ResourceCreate struct {
+	Filename string `jsonapi:"attr,filename"`
+	Blob     []byte `jsonapi:"attr,blob"`
+	Type     string `jsonapi:"attr,type"`
+	Size     int64  `jsonapi:"attr,size"`
 
-	err := r.ParseMultipartForm(5 << 20)
-
-	if err != nil {
-		e.ErrorHandler(w, "OVERLOAD_MAX_SIZE", "The max size of resource is 5Mb.")
-		return
-	}
-
-	file, handler, err := r.FormFile("file")
-
-	if err != nil {
-		e.ErrorHandler(w, "REQUEST_BODY_ERROR", "Bad request")
-		return
-	}
-
-	defer file.Close()
-
-	filename := handler.Filename
-	filetype := handler.Header.Get("Content-Type")
-	size := handler.Size
-
-	fileBytes, err := ioutil.ReadAll(file)
-
-	if err != nil {
-		e.ErrorHandler(w, "UPLOAD_FILE_ERROR", "Read file error")
-		return
-	}
-
-	resource, err := store.CreateResource(userId, filename, fileBytes, filetype, size)
-
-	if err != nil {
-		e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
-		return
-	}
-
-	json.NewEncoder(w).Encode(Response{
-		Succeed: true,
-		Message: "Upload file succeed",
-		Data:    resource,
-	})
+	CreatorId int `jsonapi:"attr,creatorId"`
 }
 
-func handleDeleteResource(w http.ResponseWriter, r *http.Request) {
-	vars := mux.Vars(r)
-	resourceId := vars["id"]
-
-	err := store.DeleteResourceById(resourceId)
-
-	if err != nil {
-		e.ErrorHandler(w, "DATABASE_ERROR", err.Error())
-		return
-	}
-
-	json.NewEncoder(w).Encode(Response{
-		Succeed: true,
-		Message: "",
-		Data:    nil,
-	})
+type ResourceFind struct {
+	Id        *int
+	CreatorId *int
+	Filename  *string
 }
 
-func handleGetResource(w http.ResponseWriter, r *http.Request) {
-	vars := mux.Vars(r)
-	resourceId := vars["id"]
-	filename := vars["filename"]
-
-	etag := `"` + resourceId + "/" + filename + `"`
-	w.Header().Set("Etag", etag)
-	w.Header().Set("Cache-Control", "max-age=2592000")
-
-	if match := r.Header.Get("If-None-Match"); match != "" {
-		if strings.Contains(match, etag) {
-			w.WriteHeader(http.StatusNotModified)
-			return
-		}
-	}
-
-	resource, err := store.GetResourceByIdAndFilename(resourceId, filename)
-
-	if err != nil {
-		e.ErrorHandler(w, "RESOURCE_NOT_FOUND", err.Error())
-		return
-	}
-
-	w.WriteHeader(http.StatusOK)
-	w.Header().Set("Content-Type", "application/octet-stream")
-	w.Write(resource.Blob)
+type ResourceDelete struct {
+	Id int
 }
 
-func RegisterResourceRoutes(r *mux.Router) {
-	resourceRouter := r.PathPrefix("/").Subrouter()
-
-	resourceRouter.Use(AuthCheckerMiddleWare)
-
-	resourceRouter.HandleFunc("/api/resource/all", handleGetMyResources).Methods("GET")
-	resourceRouter.HandleFunc("/api/resource/", handleUploadResource).Methods("PUT")
-	resourceRouter.HandleFunc("/api/resource/{id}", handleDeleteResource).Methods("DELETE")
-	resourceRouter.HandleFunc("/r/{id}/{filename}", handleGetResource).Methods("GET")
+type ResourceService interface {
+	CreateResource(create *ResourceCreate) (*Resource, error)
+	FindResourceList(find *ResourceFind) ([]*Resource, error)
+	FindResource(find *ResourceFind) (*Resource, error)
+	DeleteResource(delete *ResourceDelete) error
 }