"""
Helpers/utils for working with tornado asynchronous stuff
"""

import asyncio
import contextlib
import logging
import sys
import threading

import tornado.concurrent
import tornado.ioloop

log = logging.getLogger(__name__)


def aioloop(io_loop, warn=False):
    """
    Ensure the ioloop is an asyncio loop not a tornado ioloop.
    """
    if isinstance(io_loop, asyncio.AbstractEventLoop):
        return io_loop
    elif isinstance(io_loop, tornado.ioloop.IOLoop):
        if warn:
            import traceback

            log.warning("Passed tornado loop %s", "".join(traceback.format_stack()))
        return io_loop.asyncio_loop
    else:
        raise RuntimeError("Loop must be AbstractEventLoop (prefered) or IOLoop")


@contextlib.contextmanager
def current_ioloop(io_loop):
    """
    A context manager that will set the current ioloop to io_loop for the context
    """
    try:
        # Use instance=False to avoid auto-creating a default IOLoop that leaks FDs
        orig_loop = tornado.ioloop.IOLoop.current(instance=False)
    except RuntimeError:
        orig_loop = None

    # Normalize io_loop to asyncio loop
    asyncio_loop = aioloop(io_loop)
    asyncio.set_event_loop(asyncio_loop)
    try:
        yield
    finally:
        if orig_loop:
            asyncio.set_event_loop(aioloop(orig_loop))
        else:
            asyncio.set_event_loop(None)


