feature: allow file upload for getuploadparams* apis

Fixes #144

Signed-off-by: Abhishek Kumar <[email protected]>

Author: shwstppr

cmd/api.go

@@ -18,8 +18,15 @@
 package cmd
 
 import (
+	"bytes"
 	"errors"
 	"fmt"
+	"io"
+	"mime/multipart"
+	"net/http"
+	"os"
+	"path/filepath"
+	"reflect"
 	"strings"
 )
 
@@ -111,9 +118,116 @@ func init() {
 
 			if len(response) > 0 {
 				printResult(r.Config.Core.Output, response, filterKeys, excludeKeys)
+				if r.Config.HasShell {
+					apiName := strings.ToLower(api.Name)
+					if apiName == "getuploadparamsforiso" ||
+						apiName == "getuploadparamsforvolume" ||
+						apiName == "getuploadparamsfotemplate" {
+						promptForFileUpload(r, apiName, response)
+					}
+				}
 			}
-
 			return nil
 		},
 	}
 }
+
+func promptForFileUpload(r *Request, api string, response map[string]interface{}) {
+	fmt.Print("Enter path of the file(s) to upload (comma-separated): ")
+	var filePaths string
+	fmt.Scanln(&filePaths)
+	filePathsList := strings.FieldsFunc(filePaths, func(r rune) bool { return r == ',' })
+
+	var missingFiles []string
+	var validFiles []string
+	for _, filePath := range filePathsList {
+		filePath = strings.TrimSpace(filePath)
+		if filePath == "" {
+			continue
+		}
+		if _, err := os.Stat(filePath); os.IsNotExist(err) {
+			missingFiles = append(missingFiles, filePath)
+		} else {
+			validFiles = append(validFiles, filePath)
+		}
+	}
+	if len(missingFiles) > 0 {
+		fmt.Println("File(s) do not exist or are not accessible:", strings.Join(missingFiles, ", "))
+		return
+	}
+	if len(validFiles) == 0 {
+		fmt.Println("No valid files to upload.")
+		return
+	}
+	paramsRaw, ok := response["getuploadparams"]
+	if !ok || reflect.TypeOf(paramsRaw).Kind() != reflect.Map {
+		fmt.Println("Invalid response format for getuploadparams.")
+		return
+	}
+	params := paramsRaw.(map[string]interface{})
+	requiredKeys := []string{"postURL", "metadata", "signature", "expires"}
+	for _, key := range requiredKeys {
+		if _, ok := params[key]; !ok {
+			fmt.Printf("Missing required key '%s' in getuploadparams response.\n", key)
+			return
+		}
+	}
+
+	postURL, _ := params["postURL"].(string)
+	signature, _ := params["signature"].(string)
+	expires, _ := params["expires"].(string)
+	metadata, _ := params["metadata"].(string)
+
+	fmt.Println("Uploading files for", api, ":", validFiles)
+	spinner := r.Config.StartSpinner("uploading files, please wait...")
+	defer r.Config.StopSpinner(spinner)
+
+	for _, filePath := range validFiles {
+		if err := uploadFile(postURL, filePath, signature, expires, metadata); err != nil {
+			fmt.Println("Error uploading", filePath, ":", err)
+		}
+	}
+}
+
+// uploadFile uploads a single file to the given postURL with the required headers.
+func uploadFile(postURL, filePath, signature, expires, metadata string) error {
+	file, err := os.Open(filePath)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	var body bytes.Buffer
+	writer := multipart.NewWriter(&body)
+	part, err := writer.CreateFormFile("file", filepath.Base(filePath))
+	if err != nil {
+		return err
+	}
+	if _, err := io.Copy(part, file); err != nil {
+		return err
+	}
+	writer.Close()
+
+	req, err := http.NewRequest("POST", postURL, &body)
+	if err != nil {
+		return err
+	}
+	req.Header.Set("Content-Type", writer.FormDataContentType())
+	req.Header.Set("x-signature", signature)
+	req.Header.Set("x-expires", expires)
+	req.Header.Set("x-metadata", metadata)
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusCreated {
+		respBody, _ := io.ReadAll(resp.Body)
+		return fmt.Errorf("upload failed: %s", string(respBody))
+	}
+	fmt.Println("Upload successful for:", filePath)
+	return nil
+}