anonstream/anonstream/control/spec/methods/chat.py

56 行
1.8 KiB
Python

# SPDX-FileCopyrightText: 2022 n9k <https://git.076.ne.jp/ninya9k>
# SPDX-License-Identifier: AGPL-3.0-or-later
import itertools
from anonstream.chat import delete_chat_messages
from anonstream.control.spec import NoParse
from anonstream.control.spec.common import Str, End, Args
from anonstream.control.spec.utils import get_item, json_dumps_contiguous
class ArgsSeqs(Args):
def consume(self, words, index):
seqs = []
for i in itertools.count():
seq_string = get_item(index + i, words)
try:
seq = int(seq_string)
except TypeError as e:
if not seqs:
raise NoParse('incomplete: expected SEQ') from e
else:
break
except ValueError as e:
raise NoParse(
'could not decode {word!r} as base-10 integer'
) from e
else:
seqs.append(seq)
return self.spec, i + 1, seqs
async def cmd_chat_help():
normal = ['chat', 'help']
response = (
'Usage: chat {show [MESSAGES] | delete SEQS}\n'
'Commands:\n'
#' chat show [MESSAGES]......show chat messages\n'
' chat delete SEQS..........delete chat messages\n'
'Definitions:\n'
#' MESSAGES..................undefined\n'
' SEQS......................=SEQ [SEQ...]\n'
' SEQ.......................a chat message\'s seq, base-10 integer\n'
)
return normal, response
async def cmd_chat_delete(*seqs):
delete_chat_messages(seqs)
normal = ['chat', 'delete', *map(str, seqs)]
response = ''
return normal, response
SPEC = Str({
None: End(cmd_chat_help),
'help': End(cmd_chat_help),
'delete': ArgsSeqs(End(cmd_chat_delete)),
})