Add enter command for portal zone transitions
Implements portal-based zone transitions with the enter command. Players can enter portals at their position to move to target zones with specified coordinates. Includes departure/arrival messaging to nearby players and automatic look output in the destination zone. Portals are matched by partial name or exact alias match.
This commit is contained in:
parent
557fffe5fa
commit
aa720edae5
3 changed files with 477 additions and 0 deletions
76
src/mudlib/commands/portals.py
Normal file
76
src/mudlib/commands/portals.py
Normal file
|
|
@ -0,0 +1,76 @@
|
||||||
|
"""Portal commands for zone transitions."""
|
||||||
|
|
||||||
|
from mudlib.commands import CommandDefinition, register
|
||||||
|
from mudlib.commands.movement import send_nearby_message
|
||||||
|
from mudlib.player import Player
|
||||||
|
from mudlib.portal import Portal
|
||||||
|
from mudlib.zone import Zone
|
||||||
|
from mudlib.zones import get_zone
|
||||||
|
|
||||||
|
|
||||||
|
async def cmd_enter(player: Player, args: str) -> None:
|
||||||
|
"""Enter a portal to transition to another zone.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
player: The player executing the command
|
||||||
|
args: Portal name or alias to enter
|
||||||
|
"""
|
||||||
|
if not args.strip():
|
||||||
|
await player.send("Enter what?\r\n")
|
||||||
|
return
|
||||||
|
|
||||||
|
zone = player.location
|
||||||
|
if not isinstance(zone, Zone):
|
||||||
|
await player.send("You are nowhere.\r\n")
|
||||||
|
return
|
||||||
|
|
||||||
|
# Find portal at player's position
|
||||||
|
portals_here = [
|
||||||
|
obj for obj in zone.contents_at(player.x, player.y) if isinstance(obj, Portal)
|
||||||
|
]
|
||||||
|
|
||||||
|
# Match by name or alias (exact match only)
|
||||||
|
target_name = args.strip().lower()
|
||||||
|
portal = None
|
||||||
|
for p in portals_here:
|
||||||
|
if p.name.lower() == target_name:
|
||||||
|
portal = p
|
||||||
|
break
|
||||||
|
if target_name in (a.lower() for a in p.aliases):
|
||||||
|
portal = p
|
||||||
|
break
|
||||||
|
|
||||||
|
if not portal:
|
||||||
|
await player.send("You don't see that here.\r\n")
|
||||||
|
return
|
||||||
|
|
||||||
|
# Look up target zone
|
||||||
|
target_zone = get_zone(portal.target_zone)
|
||||||
|
if not target_zone:
|
||||||
|
await player.send("The portal doesn't lead anywhere.\r\n")
|
||||||
|
return
|
||||||
|
|
||||||
|
# Send departure message to nearby players in old zone
|
||||||
|
await send_nearby_message(
|
||||||
|
player, player.x, player.y, f"{player.name} enters {portal.name}.\r\n"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Move player to target zone
|
||||||
|
player.move_to(target_zone, x=portal.target_x, y=portal.target_y)
|
||||||
|
|
||||||
|
# Send arrival message to nearby players in new zone
|
||||||
|
await send_nearby_message(
|
||||||
|
player,
|
||||||
|
player.x,
|
||||||
|
player.y,
|
||||||
|
f"{player.name} arrives from {portal.name}.\r\n",
|
||||||
|
)
|
||||||
|
|
||||||
|
# Show new zone to player
|
||||||
|
from mudlib.commands.look import cmd_look
|
||||||
|
|
||||||
|
await cmd_look(player, "")
|
||||||
|
|
||||||
|
|
||||||
|
# Register the enter command
|
||||||
|
register(CommandDefinition("enter", cmd_enter))
|
||||||
284
tests/test_enter_portal.py
Normal file
284
tests/test_enter_portal.py
Normal file
|
|
@ -0,0 +1,284 @@
|
||||||
|
"""Tests for portal enter command."""
|
||||||
|
|
||||||
|
from unittest.mock import AsyncMock, MagicMock
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from mudlib.commands import _registry
|
||||||
|
from mudlib.player import Player
|
||||||
|
from mudlib.portal import Portal
|
||||||
|
from mudlib.zone import Zone
|
||||||
|
from mudlib.zones import register_zone, zone_registry
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def mock_writer():
|
||||||
|
writer = MagicMock()
|
||||||
|
writer.write = MagicMock()
|
||||||
|
writer.drain = AsyncMock()
|
||||||
|
return writer
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def mock_reader():
|
||||||
|
return MagicMock()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def test_zone():
|
||||||
|
terrain = [["." for _ in range(10)] for _ in range(10)]
|
||||||
|
return Zone(
|
||||||
|
name="testzone",
|
||||||
|
width=10,
|
||||||
|
height=10,
|
||||||
|
toroidal=True,
|
||||||
|
terrain=terrain,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def target_zone():
|
||||||
|
terrain = [["." for _ in range(10)] for _ in range(10)]
|
||||||
|
return Zone(
|
||||||
|
name="targetzone",
|
||||||
|
width=10,
|
||||||
|
height=10,
|
||||||
|
toroidal=True,
|
||||||
|
terrain=terrain,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def player(mock_reader, mock_writer, test_zone):
|
||||||
|
p = Player(
|
||||||
|
name="TestPlayer",
|
||||||
|
x=5,
|
||||||
|
y=5,
|
||||||
|
reader=mock_reader,
|
||||||
|
writer=mock_writer,
|
||||||
|
location=test_zone,
|
||||||
|
)
|
||||||
|
return p
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(autouse=True)
|
||||||
|
def clear_zones():
|
||||||
|
"""Clear zone registry before and after each test."""
|
||||||
|
zone_registry.clear()
|
||||||
|
yield
|
||||||
|
zone_registry.clear()
|
||||||
|
|
||||||
|
|
||||||
|
# --- cmd_enter ---
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_enter_moves_to_target_zone(player, test_zone, target_zone):
|
||||||
|
"""enter portal moves player to target zone."""
|
||||||
|
from mudlib.commands.portals import cmd_enter
|
||||||
|
|
||||||
|
register_zone("targetzone", target_zone)
|
||||||
|
Portal(
|
||||||
|
name="shimmering doorway",
|
||||||
|
aliases=["doorway"],
|
||||||
|
location=test_zone,
|
||||||
|
x=5,
|
||||||
|
y=5,
|
||||||
|
target_zone="targetzone",
|
||||||
|
target_x=3,
|
||||||
|
target_y=7,
|
||||||
|
)
|
||||||
|
|
||||||
|
await cmd_enter(player, "doorway")
|
||||||
|
assert player.location is target_zone
|
||||||
|
assert player.x == 3
|
||||||
|
assert player.y == 7
|
||||||
|
assert player in target_zone.contents
|
||||||
|
assert player not in test_zone.contents
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_enter_sends_departure_message(
|
||||||
|
player, test_zone, target_zone, mock_writer
|
||||||
|
):
|
||||||
|
"""enter portal sends departure message to old zone."""
|
||||||
|
from mudlib.commands.portals import cmd_enter
|
||||||
|
|
||||||
|
register_zone("targetzone", target_zone)
|
||||||
|
Portal(
|
||||||
|
name="portal",
|
||||||
|
location=test_zone,
|
||||||
|
x=5,
|
||||||
|
y=5,
|
||||||
|
target_zone="targetzone",
|
||||||
|
target_x=3,
|
||||||
|
target_y=7,
|
||||||
|
)
|
||||||
|
other_player = Player(
|
||||||
|
name="OtherPlayer",
|
||||||
|
x=5,
|
||||||
|
y=5,
|
||||||
|
reader=MagicMock(),
|
||||||
|
writer=MagicMock(write=MagicMock(), drain=AsyncMock()),
|
||||||
|
location=test_zone,
|
||||||
|
)
|
||||||
|
|
||||||
|
await cmd_enter(player, "portal")
|
||||||
|
|
||||||
|
# Check that other player received departure message
|
||||||
|
other_writer = other_player.writer
|
||||||
|
calls = [call[0][0] for call in other_writer.write.call_args_list]
|
||||||
|
assert any("TestPlayer" in call and "enter" in call.lower() for call in calls)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_enter_sends_arrival_message(player, test_zone, target_zone):
|
||||||
|
"""enter portal sends arrival message to new zone."""
|
||||||
|
from mudlib.commands.portals import cmd_enter
|
||||||
|
|
||||||
|
register_zone("targetzone", target_zone)
|
||||||
|
Portal(
|
||||||
|
name="portal",
|
||||||
|
location=test_zone,
|
||||||
|
x=5,
|
||||||
|
y=5,
|
||||||
|
target_zone="targetzone",
|
||||||
|
target_x=3,
|
||||||
|
target_y=7,
|
||||||
|
)
|
||||||
|
other_player = Player(
|
||||||
|
name="OtherPlayer",
|
||||||
|
x=3,
|
||||||
|
y=7,
|
||||||
|
reader=MagicMock(),
|
||||||
|
writer=MagicMock(write=MagicMock(), drain=AsyncMock()),
|
||||||
|
location=target_zone,
|
||||||
|
)
|
||||||
|
|
||||||
|
await cmd_enter(player, "portal")
|
||||||
|
|
||||||
|
# Check that other player in target zone received arrival message
|
||||||
|
other_writer = other_player.writer
|
||||||
|
calls = [call[0][0] for call in other_writer.write.call_args_list]
|
||||||
|
assert any("TestPlayer" in call and "arrive" in call.lower() for call in calls)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_enter_shows_look_in_new_zone(
|
||||||
|
player, test_zone, target_zone, mock_writer
|
||||||
|
):
|
||||||
|
"""enter portal triggers look command in new zone."""
|
||||||
|
from mudlib.commands.portals import cmd_enter
|
||||||
|
|
||||||
|
register_zone("targetzone", target_zone)
|
||||||
|
Portal(
|
||||||
|
name="portal",
|
||||||
|
location=test_zone,
|
||||||
|
x=5,
|
||||||
|
y=5,
|
||||||
|
target_zone="targetzone",
|
||||||
|
target_x=3,
|
||||||
|
target_y=7,
|
||||||
|
)
|
||||||
|
|
||||||
|
await cmd_enter(player, "portal")
|
||||||
|
|
||||||
|
# Check that look output was sent (viewport grid should be in output)
|
||||||
|
output = "".join([call[0][0] for call in mock_writer.write.call_args_list])
|
||||||
|
# Should have the @ symbol for player position
|
||||||
|
assert "@" in output
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_enter_no_args(player, mock_writer):
|
||||||
|
"""enter with no arguments gives usage hint."""
|
||||||
|
from mudlib.commands.portals import cmd_enter
|
||||||
|
|
||||||
|
await cmd_enter(player, "")
|
||||||
|
output = mock_writer.write.call_args_list[-1][0][0]
|
||||||
|
assert "enter what" in output.lower() or "usage" in output.lower()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_enter_portal_not_found(player, test_zone, mock_writer):
|
||||||
|
"""enter with portal not at position gives feedback."""
|
||||||
|
from mudlib.commands.portals import cmd_enter
|
||||||
|
|
||||||
|
await cmd_enter(player, "doorway")
|
||||||
|
output = mock_writer.write.call_args_list[-1][0][0]
|
||||||
|
assert "don't see" in output.lower()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_enter_target_zone_not_found(player, test_zone, mock_writer):
|
||||||
|
"""enter with invalid target zone gives graceful error."""
|
||||||
|
from mudlib.commands.portals import cmd_enter
|
||||||
|
|
||||||
|
Portal(
|
||||||
|
name="portal",
|
||||||
|
location=test_zone,
|
||||||
|
x=5,
|
||||||
|
y=5,
|
||||||
|
target_zone="nonexistent",
|
||||||
|
target_x=3,
|
||||||
|
target_y=7,
|
||||||
|
)
|
||||||
|
|
||||||
|
await cmd_enter(player, "portal")
|
||||||
|
output = mock_writer.write.call_args_list[-1][0][0]
|
||||||
|
assert "doesn't lead anywhere" in output.lower() or "nowhere" in output.lower()
|
||||||
|
# Player should still be in original zone
|
||||||
|
assert player.location is test_zone
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_enter_matches_aliases(player, test_zone, target_zone):
|
||||||
|
"""enter matches portal aliases."""
|
||||||
|
from mudlib.commands.portals import cmd_enter
|
||||||
|
|
||||||
|
register_zone("targetzone", target_zone)
|
||||||
|
Portal(
|
||||||
|
name="shimmering doorway",
|
||||||
|
aliases=["door", "doorway"],
|
||||||
|
location=test_zone,
|
||||||
|
x=5,
|
||||||
|
y=5,
|
||||||
|
target_zone="targetzone",
|
||||||
|
target_x=3,
|
||||||
|
target_y=7,
|
||||||
|
)
|
||||||
|
|
||||||
|
await cmd_enter(player, "door")
|
||||||
|
assert player.location is target_zone
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_enter_portal_elsewhere_in_zone(player, test_zone, mock_writer):
|
||||||
|
"""enter doesn't find portals not at player position."""
|
||||||
|
from mudlib.commands.portals import cmd_enter
|
||||||
|
|
||||||
|
Portal(
|
||||||
|
name="distant portal",
|
||||||
|
location=test_zone,
|
||||||
|
x=8,
|
||||||
|
y=8,
|
||||||
|
target_zone="targetzone",
|
||||||
|
target_x=3,
|
||||||
|
target_y=7,
|
||||||
|
)
|
||||||
|
|
||||||
|
await cmd_enter(player, "portal")
|
||||||
|
output = mock_writer.write.call_args_list[-1][0][0]
|
||||||
|
assert "don't see" in output.lower()
|
||||||
|
# Player should still be in original zone
|
||||||
|
assert player.location is test_zone
|
||||||
|
|
||||||
|
|
||||||
|
# --- command registration ---
|
||||||
|
|
||||||
|
|
||||||
|
def test_enter_command_registered():
|
||||||
|
"""enter command is registered."""
|
||||||
|
import mudlib.commands.portals # noqa: F401
|
||||||
|
|
||||||
|
assert "enter" in _registry
|
||||||
117
tests/test_two_way_portals.py
Normal file
117
tests/test_two_way_portals.py
Normal file
|
|
@ -0,0 +1,117 @@
|
||||||
|
"""Tests for two-way portal transitions."""
|
||||||
|
|
||||||
|
from unittest.mock import AsyncMock, MagicMock
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from mudlib.player import Player
|
||||||
|
from mudlib.portal import Portal
|
||||||
|
from mudlib.zone import Zone
|
||||||
|
from mudlib.zones import register_zone, zone_registry
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def mock_writer():
|
||||||
|
writer = MagicMock()
|
||||||
|
writer.write = MagicMock()
|
||||||
|
writer.drain = AsyncMock()
|
||||||
|
return writer
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def mock_reader():
|
||||||
|
return MagicMock()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def zone_a():
|
||||||
|
terrain = [["." for _ in range(10)] for _ in range(10)]
|
||||||
|
return Zone(
|
||||||
|
name="zone_a",
|
||||||
|
width=10,
|
||||||
|
height=10,
|
||||||
|
toroidal=True,
|
||||||
|
terrain=terrain,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def zone_b():
|
||||||
|
terrain = [["." for _ in range(10)] for _ in range(10)]
|
||||||
|
return Zone(
|
||||||
|
name="zone_b",
|
||||||
|
width=10,
|
||||||
|
height=10,
|
||||||
|
toroidal=True,
|
||||||
|
terrain=terrain,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def player(mock_reader, mock_writer, zone_a):
|
||||||
|
p = Player(
|
||||||
|
name="TestPlayer",
|
||||||
|
x=2,
|
||||||
|
y=2,
|
||||||
|
reader=mock_reader,
|
||||||
|
writer=mock_writer,
|
||||||
|
location=zone_a,
|
||||||
|
)
|
||||||
|
return p
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(autouse=True)
|
||||||
|
def clear_zones():
|
||||||
|
"""Clear zone registry before and after each test."""
|
||||||
|
zone_registry.clear()
|
||||||
|
yield
|
||||||
|
zone_registry.clear()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_two_way_portal_transitions(player, zone_a, zone_b):
|
||||||
|
"""Portals work bidirectionally between zones."""
|
||||||
|
from mudlib.commands.portals import cmd_enter
|
||||||
|
|
||||||
|
# Register zones
|
||||||
|
register_zone("zone_a", zone_a)
|
||||||
|
register_zone("zone_b", zone_b)
|
||||||
|
|
||||||
|
# Create portal in zone A pointing to zone B
|
||||||
|
Portal(
|
||||||
|
name="doorway to B",
|
||||||
|
location=zone_a,
|
||||||
|
x=2,
|
||||||
|
y=2,
|
||||||
|
target_zone="zone_b",
|
||||||
|
target_x=7,
|
||||||
|
target_y=7,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Create portal in zone B pointing to zone A
|
||||||
|
Portal(
|
||||||
|
name="doorway to A",
|
||||||
|
location=zone_b,
|
||||||
|
x=7,
|
||||||
|
y=7,
|
||||||
|
target_zone="zone_a",
|
||||||
|
target_x=2,
|
||||||
|
target_y=2,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Player starts in zone A at (2, 2)
|
||||||
|
assert player.location is zone_a
|
||||||
|
assert player.x == 2
|
||||||
|
assert player.y == 2
|
||||||
|
|
||||||
|
# Enter portal to zone B
|
||||||
|
await cmd_enter(player, "doorway to B")
|
||||||
|
assert player.location is zone_b
|
||||||
|
assert player.x == 7
|
||||||
|
assert player.y == 7
|
||||||
|
|
||||||
|
# Enter portal back to zone A
|
||||||
|
await cmd_enter(player, "doorway to A")
|
||||||
|
assert player.location is zone_a
|
||||||
|
assert player.x == 2
|
||||||
|
assert player.y == 2
|
||||||
Loading…
Reference in a new issue