itch-dl/itch_dl/config.py

import os
import json
import logging
import platform
from typing import Optional

import requests
from pydantic import BaseModel

from . import __version__


class Settings(BaseModel):
    """Available settings for itch-dl. Make sure all of them
    have default values, as the config file may not exist."""

    api_key: Optional[str] = None
    user_agent: str = f"python-requests/{requests.__version__} itch-dl/{__version__}"


def create_and_get_config_path() -> str:
    """Returns the configuration directory in the appropriate
    location for the current OS. The directory may not exist."""
    system = platform.system()
    if system == "Linux":
        base_path = os.environ.get("XDG_CONFIG_HOME") or os.path.expanduser("~/.config/")
    elif system == "Darwin":
        base_path = os.path.expanduser("~/Library/Application Support/")
    elif system == "Windows":
        base_path = os.environ.get("APPDATA") or os.path.expanduser("~/AppData/Roaming/")
    else:
        raise NotImplementedError(f"Unknown platform: {system}")

    return os.path.join(base_path, "itch-dl")


def load_config(profile: Optional[str] = None) -> Settings:
    """Loads the configuration from the file system if it exists,
    the returns a Settings object."""
    config_path = create_and_get_config_path()
    config_file_path = os.path.join(config_path, "config.json")
    profile_file_path = os.path.join(config_path, "profiles", profile or "")

    if os.path.isfile(config_file_path):
        logging.debug(f"Found config file: {config_file_path}")
        with open(config_file_path) as f:
            config_data = json.load(f)
    else:
        config_data = {}

    if os.path.isfile(profile_file_path):
        logging.debug(f"Found profile: {profile_file_path}")
        with open(config_file_path) as f:
            profile_data = json.load(f)

        config_data.update(profile_data)

    return Settings(**config_data)