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)