1from typing import List
 2
 3from hcraft.elements import Item, Zone
 4from hcraft.task import PlaceItemTask
 5from hcraft.transformation import Transformation, Use, Yield, PLAYER, CURRENT_ZONE
 6
 7from hcraft.examples.minicraft.minicraft import MiniCraftEnv
 8
 9
10MINICRAFT_NAME = "Unlock"
11__doc__ = MiniCraftEnv.description(MINICRAFT_NAME, for_module_header=True)
12
13
14class MiniHCraftUnlock(MiniCraftEnv):
15    MINICRAFT_NAME = MINICRAFT_NAME
16    __doc__ = MiniCraftEnv.description(MINICRAFT_NAME)
17
18    START = Zone("start_room")
19    """Start room."""
20
21    KEY = Item("key")
22    """Key used to unlock the door."""
23
24    OPEN_DOOR = Item("open_door")
25    """Open door between the two rooms."""
26    LOCKED_DOOR = Item("locked_door")
27    """Locked door between the two rooms, can be unlocked with a key."""
28
29    def __init__(self, **kwargs) -> None:
30        self.task = PlaceItemTask(self.OPEN_DOOR)
31        super().__init__(
32            self.MINICRAFT_NAME,
33            purpose=self.task,
34            start_zone=self.START,
35            **kwargs,
36        )
37
38    def build_transformations(self) -> List[Transformation]:
39        transformations = []
40
41        search_for_key = Transformation(
42            "search_for_key",
43            inventory_changes=[
44                Yield(CURRENT_ZONE, self.KEY, create=1, max=0),
45                Yield(PLAYER, self.KEY, create=0, max=0),
46            ],
47            zone=self.START,
48        )
49        transformations.append(search_for_key)
50
51        pickup = Transformation(
52            "pickup_key",
53            inventory_changes=[
54                Use(CURRENT_ZONE, self.KEY, consume=1),
55                Yield(PLAYER, self.KEY, create=1),
56            ],
57        )
58        put_down = Transformation(
59            "put_down_key",
60            inventory_changes=[
61                Use(PLAYER, self.KEY, consume=1),
62                Yield(CURRENT_ZONE, self.KEY, create=1),
63            ],
64        )
65        transformations += [pickup, put_down]
66
67        search_for_door = Transformation(
68            "search_for_door",
69            inventory_changes=[
70                Yield(CURRENT_ZONE, self.LOCKED_DOOR, max=0),
71                Yield(CURRENT_ZONE, self.OPEN_DOOR, create=0, max=0),
72            ],
73            zone=self.START,
74        )
75        transformations.append(search_for_door)
76
77        unlock_door = Transformation(
78            "unlock_door",
79            inventory_changes=[
80                Use(PLAYER, self.KEY),
81                Use(CURRENT_ZONE, self.LOCKED_DOOR, consume=1),
82                Yield(CURRENT_ZONE, self.OPEN_DOOR, create=1),
83            ],
84        )
85        transformations.append(unlock_door)
86
87        return transformations