# python
# file: config/paths.py
import os

# Constant directory names
DATA_DIR_NAME = 'data'
TEST_DATA_DIR_NAME = 'test_data'

# Project root (two levels up from this file)
PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

def get_database_dir(db_env: str | None = None) -> str:
    """
    Return the absolute base directory path for the given DB env.
    db_env: 'prod' uses `data/db`, anything else uses `test_data/db`.
    """
    env = (db_env or os.environ.get('DB_ENV', 'prod')).lower()
    base_name = DATA_DIR_NAME if env == 'prod' else TEST_DATA_DIR_NAME
    return os.path.join(PROJECT_ROOT, base_name, 'db')

def get_user_image_dir(username: str | None) -> str:
    """
    Return the absolute directory path for storing images for a specific user.
    """
    if username:
        return os.path.join(PROJECT_ROOT, DATA_DIR_NAME, 'images', username)
    return os.path.join(PROJECT_ROOT, 'resources', 'images')
