"""Manages the asyncio loop."""

import asyncio
import bpy


def kick_async_loop(*args):
    loop = asyncio.get_event_loop()

    if loop.is_closed():
        print('{}: loop closed, stopping'.format(__name__))
        stop_async_loop()
        return

    if not asyncio.Task.all_tasks():
        print('{}: no more scheduled tasks, stopping'.format(__name__))
        stop_async_loop()
        return

    # Perform a single async loop step
    async def do_nothing():
        pass

    loop.run_until_complete(do_nothing())


def async_loop_handler() -> callable:
    name = kick_async_loop.__name__
    for handler in bpy.app.handlers.scene_update_pre:
        if getattr(handler, '__name__', '') == name:
            return handler
    return None


def ensure_async_loop():
    if async_loop_handler() is not None:
        return
    bpy.app.handlers.scene_update_pre.append(kick_async_loop)


def stop_async_loop():
    handler = async_loop_handler()
    if handler is None:
        return
    bpy.app.handlers.scene_update_pre.remove(handler)

    # Optional: cancel all pending tasks.
    # for task in asyncio.Task.all_tasks():
    #     task.cancel()