anonstream/anonstream/control/parse.py

46 行
1.5 KiB
Python

# SPDX-FileCopyrightText: 2022 n9k <https://git.076.ne.jp/ninya9k>
# SPDX-License-Identifier: AGPL-3.0-or-later
from anonstream.control.spec import ParseException, Parsed
from anonstream.control.spec.common import Str
from anonstream.control.spec.methods.allowedness import SPEC as SPEC_ALLOWEDNESS
from anonstream.control.spec.methods.chat import SPEC as SPEC_CHAT
from anonstream.control.spec.methods.emote import SPEC as SPEC_EMOTE
from anonstream.control.spec.methods.help import SPEC as SPEC_HELP
from anonstream.control.spec.methods.quit import SPEC as SPEC_QUIT
from anonstream.control.spec.methods.title import SPEC as SPEC_TITLE
from anonstream.control.spec.methods.user import SPEC as SPEC_USER
SPEC = Str({
'help': SPEC_HELP,
'quit': SPEC_QUIT,
'title': SPEC_TITLE,
'chat': SPEC_CHAT,
'user': SPEC_USER,
'allowednesss': SPEC_ALLOWEDNESS,
'emote': SPEC_EMOTE,
})
async def parse(request):
words = request.split()
if not words:
normal, response = None, ''
else:
spec = SPEC
index = 0
args = []
while True:
try:
spec, n_consumed, more_args = spec.consume(words, index)
except ParseException as e:
normal, response = None, e.args[0] + '\n'
break
except Parsed as e:
fn, *_ = e.args
normal, response = await fn(*args)
break
else:
index += n_consumed
args.extend(more_args)
return normal, response