from typing import Dict, Optional, Any, List

from rich import align
from rich.console import Console
from rich.table import Table
from rich.text import Text

import opik.config as config
from opik.config import OpikConfig

DEFAULT_KEY_COLOR = "green"
DEFAULT_VALUE_COLOR = "blue"
DEFAULT_ERROR_COLOR = "red"

console = Console()


def make_key_text(key: str) -> Text:
    return Text(key, style=DEFAULT_KEY_COLOR)


def make_value_text(value: str) -> Text:
    return Text(value, style=DEFAULT_VALUE_COLOR)


def print_header(text: str) -> None:
    header_text = f"*** {text.upper()} ***"
    header_text = Text(header_text, style="bold magenta")
    header_text = align.Align.left(header_text)

    console.print()
    console.print(header_text)


def print_installed_packages(packages: Dict[str, str]) -> None:
    for name, version in sorted(packages.items()):
        name = make_key_text(name)
        version = make_value_text(version)

        console.print(name, "==", version, sep="")


def print_versions(python_version: str, opik_version: str) -> None:
    python_version_label = make_key_text("Python version:")
    python_version = make_value_text(python_version)
    opik_version_label = make_key_text("Opik version:")
    opik_version = make_value_text(opik_version)

    console.print(python_version_label, python_version)
    console.print(opik_version_label, opik_version)


def print_config_file_details(config: OpikConfig) -> None:
    file_path_label = make_key_text("Config file path:")
    file_path = make_value_text(str(config.config_file_fullpath))

    is_exists_label = make_key_text("Config file exists:")
    is_exists = make_value_text("yes" if config.config_file_exists else "no")

    console.print(file_path_label, file_path)
    console.print(is_exists_label, is_exists)


def print_current_config(config: config.OpikConfig) -> None:
    table = Table(show_header=True, header_style="bold magenta")
    table.add_column("Setting", style=DEFAULT_KEY_COLOR)
    table.add_column("Value", style=DEFAULT_VALUE_COLOR)

    current_config_values = config.as_dict(mask_api_key=True)
    for key, value in sorted(current_config_values.items()):
        if key != "sentry_dsn":
            table.add_row(key, str(value))

    console.print(table)


def print_config_scan_results(
    misconfiguration_detected: bool, error_message: Optional[str]
) -> None:
    is_misconfigured_text = Text(
        "found" if misconfiguration_detected else "not found",
        style=DEFAULT_ERROR_COLOR if misconfiguration_detected else DEFAULT_VALUE_COLOR,
    )
    issues_found_label = make_key_text("Configuration issues:")

    console.print(issues_found_label, is_misconfigured_text)

    if not misconfiguration_detected:
        return

    err_msg = Text(error_message, style=DEFAULT_ERROR_COLOR)
    console.print(err_msg)


def print_backend_workspace_availability(
    is_available: bool,
    err_msg: Optional[str],
) -> None:
    is_available_text = Text(
        "yes" if is_available else "no",
        style=DEFAULT_VALUE_COLOR if is_available else DEFAULT_ERROR_COLOR,
    )
    is_available_label = make_key_text("Backend workspace available:")

    console.print(is_available_label, is_available_text)

    if err_msg:
        print_error_message(err_msg)


def print_error_message(err_msg: str) -> None:
    err_msg = Text(err_msg, style=DEFAULT_ERROR_COLOR)
    console.print(err_msg)


def print_opik_permissions_url(opik_url: str) -> None:
    opik_url_label = make_key_text("OPIK url:")
    opik_url = make_value_text(opik_url)
    console.print(opik_url_label, opik_url)


def print_user_permissions(user_permissions: Dict[str, Any]) -> None:
    user_label = make_key_text("User:")
    user_name = make_value_text(user_permissions["user_name"])
    workspace_label = make_key_text("Workspace:")
    workspace_name = make_value_text(user_permissions["workspace_name"])
    console.print(user_label, user_name)
    console.print(workspace_label, workspace_name)

    # print permissions
    permissions: List[Dict[str, str]] = user_permissions["permissions"]

    table = Table(show_header=True, header_style="bold magenta")
    table.add_column("Permission", style=DEFAULT_KEY_COLOR)
    table.add_column("Available", style=DEFAULT_VALUE_COLOR)

    for permission in permissions:
        value = permission["permission_value"] == "true"
        table.add_row(permission["permission_name"], str(value))

    console.print(table)