dependencies: refactor to use methods properly

1 files changed, 92 insertions, 85 deletions

diff --git a/mesonbuild/dependencies/hdf5.py b/mesonbuild/dependencies/hdf5.py
index 1ceb15f..b270f58 100644
--- a/mesonbuild/dependencies/hdf5.py
+++ b/mesonbuild/dependencies/hdf5.py
@@ -15,11 +15,13 @@
 # This file contains the detection logic for miscellaneous external dependencies.
 
 import subprocess
+import shutil
 from pathlib import Path
 
 from .. import mlog
-from ..mesonlib import split_args
-from .base import DependencyException, ExternalDependency, ExternalProgram, PkgConfigDependency
+from ..mesonlib import split_args, listify
+from .base import (DependencyException, DependencyMethods, ExternalDependency, ExternalProgram,
+                   PkgConfigDependency)
 
 class HDF5Dependency(ExternalDependency):
 
@@ -29,95 +31,100 @@ class HDF5Dependency(ExternalDependency):
         kwargs['required'] = False
         kwargs['silent'] = True
         self.is_found = False
+        methods = listify(self.methods)
 
-        # 1. pkg-config
-        pkgconfig_files = ['hdf5', 'hdf5-serial']
-        # some distros put hdf5-1.2.3.pc with version number in .pc filename.
-        try:
-            ret = subprocess.run(['pkg-config', '--list-all'], stdout=subprocess.PIPE, stderr=subprocess.DEVNULL,
-                                 universal_newlines=True)
-            if ret.returncode == 0:
-                for pkg in ret.stdout.split('\n'):
-                    if pkg.startswith(('hdf5')):
-                        pkgconfig_files.append(pkg.split(' ', 1)[0])
-                pkgconfig_files = list(set(pkgconfig_files))  # dedupe
-        except FileNotFoundError:
-            # pkg-config was not available
-            pass
         if language not in ('c', 'cpp', 'fortran'):
             raise DependencyException('Language {} is not supported with HDF5.'.format(language))
 
-        for pkg in pkgconfig_files:
-            pkgdep = PkgConfigDependency(pkg, environment, kwargs, language=self.language)
-            if not pkgdep.found():
-                continue
-
-            self.compile_args = pkgdep.get_compile_args()
-            # some broken pkgconfig don't actually list the full path to the needed includes
-            newinc = []
-            for arg in self.compile_args:
-                if arg.startswith('-I'):
-                    stem = 'static' if kwargs.get('static', False) else 'shared'
-                    if (Path(arg[2:]) / stem).is_dir():
-                        newinc.append('-I' + str(Path(arg[2:]) / stem))
-            self.compile_args += newinc
-
-            # derive needed libraries by language
-            pd_link_args = pkgdep.get_link_args()
-            link_args = []
-            for larg in pd_link_args:
-                lpath = Path(larg)
-                # some pkg-config hdf5.pc (e.g. Ubuntu) don't include the commonly-used HL HDF5 libraries,
-                # so let's add them if they exist
-                # additionally, some pkgconfig HDF5 HL files are malformed so let's be sure to find HL anyway
-                if lpath.is_file():
-                    hl = []
-                    if language == 'cpp':
-                        hl += ['_hl_cpp', '_cpp']
-                    elif language == 'fortran':
-                        hl += ['_hl_fortran', 'hl_fortran', '_fortran']
-                    hl += ['_hl']  # C HL library, always needed
-
-                    suffix = '.' + lpath.name.split('.', 1)[1]  # in case of .dll.a
-                    for h in hl:
-                        hlfn = lpath.parent / (lpath.name.split('.', 1)[0] + h + suffix)
-                        if hlfn.is_file():
-                            link_args.append(str(hlfn))
-                    # HDF5 C libs are required by other HDF5 languages
-                    link_args.append(larg)
-                else:
-                    link_args.append(larg)
+        if set([DependencyMethods.AUTO, DependencyMethods.PKGCONFIG]).intersection(methods):
+            PCEXE = shutil.which('pkg-config')
+            if PCEXE:
+                pkgconfig_files = ['hdf5', 'hdf5-serial']
+                # some distros put hdf5-1.2.3.pc with version number in .pc filename.
+                ret = subprocess.run([PCEXE, '--list-all'], stdout=subprocess.PIPE, stderr=subprocess.DEVNULL,
+                                     universal_newlines=True)
+                if ret.returncode == 0:
+                    for pkg in ret.stdout.split('\n'):
+                        if pkg.startswith(('hdf5')):
+                            pkgconfig_files.append(pkg.split(' ', 1)[0])
+                    pkgconfig_files = list(set(pkgconfig_files))  # dedupe
+
+            for pkg in pkgconfig_files:
+                pkgdep = PkgConfigDependency(pkg, environment, kwargs, language=self.language)
+                if not pkgdep.found():
+                    continue
+
+                self.compile_args = pkgdep.get_compile_args()
+                # some broken pkgconfig don't actually list the full path to the needed includes
+                newinc = []
+                for arg in self.compile_args:
+                    if arg.startswith('-I'):
+                        stem = 'static' if kwargs.get('static', False) else 'shared'
+                        if (Path(arg[2:]) / stem).is_dir():
+                            newinc.append('-I' + str(Path(arg[2:]) / stem))
+                self.compile_args += newinc
+
+                # derive needed libraries by language
+                pd_link_args = pkgdep.get_link_args()
+                link_args = []
+                for larg in pd_link_args:
+                    lpath = Path(larg)
+                    # some pkg-config hdf5.pc (e.g. Ubuntu) don't include the commonly-used HL HDF5 libraries,
+                    # so let's add them if they exist
+                    # additionally, some pkgconfig HDF5 HL files are malformed so let's be sure to find HL anyway
+                    if lpath.is_file():
+                        hl = []
+                        if language == 'cpp':
+                            hl += ['_hl_cpp', '_cpp']
+                        elif language == 'fortran':
+                            hl += ['_hl_fortran', 'hl_fortran', '_fortran']
+                        hl += ['_hl']  # C HL library, always needed
+
+                        suffix = '.' + lpath.name.split('.', 1)[1]  # in case of .dll.a
+                        for h in hl:
+                            hlfn = lpath.parent / (lpath.name.split('.', 1)[0] + h + suffix)
+                            if hlfn.is_file():
+                                link_args.append(str(hlfn))
+                        # HDF5 C libs are required by other HDF5 languages
+                        link_args.append(larg)
+                    else:
+                        link_args.append(larg)
 
