d_gag.py

# -*- coding:utf-8 -*-
from encodings import utf_8
from pickle import FALSE
from select import select
from telegram import Update
from telegram.ext import CallbackContext
import redis
import json
c_GAGTYPES = {
    '胡萝卜口塞': [1, 0],  # 名稱:[一次加多少,多少絨度可以被使用]
    '口塞球': [3, 10],
    '充气口塞球': [5, 100],
    '深喉口塞': [7, 500],
    '金属开口器': [9, 1000],
    '炮机口塞': [11, 1500],
    '超级口塞': [65536, 4294967296],
}
# redis 記錄格式： [[剩餘次數,用具名稱],[參與人員陣列]]


def canUse(point: int) -> list[str]:
    """根據當前絨度列出支援使用的道具"""
    gagKeys = list(c_GAGTYPES.keys())
    canUseNames: list[str] = []
    for gagName in gagKeys:
        gagInfo = c_GAGTYPES[gagName]
        needPoint = gagInfo[1]
        if point >= needPoint:
            canUseNames.append(gagName)
    return canUseNames


def rpoint(redisConnect: redis.Redis, user: str, change: int) -> int:
    """查詢或修改絨度"""
    rediskey: str = 'rpt_' + str(user)
    point = 0
    rpointInfo = redisConnect.get(rediskey)
    if rpointInfo != None and len(rpointInfo) > 0:
        rpointInfo = rpointInfo.decode()
        point = int(rpointInfo)
    if change != 0:
        point += change
        if point < 0:
            point = 0
        redisConnect.set(rediskey, str(point))
    return point


def enable(update: Update, context: CallbackContext, redisPool: redis.ConnectionPool, switchMode: bool):
    """功能开关"""
    fromUser: str = '@'+update.message.from_user.username
    redisConnect = redis.Redis(connection_pool=redisPool)
    rediskey: str = 'gagcan_' + str(fromUser)
    enableInfo = redisConnect.get(rediskey)
    enableMode = 0
    alert = ''
    if enableInfo != None and len(enableInfo) > 0:
        enableInfo = enableInfo.decode()
        enableMode = int(enableInfo)
    enableAltStr = '被他人佩戴口塞。'
    if enableMode < 0 or enableMode > 1:
        redisConnect.close()
        alert = fromUser+' 你被超级管理员「滥权」了！被禁止修改该开关！'
        if enableMode < 0:
            alert += '\n目前 '+fromUser+' 强制禁止'+enableAltStr
        elif enableMode > 1:
            alert += '\n目前 '+fromUser+' 强制允许'+enableAltStr
        print(update.effective_chat.id,update.effective_chat.title,update.message.from_user.id,update.message.from_user.username, alert)
        context.bot.send_message(
            chat_id=update.effective_chat.id, text=alert)
        return
    if switchMode:
        alert = '现已允许 '+fromUser+' '+enableAltStr+'\n该操作对所有本 bot 所在群组均有效。'
        enableMode = 1
    else:
        alert = '现已禁止 '+fromUser+' '+enableAltStr+'\n仅对之后的佩戴操作生效，不会移除当前佩戴的口塞。'
        enableMode = 0
    redisConnect.set(rediskey, str(enableMode))
    redisConnect.close()
    print(update.effective_chat.id,update.effective_chat.title,update.message.from_user.id,update.message.from_user.username, alert)
    context.bot.send_message(
        chat_id=update.effective_chat.id, text=alert)


def help(update: Update, context: CallbackContext, c_CHAR: list[list[str]]):
    if update.message.chat.type != 'private':
        alert = '点此了解：\nhttps://t.me/rbq_ch_bot_ch/4'
        context.bot.send_message(chat_id=update.effective_chat.id, text=alert)
        return
    f = open('help_gag.txt', 'r', encoding='utf_8')
    txt = f.read()
    f.close()
    # if hashlib.md5(txt.encode()).hexdigest() != 'da03afce333339838aaacf472be6d9e7':
    #     txt = '文件保护功能检测到错误，请联系实例维护者。'
    t1Arr: list[str] = c_CHAR[0]
    t2Arr: list[str] = c_CHAR[1]
    t1: str = '「'+('」、「'.join(t1Arr))+'」'
    t2: str = '「'+('」、「'.join(t2Arr))+'」'
    gagTypesKeys = list(c_GAGTYPES.keys())
    gagInfos: list[str] = []
    for gagName in gagTypesKeys:
        gagInfoArr = c_GAGTYPES[gagName]
        addNum: str = str(gagInfoArr[0])
        canNum: str = str(gagInfoArr[1])
        newLine: str = '「'+gagName+'」，需要对方绒度 '+canNum+' ，增加所需挣扎次数 '+addNum+' 。'
        gagInfos.append(newLine)
    t3: str = '\n'.join(gagInfos)
    txt = txt.replace('<t1>', t1).replace('<t2>', t2).replace('<t3>', t3)
    context.bot.send_message(chat_id=update.effective_chat.id, text=txt)


