from pyrogram import Client
from pagermaid.listener import listener
from pagermaid.utils import Message, execute


@listener(command="cal",
          description="计算\n示例：\n`-cal 1+1`加法\n`-cal 2-1`减法\n`-cal 1*2`乘法\n`-cal 4/2`除法\n`-cal 4^2`幂运算\n`-cal sqrt(4)`开方",
          parameters="<基本运算>")
async def cal(_: Client, message: Message):
    command = message.arguments
    if not command:
        await message.edit("`出错了呜呜呜 ~ 无效的参数。`")
        return

    await message.edit(f"{command}")
    cmd = f'echo "scale=4;{command}" | bc'
    result = await execute(cmd)

    if result:
        await message.edit(f"{command}=\n`{result}`")
    else:
        return

@listener(command="con",
          description="换算\n示例：\n`-con 2 99`将99转换为2进制",
          parameters="<进制(数字)> <数值>")
async def con(_: Client, message: Message):
    command = message.arguments.split()
    if not command:
        await message.edit("`出错了呜呜呜 ~ 无效的参数。`")
        return

    obase = command[0].upper().strip()
    num = command[1].upper().strip()
    await message.edit(f"{num}")
    cmd = f'echo "obase={obase};{num}" | bc'
    result = await execute(cmd)

    if result:
        await message.edit(f"{num}=\n`{result}`\n{obase}进制")
    else:
        return