"""SnapForm Job Server without Blender dependencies."""

from blendjob import JobServer

from .jobs import generate_depth as _generate_depth
from .jobs import generate_surface as _generate_surface
from .jobs.background import run as _remove_background
from .jobs.balloon import run as _analyze_balloon
from .jobs.debug_ben2 import run as _debug_ben2
from .jobs.debug_da3 import run as _debug_da3
from .jobs.debug_moge2 import run as _debug_moge2
from .jobs.frames import run as _extract_frames
from .model_manager import ModelManager, ModelSessionCache


server = JobServer("SnapForm Job Server")


@server.resource("model_manager")
def create_model_manager(job_server):
    return ModelManager(job_server.storage_root / "models")


@server.job("download-model")
def download_model(context, parameters):
    manager = context.resource("model_manager")
    return manager.download(context, parameters["model"])


@server.job("download-required-models")
def download_required_models(context, parameters):
    manager = context.resource("model_manager")
    return manager.download_missing(context, parameters["models"])


@server.job("analyze-balloon")
def analyze_balloon(context, parameters):
    return _analyze_balloon(context, parameters)


@server.job("remove-background")
def remove_background(context, parameters):
    return _remove_background(context, parameters)


@server.job("extract-frames")
def extract_frames(context, parameters):
    return _extract_frames(context, parameters)


@server.job("generate-depth")
def generate_depth(context, parameters):
    return _generate_depth(context, parameters)


@server.job("generate-surface")
def generate_surface(context, parameters):
    return _generate_surface(context, parameters)


@server.job("debug-da3")
def debug_da3(context, parameters):
    return _debug_da3(context, parameters)


@server.job("debug-moge2")
def debug_moge2(context, parameters):
    return _debug_moge2(context, parameters)


@server.job("debug-ben2")
def debug_ben2(context, parameters):
    return _debug_ben2(context, parameters)


__all__ = ("ModelManager", "ModelSessionCache", "server")