"""Configuration module for shell-speak."""

import os
from pathlib import Path
from typing import Optional


def get_data_dir() -> Path:
    """Get the data directory for shell-speak."""
    return Path(os.environ.get("SHELL_SPEAK_DATA_DIR", "~/.local/share/shell-speak")).expanduser()


def get_history_file() -> Path:
    """Get the path to the command history file."""
    return Path(os.environ.get("SHELL_SPEAK_HISTORY_FILE", "~/.local/share/shell-speak/history.json")).expanduser()


def get_corrections_file() -> Path:
    """Get the path to the user corrections file."""
    return Path(os.environ.get("SHELL_SPEAK_CORRECTIONS_FILE", "~/.local/share/shell-speak/corrections.json")).expanduser()


def ensure_data_dir() -> Path:
    """Ensure the data directory exists."""
    data_dir = get_data_dir()
    data_dir.mkdir(parents=True, exist_ok=True)
    return data_dir


DEFAULT_TOOLS = ["docker", "kubectl", "git", "unix"]