game.py

from dataclasses import dataclass, field, make_dataclass, asdict
from typing import List, Dict
import pickle
import argparse

from oso import Oso

RESET = "\001\x1b[0m\002"
FG_RED = "\001\x1b[31m\002"
FG_GREEN = "\001\x1b[32m\002"
FG_BLUE = "\001\x1b[34m\002"
FG_YELLOW = "\001\x1b[33m\002"


@dataclass
class Game:
    time: int = 0
    id_count: int = 200

    def write(self, fmt, *args):
        print(fmt.format(*args), end="")
        return True

    def red(self, arg):
        return FG_RED + str(arg) + RESET

    def green(self, arg):
        return FG_GREEN + str(arg) + RESET

    def blue(self, arg):
        return FG_BLUE + str(arg) + RESET

    def yellow(self, arg):
        return FG_YELLOW + str(arg) + RESET

    def print_map(self):
        gate = "_ " if PASSAGES.get("garden gate").locked else "  "
        lib_door = "_ " if PASSAGES.get("large oak door").locked else "  "
        trap_door = "_ " if PASSAGES.get("trap door").locked else "  "
        rooms = []
        for room in ROOMS.elems:
            val = "x " if PLAYER.room == room.id else "  "
            rooms.insert(room.id, val)

        print("           ______     _____________________________")
        print("          |      |___|                             |")
        print("          |       ___     {}     north forest      |".format(rooms[9]))
        print("          |      |   |_____________________________|")
        print("          | farm |")
        print("          | plot |    _____________________________")
        print(" ______   |      |   |         attic     {}        |".format(rooms[6]))
        print("| wood |  |      |   |____ {}______________________|".format(trap_door))
        print("| shed |__|      |   |             |               |")
        print("|  {}   __       |___|  kitchen    |   library     |".format(rooms[8]))
        print(
            "|______|  |       ___       {}     |   {}          |".format(
                rooms[3], rooms[5]
            )
        )
        print("          |      |   |   __________|______ {}______|".format(lib_door))
        print("          |  {}  |   |  |                          |".format(rooms[7]))
        print("          |      |   |  |     living room   {}     |".format(rooms[4]))
        print("          |      |   |  |_______________________   |")
        print("          |      |   |                          |  |")
        print("          |      |   |    {}      foyer         |  |".format(rooms[2]))
        print("          |__ {}_|   |_______   ___________________|".format(gate))
        print("             | |_____|                             |")
        print("             |_______      {}           garden     |".format(rooms[1]))
        print("                     |_____________   _____________|")
        print("                                    {}".format(rooms[0]))
        print("                                 clearing")

        return True

    def tick(self):
        self.time += 1
        return True

    def create_object(self, name):
        obj = make_object(id=self.id_count, desc=name, classes=[Takeable])
        self.id_count += 1
        OBJECTS.add_item(obj)
        ROOMS.get_by_id(PLAYER.room).add_object(obj.id)
        return True

    # def save(self):
    #     with open("polar_adventure.pickle", "wb") as f:
    #         pickle.dump(PLAYER, f)
    #         pickle.dump(OBJECTS, f)
    #         pickle.dump(ROOMS, f)
    #         pickle.dump(PASSAGES, f)
    #     print("Your game has been saved to polar_adventure.pickle.")
    #     print("You can load your saved game next time using the `--load` flag.")

    # def load_saved(self, fname):
    #     with open(fname, "rb") as f:
    #         PLAYER = pickle.load(f)
    #         OBJECTS = pickle.load(f)
    #         ROOMS = pickle.load(f)
    #         PASSAGES = pickle.load(f)


@dataclass
class Room:
    id: int
    objects: List[int]
    passages: List[int]
    desc: str

    def remove_object(self, obj_id):
        self.objects.remove(obj_id)
        return True

    def add_object(self, obj_id):
        self.objects.append(obj_id)
        return True


@dataclass
class Passage:
    id: int
    rooms: Dict[int, str]
    desc: str
    locked: bool

    def unlock(self):
        self.locked = False
        return True

    def room_ids(self):
        return list(self.rooms.keys())

    def get_direction(self, room_id):
        return self.rooms.get(room_id)


@dataclass
class Object:
    id: int
    desc: str
    classes: List


@dataclass
class Animal:
    favorite_item: str


@dataclass
class Container:
    is_open: bool
    objects: List[int] = field(default_factory=list)

    def open(self):
        self.is_open = True
        return True

    def close(self):
        self.is_open = False
        return True

    def remove_object(self, obj_id):
        self.objects.remove(obj_id)
        return True

    def add_object(self, obj_id):
        self.objects.append(obj_id)
        return True


class Food:
    pass


@dataclass
class Takeable:
    pass


@dataclass
class Mushroomy:
    pass


