mcism_resource_manage_image.go

package main

import (
	"encoding/json"
	"fmt"
	"net/http"
	"strings"

	"github.com/labstack/echo"
)

type nsReq struct {
	Id          string `json:"id"`
	Name        string `json:"name"`
	Description string `json:"description"`
}

type nsInfo struct {
	Id          string `json:"id"`
	Name        string `json:"name"`
	Description string `json:"description"`
}

type imageReq struct {
	Id           string `json:"id"`
	Name         string `json:"name"`
	CreationDate string `json:"creationDate"`
	Csp          string `json:"csp"`
	CspImageId   string `json:"cspImageId"`
	Description  string `json:"description"`
}

type imageInfo struct {
	Id           string `json:"id"`
	Name         string `json:"name"`
	CreationDate string `json:"creationDate"`
	Csp          string `json:"csp"`
	CspImageId   string `json:"cspImageId"`
	Description  string `json:"description"`
}

// MCIS API Proxy: Ns
func restPostNs(c echo.Context) error {

	u := &nsReq{}
	if err := c.Bind(u); err != nil {
		return err
	}

	fmt.Println("[Creating Ns]")
	createNs(u)
	return c.JSON(http.StatusCreated, u)

}

func restGetNs(c echo.Context) error {
	id := c.Param("nsId")

	content := nsInfo{}

	fmt.Println("[Get ns for id]" + id)
	key := "/ns/" + id
	fmt.Println(key)

	keyValue, _ := store.Get(key)
	fmt.Println("<" + keyValue.Key + "> \n" + keyValue.Value)
	fmt.Println("===============================================")

	json.Unmarshal([]byte(keyValue.Value), &content)
	content.Id = id // Optional. Can be omitted.

	return c.JSON(http.StatusOK, &content)

}

func restGetAllNs(c echo.Context) error {

	var content struct {
		//Name string     `json:"name"`
		Ns []nsInfo `json:"ns"`
	}

	nsList := getNsList()

	for _, v := range nsList {

		key := "/ns/" + v
		fmt.Println(key)
		keyValue, _ := store.Get(key)
		fmt.Println("<" + keyValue.Key + "> \n" + keyValue.Value)
		nsTmp := nsInfo{}
		json.Unmarshal([]byte(keyValue.Value), &nsTmp)
		nsTmp.Id = v
		content.Ns = append(content.Ns, nsTmp)

	}
	fmt.Printf("content %+v\n", content)

	return c.JSON(http.StatusOK, &content)

}

func restPutNs(c echo.Context) error {
	return nil
}

func restDelNs(c echo.Context) error {

	id := c.Param("nsId")

	err := delNs(id)
	if err != nil {
		cblog.Error(err)
		mapA := map[string]string{"message": "Failed to delete the ns"}
		return c.JSON(http.StatusFailedDependency, &mapA)
	}

	mapA := map[string]string{"message": "The ns has been deleted"}
	return c.JSON(http.StatusOK, &mapA)
}

func restDelAllNs(c echo.Context) error {

	nsList := getNsList()

	for _, v := range nsList {
		err := delNs(v)
		if err != nil {
			cblog.Error(err)
			mapA := map[string]string{"message": "Failed to delete All nss"}
			return c.JSON(http.StatusFailedDependency, &mapA)
		}
	}

	mapA := map[string]string{"message": "All nss has been deleted"}
	return c.JSON(http.StatusOK, &mapA)

}

func createNs(u *nsReq) {

	u.Id = genUuid()

	// TODO here: implement the logic

	fmt.Println("=========================== PUT createNs")
	Key := "/ns/" + u.Id
	mapA := map[string]string{"name": u.Name, "description": u.Description}
	Val, _ := json.Marshal(mapA)
	err := store.Put(string(Key), string(Val))
	if err != nil {
		cblog.Error(err)
	}
	keyValue, _ := store.Get(string(Key))
	fmt.Println("<" + keyValue.Key + "> \n" + keyValue.Value)
	fmt.Println("===========================")

}

func getNsList() []string {

	fmt.Println("[Get nss")
	key := "/ns"
	fmt.Println(key)

	keyValue, _ := store.GetList(key, true)
	var nsList []string
	for _, v := range keyValue {
		//if !strings.Contains(v.Key, "vm") {
		nsList = append(nsList, strings.TrimPrefix(v.Key, "/ns/"))
		//}
	}
	for _, v := range nsList {
		fmt.Println("<" + v + "> \n")
	}
	fmt.Println("===============================================")
	return nsList

}

func delNs(Id string) error {

	fmt.Println("[Delete ns] " + Id)

	key := "/ns/" + Id
	fmt.Println(key)

	// delete mcis info
	err := store.Delete(key)
	if err != nil {
		cblog.Error(err)
		return err
	}

	return nil
}

/* FYI
e.POST("/resources/image", restPostImage)
e.GET("/resources/image/:id", restGetImage)
e.GET("/resources/image", restGetAllImage)
e.PUT("/resources/image/:id", restPutImage)
e.DELETE("/resources/image/:id", restDelImage)
e.DELETE("/resources/image", restDelAllImage)
*/

// MCIS API Proxy: Image
func restPostImage(c echo.Context) error {

	nsId := c.Param("nsId")

	u := &imageReq{}
	if err := c.Bind(u); err != nil {
		return err
	}

	action := c.QueryParam("action")
	fmt.Println("[POST Image requested action: " + action)
	if action == "create" {
		fmt.Println("[Creating Image]")
		createImage(nsId, u)
		return c.JSON(http.StatusCreated, u)

	} else { //if action == "register" {
		fmt.Println("[Registering Image]")
		registerImage(nsId, u)
		return c.JSON(http.StatusCreated, u)

	}

}

