Merge pull request #16 from ermetic-research/feat/kspm

Add KSPM module support in CLI and parsing

README.md

@@ -135,7 +135,7 @@ cnappgoat provision <scenario-1> <scenario-2> <scenario-N>
 
 #### Flags:
 
-- `--module` - Filters scenarios by module (e.g. CSPM, CIEM, CWPP, DSPM, etc.)
+- `--module` - Filters scenarios by module (e.g. CSPM, CIEM, CWPP, DSPM, KSPM etc.)
 - `--platform` - Filters scenarios by platform (e.g. AWS, Azure, GCP)
 - `--force` - Enables force mode (unlock locked stacks with pulumi cancel)
 - `--debug` - Enables debug logging.

cmd/commands/common/command.go

@@ -66,7 +66,7 @@ func CommandFlags() []cli.Flag {
 		},
 		&cli.StringFlag{
 			Name:    "module",
-			Usage:   "CNAPPgoat module to operate on, e.g. 'CIEM, 'CSPM', 'CWPP', 'DSPM', 'IAC'",
+			Usage:   "CNAPPgoat module to operate on, e.g. 'CIEM, 'CSPM', 'CWPP', 'DSPM', 'KSPM'",
 			Aliases: []string{"m"},
 		},
 		&cli.StringFlag{

scenario.go

@@ -53,6 +53,7 @@ const (
 	CWPP Module = "CWPP"
 	DSPM Module = "DSPM"
 	IAC  Module = "IAC"
+	KSPM Module = "KSPM"
 )
 
 type Platform string
@@ -83,6 +84,8 @@ func (m Module) String() string {
 		return "DSPM"
 	case IAC:
 		return "IAC"
+	case KSPM:
+		return "KSPM"
 	default:
 		return ""
 	}
@@ -171,6 +174,8 @@ func ModuleFromString(name string) (Module, error) {
 		return DSPM, nil
 	case strings.ToLower(IAC.String()):
 		return IAC, nil
+	case strings.ToLower(KSPM.String()):
+		return KSPM, nil
 	default:
 		return "", errors.New("unknown module name: " + name)
 	}