forked from uniborg/uniborg
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
from uniborg import util
|
|
|
|
|
|
def self_reply_cmd(borg, pattern):
|
|
def wrapper(function):
|
|
@borg.on(util.admin_cmd(pattern))
|
|
async def wrapped(event, *args, **kwargs):
|
|
await event.delete()
|
|
target = await util.get_target_message(borg, event)
|
|
if not target:
|
|
return
|
|
return await function(event, target, *args, **kwargs)
|
|
return wrapped
|
|
return wrapper
|
|
|
|
|
|
def self_reply_selector(borg, cmd):
|
|
def wrapper(function):
|
|
@borg.on(util.admin_cmd(cmd + r"( [+-]?\d+)?$"))
|
|
async def wrapped(event, *args, **kwargs):
|
|
await event.delete()
|
|
reply = await event.get_reply_message()
|
|
if not reply:
|
|
return
|
|
num_offset = int(event.pattern_match.group(1) or 0)
|
|
reverse = num_offset > 0
|
|
|
|
targets = [reply] if reverse else []
|
|
targets.extend(await borg.get_messages(
|
|
await event.get_input_chat(),
|
|
limit=abs(num_offset),
|
|
offset_id=reply.id,
|
|
reverse=reverse
|
|
))
|
|
if not reverse:
|
|
# reverse the list because we want things to always be in
|
|
# history order
|
|
targets.reverse()
|
|
targets.append(reply)
|
|
|
|
return await function(event, targets, num_offset, *args, **kwargs)
|
|
return wrapped
|
|
return wrapper
|