from pathlib import Path
from typing import List

import pytest

from build_helpers.build_helpers import find, find_version, matches


@pytest.mark.parametrize(
    "path_rel,include_files,include_dirs,excludes,scan_exclude,expected",
    [
        pytest.param("test_files", [], [], [], None, [], id="none"),
        pytest.param(
            "test_files",
            [".*"],
            [],
            [],
            [],
            [
                "a/b/bad_dir/.gitkeep",
                "a/b/file2.txt",
                "a/b/file1.txt",
                "a/b/junk.txt",
                "c/bad_dir/.gitkeep",
                "c/file2.txt",
                "c/file1.txt",
                "c/junk.txt",
            ],
            id="all",
        ),
        pytest.param(
            "test_files",
            [".*"],
            [],
            ["^a/.*"],
            [],
            ["c/bad_dir/.gitkeep", "c/file2.txt", "c/file1.txt", "c/junk.txt"],
            id="filter_a",
        ),
        pytest.param(
            "test_files",
            [".*"],
            [],
            [],
            ["^a/.*"],
            ["c/bad_dir/.gitkeep", "c/file2.txt", "c/file1.txt", "c/junk.txt"],
            id="do_not_scan_a",
        ),
        pytest.param(
            "test_files",
            ["^a/.*"],
            [],
            [],
            [],
            ["a/b/bad_dir/.gitkeep", "a/b/file2.txt", "a/b/file1.txt", "a/b/junk.txt"],
            id="include_a",
        ),
        pytest.param(
            "test_files",
            ["^a/.*"],
            [],
            [".*/file1\\.txt"],
            [],
            ["a/b/bad_dir/.gitkeep", "a/b/file2.txt", "a/b/junk.txt"],
            id="include_a,exclude_file1",
        ),
        pytest.param(
            "test_files",
            [".*"],
            [],
            ["^.*/junk.txt$"],
            [],
            [
                "a/b/bad_dir/.gitkeep",
                "a/b/file2.txt",
                "a/b/file1.txt",
                "c/bad_dir/.gitkeep",
                "c/file2.txt",
                "c/file1.txt",
            ],
            id="no_junk",
        ),
        pytest.param(
            "test_files",
            ["^.*/junk.txt"],
            [],
            [],
            [],
            ["a/b/junk.txt", "c/junk.txt"],
            id="junk_only",
        ),
        pytest.param("test_files", [], ["^a$"], [], [], ["a"], id="exact_a"),
        pytest.param(
            "test_files",
            [],
            [".*bad_dir$"],
            [],
            [],
            ["a/b/bad_dir", "c/bad_dir"],
            id="bad_dirs",
        ),
    ],
)
def test_find(
    path_rel: str,
    include_files: List[str],
    include_dirs: List[str],
    excludes: List[str],
    scan_exclude: List[str],
    expected: List[str],
) -> None:
    basedir = Path(__file__).parent.absolute()
    path = basedir / path_rel
    ret = find(
        root=path,
        excludes=excludes,
        include_files=include_files,
        include_dirs=include_dirs,
        scan_exclude=scan_exclude,
    )

    ret_set = set([str(x) for x in ret])
    expected_set = set([str(Path(x)) for x in expected])
    assert ret_set == expected_set


@pytest.mark.parametrize(
    "patterns,query,expected",
    [
        (["^a/.*"], Path("a") / "b.txt", True),
        (["^/foo/bar/.*"], Path("/foo") / "bar" / "blag", True),
    ],
)
def test_matches(patterns: List[str], query: Path, expected: bool) -> None:
    ret = matches(patterns, query)
    assert ret == expected


def test_find_version() -> None:
    version = find_version("omegaconf", "version.py")
    # Ensure `version` is a string starting with a digit.
    assert isinstance(version, str) and version and "0" <= version[0] <= "9"