#!/usr/bin/env python3
"""
    Check if all script dependencies are available.
    Does not go into venvs, dependencies need to be installed globally.
    Prints a list of which commands can be run and which can't.
    Basic binaries such as gnu coreutils are expected and not checked for.
"""

import os
import shutil
import subprocess
import configparser
from pathlib import Path
from typing import Dict, List, Tuple
import importlib

CHECK = "✅"
CROSS = "❌"

def check_command(cmd: str) -> Tuple[bool, str|None]:
    if shutil.which(cmd) is not None:
        return (True, None)
    else:
        return (False, f"Command '{cmd}' not found")

def check_env_var(var: str) -> Tuple[bool, str|None]:
    if var in os.environ:
        return (True, None)
    else:
        return (False, f"Environment variable '{var}' not set")

def check_python_lib(lib: str) -> Tuple[bool, str|None]:
    try:
        importlib.import_module(lib)
        return (True, None)
    except ImportError:
        return (False, f"Python library '{lib}' not found")

def check_stats_commands() -> List[Tuple[bool, str]]:
    """
    Checks if all the commands that are configured for the stats daemon
    can be executed.
    """
    results = []
    config_home = os.environ.get("XDG_CONFIG_HOME", os.path.expanduser("~/.config"))
    config_path = Path(config_home) / "stats/config.ini"

    if not config_path.exists():
        return [(False, f"Stats config not found at {config_path}")]

    config = configparser.ConfigParser(interpolation=None)
    config.read(config_path)

    for section in config.sections():
        if section.startswith('stat.'):
            cmd = config[section]['command']
            try:
                subprocess.run(cmd, shell=True, capture_output=True, check=True)
                results.append((True, None))
            except subprocess.CalledProcessError:
                results.append((False, f"Command '{cmd}' failed to run"))

    return results

def check_scripts() -> Dict[str, List[Tuple[bool, str|None]]]:
    """Run all checks for each script."""
    checks = {
        "e-ink": [
            check_command("python3"),
            check_python_lib("PIL"),
        ],
        "indeel": [
            check_command("file"),
            check_command("magick"),
            check_command("dctfilename"),
            check_command("webpmux"),
            check_command("gifski"),
        ],
        "mover": [],
        "pushover": [
            check_env_var("PUSHOVER_APP_TOKEN"),
            check_env_var("PUSHOVER_USER_TOKEN"),
            check_command("curl"),
            check_python_lib("requests"),
        ],
        "stats": [
            check_command("python3"),
            check_command("sensors"),
            check_command("jq"),
            check_command("top"),
            check_command("free"),
            check_command("df"),
            check_python_lib("configparser"),
            check_python_lib("http"),
            *check_stats_commands(),
        ],
        "stats-collector": [
            check_command("python3"),
            check_python_lib("sqlalchemy"),
            check_python_lib("requests"),
        ],
        "syncwall": [
            check_command("wget"),
            check_command("wal"),
        ],
        "tclean": [
            check_command("python3"),
            check_command("transmission-remote"),
            check_python_lib("argparse"),
            check_python_lib("datetime"),
        ],
        "vc": [
            check_command("ffmpeg"),
            check_command("yt-dlp"),
            check_python_lib("argparse"),
        ],
        "vcmp": [
            check_command("ffmpeg"),
            check_command("ffprobe"),
            check_command("nice"),
            check_python_lib("pathlib"),
            check_python_lib("argparse"),
            check_python_lib("logging"),
            check_python_lib("tempfile"),
        ],
    }
    return checks

def main():
    results = check_scripts()

    for script, checks in results.items():
        all_passed = all(result[0] for result in checks) if checks else True
        status = CHECK if all_passed else CROSS
        print(f"{status} {script}")

        for passed, message in checks:
            status = CHECK if passed else CROSS
            if message:
                print(f"\t{status} {message}")

if __name__ == "__main__":
    main()