def add(update: Update, context: CallbackContext, redisPool0: redis.ConnectionPool, c_CHAR: list[list[str]]):
    """為他人佩戴"""
    argsLen = len(context.args)
    if argsLen == 0:
        alert = '使用方式： `/gag @username 口塞名`\n有关详细使用规则和可用口塞名，请发送 `/gag help` 了解。'
        print(update.effective_chat.id,update.effective_chat.title,update.message.from_user.id,update.message.from_user.username, alert)
        context.bot.send_message(
            chat_id=update.effective_chat.id, text=alert)
        return
    gagTypesKeys = list(c_GAGTYPES.keys())
    gagName: str = gagTypesKeys[0]
    if argsLen == 2:
        if gagName in gagTypesKeys:
            gagName = context.args[1]
        else:
            alert = '你使了我这里没有的口塞呢……\n要了解都有哪些口塞可以使用及规则，请发送 `/gag help` 了解。'
            print(update.effective_chat.id,update.effective_chat.title,update.message.from_user.id,update.message.from_user.username, alert)
            context.bot.send_message(
                chat_id=update.effective_chat.id, text=alert)
            return
    selectGagInfo = c_GAGTYPES[gagName]
    selectGagAdd: int = selectGagInfo[0]
    selectGagNeed: int = selectGagInfo[1]
    toUser: str = context.args[0]
    if toUser == 'help':
        help(update, context, c_CHAR)
        return
    elif toUser == 'on':
        enable(update, context, redisPool0, True)
        return
    elif toUser == 'off':
        enable(update, context, redisPool0, False)
        return
    elif toUser[0] != '@':
        return
    fromUser: str = '@'+update.message.from_user.username
    chatID: int = update.message.chat.id
    redisConnect = redis.Redis(connection_pool=redisPool0)
    rediskey: str = 'gagcan_' + str(toUser)
    gagInfo = redisConnect.get(rediskey)
    isOK = False
    if gagInfo != None and len(gagInfo) > 0:
        if int(gagInfo.decode()) > 0:
            isOK = True
    if isOK == False:
        redisConnect.close()
        alert = fromUser+' 抱歉， '+toUser+' 目前禁止被其他人安装口塞。\n' + \
            toUser+' 必须使用 `/gag on` 指令来允许其他人这样做。'
        print(update.effective_chat.id,update.effective_chat.title,update.message.from_user.id,update.message.from_user.username, alert)
        context.bot.send_message(
            chat_id=update.effective_chat.id, text=alert)
        return
    rediskey = 'gag_' + str(chatID) + '_' + str(toUser)
    gagInfo = redisConnect.get(rediskey)
    alert: str = ''
    if gagInfo != None and len(gagInfo) > 0:
        gagInfo = gagInfo.decode()
        if gagInfo == '0':
            redisConnect.close()
            alert = fromUser+' 抱歉， '+toUser+' 刚刚挣脱口塞！请给对方 1 分钟的休息时间！'
            print(update.effective_chat.id,update.effective_chat.title,update.message.from_user.id,update.message.from_user.username, alert)
            context.bot.send_message(
                chat_id=update.effective_chat.id, text=alert)
            return
        infoArr = json.loads(gagInfo)
        infoArrInf = infoArr[0]
        names: list[str] = infoArr[1]
        gagTotal: int = infoArrInf[0]
        gagNameFrom: str = infoArrInf[1]
        if gagName == gagNameFrom:
            isRepeat = False
            for name in names:
                if name == fromUser:
                    isRepeat = True
                    break
            if isRepeat:
                alert = '抱歉， '+fromUser+' 。你已经为 '+toUser + \
                    ' 佩戴/加固过 '+gagName+' 了，请在 '+toUser+' 挣脱后再试。'
            else:
                gagTotal += selectGagAdd
                infoArrInf[0] = gagTotal
                infoArr[0] = infoArrInf
                names.append(fromUser)
                infoArr[1] = names
                gagInfo = json.dumps(infoArr)
                redisConnect.set(rediskey, gagInfo, ex=600)
                alert = fromUser+' 加固了 '+toUser+' 的 '+gagName+' ！'
            alert += '\n'+toUser+' 目前佩戴着被 ' + \
                (' 、 '.join(names))+' 安装或加固的 '+gagName + \
                ' ，还需要挣扎 '+str(gagTotal)+' 次才能把它挣脱！'
        else:
            alert = '抱歉， '+fromUser+' 。由于 '+toUser + \
                    ' 已经佩戴 '+gagNameFrom+' 了，已经没有地方再装个 '+gagName+' 啦。请在 '+toUser+' 挣脱后再试。'
    else:
        point: int = rpoint(redisConnect, toUser, 0)
        if point < selectGagNeed:
            redisConnect.close()
            canUseNames: list[str] = canUse(point)
            alert = fromUser+' 抱歉，由于 '+toUser+' 的绒度不够，你不能给 '+toUser+' 佩戴 ' + \
                gagName+' ，需要等待对方绒度达到 '+str(selectGagNeed)+' 才行。\n'+toUser+' 当前绒度为 ' + \
                str(point)+' ，目前可以使用的口塞有 '+(' 、 '.join(canUseNames)) + \
                ' 。\n要查询自己的绒度，使用 `/rbqpoint` ；要查询每个绒度对应可用口塞，使用 `/gag help` 。'
            print(update.effective_chat.id,update.effective_chat.title,update.message.from_user.id,update.message.from_user.username, alert)
            context.bot.send_message(
                chat_id=update.effective_chat.id, text=alert)
            return
        gagInfo = json.dumps([[selectGagAdd, gagName], [fromUser]])
        redisConnect.set(rediskey, gagInfo, ex=600)
        alert = fromUser+' 为 '+toUser+' 戴上了 '+gagName+' ！\n'+toUser+' 必须挣扎 '+str(selectGagAdd)+' 次才能挣脱它！\n其他人可以继续用同样指令加固 '+toUser+' 的 '+gagName+' （但同一个人只能在对方挣脱后才能再次为对方佩戴或加固）。'
        if fromUser == toUser:
            alert += '\n咦？！居然自己给自己戴？真是个可爱的绒布球呢！'
        alert += '\nhttps://t.me/rbq_ch_bot_ch/5'
    redisConnect.close()
    print(update.effective_chat.id,update.effective_chat.title,update.message.from_user.id,update.message.from_user.username, alert)
    context.bot.send_message(chat_id=update.effective_chat.id, text=alert)


