"""Backward-compatibility shim for USDU helpers.

Route handlers and job logic now live in:
- upscale.job_store
- upscale.job_timeout
- upscale.payload_parsers
- api.usdu_routes
"""

from ..upscale.conditioning import clone_conditioning, clone_control_chain
from ..upscale.job_store import (
    MAX_PAYLOAD_SIZE,
    _cleanup_job,
    _drain_results_queue,
    _get_completed_count,
    _init_job_queue,
    _mark_task_completed,
    ensure_tile_jobs_initialized,
    init_dynamic_job,
    init_static_job_batched,
)
from ..upscale.job_timeout import _check_and_requeue_timed_out_workers
from ..upscale.payload_parsers import _parse_tiles_from_form
from .logging import debug_log
from .network import get_client_session


async def _send_heartbeat_to_master(multi_job_id, master_url, worker_id):
    """Send heartbeat to master from worker-side processing loops."""
    try:
        data = {'multi_job_id': multi_job_id, 'worker_id': str(worker_id)}
        session = await get_client_session()
        url = f"{master_url}/distributed/heartbeat"
        async with session.post(url, json=data) as response:
            response.raise_for_status()
    except Exception as e:
        debug_log(f"Heartbeat failed: {e}")


__all__ = [
    "MAX_PAYLOAD_SIZE",
    "_check_and_requeue_timed_out_workers",
    "_cleanup_job",
    "_drain_results_queue",
    "_get_completed_count",
    "_init_job_queue",
    "_mark_task_completed",
    "_parse_tiles_from_form",
    "_send_heartbeat_to_master",
    "clone_conditioning",
    "clone_control_chain",
    "ensure_tile_jobs_initialized",
    "init_dynamic_job",
    "init_static_job_batched",
]