msubprojects: Allow comma separated list of types

2 files changed, 13 insertions, 6 deletions

diff --git a/mesonbuild/msubprojects.py b/mesonbuild/msubprojects.py
index 22729d4..6329feb 100755
--- a/mesonbuild/msubprojects.py
+++ b/mesonbuild/msubprojects.py
@@ -2,8 +2,8 @@ import os, subprocess
 import argparse
 
 from . import mlog
-from .mesonlib import quiet_git, verbose_git, GitException, Popen_safe
-from .wrap.wrap import API_ROOT, Resolver, WrapException
+from .mesonlib import quiet_git, verbose_git, GitException, Popen_safe, MesonException
+from .wrap.wrap import API_ROOT, Resolver, WrapException, ALL_TYPES
 from .wrap import wraptool
 
 def update_wrapdb_file(wrap, repo_dir, options):
@@ -257,9 +257,8 @@ def foreach(wrap, repo_dir, options):
 def add_common_arguments(p):
     p.add_argument('--sourcedir', default='.',
                    help='Path to source directory')
-    p.add_argument('--type', default='',
-                   choices=['file', 'git', 'hg', 'svn'],
-                   help='Only subprojects of given type (default: all)')
+    p.add_argument('--types', default='',
+                   help='Comma-separated list of subproject types. Supported types are: {} (default: all)'.format(', '.join(ALL_TYPES)))
 
 def add_subprojects_argument(p):
     p.add_argument('subprojects', nargs='*',
@@ -318,9 +317,13 @@ def run(options):
         wraps = [wrap for name, wrap in r.wraps.items() if name in options.subprojects]
     else:
         wraps = r.wraps.values()
+    types = [t.strip() for t in options.types.split(',')]
+    for t in types:
+        if t not in ALL_TYPES:
+            raise MesonException('Unknown subproject type {!r}, supported types are: {}'.format(t, ', '.join(ALL_TYPES)))
     failures = []
     for wrap in wraps:
-        if options.type and wrap.type != options.type:
+        if wrap.type not in types:
             continue
         dirname = os.path.join(subprojects_dir, wrap.directory)
         if not options.subprojects_func(wrap, dirname, options):
diff --git a/mesonbuild/wrap/wrap.py b/mesonbuild/wrap/wrap.py
index 969c82d..98b30ee 100644
--- a/mesonbuild/wrap/wrap.py
+++ b/mesonbuild/wrap/wrap.py
@@ -49,6 +49,8 @@ REQ_TIMEOUT = 600.0
 SSL_WARNING_PRINTED = False
 WHITELIST_SUBDOMAIN = 'wrapdb.mesonbuild.com'
 
+ALL_TYPES = ['file', 'git', 'hg', 'svn']
+
 def whitelist_wrapdb(urlstr: str) -> urllib.parse.ParseResult:
     """ raises WrapException if not whitelisted subdomain """
     url = urllib.parse.urlparse(urlstr)
@@ -106,6 +108,8 @@ class PackageDefinition:
         self.directory = self.values.get('directory', self.name)
         if os.path.dirname(self.directory):
             raise WrapException('Directory key must be a name and not a path')
+        if self.type and self.type not in ALL_TYPES:
+            raise WrapException('Unknown wrap type {!r}'.format(self.type))
 
     def guess_type(self) -> None:
         if os.path.exists(os.path.join(self.filename, '.git')):