@dataclass
class Soup:
    kind: List

    def reset(self):
        self.kind = []
        return True

    def add_ingredient(self, ingredient):
        self.kind.append(ingredient)
        return True


@dataclass
class Source:
    produces: str


@dataclass
class Wand:
    pass


@dataclass
class Wet:
    pass


@dataclass
class OnFire:
    pass


@dataclass
class Leafy:
    pass


def make_object(id, desc, classes=None, **kwargs):
    if Object not in classes:
        classes.append(Object)
    cls = make_dataclass(desc, [], bases=tuple(classes))
    return cls(id=id, desc=desc, classes=classes, **kwargs)


@dataclass
class Player:
    room: int = 1
    won: bool = False
    objects: List[int] = field(default_factory=list)

    def set_room(self, room_id):
        self.room = room_id
        return True

    def remove_object(self, obj_id):
        self.objects.remove(obj_id)
        return True

    def add_object(self, obj_id):
        self.objects.append(obj_id)
        return True


class Collection:
    def __init__(self, elems, key):
        self.elems = elems
        self.key = key

        by_id = {}
        by_key = {}

        for i, elem in enumerate(self.elems):
            by_id[elem.id] = i
            by_key[getattr(elem, key)] = i

        self.by_id = by_id
        self.by_key = by_key

    def get_by_id(self, i):
        idx = self.by_id.get(i)
        if idx is not None:
            return self.elems[idx]
        return None

    def get(self, key):
        idx = self.by_key.get(key)
        if idx is not None:
            return self.elems[idx]
        return None

    def all(self):
        return self.elems

    def add_class(self, i, class_name):
        obj = self.get_by_id(i)
        if obj:
            classes = {
                "Mushroomy": Mushroomy,
                "Takeable": Takeable,
                "Wet": Wet,
                "OnFire": OnFire,
                "Leafy": Leafy,
            }
            new_class = classes[class_name]
            new_classes = obj.classes
            new_classes.insert(0, new_class)
            attrs = asdict(obj)
            del attrs["classes"]
            new_obj = make_object(classes=new_classes, **attrs)
            idx = self.by_id.get(obj.id)
            self.elems[idx] = new_obj
            return True
        return False

    def add_item(self, item):
        self.elems.append(item)
        self.by_id[item.id] = len(self.elems) - 1
        self.by_key[getattr(item, self.key)] = len(self.elems) - 1


GAME = Game()
PLAYER = Player()
OBJECTS = Collection(
    [
        make_object(id=1, desc="dog", favorite_item="ball", classes=[Animal]),
        make_object(
            id=2, desc="cat", favorite_item="yarn ball", classes=[Animal, Takeable]
        ),
        make_object(
            id=3, desc="duck", favorite_item="bathtub", classes=[Animal, Takeable]
        ),
        make_object(id=4, desc="key", classes=[Takeable]),
        make_object(id=5, desc="map", classes=[Takeable]),
        make_object(id=6, desc="fireplace", is_open=True, classes=[Container]),
        make_object(id=7, desc="wood", classes=[Takeable]),
        make_object(id=8, desc="matchbook", classes=[Takeable]),
        make_object(id=9, desc="carrot", classes=[Takeable, Food]),
        make_object(id=10, desc="apple", classes=[Takeable, Food]),
        make_object(id=11, desc="fire", classes=[]),
        make_object(id=12, desc="ball", classes=[Takeable]),
        make_object(id=13, desc="bag of mushroom spores", classes=[Takeable]),
        make_object(id=14, desc="watch", classes=[Takeable]),
        make_object(id=15, desc="letter", classes=[Takeable]),
        make_object(
            id=16,
            desc="envelope",
            is_open=False,
            objects=[15],
            classes=[Container],
        ),
        make_object(id=17, desc="blue wand", classes=[Wand]),
        make_object(id=18, desc="red wand", classes=[Takeable, Wand]),
        make_object(id=19, desc="green wand", classes=[Takeable, Wand]),
        make_object(
            id=20, desc="pond", is_open=True, objects=[17], classes=[Container]
        ),
        make_object(
            id=21, desc="trunk", is_open=False, objects=[4], classes=[Container]
        ),
        make_object(
            id=22, desc="dresser", is_open=False, objects=[19], classes=[Container]
        ),
        make_object(id=100, desc="yarn ball", classes=[Takeable]),
        make_object(id=101, desc="bathtub", classes=[]),
        make_object(id=102, desc="potato", classes=[Takeable, Food]),
        make_object(id=103, desc="cabbage", classes=[Takeable, Food]),
        make_object(id=104, desc="onion", classes=[Takeable, Food]),
        make_object(id=105, desc="soup", kind=[], classes=[Takeable, Food, Soup]),
        make_object(id=106, desc="pot", is_open=True, classes=[Container]),
        make_object(id=107, desc="carrot patch", produces="carrot", classes=[Source]),
        make_object(id=108, desc="apple tree", produces="apple", classes=[Source]),
        make_object(id=109, desc="cabbage patch", produces="cabbage", classes=[Source]),
        make_object(id=110, desc="potato patch", produces="potato", classes=[Source]),
        make_object(id=111, desc="onion patch", produces="onion", classes=[Source]),
        make_object(id=112, desc="wood pile", produces="wood", classes=[Source]),
        make_object(id=113, desc="cook book", classes=[Takeable]),
    ],
    "desc",
)


def obj_id(desc):
    return OBJECTS.get(desc).id


ROOMS = Collection(
    [
        Room(
            id=1,
            objects=[],
            passages=[1],
            desc="The Clearing",
        ),
        Room(
            id=2,
            objects=[obj_id("dog"), obj_id("apple tree"), obj_id("envelope")],
            passages=[1, 2, 3],
            desc="The Garden",
        ),
        Room(id=3, objects=[obj_id("cook book")], passages=[2, 4, 5], desc="The Foyer"),
        Room(
            id=4,
            objects=[obj_id("matchbook"), obj_id("pot")],
            passages=[4, 6, 7],
            desc="The Kitchen",
        ),
        Room(
            id=5,
            objects=[
                obj_id("cat"),
                obj_id("trunk"),
            ],
            passages=[5, 8],
            desc="The Living Room",
        ),
        Room(
            id=6,
            objects=[
                obj_id("map"),
                obj_id("fireplace"),
                obj_id("yarn ball"),
            ],
            passages=[8],
            desc="The Library",
        ),
        Room(
            id=7,
            objects=[obj_id("bathtub"), obj_id("dresser")],
            passages=[6],
            desc="The Attic",
        ),
        Room(
            id=8,
            objects=[
                obj_id("bag of mushroom spores"),
                obj_id("duck"),
                obj_id("carrot patch"),
                obj_id("cabbage patch"),
                obj_id("potato patch"),
                obj_id("onion patch"),
                obj_id("pond"),
            ],
            passages=[3, 7, 9, 10],
            desc="The Farm Plot",
        ),
        Room(
            id=9,
            objects=[obj_id("wood pile"), obj_id("ball")],
            passages=[9],
            desc="The Woodshed",
        ),
        Room(id=10, objects=[], passages=[10], desc="The North Forest"),
    ],
    "desc",
)
PASSAGES = Collection(
    [
        Passage(id=1, rooms={1: "north", 2: "south"}, desc="iron gate", locked=False),
        Passage(id=2, rooms={2: "north", 3: "south"}, desc="front door", locked=False),
        Passage(id=3, rooms={2: "west", 8: "south"}, desc="garden gate", locked=True),
        Passage(id=4, rooms={3: "west", 4: "south"}, desc="hallway", locked=False),
        Passage(id=5, rooms={3: "east", 5: "south"}, desc="corridor", locked=False),
        Passage(id=6, rooms={4: "north", 7: "south"}, desc="trap door", locked=True),
        Passage(id=7, rooms={4: "west", 8: "east"}, desc="back door", locked=False),
        Passage(
            id=8, rooms={5: "north", 6: "south"}, desc="large oak door", locked=True
        ),
        Passage(id=9, rooms={8: "west", 9: "east"}, desc="shed door", locked=False),
        Passage(id=10, rooms={8: "northeast", 10: "west"}, desc="trail", locked=False),
    ],
    "desc",
)


def main():
    # parser = argparse.ArgumentParser(description="An epic Polar adventure.")
    # parser.add_argument(
    #     "-l", "--load", type=str, nargs=1, help="the filename of a saved game"
    # )
    # args = parser.parse_args()
    # if args.load:
    #     GAME.load_saved(args.load)

    oso = Oso()
    oso.register_class(Game)
    oso.register_class(Room)
    oso.register_class(Passage)
    oso.register_class(Player)
    oso.register_class(Collection)
    oso.register_class(Object)
    oso.register_class(Animal)
    oso.register_class(Food)
    oso.register_class(Container)
    oso.register_class(Takeable)
    oso.register_class(Mushroomy)
    oso.register_class(Soup)
    oso.register_class(Source)
    oso.register_class(Wand)
    oso.register_class(Wet)
    oso.register_class(OnFire)
    oso.register_class(Leafy)
    oso.register_constant(GAME, "GAME")
    oso.register_constant(PLAYER, "PLAYER")
    oso.register_constant(ROOMS, "Rooms")
    oso.register_constant(PASSAGES, "Passages")
    oso.register_constant(OBJECTS, "Objects")
    oso.load_file("world.polar")
    oso.load_file("commands.polar")
    oso.load_file("tests.polar")
    oso.repl()


if __name__ == "__main__":
    main()