add support for listing supported ecosystem

Signed-off-by: Keshav Priyadarshi <git@keshav.space>

Author: keshav-space

vulntotal/vulntotal-cli.py

@@ -41,7 +41,7 @@ def get_help():
    ╚██╔╝  ╚██████╔╝███████╗██║ ╚███║   ██║   ╚█████╔╝   ██║   ██║  ██║███████╗
     ╚═╝    ╚═════╝ ╚══════╝╚═╝  ╚══╝   ╚═╝    ╚════╝    ╚═╝   ╚═╝  ╚═╝╚══════╝
 
-Usage: vulntotal-cli.py [-h] [-l] [-r] [-d validator [validator ...]]
+Usage: vulntotal-cli.py [-h] [-l] [-le] [-r] [-d validator [validator ...]]
                         [-e validator [validator ...]] [-p PURL]
 
 Description:
@@ -52,6 +52,7 @@ def get_help():
 Optional arguments:
   -h, --help                        Show this help message and exit
   -l, --list                        Lists all the available validators
+  -le, --ecosystem                  Lists ecosystem supported by active vendor
   -r, --raw                         List of all the raw response from vendor
 
   -e validator [validator ...]      Enable these validator/s only
@@ -64,6 +65,7 @@ def get_help():
 
 Examples:
     python vulntotal-cli.py --list
+    python vulntotal-cli.py -le
     python vulntotal-cli.py -p 'pkg:pypi/jinja2@2.4.1'
     python vulntotal-cli.py -e osv vulnerablecode -p 'pkg:go/github.com/cloudflare/cfrpki@v1.4.1'
     python vulntotal-cli.py -d osv -p 'pkg:maven/org.apache.tomcat/tomcat@10.1.0-M8'
@@ -75,10 +77,11 @@ def get_valid_validators(validators):
     valid_validators = {}
     unknown_validators = []
     for validator in validators:
+        key = validator.lower()
         try:
-            valid_validators[validator] = VALIDATORS_REGISTRY[validator]
+            valid_validators[key] = VALIDATORS_REGISTRY[key]
         except KeyError:
-            unknown_validators.append(validator)
+            unknown_validators.append(key)
     if unknown_validators:
         raise CommandError(f"Unknown validator: {unknown_validators}")
     return valid_validators
@@ -94,9 +97,21 @@ def get_enabled_validator(validators):
 
 
 def list_validators():
-    validators = list(VALIDATORS_REGISTRY)
+    validators = [x.upper() for x in list(VALIDATORS_REGISTRY)]
     print("Currently supported validators:")
-    print("\n".join(validators))
+    print("\n".join(sorted(validators)))
+
+
+def list_supported_ecosystem(validators):
+    ecosystems = []
+    for key, validator in validators.items():
+        vendor_supported_ecosystem = validator.supported_ecosystem()
+        ecosystems.extend([x.upper() for x in vendor_supported_ecosystem.keys()])
+
+    active_validator = [x.upper() for x in validators.keys()]
+    print("Active Validators:", ", ".join(sorted(active_validator)), end="\n\n")
+    print("Ecosystem supported by active validators")
+    print("\n".join(sorted(set(ecosystems))))
 
 
 def formatted_row(validator, advisory):
@@ -123,8 +138,11 @@ def run_validators(purl, validators):
     if not validators:
         print("No validators available!")
         return
+
     print("PURL: ", purl)
-    print("Active Validators: ", ", ".join(validators.keys()))
+    active_validator = [x.upper() for x in validators.keys()]
+    print("Active Validators:", ", ".join(sorted(active_validator)), end="\n\n")
+
     table = Texttable()
     table.set_cols_dtype(["t", "t", "t", "t"])
     table.set_cols_align(["c", "l", "l", "l"])
@@ -152,6 +170,9 @@ def handler():
     parser.add_argument(
         "-l", "--list", action="store_true", help="Lists all the available validators"
     )
+    parser.add_argument(
+        "-le", "--ecosystem", action="store_true", help="Lists ecosystem supported by active vendor"
+    )
     parser.add_argument(
         "-r", "--raw", action="store_true", help="List of all the raw response from vendor"
     )
@@ -174,6 +195,16 @@ def handler():
     elif args.list:
         list_validators()
 
+    elif args.ecosystem:
+        if args.enable:
+            list_supported_ecosystem(get_enabled_validator(args.enable))
+
+        elif args.disable:
+            list_supported_ecosystem(get_undisabled_validator(args.disable))
+
+        else:
+            list_supported_ecosystem(VALIDATORS_REGISTRY)
+
     elif args.raw:
         if args.purl:
             if args.enable: