pantsbuild / pants / 20332790708

import sys

from pathlib import PurePath

from typing import Set

def resolve_pure_path(base: PurePath, relative_path: PurePath) -> PurePath:

    parts = list(base.parts)

    for component in relative_path.parts:

        if component == ".":

            pass

        elif component == "..":

            if not parts:

                raise ValueError(f"Relative path {relative_path} escapes from path {base}.")

            parts.pop()

            parts.append(component)

    return PurePath(*parts)

def extract_module_source_paths(path: PurePath, raw_content: bytes) -> Set[str]:

    import hcl2  # type: ignore[import-not-found]  # pants: no-infer-dep

    content = raw_content.decode("utf-8")

    parsed_content = hcl2.loads(content)

    paths = set()

    for wrapped_module in parsed_content.get("module", []):

        values = list(wrapped_module.values())[

        source = values.get("source", "")

        if source.startswith("./") or source.startswith("../"):

            try:

                resolved_path = resolve_pure_path(path, PurePath(source))

                paths.add(str(resolved_path))

            except ValueError:

                pass

    return paths

def main(args):

    paths = set()

    for filename in args:

        with open(filename, "rb") as f:

            content = f.read()

        paths |= extract_module_source_paths(PurePath(filename).parent, content)

    for path in paths:

        print(path)

if __name__ == "__main__":

    main(sys.argv[1:])