from __future__ import annotations

import typing

from .. import operator, entities, errors


@operator.operator_class(
    name='model',
    help='显示和切换模型列表',
    usage='!model\n!model show <模型名>\n!model set <模型名>',
    privilege=2,
)
class ModelOperator(operator.CommandOperator):
    """Model命令"""

    async def execute(self, context: entities.ExecuteContext) -> typing.AsyncGenerator[entities.CommandReturn, None]:
        content = '模型列表:\n'

        model_list = self.ap.model_mgr.model_list

        for model in model_list:
            content += f'\n名称: {model.name}\n'
            content += f'请求器: {model.requester.name}\n'

        content += f'\n当前对话使用模型: {context.query.use_model.name}\n'
        content += f'新对话默认使用模型: {self.ap.provider_cfg.data.get("model")}\n'

        yield entities.CommandReturn(text=content.strip())


@operator.operator_class(name='show', help='显示模型详情', privilege=2, parent_class=ModelOperator)
class ModelShowOperator(operator.CommandOperator):
    """Model Show命令"""

    async def execute(self, context: entities.ExecuteContext) -> typing.AsyncGenerator[entities.CommandReturn, None]:
        model_name = context.crt_params[0]

        model = None
        for _model in self.ap.model_mgr.model_list:
            if model_name == _model.name:
                model = _model
                break

        if model is None:
            yield entities.CommandReturn(error=errors.CommandError(f'未找到模型 {model_name}'))
        else:
            content = '模型详情\n'
            content += f'名称: {model.name}\n'
            if model.model_name is not None:
                content += f'请求模型名称: {model.model_name}\n'
            content += f'请求器: {model.requester.name}\n'
            content += f'密钥组: {model.token_mgr.name}\n'
            content += f'支持视觉: {model.vision_supported}\n'
            content += f'支持工具: {model.tool_call_supported}\n'

            yield entities.CommandReturn(text=content.strip())


@operator.operator_class(name='set', help='设置默认使用模型', privilege=2, parent_class=ModelOperator)
class ModelSetOperator(operator.CommandOperator):
    """Model Set命令"""

    async def execute(self, context: entities.ExecuteContext) -> typing.AsyncGenerator[entities.CommandReturn, None]:
        model_name = context.crt_params[0]

        model = None
        for _model in self.ap.model_mgr.model_list:
            if model_name == _model.name:
                model = _model
                break

        if model is None:
            yield entities.CommandReturn(error=errors.CommandError(f'未找到模型 {model_name}'))
        else:
            self.ap.provider_cfg.data['model'] = model_name
            await self.ap.provider_cfg.dump_config()
            yield entities.CommandReturn(text=f'已设置当前使用模型为 {model_name}，重置会话以生效')