"""Company context helpers for multi-company users."""

from flask import session
from flask_login import current_user


def get_active_company_id():
    """Return the active company ID from session, falling back to users.company_id."""
    if not current_user.is_authenticated:
        return None
    active_id = session.get('active_company_id')
    if active_id:
        return active_id
    return current_user.company_id