Lonami
/
uniborg
forked from kate/uniborg
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

re: add message header to prevent abuse

This commit is contained in:
Dan Elkouby 2018-05-13 13:25:52 +03:00
parent a99848267f
commit dbf561880e
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
stdplugins/re.py
View File

@ -5,7 +5,7 @@ import regex
from telethon import events, utils from telethon import events, utils
from telethon.tl import types, functions from telethon.tl import types, functions
HEADER = "[[regex]]\n"
KNOWN_RE_BOTS = re.compile( KNOWN_RE_BOTS = re.compile(
r'(regex|moku|BananaButler_)bot', r'(regex|moku|BananaButler_)bot',
flags=re.IGNORECASE flags=re.IGNORECASE
@ -42,8 +42,10 @@ def doit(chat_id, match, original):
def actually_doit(original): def actually_doit(original):
try: try:
s, i = regex.subn(fr, to, original.message, s = original.message
count=count, flags=flags) if s.startswith(HEADER):
s = s[len(HEADER):]
s, i = regex.subn(fr, to, s, count=count, flags=flags)
if i > 0: if i > 0:
return original, s return original, s
except Exception as e: except Exception as e:
@ -90,6 +92,7 @@ async def on_regex(event):
m, s = doit(chat_id, event.pattern_match, await event.reply_message) m, s = doit(chat_id, event.pattern_match, await event.reply_message)
if m is not None: if m is not None:
s = f"{HEADER}{s}"
out = await borg.send_message(await event.input_chat, s, reply_to=m.id) out = await borg.send_message(await event.input_chat, s, reply_to=m.id)
last_msgs[chat_id].appendleft(out) last_msgs[chat_id].appendleft(out)
elif s is not None: elif s is not None: