mirror of
https://github.com/usememos/memos.git
synced 2024-12-19 09:02:49 +03:00
131 lines
2.8 KiB
Go
131 lines
2.8 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"memos/api/e"
|
|
"memos/store"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
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,
|
|
})
|
|
}
|
|
|
|
func handleUploadResource(w http.ResponseWriter, r *http.Request) {
|
|
userId, _ := GetUserIdInSession(r)
|
|
|
|
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,
|
|
})
|
|
}
|
|
|
|
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,
|
|
})
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
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")
|
|
}
|