SmallSailing

#!/usr/bin/env python3
# Copyright (c) 2019-2024 Benjamin Holt -- MIT License

"""
Tiny, seemingly pointless, interactive fiction game to play with `smallmachine`
"""
import random
import time

import smallmachine as sm
#####


###  Helpers  ###
def adlib(x, joiner=" "):
    "Dynamically assemble messages from nested collections of parts.  Tuples are pieces to be strung together, lists are variants to choose among; anything else is used as a string"
    if type(x) is tuple:
        return joiner.join( adlib(i) for i in x )  # Joining with "|" can be helpful to see how messages get put together
    if type(x) is list:
        return adlib(random.choice(x))
    return str(x)

class Location:
    def __init__(self, name, description):
        self.name = name
        self.description = description

    # Demonstrate a simple enter-action pattern
    def enter(self, **_):
        """Enter action: describe the location"""
        return adlib(self.description)

    def __str__(self):
        return self.name

    @classmethod
    def enter_action(cls, state, destination, **ctx):
        l = destination if destination is not ... else state
        if isinstance(l, cls):
            return l.enter(state=state, destination=destination, **ctx)


def input_in(*args):
    return lambda input, **_: input in args
#####


###  Fixtures  ###
above = Location(
    "above", 
    ("You are on the deck of a small sailboat on a", ["calm", "serene", "blue", "clear", "glistening",], "sea; an open hatch leads down.")
)

below = Location(
    "below", 
    ("You are in the", ["cozy", "homey", "snug",], "cabin of a small boat, with just enough room for a bunk and a tiny desk with a logbook; a hatch leads up.")
)

messages = {
    "sail": ("You", ["set", "adjust", "tack",], "your sail", ["to", "toward", "for"], "{}."),
    "sleep": ("The bunk is", ["soft", "comfortable", "warm", "cozy",], "and you", ["rest", "sleep", "snooze", "nap", "doze",], ["well.", "deeply.", "blissfully.", "nicely"]),
    "log": [("Weather was", ["fair.", "good.", "lovely.",]),
        (
            ["Good", "Quick", "Slow",], "sailing",
            [
                "today", 
                ("this", ["morning.", "afternoon.", "evening.",])
            ]
        ),
    ],
}
#####


###  Actions  ###
def sail_action(**_):
    s = input("Where to? > ")
    return adlib(messages["sail"]).format(s)

log_entries = [(["Fair", "Nice", "Brisk",], "weather."),]  # Put one bogus entry in 'cos choose (in adlib) can't handle an empty list
def write_action(**_):
    s = input("What do you want to say? > ")
    log_entries.append(s)
    return "Written"

def explode_action(**_):
    raise ValueError("Ka-Boom!")
#####


###  Rules  ###
rules = {
    "start": [
        ("start", lambda input, **_: input != "crash", Location.enter_action, above),
    ],
    above: [
        ("go below", input_in("d", "down", "below"), Location.enter_action, below),
        ("sail", input_in("s", "sail"), sail_action, ...),
    ],
    below: [
        ("go above", input_in("u", "up", "above"), Location.enter_action, above),
        ("read log", input_in("r", "read", "read logbook"), lambda **_: adlib([messages["log"], log_entries]), ...),
        ("write log", input_in("w", "write", "log"), write_action, ...),
        ("sleep", input_in("s", "sleep", "bunk", "lie down", "lay down", "nap"), lambda **_: adlib(messages["sleep"]), ...),
    ],
    ...: [
        ("look", input_in("l", "look"), Location.enter_action, ...),
        ("kaboom", "kaboom", explode_action, ...),
        ("warp", "warp", lambda **_: "Warp out of this dimension", "elsewhere"),
        ("anything except crash", lambda input, **_: input != "crash", lambda **_: "Sorry, you can't do that.", ...),
    ]
}
#####


###  World  ###
world = sm.StateMachine(rules, "start")
# world = sm.StateMachine(rules, "start", tracer=True)
# world = sm.StateMachine(rules, "start", tracer=sm.PrefixTracer("***"))
#####


###  Main  ###
print("Smooth Sailing", flush=True)
time.sleep(0.1)  # HACK: wait for flush; sometimes prompt prints out-of-order with print output in spite of flush=True
out = world(input("Press enter to start. "))
if out:
    print(out, flush=True)
while True:
    time.sleep(0.1)  # HACK: wait for flush
    out = world(input("> "))
    if out:
        print(out, flush=True)
#####