Wire target resolution into thing commands

Replace local exact-match helpers with targeting module calls for
prefix matching and ordinal disambiguation. Works in get, drop, and
container extraction (get X from Y).
This commit is contained in:
Jared Miller 2026-02-14 01:20:45 -05:00
parent 5a0c1b2151
commit aca9864881
Signed by: shmup
GPG key ID: 22B5C6D66A38B06C
2 changed files with 141 additions and 32 deletions

View file

@ -3,34 +3,25 @@
from mudlib.commands import CommandDefinition, register from mudlib.commands import CommandDefinition, register
from mudlib.container import Container from mudlib.container import Container
from mudlib.player import Player from mudlib.player import Player
from mudlib.targeting import find_in_inventory, find_thing_on_tile
from mudlib.thing import Thing from mudlib.thing import Thing
from mudlib.zone import Zone from mudlib.zone import Zone
def _find_thing_at(name: str, zone: Zone, x: int, y: int) -> Thing | None: def _find_thing_at(name: str, zone: Zone, x: int, y: int) -> Thing | None:
"""Find a thing on the ground matching name or alias.""" """Find a thing on the ground matching name or alias.
name_lower = name.lower()
for obj in zone.contents_at(x, y): Deprecated: Use find_thing_on_tile from mudlib.targeting instead.
if not isinstance(obj, Thing): """
continue return find_thing_on_tile(name, zone, x, y)
if obj.name.lower() == name_lower:
return obj
if name_lower in (a.lower() for a in obj.aliases):
return obj
return None
def _find_thing_in_inventory(name: str, player: Player) -> Thing | None: def _find_thing_in_inventory(name: str, player: Player) -> Thing | None:
"""Find a thing in the player's inventory matching name or alias.""" """Find a thing in the player's inventory matching name or alias.
name_lower = name.lower()
for obj in player.contents: Deprecated: Use find_in_inventory from mudlib.targeting instead.
if not isinstance(obj, Thing): """
continue return find_in_inventory(name, player)
if obj.name.lower() == name_lower:
return obj
if name_lower in (a.lower() for a in obj.aliases):
return obj
return None
def _format_thing_name(thing: Thing) -> str: def _format_thing_name(thing: Thing) -> str:
@ -113,18 +104,11 @@ async def _handle_take_from(player: Player, args: str) -> None:
await player.send(f"The {container_obj.name} is closed.\r\n") await player.send(f"The {container_obj.name} is closed.\r\n")
return return
# Find thing in container # Find thing in container using targeting (supports prefix and ordinals)
thing = None from mudlib.targeting import resolve_target
thing_name_lower = thing_name.lower()
for obj in container_obj.contents: container_things = [obj for obj in container_obj.contents if isinstance(obj, Thing)]
if not isinstance(obj, Thing): thing = resolve_target(thing_name, container_things)
continue
if obj.name.lower() == thing_name_lower:
thing = obj
break
if thing_name_lower in (a.lower() for a in obj.aliases):
thing = obj
break
if thing is None: if thing is None:
await player.send(f"The {container_obj.name} doesn't contain that.\r\n") await player.send(f"The {container_obj.name} doesn't contain that.\r\n")

View file

@ -0,0 +1,125 @@
"""Tests for targeting integration in thing commands."""
import pytest
from mudlib.commands.things import cmd_drop, cmd_get
from mudlib.container import Container
from mudlib.thing import Thing
@pytest.mark.asyncio
async def test_get_prefix_match(player, test_zone):
"""Test that 'get gob' prefix matches 'goblin figurine' on ground."""
figurine = Thing(name="goblin figurine", x=0, y=0)
figurine.location = test_zone
test_zone._contents.append(figurine)
await cmd_get(player, "gob")
# Check thing moved to player inventory
assert figurine in player.contents
assert figurine not in test_zone._contents
@pytest.mark.asyncio
async def test_get_ordinal_from_ground(player, test_zone):
"""Test that 'get 2.sword' gets second sword from ground."""
sword1 = Thing(name="sword", x=0, y=0)
sword1.location = test_zone
test_zone._contents.append(sword1)
sword2 = Thing(name="sword", x=0, y=0)
sword2.location = test_zone
test_zone._contents.append(sword2)
await cmd_get(player, "2.sword")
# Check second sword moved to player inventory
assert sword2 in player.contents
assert sword1 not in player.contents
@pytest.mark.asyncio
async def test_drop_prefix_match(player, test_zone):
"""Test that 'drop sw' prefix matches sword in inventory."""
sword = Thing(name="sword", x=0, y=0)
sword.move_to(player)
await cmd_drop(player, "sw")
# Check sword moved to ground
assert sword not in player.contents
assert sword in test_zone._contents
assert sword.x == 0
assert sword.y == 0
@pytest.mark.asyncio
async def test_get_ordinal_from_container(player, test_zone):
"""Test that 'get 2.sword from chest' gets second sword from container."""
chest = Container(name="chest", x=0, y=0, capacity=100)
chest.location = test_zone
chest.closed = False
test_zone._contents.append(chest)
sword1 = Thing(name="sword", x=0, y=0)
sword1.move_to(chest)
sword2 = Thing(name="sword", x=0, y=0)
sword2.move_to(chest)
await cmd_get(player, "2.sword from chest")
# Check second sword moved to player inventory
assert sword2 in player.contents
assert sword1 not in player.contents
assert sword1 in chest.contents
@pytest.mark.asyncio
async def test_get_no_match(player, test_zone):
"""Test that get fails gracefully when no match is found."""
await cmd_get(player, "nonexistent")
# Check for error message
written = "".join(call.args[0] for call in player.writer.write.call_args_list)
assert "don't see" in written.lower()
@pytest.mark.asyncio
async def test_drop_no_match(player, test_zone):
"""Test that drop fails gracefully when no match is found."""
await cmd_drop(player, "nonexistent")
# Check for error message
written = "".join(call.args[0] for call in player.writer.write.call_args_list)
assert "not carrying" in written.lower()
@pytest.mark.asyncio
async def test_get_alias_match(player, test_zone):
"""Test that aliases work with targeting."""
figurine = Thing(name="goblin figurine", x=0, y=0, aliases=["fig", "goblin"])
figurine.location = test_zone
test_zone._contents.append(figurine)
await cmd_get(player, "fig")
assert figurine in player.contents
@pytest.mark.asyncio
async def test_drop_ordinal(player, test_zone):
"""Test that drop works with ordinal disambiguation."""
sword1 = Thing(name="sword", x=0, y=0)
sword1.move_to(player)
sword2 = Thing(name="sword", x=0, y=0)
sword2.move_to(player)
await cmd_drop(player, "2.sword")
# Check second sword moved to ground
assert sword2 not in player.contents
assert sword1 in player.contents
assert sword2 in test_zone._contents