import asyncio
from typing import Any, Dict, List, Union, Optional

from nonebot.config import Config
from nonebot.adapters import Bot as BaseBot
from nonebot.drivers import Driver, WebSocket

from .event import Event
from .message import Message, MessageSegment


def get_auth_bearer(access_token: Optional[str] = ...) -> Optional[str]:
    ...


async def _check_reply(bot: "Bot", event: Event):
    ...


def _check_at_me(bot: "Bot", event: Event):
    ...


def _check_nickname(bot: "Bot", event: Event):
    ...


def _handle_api_result(result: Optional[Dict[str, Any]]) -> Any:
    ...


class ResultStore:
    _seq: int = ...
    _futures: Dict[int, asyncio.Future] = ...

    @classmethod
    def get_seq(cls) -> int:
        ...

    @classmethod
    def add_result(cls, result: Dict[str, Any]):
        ...

    @classmethod
    async def fetch(cls, seq: int, timeout: Optional[float]) -> Dict[str, Any]:
        ...


class Bot(BaseBot):

    def __init__(self,
                 driver: Driver,
                 connection_type: str,
                 config: Config,
                 self_id: str,
                 *,
                 websocket: WebSocket = None):
        ...

    def type(self) -> str:
        ...

    @classmethod
    async def check_permission(cls, driver: Driver, connection_type: str,
                               headers: dict, body: Optional[dict]) -> str:
        ...

    async def handle_message(self, message: dict):
        ...

    async def call_api(self, api: str, *, **data) -> Any:
        ...

    async def send(self, event: Event, message: Union[str, Message,
                                                      MessageSegment],
                   **kwargs) -> Any:
        ...

    async def send_private_msg(
        self,
        *,
        user_id: int,
        message: Union[str, Message],
        auto_escape: bool = ...,
    ) -> Dict[str, Any]:
        """
        :说明:

          发送私聊消息。

        :参数:

          * ``user_id``: 对方 QQ 号
          * ``message``: 要发送的内容
          * ``auto_escape``: 消息内容是否作为纯文本发送（即不解析 CQ 码），只在 ``message`` 字段是字符串时有效

        """
        ...

    async def send_group_msg(
        self,
        *,
        group_id: int,
        message: Union[str, Message],
        auto_escape: bool = ...,
    ) -> Dict[str, Any]:
        """
        :说明:

          发送群消息。

        :参数:

          * ``group_id``: 群号
          * ``message``: 要发送的内容
          * ``auto_escape``: 消息内容是否作为纯文本发送（即不解析 CQ 码），只在 ``message`` 字段是字符串时有效

        """
        ...

    async def send_msg(
        self,
        *,
        message_type: Optional[str] = ...,
        user_id: Optional[int] = ...,
        group_id: Optional[int] = ...,
        message: Union[str, Message],
        auto_escape: bool = ...,
    ) -> Dict[str, Any]:
        """
        :说明:

          发送消息。

        :参数:

          * ``message_type``: 消息类型，支持 ``private``、``group``，分别对应私聊、群组、讨论组，如不传入，则根据传入的 ``*_id`` 参数判断
          * ``user_id``: 对方 QQ 号（消息类型为 ``private`` 时需要）
          * ``group_id``: 群号（消息类型为 ``group`` 时需要）
          * ``message``: 要发送的内容
          * ``auto_escape``: 消息内容是否作为纯文本发送（即不解析 CQ 码），只在 ``message`` 字段是字符串时有效

        """
        ...

    async def delete_msg(
        self,
        *,
        message_id: int,
    ) -> None:
        """
        :说明:

          撤回消息。

        :参数:

          * ``message_id``: 消息 ID

        """
        ...

    async def get_msg(
        self,
        *,
        message_id: int,
    ) -> Dict[str, Any]:
        """
        :说明:

          获取消息。

        :参数:

          * ``message_id``: 消息 ID

        """
        ...

    async def get_forward_msg(
        self,
        *,
        id: int,
    ) -> None:
        """
        :说明:

          获取合并转发消息。

        :参数:

          * ``id``: 合并转发 ID

        """
        ...

    async def send_like(
        self,
        *,
        user_id: int,
        times: int = ...,
    ) -> None:
        """
        :说明:

          发送好友赞。

        :参数:

          * ``user_id``: 对方 QQ 号
          * ``times``: 赞的次数，每个好友每天最多 10 次

        """
        ...

    async def set_group_kick(
        self,
        *,
        group_id: int,
        user_id: int,
        reject_add_request: bool = ...,
    ) -> None:
        """
        :说明:

          群组踢人。

        :参数:

          * ``group_id``: 群号
          * ``user_id``: 要踢的 QQ 号
          * ``reject_add_request``: 拒绝此人的加群请求

        """
        ...

    async def set_group_ban(
        self,
        *,
        group_id: int,
        user_id: int,
        duration: int = ...,
    ) -> None:
        """
        :说明:

          群组单人禁言。

        :参数:

          * ``group_id``: 群号
          * ``user_id``: 要禁言的 QQ 号
          * ``duration``: 禁言时长，单位秒，``0`` 表示取消禁言

        """
        ...

    async def set_group_anonymous_ban(
        self,
        *,
        group_id: int,
        anonymous: Optional[Dict[str, Any]] = ...,
        anonymous_flag: Optional[str] = ...,
        duration: int = ...,
    ) -> None:
        """
        :说明:

          群组匿名用户禁言。

        :参数:

          * ``group_id``: 群号
          * ``anonymous``: 可选，要禁言的匿名用户对象（群消息上报的 ``anonymous`` 字段）
          * ``anonymous_flag``: 可选，要禁言的匿名用户的 flag（需从群消息上报的数据中获得）
          * ``duration``: 禁言时长，单位秒，无法取消匿名用户禁言

        """
        ...

    async def set_group_whole_ban(
        self,
        *,
        group_id: int,
        enable: bool = ...,
    ) -> None:
        """
        :说明:

          群组全员禁言。

        :参数:

          * ``group_id``: 群号
          * ``enable``: 是否禁言

        """
        ...

    async def set_group_admin(
        self,
        *,
        group_id: int,
        user_id: int,
        enable: bool = ...,
    ) -> None:
        """
        :说明:

          群组设置管理员。

        :参数:

          * ``group_id``: 群号
          * ``user_id``: 要设置管理员的 QQ 号
          * ``enable``: ``True`` 为设置，``False`` 为取消

        """
        ...

    async def set_group_anonymous(
        self,
        *,
        group_id: int,
        enable: bool = ...,
    ) -> None:
        """
        :说明:

          群组匿名。

        :参数:

          * ``group_id``: 群号
          * ``enable``: 是否允许匿名聊天

        """
        ...

    async def set_group_card(
        self,
        *,
        group_id: int,
        user_id: int,
        card: str = ...,
    ) -> None:
        """
        :说明:

          设置群名片（群备注）。

        :参数:

          * ``group_id``: 群号
          * ``user_id``: 要设置的 QQ 号
          * ``card``: 群名片内容，不填或空字符串表示删除群名片

        """
        ...

    async def set_group_name(
        self,
        *,
        group_id: int,
        group_name: str,
    ) -> None:
        """
        :说明:

          设置群名。

        :参数:

          * ``group_id``: 群号
          * ``group_name``: 新群名

        """
        ...

    async def set_group_leave(
        self,
        *,
        group_id: int,
        is_dismiss: bool = ...,
    ) -> None:
        """
        :说明:

          退出群组。

        :参数:

          * ``group_id``: 群号
          * ``is_dismiss``: 是否解散，如果登录号是群主，则仅在此项为 True 时能够解散

        """
        ...

    async def set_group_special_title(
        self,
        *,
        group_id: int,
        user_id: int,
        special_title: str = ...,
        duration: int = ...,
    ) -> None:
        """
        :说明:

          设置群组专属头衔。

        :参数:

          * ``group_id``: 群号
          * ``user_id``: 要设置的 QQ 号
          * ``special_title``: 专属头衔，不填或空字符串表示删除专属头衔
          * ``duration``: 专属头衔有效期，单位秒，-1 表示永久，不过此项似乎没有效果，可能是只有某些特殊的时间长度有效，有待测试

        """
        ...

    async def set_friend_add_request(
        self,
        *,
        flag: str,
        approve: bool = ...,
        remark: str = ...,
    ) -> None:
        """
        :说明:

          处理加好友请求。

        :参数:

          * ``flag``: 加好友请求的 flag（需从上报的数据中获得）
          * ``approve``: 是否同意请求
          * ``remark``: 添加后的好友备注（仅在同意时有效）

        """
        ...

    async def set_group_add_request(
        self,
        *,
        flag: str,
        sub_type: str,
        approve: bool = ...,
        reason: str = ...,
    ) -> None:
        """
        :说明:

          处理加群请求／邀请。

        :参数:

          * ``flag``: 加群请求的 flag（需从上报的数据中获得）
          * ``sub_type``: ``add`` 或 ``invite``，请求类型（需要和上报消息中的 ``sub_type`` 字段相符）
          * ``approve``: 是否同意请求／邀请
          * ``reason``: 拒绝理由（仅在拒绝时有效）

        """
        ...

    async def get_login_info(self) -> Dict[str, Any]:
        """
        :说明:

          获取登录号信息。

        """
        ...

    async def get_stranger_info(
        self,
        *,
        user_id: int,
        no_cache: bool = ...,
    ) -> Dict[str, Any]:
        """
        :说明:

          获取陌生人信息。

        :参数:

          * ``user_id``: QQ 号
          * ``no_cache``: 是否不使用缓存（使用缓存可能更新不及时，但响应更快）

        """
        ...

    async def get_friend_list(self) -> List[Dict[str, Any]]:
        """
        :说明:

          获取好友列表。

        """
        ...

    async def get_group_info(
        self,
        *,
        group_id: int,
        no_cache: bool = ...,
    ) -> Dict[str, Any]:
        """
        :说明:

          获取群信息。

        :参数:

          * ``group_id``: 群号
          * ``no_cache``: 是否不使用缓存（使用缓存可能更新不及时，但响应更快）

        """
        ...

    async def get_group_list(self) -> List[Dict[str, Any]]:
        """
        :说明:

          获取群列表。

        """
        ...

    async def get_group_member_info(
        self,
        *,
        group_id: int,
        user_id: int,
        no_cache: bool = ...,
    ) -> Dict[str, Any]:
        """
        :说明:

          获取群成员信息。

        :参数:

          * ``group_id``: 群号
          * ``user_id``: QQ 号
          * ``no_cache``: 是否不使用缓存（使用缓存可能更新不及时，但响应更快）

        """
        ...

    async def get_group_member_list(
        self,
        *,
        group_id: int,
    ) -> List[Dict[str, Any]]:
        """
        :说明:

          获取群成员列表。

        :参数:

          * ``group_id``: 群号

        """
        ...

    async def get_group_honor_info(
        self,
        *,
        group_id: int,
        type: str = ...,
    ) -> Dict[str, Any]:
        """
        :说明:

          获取群荣誉信息。

        :参数:

          * ``group_id``: 群号
          * ``type``: 要获取的群荣誉类型，可传入 ``talkative`` ``performer`` ``legend`` ``strong_newbie`` ``emotion`` 以分别获取单个类型的群荣誉数据，或传入 ``all`` 获取所有数据

        """
        ...

    async def get_cookies(
        self,
        *,
        domain: str = ...,
    ) -> Dict[str, Any]:
        """
        :说明:

          获取 Cookies。

        :参数:

          * ``domain``: 需要获取 cookies 的域名

        """
        ...

    async def get_csrf_token(self) -> Dict[str, Any]:
        """
        :说明:

          获取 CSRF Token。

        """
        ...

    async def get_credentials(
        self,
        *,
        domain: str = ...,
    ) -> Dict[str, Any]:
        """
        :说明:

          获取 QQ 相关接口凭证。

        :参数:

          * ``domain``: 需要获取 cookies 的域名

        """
        ...

    async def get_record(
        self,
        *,
        file: str,
        out_format: str,
    ) -> Dict[str, Any]:
        """
        :说明:

          获取语音。

        :参数:

          * ``file``: 收到的语音文件名（CQ 码的 ``file`` 参数），如 ``0B38145AA44505000B38145AA4450500.silk``
          * ``out_format``: 要转换到的格式，目前支持 ``mp3``、``amr``、``wma``、``m4a``、``spx``、``ogg``、``wav``、``flac``

        """
        ...

    async def get_image(
        self,
        *,
        file: str,
    ) -> Dict[str, Any]:
        """
        :说明:

          获取图片。

        :参数:

          * ``file``: 收到的图片文件名（CQ 码的 ``file`` 参数），如 ``6B4DE3DFD1BD271E3297859D41C530F5.jpg``

        """
        ...

    async def can_send_image(self) -> Dict[str, Any]:
        """
        :说明:

          检查是否可以发送图片。

        """
        ...

    async def can_send_record(self) -> Dict[str, Any]:
        """
        :说明:

          检查是否可以发送语音。

        """
        ...

    async def get_status(self) -> Dict[str, Any]:
        """
        :说明:

          获取插件运行状态。

        """
        ...

    async def get_version_info(self) -> Dict[str, Any]:
        """
        :说明:

          获取版本信息。

        """
        ...

    async def set_restart(
        self,
        *,
        delay: int = ...,
    ) -> None:
        """
        :说明:

          重启 OneBot 实现。

        :参数:

          * ``delay``: 要延迟的毫秒数，如果默认情况下无法重启，可以尝试设置延迟为 2000 左右

        """
        ...

    async def clean_cache(self) -> None:
        """
        :说明:

          清理数据目录。

        """
        ...