from typing import Callable, Union, Dict, List

from apscheduler.triggers.combining import AndTrigger, OrTrigger
from apscheduler.triggers.cron import CronTrigger
from apscheduler.triggers.date import DateTrigger
from apscheduler.triggers.interval import IntervalTrigger

from .component_matches import *


def convert2lst(elements: Union[str, list, tuple]) -> list:
    if isinstance(elements, str):
        return [elements]
    elif isinstance(elements, tuple):
        return list(elements)
    return elements


class Command:
    def __init__(self,
                 bind_prefix: str,
                 alias: Union[str, list, tuple, dict] = None,
                 desc: str = None,
                 recommend_modules: Union[str, list, tuple] = None,
                 developers: Union[str, list, tuple] = None,
                 required_admin: bool = False,
                 base: bool = False,
                 required_superuser: bool = False,
                 available_for: Union[str, list, tuple] = '*',
                 exclude_from: Union[str, list, tuple] = ''):
        self.bind_prefix: str = bind_prefix
        if isinstance(alias, str):
            alias = {alias: bind_prefix}
        elif isinstance(alias, (tuple, list)):
            alias = {x: bind_prefix for x in alias}
        self.alias: Dict[str, str] = alias
        self.desc: str = desc
        self.recommend_modules: List[str] = convert2lst(recommend_modules)
        self.developers: List[str] = convert2lst(developers)
        self.required_admin: bool = required_admin
        self.base: bool = base
        self.required_superuser: bool = required_superuser
        self.available_for: List[str] = convert2lst(available_for)
        self.exclude_from: List[str] = convert2lst(exclude_from)
        self.match_list = CommandMatches()


class RegexCommand:
    def __init__(self,
                 bind_prefix: str,
                 alias: Union[str, list, tuple, dict] = None,
                 desc: str = None,
                 recommend_modules: Union[str, list, tuple] = None,
                 developers: Union[str, list, tuple] = None,
                 required_admin: bool = False,
                 base: bool = False,
                 required_superuser: bool = False,
                 available_for: Union[str, list, tuple] = '*',
                 exclude_from: Union[str, list, tuple] = ''):
        self.bind_prefix: str = bind_prefix
        if isinstance(alias, str):
            alias = {alias: bind_prefix}
        elif isinstance(alias, (tuple, list)):
            alias = {x: bind_prefix for x in alias}
        self.alias: Dict[str, str] = alias
        self.desc: str = desc
        self.recommend_modules: List[str] = convert2lst(recommend_modules)
        self.developers: List[str] = convert2lst(developers)
        self.required_admin: bool = required_admin
        self.base: bool = base
        self.required_superuser: bool = required_superuser
        self.available_for: List[str] = convert2lst(available_for)
        self.exclude_from: List[str] = convert2lst(exclude_from)
        self.match_list = RegexMatches()


class Schedule:
    def __init__(self,
                 function: Callable,
                 trigger: [AndTrigger, OrTrigger, DateTrigger, CronTrigger, IntervalTrigger],
                 bind_prefix: str,
                 desc: str = None,
                 alias: Union[str, list, tuple, dict] = None,
                 recommend_modules: Union[str, list, tuple] = None,
                 developers: Union[str, list, tuple] = None,
                 required_superuser: bool = False,
                 available_for: Union[str, list, tuple] = '*',
                 exclude_from: Union[str, list, tuple] = ''):
        self.function: Callable = function
        self.trigger: [AndTrigger, OrTrigger, DateTrigger, CronTrigger, IntervalTrigger] = trigger
        self.bind_prefix: str = bind_prefix
        self.desc: str = desc
        if isinstance(alias, str):
            alias = {alias: bind_prefix}
        elif isinstance(alias, (tuple, list)):
            alias = {x: bind_prefix for x in alias}
        self.alias: Dict[str, str] = alias
        self.recommend_modules: List[str] = convert2lst(recommend_modules)
        self.developers: List[str] = convert2lst(developers)
        self.required_superuser: bool = required_superuser
        self.available_for: List[str] = convert2lst(available_for)
        self.exclude_from: List[str] = convert2lst(exclude_from)


class StartUp:
    def __init__(self,
                 function: Callable,
                 bind_prefix: str,
                 desc: str = None,
                 alias: Union[str, list, tuple, dict] = None,
                 recommend_modules: Union[str, list, tuple] = None,
                 developers: Union[str, list, tuple] = None,
                 required_superuser: bool = False,
                 available_for: Union[str, list, tuple] = '*',
                 exclude_from: Union[str, list, tuple] = ''):
        self.function: Callable = function
        self.bind_prefix: str = bind_prefix
        self.desc: str = desc
        if isinstance(alias, str):
            alias = {alias: bind_prefix}
        elif isinstance(alias, (tuple, list)):
            alias = {x: bind_prefix for x in alias}
        self.alias: Dict[str, str] = alias
        self.recommend_modules: List[str] = convert2lst(recommend_modules)
        self.developers = convert2lst(developers)
        self.required_superuser: bool = required_superuser
        self.available_for: List[str] = convert2lst(available_for)
        self.exclude_from: List[str] = convert2lst(exclude_from)


__all__ = ["Command", "RegexCommand", "Schedule", "StartUp", "AndTrigger", "OrTrigger", "DateTrigger",
           "CronTrigger", "IntervalTrigger"]