config.py

from __future__ import annotations

import enum
import logging
import shutil
import sys
from pathlib import Path
from typing import Any

import yaml
from pydantic import BaseSettings, root_validator, validator

logger = logging.getLogger("ClimateToken")


class ExecutionMode(enum.Enum):
    DEV = "dev"
    REGISTRY = "registry"
    CLIENT = "client"
    EXPLORER = "explorer"


class ServerPort(enum.Enum):
    DEV = 31999
    CLIMATE_WAREHOUSE = 31310
    CLIMATE_PORTAL = 31311
    CLIMATE_TOKEN_REGISTRY = 31312
    CLIMATE_EXPLORER = 31313
    CLIMATE_TOKEN_CLIENT = 31314


class Settings(BaseSettings):
    _HIDDEN_FIELDS = ["MODE", "CHIA_ROOT", "CONFIG_PATH", "SERVER_PORT"]

    # Hidden configs: not exposed in config.yaml
    MODE: ExecutionMode
    CHIA_ROOT: Path = Path("~/.chia/mainnet")
    CONFIG_PATH: Path = Path("climate_token/config/config.yaml")
    SERVER_PORT: int | None = 31999

    # Visible configs: configurable through config.yaml
    LOG_PATH: Path = Path("climate_token/log/debug.log")
    DB_PATH: Path = Path("climate_explorer/db/climate_activity_CHALLENGE.sqlite")

    CLIMATE_EXPLORER_SERVER_HOST: str = "0.0.0.0"
    BLOCK_START: int = 1_500_000
    BLOCK_RANGE: int = 10_000
    MIN_DEPTH: int = 4
    # we always look back ~36 hours since the climate warehouse waits 24 hours before
    # setting metadata to be readable
    LOOKBACK_DEPTH: int = 6_912
    # fee is in mojos
    DEFAULT_FEE: int = 1_000_000_000
    CADT_API_SERVER_HOST: str = "https://observer.climateactiondata.org/api"
    CADT_API_KEY: str | None = None
    CHIA_HOSTNAME: str = "localhost"
    CHIA_WALLET_HOSTNAME: str | None = None
    CHIA_FULL_NODE_HOSTNAME: str | None = None
    CHIA_FULL_NODE_RPC_PORT: int = 8555
    CHIA_WALLET_RPC_PORT: int = 9256
    CLIMATE_EXPLORER_PORT: int | None = None
    CLIMATE_TOKEN_CLIENT_PORT: int | None = None
    CLIMATE_TOKEN_REGISTRY_PORT: int | None = None
    DEV_PORT: int | None = None
    SCAN_ALL_ORGANIZATIONS: bool | None = False

    _instance: Settings | None = None

    @classmethod
    def get_instance(cls) -> Settings:
        if cls._instance is None:
            cls._instance = get_settings()
        return cls._instance

    @root_validator(skip_on_failure=True)
    def configure_port(cls, values: dict[str, Any]) -> dict[str, Any]:
        if values["MODE"] == ExecutionMode.REGISTRY:
            values["SERVER_PORT"] = values.get("CLIMATE_TOKEN_REGISTRY_PORT", ServerPort.CLIMATE_TOKEN_REGISTRY.value)
        elif values["MODE"] == ExecutionMode.CLIENT:
            values["SERVER_PORT"] = values.get("CLIMATE_TOKEN_CLIENT_PORT", ServerPort.CLIMATE_TOKEN_CLIENT.value)
        elif values["MODE"] == ExecutionMode.EXPLORER:
            values["SERVER_PORT"] = values.get("CLIMATE_EXPLORER_PORT", ServerPort.CLIMATE_EXPLORER.value)
        elif values["MODE"] == ExecutionMode.DEV:
            values["SERVER_PORT"] = values.get("DEV_PORT", ServerPort.DEV.value)
        else:
            raise ValueError(f"Invalid mode {values['MODE']}!")

        return values

    @validator("CHIA_ROOT", pre=True)
    def expand_root(cls, v: str) -> Path:
        return Path(v).expanduser()

    @validator("CONFIG_PATH", "LOG_PATH", "DB_PATH")
    def prepend_root(cls, v: str, values: dict[str, Any]) -> Path:
        full_dir: Path = values["CHIA_ROOT"] / v
        parent: Path = full_dir.parent
        parent.mkdir(parents=True, exist_ok=True)
        return full_dir


def get_settings() -> Settings:
    in_pyinstaller: bool = getattr(sys, "frozen", False)

    default_env_file: Path
    default_config_file: Path
    if in_pyinstaller:
        base_path = getattr(sys, "_MEIPASS")
        default_env_file = Path(base_path).joinpath(".env")
        default_config_file = Path(base_path).joinpath("config.yaml")
    else:
        default_env_file = Path(".env")
        default_config_file = Path("config.yaml")

    default_settings = Settings(_env_file=default_env_file)  # type: ignore[call-arg]
    config_file: Path = default_settings.CONFIG_PATH

    if not config_file.is_file():
        config_file.parent.mkdir(parents=True, exist_ok=True)
        shutil.copy(default_config_file, config_file)

    with open(config_file) as f:
        settings_dict = yaml.safe_load(f)

    settings_dict = default_settings.dict() | (settings_dict or {})
    Settings(**settings_dict)
    Settings._instance = Settings(**settings_dict)

    return Settings._instance


settings = get_settings()