#!/usr/bin/env python3
#
# Verifies dotfiles installed binaries correctly
#
from distutils.spawn import find_executable
from typing import List, Text
from subprocess import run
import pytest


# --------------------------------------------------------------------------- #
# Helper functions
# --------------------------------------------------------------------------- #
def in_path(binary: Text) -> bool:
    """
    Check whether `binary` is in PATH
    """
    return find_executable(binary) is not None


def in_shell_path(shell: Text, binary: Text) -> bool:
    """
    Check whether `binary` is in interactive shell's PATH
    """
    command = f"{shell} -i -c 'command -v {binary}'"
    try:
        result = run(command, shell=True)
        return result.returncode == 0
    except Exception:
        return False


# --------------------------------------------------------------------------- #
# Test fixtures
# --------------------------------------------------------------------------- #
shells: List[Text] = [
    "sh",
    "bash",
    "zsh",
]

binaries: List[Text] = [
    # extend shells
    *shells,
    # tools
    "git",
    "gh",
    "aws",
    "gcloud",
    "terraform",
    "kubectl",
    "docker",
    "docker-compose",
    "screenfetch",
    # language: python
    "pyenv",
    "python",
    "python3",
    "pip",
    "pip3",
    "poetry",
    # langauge: js
    "node",
    "npm",
    "yarn",
    # language: java
    "java",
]


# --------------------------------------------------------------------------- #
# Tests
# --------------------------------------------------------------------------- #
@pytest.mark.parametrize("shell", shells)
def test_shells(shell: Text):
    """Assert all expected shells are in PATH."""
    assert in_path(shell)


@pytest.mark.parametrize("binary", binaries)
@pytest.mark.parametrize("shell", shells)
def test_binaries(shell: Text, binary: Text):
    """Asserts all expected binaries are in all shells."""
    assert in_shell_path(shell, binary)