fix: Improve parsing of boolean action parameters

fortify · Oct 30, 2024 · d3b6f4c · d3b6f4c
1 parent 27d5a1a
commit d3b6f4c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/...fcli-common/src/main/java/com/fortify/cli/common/action/runner/ActionParameterHelper.java b/...fcli-common/src/main/java/com/fortify/cli/common/action/runner/ActionParameterHelper.java
@@ -39,6 +39,7 @@ private static final void addOptionDescriptor(List<IOptionDescriptor> result, Ac
                 .name(getOptionName(parameter.getName()))
                 .aliases(getOptionAliases(parameter.getCliAliasesArray()))
                 .description(parameter.getDescription())
+                .bool(parameter.getType()!=null && parameter.getType().equals("boolean"))
                 .build());
     }
 

diff --git a/fcli-core/fcli-common/src/main/java/com/fortify/cli/common/cli/util/SimpleOptionsParser.java b/fcli-core/fcli-common/src/main/java/com/fortify/cli/common/cli/util/SimpleOptionsParser.java
@@ -45,6 +45,7 @@ public static interface IOptionDescriptor {
         String getName();
         List<String> getAliases();
         String getDescription();
+        boolean isBool();
 
         default String getOptionNamesAndAliasesString(String delimiter) {
             var name = getName();
@@ -59,6 +60,7 @@ public static class OptionDescriptor implements IOptionDescriptor {
         private final String name;
         private final List<String> aliases;
         private final String description;
+        private final boolean bool;
     }
 
     @Data
@@ -107,13 +109,16 @@ private Map<String, String> parseArgs(List<String> validationErrors, String[] ar
                 var argWithPossibleValue = argsDeque.pop();
                 var argElts = argWithPossibleValue.split("=", 2);
                 var arg = argElts[0];
-                if ( !optionsByNameAndAliases.containsKey(arg) ) {
+                var optionDescriptor = optionsByNameAndAliases.get(arg);
+                if ( optionDescriptor==null ) {
                     validationErrors.add("Unknown command line option: "+arg);
                 } else if ( argElts.length==2 ) {
                     result.put(arg, argElts[1]);
                 } else {
                     var nextArg = argsDeque.peek(); 
-                    var value = optionsByNameAndAliases.containsKey(nextArg) ? null : argsDeque.pop();
+                    var value = nextArg==null || optionsByNameAndAliases.containsKey(nextArg) 
+                            ? (optionDescriptor.isBool() ? "true" : null)
+                            : argsDeque.pop();
                     result.put(arg, value);
                 }
             }

diff --git a/fcli-core/fcli-fod/src/main/resources/com/fortify/cli/fod/actions/zip/setup-release.yaml b/fcli-core/fcli-fod/src/main/resources/com/fortify/cli/fod/actions/zip/setup-release.yaml
@@ -62,11 +62,13 @@ parameters:
   - group: sast_setup_opts
     required: false
     name: use-aviator
-    description: "See `fcli fod sast-scan setup`"      
+    description: "See `fcli fod sast-scan setup`"
+    type: boolean      
   - group: sast_setup_opts
     required: false
     name: oss
     description: "See `fcli fod sast-scan setup`" 
+    type: boolean
   - name: attributes
     required: false
     cliAliases: attrs