--- /dev/null
+package api
+
+import (
+ "bytes"
+ "encoding/base64"
+ "encoding/json"
+ "fmt"
+ "io"
+ "io/ioutil"
+ "net/http"
+ "os"
+ "os/user"
+ "log"
+ "path"
+ "strconv"
+
+ image "bpa-restapi-agent/internal/app"
+
+ "github.com/gorilla/mux"
+)
+
+// imageHandler is used to store backend implementations objects
+// Also simplifies mocking for unit testing purposes
+type imageHandler struct {
+ // Interface that implements Image operations
+ // We will set this variable with a mock interface for testing
+ client image.ImageManager
+ dirPath string
+}
+
+// CreateHandler handles creation of the image entry in the database
+
+func (h imageHandler) createHandler(w http.ResponseWriter, r *http.Request) {
+ var v image.Image
+
+ // Implemenation using multipart form
+ // Review and enable/remove at a later date
+ // Set Max size to 16mb here
+ err := r.ParseMultipartForm(16777216)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusUnprocessableEntity)
+ return
+ }
+
+ jsn := bytes.NewBuffer([]byte(r.FormValue("metadata")))
+ err = json.NewDecoder(jsn).Decode(&v)
+ switch {
+ case err == io.EOF:
+ http.Error(w, "Empty body", http.StatusBadRequest)
+ return
+ case err != nil:
+ http.Error(w, err.Error(), http.StatusUnprocessableEntity)
+ return
+ }
+
+ // Name is required.
+ if v.ImageName == "" {
+ http.Error(w, "Missing name in POST request", http.StatusBadRequest)
+ return
+ }
+
+ // Owner is required.
+ if v.Owner == "" {
+ http.Error(w, "Missing Owner in POST request", http.StatusBadRequest)
+ return
+ }
+
+ if v.ImageLength == 0 {
+ e := "Improper upload length"
+ w.WriteHeader(http.StatusBadRequest)
+ w.Write([]byte(e))
+ return
+ }
+
+ //Create file directory
+ dir, err := createFileDir(v.Type)
+ if err != nil {
+ log.Fatal("Error creating file server directory", err)
+ }
+
+ //Read the file section and ignore the header
+ file, _, err := r.FormFile("file")
+ if err != nil {
+ http.Error(w, "Unable to process file", http.StatusUnprocessableEntity)
+ return
+ }
+
+ defer file.Close()
+
+ //Convert the file content to base64 for storage
+ content, err := ioutil.ReadAll(file)
+ if err != nil {
+ http.Error(w, "Unable to read file", http.StatusUnprocessableEntity)
+ return
+ }
+
+ v.Config = base64.StdEncoding.EncodeToString(content)
+
+ ret, err := h.client.Create(v)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+ h.dirPath = dir
+ filePath := path.Join(h.dirPath, v.ImageName)
+ file1, err := os.Create(filePath)
+ if err != nil {
+ e := "Error creating file in filesystem"
+ log.Printf("%s %s\n", e, err)
+ w.WriteHeader(http.StatusInternalServerError)
+ return
+ }
+
+ defer file1.Close()
+
+ w.Header().Set("Content-Type", "application/json")
+ w.WriteHeader(http.StatusCreated)
+ err = json.NewEncoder(w).Encode(ret)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+}
+
+// Create file
+
+func createFileDir(dirName string) (string, error) {
+ u, err := user.Current()
+ if err != nil {
+ log.Println("Error while fetching user home directory", err)
+ return "", err
+ }
+ home := u.HomeDir
+ dirPath := path.Join(home, "images", dirName)
+ err = os.MkdirAll(dirPath, 0744)
+ if err != nil {
+ log.Println("Error while creating file server directory", err)
+ return "", err
+ }
+ return dirPath, nil
+}
+
+// getHandler handles GET operations on a particular name
+// Returns an Image
+func (h imageHandler) getHandler(w http.ResponseWriter, r *http.Request) {
+ vars := mux.Vars(r)
+ // ownerName := vars["owner"]
+ // clusterName := vars["clustername"]
+ imageName := vars["imgname"]
+
+ ret, err := h.client.Get(imageName)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+
+ w.Header().Set("Content-Type", "application/json")
+ w.WriteHeader(http.StatusOK)
+ err = json.NewEncoder(w).Encode(ret)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+}
+
+// deleteHandler handles DELETE operations on a particular record
+func (h imageHandler) deleteHandler(w http.ResponseWriter, r *http.Request) {
+ vars := mux.Vars(r)
+ // ownerName := vars["owner"]
+ // clusterName := vars["clustername"]
+ imageName := vars["imgname"]
+
+ err := h.client.Delete(imageName)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+
+ w.WriteHeader(http.StatusNoContent)
+}
+
+// UpdateHandler handles Update operations on a particular image
+func (h imageHandler) updateHandler(w http.ResponseWriter, r *http.Request) {
+ var v image.Image
+ vars := mux.Vars(r)
+ imageName := vars["imgname"]
+
+ err := r.ParseMultipartForm(16777216)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusUnprocessableEntity)
+ return
+ }
+
+ jsn := bytes.NewBuffer([]byte(r.FormValue("metadata")))
+ err = json.NewDecoder(jsn).Decode(&v)
+ switch {
+ case err == io.EOF:
+ http.Error(w, "Empty body", http.StatusBadRequest)
+ return
+ case err != nil:
+ http.Error(w, err.Error(), http.StatusUnprocessableEntity)
+ return
+ }
+
+ // Name is required.
+ if v.ImageName == "" {
+ http.Error(w, "Missing name in PUT request", http.StatusBadRequest)
+ return
+ }
+
+ // Owner is required.
+ if v.Owner == "" {
+ http.Error(w, "Missing Owner in PUT request", http.StatusBadRequest)
+ return
+ }
+
+ //Read the file section and ignore the header
+ file, _, err := r.FormFile("file")
+ if err != nil {
+ http.Error(w, "Unable to process file", http.StatusUnprocessableEntity)
+ return
+ }
+
+ defer file.Close()
+
+ //Convert the file content to base64 for storage
+ content, err := ioutil.ReadAll(file)
+ if err != nil {
+ http.Error(w, "Unable to read file", http.StatusUnprocessableEntity)
+ return
+ }
+
+ v.Config = base64.StdEncoding.EncodeToString(content)
+
+ ret, err := h.client.Update(imageName, v)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+
+ w.Header().Set("Content-Type", "application/json")
+ w.WriteHeader(http.StatusCreated)
+ err = json.NewEncoder(w).Encode(ret)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+}
+
+// File upload is handled by the patchHandler
+
+func (h imageHandler) patchHandler(w http.ResponseWriter, r *http.Request) {
+ log.Println("going to patch file")
+ vars := mux.Vars(r)
+ imageName := vars["imgname"]
+ file, err := h.client.Get(imageName)
+ if err != nil {
+ http.Error(w, err.Error(), http.StatusInternalServerError)
+ return
+ }
+ if *file.UploadComplete == true {
+ e := "Upload already completed"
+ w.WriteHeader(http.StatusUnprocessableEntity)
+ w.Write([]byte(e))
+ return
+ }
+ off, err := strconv.Atoi(r.Header.Get("Upload-Offset"))
+ if err != nil {
+ log.Println("Improper upload offset", err)
+ w.WriteHeader(http.StatusBadRequest)
+ return
+ }
+ log.Printf("Upload offset %d\n", off)
+ if *file.ImageOffset != off {
+ e := fmt.Sprintf("Expected Offset %d got offset %d", *file.ImageOffset, off)
+ w.WriteHeader(http.StatusConflict)
+ w.Write([]byte(e))
+ return
+ }
+
+ log.Println("Content length is", r.Header.Get("Content-Length"))
+ clh := r.Header.Get("Content-Length")
+ cl, err := strconv.Atoi(clh)
+ if err != nil {
+ log.Println("unknown content length")
+ w.WriteHeader(http.StatusInternalServerError)
+ return
+ }
+
+ if cl != (file.ImageLength - *file.ImageOffset) {
+ e := fmt.Sprintf("Content length doesn't match upload length. Expected content length %d got %d", file.ImageLength-*file.ImageOffset, cl)
+ log.Println(e)
+ w.WriteHeader(http.StatusBadRequest)
+ w.Write([]byte(e))
+ return
+ }
+
+ body, err := ioutil.ReadAll(r.Body)
+ if err != nil {
+ log.Printf("Received file partially %s\n", err)
+ log.Println("Size of received file ", len(body))
+ }
+
+ u, err := user.Current()
+ if err != nil {
+ log.Println("Error while fetching user home directory", err)
+ return
+ }
+ home := u.HomeDir
+ dir := path.Join(home, "images", file.Type)
+ h.dirPath = dir
+ fp := fmt.Sprintf("%s/%s", h.dirPath, imageName)
+ f, err := os.OpenFile(fp, os.O_APPEND|os.O_WRONLY, 0644)
+ if err != nil {
+ log.Printf("unable to open file %s\n", err)
+ w.WriteHeader(http.StatusInternalServerError)
+ return
+ }
+ defer f.Close()
+
+ n, err := f.WriteAt(body, int64(off))
+ if err != nil {
+ log.Printf("unable to write %s", err)
+ w.WriteHeader(http.StatusInternalServerError)
+ return
+ }
+ log.Println("number of bytes written ", n)
+ no := *file.ImageOffset + n
+ file.ImageOffset = &no
+
+ uo := strconv.Itoa(*file.ImageOffset)
+ w.Header().Set("Upload-Offset", uo)
+ if *file.ImageOffset == file.ImageLength {
+ log.Println("upload completed successfully")
+ *file.UploadComplete = true
+ }
+
+ // err = h.updateFile(file)
+ // if err != nil {
+ // log.Println("Error while updating file", err)
+ // w.WriteHeader(http.StatusInternalServerError)
+ // return
+ // }
+ w.WriteHeader(http.StatusNoContent)
+
+ return
+
+}
Code Review / icn.git / blobdiff