main.py

import asyncio
import logging
import os
import sys

from common.constants import IS_WINDOWS
from common.logger import init_logging
from common.scheduler import scheduler
from common.tasks import ArkViewer
from common.version import VERSION

init_logging()


log = logging.getLogger("arkview.main")


class Manager:
    """Compile with 'pyinstaller.exe --clean app.spec'"""

    def __init__(self, loop: asyncio.AbstractEventLoop) -> None:
        self.loop: asyncio.AbstractEventLoop = loop
        self.handler = ArkViewer()

    async def start(self) -> None:
        log.info(f"Version: {VERSION}")
        scheduler.start()
        scheduler.remove_all_jobs()
        success = await self.handler.initialize()
        if not success:
            input("Initialization failed. Press any key to exit...")
            self.loop.stop()

    async def shutdown(self) -> None:
        scheduler.remove_all_jobs()
        scheduler.shutdown(wait=False)

        tasks = [t for t in asyncio.all_tasks() if t is not asyncio.current_task()]
        [task.cancel() for task in tasks]

        log.info("Cancelling outstanding tasks")
        await asyncio.gather(*tasks, return_exceptions=False)

        log.info("Shutting down asyncgens...")
        try:
            await self.loop.shutdown_asyncgens()
            await asyncio.sleep(1)
            self.loop.stop()
        except RuntimeError:
            pass

    @classmethod
    def run(cls) -> None:
        log.info(f"Starting ArkViewer with PID {os.getpid()}")

        loop = asyncio.ProactorEventLoop() if IS_WINDOWS else asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
        arkview = cls(loop)

        try:
            loop.create_task(arkview.start())
            loop.run_forever()
        except KeyboardInterrupt:
            print("CTRL+C received, shutting down...")
        except Exception as e:
            log.critical("Fatal error!", exc_info=e)
        finally:
            log.info("Shutting down...")
            if not loop.is_closed():
                loop.run_until_complete(arkview.shutdown())
                loop.run_until_complete(loop.shutdown_asyncgens())
                asyncio.set_event_loop(None)
                loop.stop()
                loop.close()

                log.info("Goodbye.")
                sys.exit()


if __name__ == "__main__":
    Manager.run()