func restGetImage(c echo.Context) error {

	nsId := c.Param("nsId")

	id := c.Param("imageId")

	content := imageInfo{}
	/*
		var content struct {
			Id    string `json:"id"`
			Name         string `json:"name"`
			CreationDate string `json:"creationDate"`
			Csp          string `json:"csp"`
			CspImageId   string `json:"cspImageId"`
			Description  string `json:"description"`
		}
	*/

	fmt.Println("[Get image for id]" + id)
	key := "/ns/" + nsId + "/resources/image/" + id
	fmt.Println(key)

	keyValue, _ := store.Get(key)
	fmt.Println("<" + keyValue.Key + "> \n" + keyValue.Value)
	fmt.Println("===============================================")

	json.Unmarshal([]byte(keyValue.Value), &content)
	content.Id = id // Optional. Can be omitted.

	return c.JSON(http.StatusOK, &content)

}

func restGetAllImage(c echo.Context) error {

	nsId := c.Param("nsId")

	var content struct {
		//Name string     `json:"name"`
		Image []imageInfo `json:"image"`
	}

	imageList := getImageList(nsId)

	for _, v := range imageList {

		key := "/ns/" + nsId + "/resources/image/" + v
		fmt.Println(key)
		keyValue, _ := store.Get(key)
		fmt.Println("<" + keyValue.Key + "> \n" + keyValue.Value)
		imageTmp := imageInfo{}
		json.Unmarshal([]byte(keyValue.Value), &imageTmp)
		imageTmp.Id = v
		content.Image = append(content.Image, imageTmp)

	}
	fmt.Printf("content %+v\n", content)

	return c.JSON(http.StatusOK, &content)

}

func restPutImage(c echo.Context) error {
	//nsId := c.Param("nsId")

	return nil
}

func restDelImage(c echo.Context) error {

	nsId := c.Param("nsId")
	id := c.Param("imageId")

	err := delImage(nsId, id)
	if err != nil {
		cblog.Error(err)
		mapA := map[string]string{"message": "Failed to delete the image"}
		return c.JSON(http.StatusFailedDependency, &mapA)
	}

	mapA := map[string]string{"message": "The image has been deleted"}
	return c.JSON(http.StatusOK, &mapA)
}

func restDelAllImage(c echo.Context) error {

	nsId := c.Param("nsId")

	imageList := getImageList(nsId)

	for _, v := range imageList {
		err := delImage(nsId, v)
		if err != nil {
			cblog.Error(err)
			mapA := map[string]string{"message": "Failed to delete All images"}
			return c.JSON(http.StatusFailedDependency, &mapA)
		}
	}

	mapA := map[string]string{"message": "All images has been deleted"}
	return c.JSON(http.StatusOK, &mapA)

}

func createImage(nsId string, u *imageReq) {

	u.Id = genUuid()

	// TODO here: implement the logic
	// Option 1. Let the user upload an image file.
	// Option 2. Let the user specify the URL of an image file.
	// Option 3. Let the user snapshot specific VM for the new image file.

	// cb-store
	fmt.Println("=========================== PUT createImage")
	Key := "/ns/" + nsId + "/resources/image/" + u.Id
	mapA := map[string]string{"name": u.Name, "description": u.Description, "creationDate": u.CreationDate, "csp": u.Csp, "cspImageId": u.CspImageId}
	Val, _ := json.Marshal(mapA)
	err := store.Put(string(Key), string(Val))
	if err != nil {
		cblog.Error(err)
	}
	keyValue, _ := store.Get(string(Key))
	fmt.Println("<" + keyValue.Key + "> \n" + keyValue.Value)
	fmt.Println("===========================")

}

func registerImage(nsId string, u *imageReq) {

	u.Id = genUuid()

	// TODO here: implement the logic
	// - Fetch the image info from CSP.

	// cb-store
	fmt.Println("=========================== PUT registerImage")
	Key := "/ns/" + nsId + "/resources/image/" + u.Id
	mapA := map[string]string{"name": u.Name, "description": u.Description, "creationDate": u.CreationDate, "csp": u.Csp, "cspImageId": u.CspImageId}
	Val, _ := json.Marshal(mapA)
	err := store.Put(string(Key), string(Val))
	if err != nil {
		cblog.Error(err)
	}
	keyValue, _ := store.Get(string(Key))
	fmt.Println("<" + keyValue.Key + "> \n" + keyValue.Value)
	fmt.Println("===========================")

}

func getImageList(nsId string) []string {

	fmt.Println("[Get images")
	key := "/ns/" + nsId + "/resources/image"
	fmt.Println(key)

	keyValue, _ := store.GetList(key, true)
	var imageList []string
	for _, v := range keyValue {
		//if !strings.Contains(v.Key, "vm") {
		imageList = append(imageList, strings.TrimPrefix(v.Key, "/ns/"+nsId+"/resources/image/"))
		//}
	}
	for _, v := range imageList {
		fmt.Println("<" + v + "> \n")
	}
	fmt.Println("===============================================")
	return imageList

}

func delImage(nsId string, Id string) error {

	fmt.Println("[Delete image] " + Id)

	key := "/ns/" + nsId + "/resources/image/" + Id
	fmt.Println(key)

	// delete mcis info
	err := store.Delete(key)
	if err != nil {
		cblog.Error(err)
		return err
	}

	return nil
}