Added some configurable options for ProxMox API via CloudConfig

cmd/cmd_instance.go

@@ -96,6 +96,8 @@ func instanceCreateCommandHandler(cmd *cobra.Command, args []string) {
 	if err != nil {
 		exitWithError(err.Error())
 	}
+
+	fmt.Printf("%s instance '%s' created...\n", c.CloudConfig.Platform, c.RunConfig.InstanceName)
 }
 
 func instanceListCommand() *cobra.Command {

go.mod

@@ -23,6 +23,7 @@ require (
 	github.com/docker/distribution v2.8.0+incompatible
 	github.com/docker/docker v20.10.7+incompatible
 	github.com/docker/go-connections v0.4.0 // indirect
+	github.com/docker/go-units v0.4.0 // indirect
 	github.com/dustin/go-humanize v1.0.0
 	github.com/form3tech-oss/jwt-go v3.2.5+incompatible // indirect
 	github.com/go-errors/errors v1.0.1

proxmox/proxmox_checks.go

@@ -8,6 +8,7 @@ import (
 	"fmt"
 	"io/ioutil"
 	"net/http"
+	"strings"
 )
 
 type pData struct {
@@ -22,12 +23,23 @@ type sData struct {
 	Active  int    `json:"active"`
 	Enabled int    `json:"enabled"`
 	Storage string `json:"storage"`
+	Content string `json:"content"`
 }
 
 type sDataArr struct {
 	Data []sData `json:"data"`
 }
 
+type bData struct {
+	Active int    `json:"active"`
+	Type   string `json:"type"`
+	Iface  string `json:"iface"`
+}
+
+type bDataArr struct {
+	Data []bData `json:"data"`
+}
+
 // CheckInit return custom error on {"data": null} or {"data": []} result come from ProxMox API /api2/json/pools
 func (p *ProxMox) CheckInit() error {
 
@@ -75,7 +87,7 @@ func (p *ProxMox) CheckInit() error {
 }
 
 // CheckStorage return error when not found configured storage or any storages via ProxMox API
-func (p *ProxMox) CheckStorage(storage string) error {
+func (p *ProxMox) CheckStorage(storage string, stype string) error {
 
 	var err error
 
@@ -87,6 +99,8 @@ func (p *ProxMox) CheckStorage(storage string) error {
 	enb := errors.New("storage is disabled: " + storage)
 	ect := errors.New("storage is not active: " + storage)
 	ecs := errors.New("not found storage: " + storage)
+	eim := errors.New("storage is not configured for containing disk images: " + storage)
+	eis := errors.New("storage is not configured for containing iso images: " + storage)
 
 	req, err := http.NewRequest("GET", p.apiURL+"/api2/json/nodes/"+p.nodeNAME+"/storage", &b)
 	if err != nil {
@@ -137,6 +151,94 @@ func (p *ProxMox) CheckStorage(storage string) error {
 				return enb
 			}
 
+			if stype == "images" {
+
+				if !strings.Contains(v.Content, stype) {
+					return eim
+				}
+
+			} else if stype == "iso" {
+
+				if !strings.Contains(v.Content, stype) {
+					return eis
+				}
+
+			} else {
+				return errors.New("unknown type of storage")
+			}
+
+			return nil
+
+		}
+	}
+
+	return ecs
+
+}
+
+// CheckBridge return error when not found configured bridge any network interfaces via ProxMox API
+func (p *ProxMox) CheckBridge(bridge string) error {
+
+	var err error
+
+	var brs bDataArr
+
+	var b bytes.Buffer
+
+	edk := errors.New("no any bridges is configured")
+	ebr := errors.New("is not a bridge: " + bridge)
+	ect := errors.New("bridge is not active: " + bridge)
+	ecs := errors.New("not found bridge: " + bridge)
+
+	req, err := http.NewRequest("GET", p.apiURL+"/api2/json/nodes/"+p.nodeNAME+"/network", &b)
+	if err != nil {
+		return err
+	}
+
+	tr := &http.Transport{
+		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
+	}
+	client := &http.Client{Transport: tr}
+
+	req.Header.Add("Authorization", "PVEAPIToken="+p.tokenID+"="+p.secret)
+
+	resp, err := client.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return err
+	}
+
+	err = json.Unmarshal(body, &brs)
+	if err != nil {
+		return err
+	}
+
+	if err == nil && brs.Data == nil {
+		return edk
+	}
+
+	debug := false
+	if debug {
+		fmt.Println(string(body))
+	}
+
+	for _, v := range brs.Data {
+
+		if v.Iface == bridge {
+
+			if v.Active != 1 {
+				return ect
+			}
+
+			if v.Type != "bridge" {
+				return ebr
+			}
+
 			return nil
 
 		}
@@ -168,7 +270,7 @@ func (p *ProxMox) CheckResult(body []byte) error {
 }
 
 // CheckResultType return error or custom error based on type of check, when {"data": null} or {"data": []} result come from ProxMox API
-func (p *ProxMox) CheckResultType(body []byte, rtype string) error {
+func (p *ProxMox) CheckResultType(body []byte, rtype string, rname string) error {
 
 	var err error
 
@@ -177,17 +279,17 @@ func (p *ProxMox) CheckResultType(body []byte, rtype string) error {
 
 	switch rtype {
 	case "createimage":
-		ecs = errors.New("can not create disk image in 'local' storage, " + edef)
+		ecs = errors.New("can not create disk image in " + rname + " storage, " + edef)
 	case "listimages":
-		ecs = errors.New("no disk images found in 'local' storage or " + edef)
+		ecs = errors.New("no disk images found in " + rname + " storage or " + edef)
 	case "createinstance":
-		ecs = errors.New("can not create machine instance with disk image from 'local' storage, " + edef)
+		ecs = errors.New("can not create machine instance with disk image from " + rname + " image/storage, " + edef)
 	case "getnextid":
 		ecs = errors.New("can not get next id, " + edef)
 	case "movdisk":
-		ecs = errors.New("can not move iso to raw disk on 'local-lvm' storage, " + edef)
+		ecs = errors.New("can not move iso to raw disk on " + rname + " storage, " + edef)
 	case "addvirtiodisk":
-		ecs = errors.New("can not add virtio disk from 'local' storage, " + edef)
+		ecs = errors.New("can not add virtio disk from " + rname + " storage, " + edef)
 	case "bootorderset":
 		ecs = errors.New("can not set boot order, " + edef)
 	}

proxmox/proxmox_image.go

@@ -66,8 +66,13 @@ func (p *ProxMox) CreateImage(ctx *lepton.Context, imagePath string) error {
 	c := ctx.Config()
 
 	imageName := c.CloudConfig.ImageName
+	isoStorageName := c.CloudConfig.IsoStorageName
 
-	err = p.CheckStorage("local")
+	if isoStorageName == "" {
+		isoStorageName = "local"
+	}
+
+	err = p.CheckStorage(isoStorageName, "iso")
 	if err != nil {
 		return err
 	}
@@ -105,7 +110,7 @@ func (p *ProxMox) CreateImage(ctx *lepton.Context, imagePath string) error {
 
 	w.Close()
 
-	req, err := http.NewRequest("POST", p.apiURL+"/api2/json/nodes/"+p.nodeNAME+"/storage/local/upload", &b)
+	req, err := http.NewRequest("POST", p.apiURL+"/api2/json/nodes/"+p.nodeNAME+"/storage/"+isoStorageName+"/upload", &b)
 	if err != nil {
 		fmt.Println(err)
 		return err
@@ -131,7 +136,7 @@ func (p *ProxMox) CreateImage(ctx *lepton.Context, imagePath string) error {
 		return err
 	}
 
-	err = p.CheckResultType(body, "createimage")
+	err = p.CheckResultType(body, "createimage", isoStorageName)
 	if err != nil {
 		return err
 	}
@@ -163,7 +168,22 @@ type ImageInfo struct {
 // ListImages lists images on ProxMox
 func (p *ProxMox) ListImages(ctx *lepton.Context) error {
 
-	req, err := http.NewRequest("GET", p.apiURL+"/api2/json/nodes/"+p.nodeNAME+"/storage/local/content", nil)
+	var err error
+
+	c := ctx.Config()
+
+	isoStorageName := c.CloudConfig.StorageName
+
+	if isoStorageName == "" {
+		isoStorageName = "local"
+	}
+
+	err = p.CheckStorage(isoStorageName, "iso")
+	if err != nil {
+		return err
+	}
+
+	req, err := http.NewRequest("GET", p.apiURL+"/api2/json/nodes/"+p.nodeNAME+"/storage/"+isoStorageName+"/content", nil)
 	if err != nil {
 		fmt.Println(err)
 		return err
@@ -187,7 +207,7 @@ func (p *ProxMox) ListImages(ctx *lepton.Context) error {
 		return err
 	}
 
-	err = p.CheckResultType(body, "listimages")
+	err = p.CheckResultType(body, "listimages", isoStorageName)
 	if err != nil {
 		return err
 	}