#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from typing import Self, Any, Callable, Optional, Sequence
from Interfaces.Application.AnPInterface import AnPInterface
from Models.QueuesModel import QueuesModel
from Utils.Common import Common

class QueuesManager:

    def __init__(self:Self, anp:AnPInterface) -> None:
        self.anp:AnPInterface = anp
        self.__queues:dict[str, QueuesModel] = {}

        self.update()

    def update(self:Self) -> None:
        key:str

        for key in ("default_queues_files", "queues_files", "default_queues", "queues"):
            self.create(self.anp.settings.get(key), True)

    def reset(self:Self) -> None:
        self.__queues = {}
        self.update()

    def close(self:Self) -> None:
        self.remove()

    def create(self:Self, inputs:Any|None, overwrite:bool = False) -> None:
        
        subinputs:dict[str, Any|None]

        for subinputs in Common.load_json(inputs, True):

            key:str
            queue:Any|None

            for key, queue in subinputs.items():
                if overwrite or key not in self.__queues:
                    self.__queues[key] = QueuesModel(key, queue)

    def add(self:Self, key:str, callback:Callable[[Callable[[], None]], None], *arguments:list[Any|None]) -> int|None:
        if key in self.__queues:
            return self.__queues[key].add(callback, *arguments)
        return None

    def next(self:Self) -> None:

        queue:QueuesModel
        
        for queue in self.__queues.values():
            queue.next()

    def cancel(self:Self, keys:Optional[str|Sequence[str]] = False, _is:Optional[int|Sequence[int]] = None) -> bool:
        if keys is None:

            queue:QueuesModel

            for queue in self.__queues.values():
                if _is is None:
                    queue.cancel_all()
                else:

                    i:int

                    for i in Common.get_keys(_is):
                        queue.cancel(i)

            return True

        done:bool = False
        key:str

        for key in Common.get_keys(keys):
            if key in self.__queues:
                if _is is None:
                    self.__queues[key].cancel_all()
                    done = True
                else:

                    i:int

                    for i in Common.get_keys(_is):
                        self.__queues[key].cancel(i)
                        done = True

        return done
    
    def remove(self:Self, keys:Optional[str|Sequence[str]] = None) -> bool:
        if keys is None:

            for queue in self.__queues.values():
                queue.cancel_all()
            self.__queues = {}

            return True

        done:bool = False
        key:str

        for key in Common.get_keys(keys):
            if key in self.__queues:
                del self.__queues[key]
                done = True

        return done