class SyncWrapper:
    """
    A wrapper to make Async classes synchronous

    This is uses as a simple wrapper, for example:

    asynchronous = AsyncClass()
    # this method would regularly return a future
    future = asynchronous.async_method()

    sync = SyncWrapper(async_factory_method, (arg1, arg2), {'kwarg1': 'val'})
    # the sync wrapper will automatically wait on the future
    ret = sync.async_method()
    """

    def __init__(
        self,
        cls,
        args=None,
        kwargs=None,
        async_methods=None,
        close_methods=None,
        loop_kwarg=None,
    ):
        self.asyncio_loop = asyncio.new_event_loop()
        self.io_loop = tornado.ioloop.IOLoop(
            asyncio_loop=self.asyncio_loop, make_current=False
        )
        if args is None:
            args = []
        if kwargs is None:
            kwargs = {}
        if async_methods is None:
            async_methods = []
        if close_methods is None:
            close_methods = []
        self.loop_kwarg = loop_kwarg
        self.cls = cls
        if loop_kwarg:
            kwargs[self.loop_kwarg] = self.io_loop
        with current_ioloop(self.io_loop):
            self.obj = cls(*args, **kwargs)
        self._async_methods = list(
            set(async_methods + getattr(self.obj, "async_methods", []))
        )
        self._close_methods = list(
            set(close_methods + getattr(self.obj, "close_methods", []))
        )

    def _populate_async_methods(self):
        """
        We need the '_coroutines' attribute on classes until we can depricate
        tornado<4.5. After that 'is_coroutine_fuction' will always be
        available.
        """
        if hasattr(self.obj, "_coroutines"):
            self._async_methods += self.obj._coroutines

    def __repr__(self):
        return f"<SyncWrapper(cls={self.cls})"

    @staticmethod
    def _loop_can_run_until_complete(loop):
        """
        Return ``True`` iff ``loop.run_until_complete(coro)`` can drive a
        freshly created coroutine to completion without immediately raising.

        ``BaseEventLoop.run_until_complete`` raises ``RuntimeError`` if the
        loop is closed or already running, but only *after* its
        ``future`` argument has been evaluated.  Constructing the
        coroutine without being able to await it leaks it through
        ``coroutine.__del__`` as ``RuntimeWarning: coroutine '...' was
        never awaited`` (see ``close()``).  We avoid that by inspecting
        the loop's state up front.
        """
        if loop is None:
            return False
        if loop.is_closed():
            return False
        if loop.is_running():
            return False
        return True

    def close(self):
        for method in self._close_methods:
            if method in self._async_methods:
                method = self._wrap(method)
            else:
                try:
                    method = getattr(self.obj, method)
                except AttributeError:
                    log.error("No sync method %s on object %r", method, self.obj)
                    continue
            try:
                method()
            except AttributeError:
                log.error("No async method %s on object %r", method, self.obj)
            except Exception as exc:  # pylint: disable=broad-except
                log.exception(
                    "Exception encountered while running stop method: %s", exc
                )
        # Shut down asyncio resources before closing the IOLoop so file descriptors
        # held by pending tasks, async generators, and the default executor are released.
        #
        # Each of the three ``run_until_complete`` calls below takes a freshly
        # constructed coroutine object as its argument.  If the loop is already
        # closed (or running) at that point ``run_until_complete`` raises
        # ``RuntimeError`` *after* the coroutine has been created but *before*
        # ``ensure_future`` wraps it — and the bare coroutine object is then
        # garbage-collected unawaited, emitting a
        # ``RuntimeWarning: coroutine '...' was never awaited`` on stderr.  On
        # Python 3.14 / Windows the batch CLI integration tests
        # (``tests/pytests/integration/cli/test_batch.py::test_batch_retcode``
        # and ``test_multiple_modules_in_batch``) gate on ``assert not
        # cmd.stderr`` and turn that warning into a hard failure.
        #
        # Gate every call on ``not _loop_can_run_until_complete(loop)`` so we
        # never even *construct* the inner coroutine when the loop can't drive
        # it to completion.
        try:
            if self._loop_can_run_until_complete(self.asyncio_loop):
                pending_tasks = [
                    task
                    for task in asyncio.all_tasks(self.asyncio_loop)
                    if not task.done()
                ]
                if pending_tasks:
                    for task in pending_tasks:
                        task.cancel()
                    gathered = asyncio.gather(*pending_tasks, return_exceptions=True)
                    try:
                        self.asyncio_loop.run_until_complete(gathered)
                    except Exception:  # pylint: disable=broad-except
                        # ``gathered`` is a Future; if run_until_complete bailed
                        # part-way we still need to make sure the Future is
                        # consumed so its exception (if any) isn't logged as
                        # unhandled.  Tasks already cancelled above.
                        if not gathered.done():
                            gathered.cancel()

            if self._loop_can_run_until_complete(self.asyncio_loop):
                shutdown_agens = self.asyncio_loop.shutdown_asyncgens()
                try:
                    self.asyncio_loop.run_until_complete(shutdown_agens)
                except Exception:  # pylint: disable=broad-except
                    shutdown_agens.close()

            if self._loop_can_run_until_complete(self.asyncio_loop):
                shutdown_exec = self.asyncio_loop.shutdown_default_executor()
                try:
                    self.asyncio_loop.run_until_complete(shutdown_exec)
                except Exception:  # pylint: disable=broad-except
                    shutdown_exec.close()

        except Exception as exc:  # pylint: disable=broad-except
            log.error("Error during asyncio shutdown: %s", exc)

        io_loop = self.io_loop
        try:
            io_loop.stop()
        except Exception as exc:  # pylint: disable=broad-except
            log.error("Error stopping IOLoop: %s", exc)
        try:
            io_loop.close(all_fds=True)
        except KeyError:
            pass
        except Exception as exc:  # pylint: disable=broad-except
            log.error("Unexpected error closing IOLoop: %s", exc)

        if not self.asyncio_loop.is_closed():
            try:
                self.asyncio_loop.close()
            except Exception as exc:  # pylint: disable=broad-except
                log.error("Error closing asyncio loop: %s", exc)

        self.obj = None
        self.io_loop = None
        self.asyncio_loop = None

    def __getattr__(self, key):
        if key in self._async_methods:
            return self._wrap(key)
        return getattr(self.obj, key)

    def _wrap(self, key):
        def wrap(*args, **kwargs):
            try:
                asyncio.get_running_loop()
            except RuntimeError:
                # asyncio.get_running_loop() raises RuntimeError
                # if there is no running loop, so we can run the method
                # directly with no detaching it to the distinct thread.
                # It will make SyncWrapper way faster for the cases
                # when there are no nested SyncWrapper objects used.
                return self.io_loop.run_sync(
                    lambda: getattr(self.obj, key)(*args, **kwargs)
                )
            results = []
            thread = threading.Thread(
                target=self._target,
                args=(key, args, kwargs, results, self.asyncio_loop),
            )
            thread.start()
            thread.join()
            if results[0]:
                return results[1]
            else:
                exc_info = results[1]
                raise exc_info[1].with_traceback(exc_info[2])

        return wrap

    def _target(self, key, args, kwargs, results, asyncio_loop):
        asyncio.set_event_loop(asyncio_loop)
        io_loop = tornado.ioloop.IOLoop.current()
        try:
            result = io_loop.run_sync(lambda: getattr(self.obj, key)(*args, **kwargs))
            results.append(True)
            results.append(result)
        except Exception:  # pylint: disable=broad-except
            results.append(False)
            results.append(sys.exc_info())

    def __enter__(self):
        if hasattr(self.obj, "__aenter__"):
            ret = self._wrap("__aenter__")()
            if ret == self.obj:
                return self
            else:
                return ret
        elif hasattr(self.obj, "__enter__"):
            ret = self.obj.__enter__()
            if ret == self.obj:
                return self
            else:
                return ret
        return self

    def __exit__(self, exc_type, exc_val, tb):
        if hasattr(self.obj, "__aexit__"):
            self._wrap("__aexit__")(exc_type, exc_val, tb)
        self.close()
