|
1 |
| -import subprocess |
2 |
| - |
3 |
| - |
4 |
| -def check_installed(tool: str, version="") -> int: |
5 |
| - if version: |
6 |
| - check_version_cmd = [f'{tool}-{version} ', '--version'] |
7 |
| - else: |
8 |
| - check_version_cmd = [tool, '--version'] |
9 |
| - try: |
10 |
| - subprocess.run(check_version_cmd, stdout=subprocess.PIPE) |
11 |
| - retval = 0 |
12 |
| - except FileNotFoundError: |
13 |
| - retval = install_clang_tools(version) |
14 |
| - return retval |
15 |
| - |
16 |
| - |
17 |
| -def install_clang_tools(version: str) -> int: |
18 |
| - if version: |
19 |
| - # clang-tools exist because install_requires=['clang-tools'] in setup.py |
20 |
| - install_tool_cmd = ['clang-tools', '-i', version] |
21 |
| - else: |
22 |
| - # install version 13 by default if clang-tools not exist. |
23 |
| - install_tool_cmd = ['clang-tools', '-i', '13'] |
24 |
| - try: |
25 |
| - subprocess.run(install_tool_cmd, stdout=subprocess.PIPE) |
26 |
| - retval = 0 |
27 |
| - except Exception: |
28 |
| - retval = 1 |
29 |
| - return retval |
30 |
| - |
31 |
| - |
32 |
| -def get_expect_version(args) -> str: |
33 |
| - for arg in args: |
34 |
| - if arg.startswith("--version"): # expect specific clang-tools version. |
35 |
| - # If --version is passed in as 2 arguments, the second is version |
36 |
| - if arg == "--version" and args.index(arg) != len(args) - 1: |
37 |
| - # when --version 14 |
38 |
| - expect_version = args[args.index(arg) + 1] |
39 |
| - else: |
40 |
| - # when --version=14 |
41 |
| - expect_version = arg.replace(" ", "").replace("=", "").replace("--version", "") |
42 |
| - return expect_version |
43 |
| - return "" |
| 1 | +import sys |
| 2 | +from pathlib import Path |
| 3 | +import logging |
| 4 | +from typing import Optional |
| 5 | + |
| 6 | +from clang_tools.install import is_installed as _is_installed, install_tool |
| 7 | + |
| 8 | + |
| 9 | +LOG = logging.getLogger(__name__) |
| 10 | + |
| 11 | + |
| 12 | +DEFAULT_CLANG_VERSION = "13" |
| 13 | + |
| 14 | + |
| 15 | +def is_installed(tool_name: str, version: str) -> Optional[Path]: |
| 16 | + """Check if tool is installed. |
| 17 | +
|
| 18 | + Checks the current python prefix and PATH via clang_tools.install.is_installed. |
| 19 | + """ |
| 20 | + # check in current python prefix (usual situation when we installed into pre-commit venv) |
| 21 | + directory = Path(sys.executable).parent |
| 22 | + path = (directory / f"{tool_name}-{version}") |
| 23 | + if path.is_file(): |
| 24 | + return path |
| 25 | + |
| 26 | + # also check using clang_tools |
| 27 | + path = _is_installed(tool_name, version) |
| 28 | + if path is not None: |
| 29 | + return Path(path) |
| 30 | + |
| 31 | + # not found |
| 32 | + return None |
| 33 | + |
| 34 | + |
| 35 | +def ensure_installed(tool_name: str, version: str = DEFAULT_CLANG_VERSION) -> Path: |
| 36 | + """ |
| 37 | + Ensure tool is available at given version. |
| 38 | + """ |
| 39 | + LOG.info("Checking for %s, version %s", tool_name, version) |
| 40 | + path = is_installed(tool_name, version) |
| 41 | + if path is not None: |
| 42 | + LOG.info("%s, version %s is already installed", tool_name, version) |
| 43 | + return path |
| 44 | + |
| 45 | + LOG.info("Installing %s, version %s", tool_name, version) |
| 46 | + directory = Path(sys.executable).parent |
| 47 | + install_tool(tool_name, version, directory=str(directory), no_progress_bar=True) |
| 48 | + return directory / f"{tool_name}-{version}" |
0 commit comments