import random
from pagermaid.listener import listener
from pagermaid.enums import Message
from pagermaid.utils import pip_install

pip_install("jieba")

import jieba
import jieba.posseg as pseg

jieba.setLogLevel(20)


def chaos(x, y, chaosrate):
    if random.random() > chaosrate:
        return x
    if x in {'[', ']'}:
        return ''
    if x in {'，'}:
        return '…'
    if x in {'!', '！', }:
        return '‼‼‼'
    if x in {'。'}:
        return '❗'
    if len(x) > 1 and random.random() < 0.1:
        return f'{x[0]}…{x}'
    if len(x) > 1 and random.random() < 0.4:
        return f'{x[0]}♥{x}'
    if y == 'n' and random.random() < 0.1:
        x = '⭕' * len(x)
        return f'…{x}'
    if x in {'\……n', '\♥n'}:
        return '\n'
    if x in {'…………'}:
        return '……'
    if y == 'n' and random.random() < 0.2:
        x = '⭕' * len(x)
    return f'……{x}'


def chs2yin(s, chaosrate=0.8):
    return ''.join(chaos(x, y, chaosrate) for x, y in pseg.cut(s))


@listener(command="yinglish",
          description="能把中文和英文翻译成淫语的翻译机！")
async def yinglish(context: Message):
    if not context.arguments:
        await context.edit("你没说话我转换个啥")
    else:
        outputtext = chs2yin(context.arguments)
        await context.edit(f"{outputtext}")