RedditVideoMakerBot/utils/checker.py

#!/usr/bin/env python
import os
from rich.console import Console
from rich.table import Table
from rich import box
import re
import dotenv
from utils.console import handle_input

console = Console()


def check_env() -> bool:
    """Checks to see what's been put in .env

    Returns:
        bool: Whether or not everything was put in properly
    """
    if not os.path.exists(".env.template"):
        console.print("[red]Couldn't find .env.template. Unable to check variables.")
        return True
    if not os.path.exists(".env"):
        console.print("[red]Couldn't find the .env file, creating one now.")
        with open(".env", "x", encoding="utf-8") as file:
            file.write("")
    success = True
    with open(".env.template", "r", encoding="utf-8") as template:
        # req_envs = [env.split("=")[0] for env in template.readlines() if "=" in env]
        matching = {}
        explanations = {}
        bounds = {}
        types = {}
        oob_errors = {}
        examples = {}
        req_envs = []
        var_optional = False
        for line in template.readlines():
            if (
                line.startswith("#") is not True
                and "=" in line
                and var_optional is not True
            ):
                req_envs.append(line.split("=")[0])
                if "#" in line:
                    examples[line.split("=")[0]] = "#".join(line.split("#")[1:]).strip()
            elif "#OPTIONAL" in line:
                var_optional = True
            elif line.startswith("#MATCH_REGEX "):
                matching[req_envs[-1]] = line.removeprefix("#MATCH_REGEX ")[:-1]
                var_optional = False
            elif line.startswith("#OOB_ERROR "):
                oob_errors[req_envs[-1]] = line.removeprefix("#OOB_ERROR ")[:-1]
                var_optional = False
            elif line.startswith("#RANGE "):
                bounds[req_envs[-1]] = tuple(
                    map(
                        lambda x: float(x) if x != "None" else None,
                        line.removeprefix("#RANGE ")[:-1].split(":"),
                    )
                )
                var_optional = False
            elif line.startswith("#MATCH_TYPE "):
                types[req_envs[-1]] = eval(
                    line.removeprefix("#MATCH_TYPE ")[:-1].split()[0]
                )
                var_optional = False
            elif line.startswith("#EXPLANATION "):
                explanations[req_envs[-1]] = line.removeprefix("#EXPLANATION ")[:-1]
                var_optional = False
            else:
                var_optional = False
    missing = set()
    incorrect = set()
    dotenv.load_dotenv()
    for env in req_envs:
        value = os.getenv(env)
        if value is None:
            missing.add(env)
            continue
        if env in matching.keys():
            re.match(matching[env], value) is None and incorrect.add(env)
        if env in bounds.keys() and env not in types.keys():
            len(value) >= bounds[env][0] or (
                len(bounds[env]) > 1 and bounds[env][1] >= len(value)
            ) or incorrect.add(env)
            continue
        if env in types.keys():
            try:
                temp = types[env](value)
                if env in bounds.keys():
                    (bounds[env][0] <= temp or incorrect.add(env)) and len(
                        bounds[env]
                    ) > 1 and (bounds[env][1] >= temp or incorrect.add(env))
            except ValueError:
                incorrect.add(env)

    if len(missing):
        table = Table(
            title="Missing variables",
            highlight=True,
            show_lines=True,
            box=box.ROUNDED,
            border_style="#414868",
            header_style="#C0CAF5 bold",
            title_justify="left",
            title_style="#C0CAF5 bold",
        )
        table.add_column("Variable", justify="left", style="#7AA2F7 bold", no_wrap=True)
        table.add_column("Explanation", justify="left", style="#BB9AF7", no_wrap=False)
        table.add_column("Example", justify="center", style="#F7768E", no_wrap=True)
        table.add_column("Min", justify="right", style="#F7768E", no_wrap=True)
        table.add_column("Max", justify="left", style="#F7768E", no_wrap=True)
        for env in missing:
            table.add_row(
                env,
                explanations[env]
                if env in explanations.keys()
                else "No explanation given",
                examples[env] if env in examples.keys() else "",
                str(bounds[env][0])
                if env in bounds.keys() and bounds[env][1] is not None
                else "",
                str(bounds[env][1])
                if env in bounds.keys()
                and len(bounds[env]) > 1
                and bounds[env][1] is not None
                else "",
            )
        console.print(table)
        success = False
    if len(incorrect):
        table = Table(
            title="Incorrect variables",
            highlight=True,
            show_lines=True,
            box=box.ROUNDED,
            border_style="#414868",
            header_style="#C0CAF5 bold",
            title_justify="left",
            title_style="#C0CAF5 bold",
        )
        table.add_column("Variable", justify="left", style="#7AA2F7 bold", no_wrap=True)
        table.add_column(
            "Current value", justify="left", style="#F7768E", no_wrap=False
        )
        table.add_column("Explanation", justify="left", style="#BB9AF7", no_wrap=False)
        table.add_column("Example", justify="center", style="#F7768E", no_wrap=True)
        table.add_column("Min", justify="right", style="#F7768E", no_wrap=True)
        table.add_column("Max", justify="left", style="#F7768E", no_wrap=True)
        for env in incorrect:
            table.add_row(
                env,
                os.getenv(env),
                explanations[env]
                if env in explanations.keys()
                else "No explanation given",
                str(types[env].__name__) if env in types.keys() else "str",
                str(bounds[env][0]) if env in bounds.keys() else "None",
                str(bounds[env][1])
                if env in bounds.keys() and len(bounds[env]) > 1
                else "None",
            )
            missing.add(env)
        console.print(table)
        success = False
    if success is True:
        return True
    console.print(
        "[green]Do you want to automatically overwrite incorrect variables and add the missing variables? (y/n)"
    )
    if not input().casefold().startswith("y"):
        console.print("[red]Aborting: Unresolved missing variables")
        return False
    if len(incorrect):
        with open(".env", "r+", encoding="utf-8") as env_file:
            lines = []
            for line in env_file.readlines():
                line.split("=")[0].strip() not in incorrect and lines.append(line)
            env_file.seek(0)
            env_file.write("\n".join(lines))
            env_file.truncate()
        console.print("[green]Successfully removed incorrectly set variables from .env")
    with open(".env", "a", encoding="utf-8") as env_file:
        for env in missing:
            env_file.write(
                env
                + "="
                + ('"' if env not in types.keys() else "")
                + str(
                    handle_input(
                        "[#F7768E bold]" + env + "[#C0CAF5 bold]=",
                        types[env] if env in types.keys() else False,
                        matching[env] if env in matching.keys() else ".*",
                        explanations[env]
                        if env in explanations.keys()
                        else "Incorrect input. Try again.",
                        bounds[env][0] if env in bounds.keys() else None,
                        bounds[env][1]
                        if env in bounds.keys() and len(bounds[env]) > 1
                        else None,
                        oob_errors[env]
                        if env in oob_errors.keys()
                        else "Input too long/short.",
                        extra_info="[#C0CAF5 bold]⮶ "
                        + (
                            explanations[env]
                            if env in explanations.keys()
                            else "No info available"
                        ),
                    )
                )
                + ('"' if env not in types.keys() else "")
                + "\n"
            )
    return True


if __name__ == "__main__":
    check_env()