def chk(update: Update, context: CallbackContext, redisPool0: redis.ConnectionPool, c_CHAR: list[list[str]]) -> bool:
    """檢查加固值是否應該變化"""
    text: str = update.message.text
    fromUser: str = '@'+update.message.from_user.username
    chatID: int = update.message.chat.id
    redisConnect = redis.Redis(connection_pool=redisPool0)
    rediskey: str = 'gag_' + str(chatID) + '_' + str(fromUser)
    gagInfo = redisConnect.get(rediskey)
    if gagInfo != None and len(gagInfo) > 0:
        gagInfo = gagInfo.decode()
        if gagInfo == '0':
            redisConnect.close()
            return False
        infoArr = json.loads(gagInfo)
        infoArrInf = infoArr[0]
        gagTotal: int = int(infoArrInf[0])
        gagName: str = infoArrInf[1]
        names = infoArr[1]
        isOK = True
        charAll = c_CHAR[0] + c_CHAR[1]
        for msgChar in text:
            isInChar = False
            for dbChar in charAll:
                if msgChar == dbChar:
                    isInChar = True
                    break
            if isInChar == False:
                isOK = False
                break
        if isOK:
            inOK1 = False
            inOK2 = False
            for msgChar in text:
                if inOK1 == False:
                    for dbChar in c_CHAR[0]:
                        if msgChar == dbChar:
                            inOK1 = True
                            break
                if inOK2 == False:
                    for dbChar in c_CHAR[1]:
                        if msgChar == dbChar:
                            inOK2 = True
                            break
                if inOK1 and inOK2:
                    break
            if inOK1 == False or inOK2 == False:
                isOK = False
        if isOK:
            gagTotal -= 1
            infoArrInf[0] = gagTotal
            infoArr[0] = infoArrInf
            gagInfo = json.dumps(infoArr)
            redisConnect.set(rediskey, gagInfo, ex=600)
            gagTotalStr: str = str(gagTotal)
            rpoint(redisConnect, fromUser, 1)
            print(fromUser+' -1 = '+gagTotalStr)
            if gagTotal > 0:
                singleNum: str = gagTotalStr[-1]
                if singleNum == '5' or singleNum == '0':
                    alert = fromUser+' 加油！还有 '+gagTotalStr+' 次！'
                    print(update.effective_chat.id,update.effective_chat.title,update.message.from_user.id,update.message.from_user.username, alert)
                    context.bot.send_message(
                        chat_id=update.effective_chat.id, text=alert)
            else:
                redisConnect.set(rediskey, '0', ex=60)
                point = rpoint(redisConnect, fromUser, 1)
                print(fromUser+' -1 = '+str(gagTotal))
                alert: str = fromUser + ' 挣脱了被 ' + \
                    (' 、 '.join(names))+' 佩戴或加固的 '+gagName + \
                    ' ！现在可以自由说话了！\n接下来 1 分钟为强制休息时间，期间不能再被佩戴任何口塞哦。\n现在 ' + \
                    fromUser+' 的绒度升到了 '+str(point)+' ！'
                print(update.effective_chat.id,update.effective_chat.title,update.message.from_user.id,update.message.from_user.username, alert)
                context.bot.send_message(
                    chat_id=update.effective_chat.id, text=alert)
        else:
            context.bot.delete_message(
                chat_id=update.message.chat_id, message_id=update.message.message_id)
            print(fromUser+' -0 = '+str(gagTotal))
        redisConnect.close()
        return True
    redisConnect.close()
    return False