+                self.link_args = link_args
+                self.version = pkgdep.get_version()
+                self.is_found = True
+                self.pcdep = pkgdep
+                return
+
+        if DependencyMethods.AUTO in methods:
+            wrappers = {'c': 'h5cc', 'cpp': 'h5c++', 'fortran': 'h5fc'}
+            comp_args = []
+            link_args = []
+            # have to always do C as well as desired language
+            for lang in set([language, 'c']):
+                prog = ExternalProgram(wrappers[lang], silent=True)
+                if not prog.found():
+                    return
+                cmd = prog.get_command() + ['-show']
+                p = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True, timeout=15)
+                if p.returncode != 0:
+                    mlog.debug('Command', mlog.bold(cmd), 'failed to run:')
+                    mlog.debug(mlog.bold('Standard output\n'), p.stdout)
+                    mlog.debug(mlog.bold('Standard error\n'), p.stderr)
+                    return
+                args = split_args(p.stdout)
+                for arg in args[1:]:
+                    if arg.startswith(('-I', '-f', '-D')) or arg == '-pthread':
+                        comp_args.append(arg)
+                    elif arg.startswith(('-L', '-l', '-Wl')):
+                        link_args.append(arg)
+                    elif Path(arg).is_file():
+                        link_args.append(arg)
+            self.compile_args = comp_args
             self.link_args = link_args
-            self.version = pkgdep.get_version()
             self.is_found = True
-            self.pcdep = pkgdep
             return
 
-        # 2. compiler wrapper fallback
-        wrappers = {'c': 'h5cc', 'cpp': 'h5c++', 'fortran': 'h5fc'}
-        comp_args = []
-        link_args = []
-        # have to always do C as well as desired language
-        for lang in set([language, 'c']):
-            prog = ExternalProgram(wrappers[lang], silent=True)
-            if not prog.found():
-                return
-            cmd = prog.get_command() + ['-show']
-            p = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True, timeout=15)
-            if p.returncode != 0:
-                mlog.debug('Command', mlog.bold(cmd), 'failed to run:')
-                mlog.debug(mlog.bold('Standard output\n'), p.stdout)
-                mlog.debug(mlog.bold('Standard error\n'), p.stderr)
-                return
-            args = split_args(p.stdout)
-            for arg in args[1:]:
-                if arg.startswith(('-I', '-f', '-D')) or arg == '-pthread':
-                    comp_args.append(arg)
-                elif arg.startswith(('-L', '-l', '-Wl')):
-                    link_args.append(arg)
-                elif Path(arg).is_file():
-                    link_args.append(arg)
-        self.compile_args = comp_args
-        self.link_args = link_args
-        self.is_found = True
+    @staticmethod
+    def get_methods():
+        return [DependencyMethods.AUTO, DependencyMethods.PKGCONFIG]