mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 19:44:20 +00:00
128 lines
4.1 KiB
Python
128 lines
4.1 KiB
Python
# Module docstring
|
|
"""
|
|
The PDA ("main" class) for the :mod:`royalnet_console` frontend.
|
|
"""
|
|
|
|
# Special imports
|
|
from __future__ import annotations
|
|
import royalnet.royaltyping as t
|
|
|
|
# External imports
|
|
import logging
|
|
import asyncio
|
|
import royalnet.engineer as engi
|
|
import click
|
|
import datetime
|
|
|
|
# Internal imports
|
|
from . import bullets
|
|
|
|
# Special global objects
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
# Code
|
|
class ConsolePDA:
|
|
"""
|
|
A PDA which handles :mod:`royalnet` input and output using a terminal as source.
|
|
"""
|
|
|
|
def __init__(self):
|
|
log.debug(f"Creating new ConsolePDA...")
|
|
|
|
self.dispenser: t.Optional[engi.Dispenser] = None
|
|
"""
|
|
The :class:`royalnet.engineer.dispenser.Dispenser` of this PDA.
|
|
"""
|
|
|
|
self.conversations: t.List[engi.Conversation] = []
|
|
"""
|
|
A :class:`list` of conversations to run before a new event is :meth:`.put` in a
|
|
:class:`~royalnet.engineer.dispenser.Dispenser`.
|
|
"""
|
|
|
|
async def run(self, cycles: t.Union[bool, int] = True) -> t.NoReturn:
|
|
"""
|
|
Run the main loop of the :class:`.ConsolePDA` for ``cycles`` cycles, or unlimited cycles if the parameter is
|
|
:data:`True`.
|
|
"""
|
|
|
|
while cycles:
|
|
message = click.prompt("", type=str, prompt_suffix=">>> ", show_default=False)
|
|
log.debug(f"Received a new input: {message!r}")
|
|
|
|
log.debug(f"Creating ConsoleMessageReceived from: {message!r}")
|
|
projectile = bullets.ConsoleMessageReceived(_text=message, _timestamp=datetime.datetime.now())
|
|
|
|
log.debug(f"Putting projectile: {projectile!r}")
|
|
await self.put_projectile(proj=projectile)
|
|
|
|
if isinstance(cycles, int):
|
|
cycles -= 1
|
|
|
|
def register_conversation(self, conv: engi.Conversation) -> None:
|
|
"""
|
|
Register a new conversation in the PDA.
|
|
|
|
:param conv: The conversation to register.
|
|
"""
|
|
log.info(f"Registering conversation: {conv!r}")
|
|
self.conversations.append(conv)
|
|
|
|
def unregister_conversation(self, conv: engi.Conversation) -> None:
|
|
"""
|
|
Unregister a conversation from the PDA.
|
|
|
|
:param conv: The conversation to unregister.
|
|
"""
|
|
log.info(f"Unregistering conversation: {conv!r}")
|
|
self.conversations.remove(conv)
|
|
|
|
def register_partial(self, part: engi.PartialCommand, names: t.List[str]) -> engi.Command:
|
|
"""
|
|
Register a new :class:`~royalnet.engineer.command.PartialCommand` in the PDA, converting it to a
|
|
:class:`royalnet.engineer.Command` in the process.
|
|
|
|
:param part: The :class:`~royalnet.engineer.command.PartialCommand` to register.
|
|
:param names: The :attr:`~royalnet.engineer.command.Command.names` to register the command with.
|
|
:return: The resulting :class:`~royalnet.engineer.command.Command`.
|
|
"""
|
|
log.debug(f"Completing partial: {part!r}")
|
|
if part.syntax:
|
|
command = part.complete(pattern=r"^{name}\s+{syntax}$", names=names)
|
|
else:
|
|
command = part.complete(pattern=r"^{name}$", names=names)
|
|
self.register_conversation(command)
|
|
return command
|
|
|
|
async def put_projectile(self, proj: engi.Projectile) -> None:
|
|
"""
|
|
Insert a new projectile into the dispenser.
|
|
|
|
:param proj: The projectile to put in the dispenser.
|
|
"""
|
|
if not self.dispenser:
|
|
log.debug(f"Dispenser not found, creating one...")
|
|
self.dispenser = engi.Dispenser()
|
|
|
|
log.debug("Getting running loop...")
|
|
loop = asyncio.get_running_loop()
|
|
|
|
for conversation in self.conversations:
|
|
log.debug(f"Creating run task for: {conversation!r}")
|
|
loop.create_task(self.dispenser.run(conversation), name=f"{repr(conversation)}")
|
|
|
|
log.debug("Running a event loop cycle...")
|
|
await asyncio.sleep(0)
|
|
|
|
log.debug(f"Putting projectile {proj!r} in dispenser {self.dispenser!r}...")
|
|
await self.dispenser.put(proj)
|
|
|
|
log.debug("Awaiting another event loop cycle...")
|
|
await asyncio.sleep(0)
|
|
|
|
|
|
# Objects exported by this module
|
|
__all__ = (
|
|
"ConsolePDA",
|
|
)
|