make wrap search custom subprojects dirs

Changes from Dylan:
  - Don't use Path
  - merge the lint fixes
  - Fix some typing issues
  - Handle non-meson projects
  - Remove some code duplication by putting `get_subproject_dir` in
    utils

1 files changed, 24 insertions, 10 deletions

diff --git a/mesonbuild/wrap/wraptool.py b/mesonbuild/wrap/wraptool.py
index 6f97fe221..85175114e 100644
--- a/mesonbuild/wrap/wraptool.py
+++ b/mesonbuild/wrap/wraptool.py
@@ -11,8 +11,6 @@ import typing as T
 from glob import glob
 from .wrap import (open_wrapdburl, read_and_decompress, WrapException, get_releases,
                    get_releases_data, parse_patch_url)
-from pathlib import Path
-
 from .. import mesonlib, msubprojects
 
 if T.TYPE_CHECKING:
@@ -91,11 +89,12 @@ def get_latest_version(name: str, allow_insecure: bool) -> T.Tuple[str, str]:
 
 def install(options: 'argparse.Namespace') -> None:
     name = options.name
-    if not os.path.isdir('subprojects'):
+    subproject_dir_name = mesonlib.get_subproject_dir()
+    if subproject_dir_name is None or not os.path.isdir(subproject_dir_name):
         raise SystemExit('Subprojects dir not found. Run this script in your source root directory.')
-    if os.path.isdir(os.path.join('subprojects', name)):
+    if os.path.isdir(os.path.join(subproject_dir_name, name)):
         raise SystemExit('Subproject directory for this project already exists.')
-    wrapfile = os.path.join('subprojects', name + '.wrap')
+    wrapfile = os.path.join(subproject_dir_name, name + '.wrap')
     if os.path.exists(wrapfile):
         raise SystemExit('Wrap file already exists.')
     (version, revision) = get_latest_version(name, options.allow_insecure)
@@ -143,11 +142,20 @@ def do_promotion(from_path: str, spdir_name: str) -> None:
         outputdir = os.path.join(spdir_name, sproj_name)
         if os.path.exists(outputdir):
             raise SystemExit(f'Output dir {outputdir} already exists. Will not overwrite.')
-        shutil.copytree(from_path, outputdir, ignore=shutil.ignore_patterns('subprojects'))
+
+        subpdir = mesonlib.get_subproject_dir()
+        if subpdir is not None:
+            ignore = shutil.ignore_patterns(subpdir)
+        else:
+            ignore = None
+
+        shutil.copytree(from_path, outputdir, ignore=ignore)
 
 def promote(options: 'argparse.Namespace') -> None:
     argument = options.project_path
-    spdir_name = 'subprojects'
+    spdir_name = mesonlib.get_subproject_dir()
+    if spdir_name is None:
+        raise SystemExit('Subproject dir not found. Run this script in your source root directory.')
     sprojs = mesonlib.detect_subprojects(spdir_name)
 
     # check if the argument is a full path to a subproject directory or wrap file
@@ -170,7 +178,9 @@ def promote(options: 'argparse.Namespace') -> None:
 
 def status(options: 'argparse.Namespace') -> None:
     print('Subproject status')
-    for w in glob('subprojects/*.wrap'):
+    subdir = mesonlib.get_subproject_dir()
+    assert subdir is not None, "This should only happen in a non-native subproject"
+    for w in glob(f'{subdir}/*.wrap'):
         name = os.path.basename(w)[:-5]
         try:
             (latest_branch, latest_revision) = get_latest_version(name, options.allow_insecure)
@@ -189,8 +199,12 @@ def status(options: 'argparse.Namespace') -> None:
 
 def update_db(options: 'argparse.Namespace') -> None:
     data = get_releases_data(options.allow_insecure)
-    Path('subprojects').mkdir(exist_ok=True)
-    with Path('subprojects/wrapdb.json').open('wb') as f:
+    subproject_dir_name = mesonlib.get_subproject_dir()
+    if subproject_dir_name is None:
+        raise SystemExit('Subproject dir not found. Run this script in your source root directory.')
+
+    os.makedirs(subproject_dir_name, exist_ok=True)
+    with open(os.path.join(subproject_dir_name, 'wrapdb.json'), 'wb') as f:
         f.write(data)
 
 def run(options: 'argparse.Namespace') -> int: