add error handling

cautils/getter/getpoliciesutils.go

@@ -21,6 +21,7 @@ func GetDefaultPath(name string) string {
 	return defaultfilePath
 }
 
+// Save control as json in file
 func SaveControlInFile(control *reporthandling.Control, pathStr string) error {
 	encodedData, err := json.Marshal(control)
 	if err != nil {

cautils/getter/loadpolicy.go

@@ -26,6 +26,7 @@ func NewLoadPolicy(filePath string) *LoadPolicy {
 	}
 }
 
+// Return control from file
 func (lp *LoadPolicy) GetControl(controlName string) (*reporthandling.Control, error) {
 
 	control := &reporthandling.Control{}

clihandler/cmd/download.go

@@ -18,7 +18,10 @@ var downloadCmd = &cobra.Command{
 	Long:  ``,
 	Args: func(cmd *cobra.Command, args []string) error {
 		if len(args) != 2 {
-			return fmt.Errorf("requires two arguments : framework <framework-name>")
+			return fmt.Errorf("requires two arguments : framework/control <framework-name>/<control-name>")
+		}
+		if !strings.EqualFold(args[0], "framework") && !strings.EqualFold(args[0], "control") {
+			return fmt.Errorf("invalid parameter '%s'. Supported parameters: framework, control", args[0])
 		}
 		return nil
 	},

clihandler/cmd/scan.go

@@ -1,6 +1,9 @@
 package cmd
 
 import (
+	"fmt"
+	"strings"
+
 	"github.com/armosec/kubescape/cautils"
 	"github.com/spf13/cobra"
 )
@@ -12,6 +15,15 @@ var scanCmd = &cobra.Command{
 	Use:   "scan <command>",
 	Short: "Scan the current running cluster or yaml files",
 	Long:  `The action you want to perform`,
+	Args: func(cmd *cobra.Command, args []string) error {
+		if len(args) == 0 {
+			return fmt.Errorf("requires one  argument: framework/control")
+		}
+		if !strings.EqualFold(args[0], "framework") && !strings.EqualFold(args[0], "control") {
+			return fmt.Errorf("invalid parameter '%s'. Supported parameters: framework, control", args[0])
+		}
+		return nil
+	},
 	Run: func(cmd *cobra.Command, args []string) {
 	},
 }