from pyrogram.enums import ParseMode

from pagermaid.listener import listener
from pagermaid.enums import Client, Message


@listener(command="duckduckgo", description="Duckduckgo 搜索", parameters="[query]")
async def duckduckgo(client: Client, message: Message):
    text = message.arguments
    if not text:
        return await message.edit("请输入查询内容")
    async with client.conversation("PagerMaid_Modify_bot") as conv:
        answer: Message = await conv.ask(f"/duckduckgo {text}")
        await conv.mark_as_read()
    await message.edit(answer.text.html, parse_mode=ParseMode.HTML)


@listener(command="caiyun", description="彩云翻译", parameters="[query]")
async def caiyun_translate(client: Client, message: Message):
    text = message.arguments
    if not text:
        return await message.edit("请输入查询内容")
    async with client.conversation("PagerMaid_Modify_bot") as conv:
        answer: Message = await conv.ask(f"/translate {text}")
        await conv.mark_as_read()
    await message.edit(answer.text)


@listener(command="weather", description="使用彩云天气 api 查询国内实时天气。", parameters="[位置]")
async def weather(client: Client, message: Message):
    text = message.arguments
    if not text:
        return await message.edit("请输入正确的地址")
    async with client.conversation("PagerMaid_Modify_bot") as conv:
        answer: Message = await conv.ask(f"/weather_api {text}")
        await conv.mark_as_read()
    await message.edit(answer.text)


@listener(
    command="weather_pic", description="使用彩云天气 api 查询国内实时天气，但是显示图片。", parameters="[位置]"
)
async def weather_pic(client: Client, message: Message):
    text = message.arguments
    if not text:
        return await message.edit("请输入正确的地址")
    async with client.conversation("PagerMaid_Modify_bot") as conv:
        answer: Message = await conv.ask(f"/weather {text}")
        await conv.mark_as_read()
    await answer.copy(
        message.chat.id,
        reply_to_message_id=message.reply_to_message_id,
        message_thread_id=message.message_thread_id,
    )
    await message.safe_delete()


@listener(
    command="weather_he", description="使用和风天气 api 查询国内实时天气，但是显示图片。", parameters="[位置]"
)
async def weather_he(client: Client, message: Message):
    text = message.arguments
    if not text:
        return await message.edit("请输入正确的地址")
    async with client.conversation("PagerMaid_Modify_bot") as conv:
        answer: Message = await conv.ask(f"/weather_he {text}")
        await conv.mark_as_read()
    await answer.copy(
        message.chat.id,
        reply_to_message_id=message.reply_to_message_id,
        message_thread_id=message.message_thread_id,
    )
    await message.safe_delete()


async def az_tts(client: Client, message: Message, mode: str):
    text = message.arguments
    if not text:
        return await message.edit("请输入需要 tts 的内容")
    async with client.conversation("PagerMaid_Modify_bot") as conv:
        answer: Message = await conv.ask(f"/tts {text} {mode}")
        await conv.mark_as_read()
    await answer.copy(
        message.chat.id,
        reply_to_message_id=message.reply_to_message_id,
        message_thread_id=message.message_thread_id,
    )
    await message.safe_delete()


@listener(
    command="tts_nan",
    description="通过 Azure 文本到语音 基于字符串生成 简体男声 语音消息。",
    parameters="[字符串]",
)
async def az_tts_nan(client: Client, message: Message):
    await az_tts(client, message, "")


@listener(
    command="tts_nv",
    description="通过 Azure 文本到语音 基于字符串生成 简体女声 语音消息。",
    parameters="[字符串]",
)
async def az_tts_nv(client: Client, message: Message):
    await az_tts(client, message, "nv")


@listener(
    command="tts_tw",
    description="通过 Azure 文本到语音 基于字符串生成 繁体男声 语音消息。",
    parameters="[字符串]",
)
async def az_tts_tw(client: Client, message: Message):
    await az_tts(client, message, "tw")


@listener(
    command="tts_ne",
    description="通过 Azure 文本到语音 基于字符串生成 简体新闻男声 语音消息。",
    parameters="[字符串]",
)
async def az_tts_ne(client: Client, message: Message):
    await az_tts(client, message, "ne")


@listener(
    command="tts_en",
    description="通过 Azure 文本到语音 基于字符串生成 英文男声 语音消息。",
    parameters="[字符串]",
)
async def az_tts_en(client: Client, message: Message):
    await az_tts(client, message, "en")


@listener(command="draw", description="使用 AI 进行绘图。", parameters="prompt")
async def draw_photo(client: Client, message: Message):
    text = message.obtain_message()
    if not text:
        return await message.edit("请输入 prompt")
    async with client.conversation("PagerMaid_Modify_bot") as conv:
        answer: Message = await conv.ask(f"/draw {text}")
        await conv.mark_as_read()
    await answer.copy(
        message.chat.id,
        reply_to_message_id=message.reply_to_message_id,
        message_thread_id=message.message_thread_id,
    )
    await message.safe_delete()