"""Chart and style helpers for Plotly."""

import json
from pathlib import Path
from typing import Any

from rich.console import Console

from openbb_cli.config.constants import STYLES_DIRECTORY

console = Console()


class Style:
    """Handle style configurations for Plotly and Rich."""

    STYLES_REPO = STYLES_DIRECTORY

    console_styles_available: dict[str, Path] = {}
    console_style: dict[str, Any] = {}

    line_color: str = ""
    up_color: str = ""
    down_color: str = ""
    up_colorway: list[str] = []
    down_colorway: list[str] = []
    up_color_transparent: str = ""
    down_color_transparent: str = ""

    line_width: float = 1.5

    def __init__(
        self,
        style: str | None = "",
        directory: Path | None = None,
    ):
        """Initialize the class."""
        self._load(directory)
        self.apply(style, directory)

    def apply(self, style: str | None = None, directory: Path | None = None) -> None:
        """Apply the style to the console."""
        if style:
            if style in self.console_styles_available:
                json_path: Path | None = self.console_styles_available[style]
            else:
                self._load(directory)
                if style in self.console_styles_available:
                    json_path = self.console_styles_available[style]
                else:
                    console.print(f"\nInvalid console style '{style}', using default.")
                    json_path = self.console_styles_available.get("dark", None)

            if json_path:
                self.console_style = self._from_json(json_path)
            else:
                console.print("Error loading default.")

    def _from_directory(self, folder: Path | None) -> None:
        """Load custom styles from folder.

        Parameters
        ----------
        folder : Path
            Path to the folder containing the stylesheets.
        """
        if not folder or not folder.exists():
            return

        for attr, ext in zip(
            ["console_styles_available"],
            [".richstyle.json"],
        ):
            for file in folder.rglob(f"*{ext}"):
                getattr(self, attr)[file.name.replace(ext, "")] = file

    def _load(self, directory: Path | None = None) -> None:
        """Load custom styles from default and user folders."""
        self._from_directory(self.STYLES_REPO)
        self._from_directory(directory)

    def _from_json(self, file: Path) -> dict[str, Any]:
        """Load style from json file."""
        with open(file) as f:
            json_style: dict = json.load(f)
            for key, value in json_style.items():
                json_style[key] = value.replace(" ", "")
            return json_style

    @property
    def available_styles(self) -> list[str]:
        """Return available styles."""
        return list(self.console_styles_available.keys())