limbo-code 0.1.0__py3-none-any.whl

limbo/ui/banner.py

@@ -0,0 +1,83 @@
+"""Startup ASCII-art banner shown on the main screen.
+
+Generated from the project's cartoon image (red shirt / yellow pants on a
+blue background) by quantizing each pixel to the nearest of four palette
+colors and mapping them to characters:
+
+- ``@`` dark outline (line art preserved with a per-cell coverage threshold)
+- ``o`` red shirt
+- ``.`` yellow pants
+- `` `` (space) blue background
+
+``startup_art_text()`` renders the same characters as a Rich ``Text`` with
+the original image colors: each cell's foreground is its palette color on
+the blue background, so the banner looks like the source cartoon.
+"""
+
+from rich.text import Text
+
+# Palette sampled from the source image (RGB).
+BLUE = (48, 105, 169)  # background
+YELLOW = (253, 231, 92)  # pants
+RED = (237, 88, 83)  # shirt
+DARK = (65, 64, 94)  # outline
+
+_FG = {"@": DARK, "o": RED, ".": YELLOW, " ": BLUE}
+
+
+def _style(char: str) -> str:
+    r, g, b = _FG[char]
+    bg = BLUE
+    return f"rgb({r},{g},{b}) on rgb({bg[0]},{bg[1]},{bg[2]})"
+
+
+def startup_art_text() -> Text:
+    """STARTUP_ART as a Rich Text colored with the original image palette.
+
+    Lines are padded to a uniform width so the blue background forms the
+    same solid rectangle as the source image.
+    """
+    lines = STARTUP_ART.splitlines()
+    width = max(len(line) for line in lines)
+    text = Text()
+    for i, line in enumerate(lines):
+        if i:
+            text.append("\n")
+        for ch in line.ljust(width):
+            text.append(ch, style=_style(ch))
+    return text
+
+
+STARTUP_ART = """\
+      @@ooooo@@ooooooooooooooooooooooooooooo@@oooooooooo@@
+       @ooooo@@ooooooooooooooooooooooooooooo@@oooooooooo@@
+       @@oooo@@ooooooooooooooooooooooooooooo@@oooooooooo@@
+       @@oooo@oooooooooooooooooooooooooooooo@@oooooooooo@@
+       @@ooo@@oooooooooooooooooooooooooooooo@@oooooooooo@@
+       @@ooo@@oooooooooooooooooooooooooooooo@@oooooooooo@@
+       @@ooo@@oooooooooooooooooooooooooooooo@@oooooooooo@@
+       @@ooo@@oooooooooooooooooooooooooooooo@@ooooooooooo@
+       @@ooo@@oooooooooooooooooooooooooooooo@@ooooooooooo@
+       @@ooo@@@ooooooooooooooooooooooooooooo@@oooooooooo@@
+        @@ooo@@@@@@@@ooooooooooooooooooooooo@@@ooooooooo@@
+         @@@@@@....@@@@@@@@@@@@@@@@@@@@@@@@@@@@@ooooooo@@
+          @@@@@................................@@@@@@@@@@
+            @@@.......@@@@.@@@@................@@@@@@@@@@@
+            @@@.......@@@@..@...................@@@@@@...@
+             @@........@@@..@...................@@......@@@
+              @@.......@@@......................@@..@.@@@@@
+         @@@@ @@...@@..@@@@.....................@@@@@@@@@@@
+  @@@          @@..@@@....@@@@...................@@@@@@@@@
+ @@     @     @@@@...@...@@@@....................@@@@@@@@
+     @@@@@      @@.@@@....@@@....................@@
+             @@  @@.........@@@@.................@@
+            @@   @@...........@@.................@@
+      @@         @@...........@@.................@@
+      @@         @@...........@@.................@@
+                 @@...........@@.................@@
+                  @...........@@.................@@
+                  @...........@@.................@@
+                  @@..........@@.................@
+                  @@...........@................@@
+                  @@...........@................@@
+                  @@...........@@...............@@"""

limbo/ui/commands.py

@@ -0,0 +1,61 @@
+"""Slash-command registry: one source for menu metadata and dispatch.
+
+A slash command is defined once — name, description, whether it takes
+arguments, and its handler. The autocomplete menu renders from the registry
+and the input dispatcher consults it, so the two can never drift.
+"""
+
+from __future__ import annotations
+
+from collections.abc import Callable
+from dataclasses import dataclass, field
+
+from limbo.skills import Skill
+
+
+@dataclass(frozen=True)
+class SlashCommand:
+    name: str
+    description: str
+    takes_args: bool = False
+    kind: str = "builtin"  # "builtin" | "skill"
+    handler: Callable[[str], None] | None = field(
+        default=None, compare=False, repr=False
+    )
+
+
+class SlashCommandRegistry:
+    """Holds the built-in commands; merges skills for display and lookup."""
+
+    def __init__(self) -> None:
+        self._commands: dict[str, SlashCommand] = {}
+
+    def register(self, command: SlashCommand) -> None:
+        self._commands[command.name] = command
+
+    def get(self, name: str) -> SlashCommand | None:
+        return self._commands.get(name)
+
+    def all(self) -> list[SlashCommand]:
+        return list(self._commands.values())
+
+    def candidates(self, skills: list[Skill]) -> list[SlashCommand]:
+        """Built-in commands plus skills (built-ins win on name collisions)."""
+        candidates = self.all()
+        for skill in skills:
+            if f"/{skill.name}" in self._commands:
+                continue
+            candidates.append(
+                SlashCommand(
+                    f"/{skill.name}",
+                    f"[skill] {skill.description}".rstrip(),
+                    takes_args=True,
+                    kind="skill",
+                )
+            )
+        return candidates
+
+    def help_text(self) -> str:
+        return "可用命令：" + " · ".join(
+            f"{command.name} {command.description}" for command in self.all()
+        )

limbo/ui/screens/__init__.py

@@ -0,0 +1 @@
+"""Limbo screens."""

limbo/ui/screens/game2048.py

@@ -0,0 +1,213 @@
+"""2048 mini-game: pure game logic + a modal screen.
+
+The screen is a modal pushed on top of the main chat screen. Textual keeps
+workers on the underlying screen running while a modal is up, so the agent
+turn continues in the background — the game never interrupts it.
+"""
+
+from __future__ import annotations
+
+import random
+from typing import Literal
+
+from textual.app import ComposeResult
+from textual.binding import Binding
+from textual.containers import Grid, Vertical
+from textual.screen import ModalScreen
+from textual.widgets import Static
+
+Direction = Literal["up", "down", "left", "right"]
+
+TARGET = 2048
+
+
+class Game2048:
+    """Pure 2048 game logic, independent of the UI."""
+
+    SIZE = 4
+
+    def __init__(self, rng: random.Random | None = None) -> None:
+        self._rng = rng or random.Random()
+        self.grid: list[list[int]] = [
+            [0] * self.SIZE for _ in range(self.SIZE)
+        ]
+        self.score = 0
+        self.won = False
+        self.spawn()
+        self.spawn()
+
+    # -- state ----------------------------------------------------------------
+
+    @property
+    def over(self) -> bool:
+        """Game over when no empty cell and no adjacent equal tiles remain."""
+        size = self.SIZE
+        for r in range(size):
+            for c in range(size):
+                if self.grid[r][c] == 0:
+                    return False
+                if c + 1 < size and self.grid[r][c] == self.grid[r][c + 1]:
+                    return False
+                if r + 1 < size and self.grid[r][c] == self.grid[r + 1][c]:
+                    return False
+        return True
+
+    def spawn(self) -> bool:
+        """Add a 2 (90%) or 4 (10%) to a random empty cell."""
+        empties = [
+            (r, c)
+            for r in range(self.SIZE)
+            for c in range(self.SIZE)
+            if self.grid[r][c] == 0
+        ]
+        if not empties:
+            return False
+        r, c = self._rng.choice(empties)
+        self.grid[r][c] = 4 if self._rng.random() < 0.1 else 2
+        return True
+
+    # -- moves ------------------------------------------------------------------
+
+    def move(self, direction: Direction) -> bool:
+        """Slide and merge the grid. Returns True if anything changed."""
+        lines = self._lines(direction)
+        gained = 0
+        new_lines = []
+        for line in lines:
+            merged, line_gained = self._merge_left(line)
+            new_lines.append(merged)
+            gained += line_gained
+        if new_lines == lines:
+            return False
+        self.score += gained
+        self._write_lines(direction, new_lines)
+        if not self.won and any(
+            value >= TARGET for row in self.grid for value in row
+        ):
+            self.won = True
+        self.spawn()
+        return True
+
+    def _lines(self, direction: Direction) -> list[list[int]]:
+        """Extract rows/columns so every move reduces to 'merge left'."""
+        size = self.SIZE
+        if direction == "left":
+            return [row[:] for row in self.grid]
+        if direction == "right":
+            return [row[::-1] for row in self.grid]
+        if direction == "up":
+            return [[self.grid[r][c] for r in range(size)] for c in range(size)]
+        return [
+            [self.grid[r][c] for r in range(size - 1, -1, -1)]
+            for c in range(size)
+        ]
+
+    def _write_lines(self, direction: Direction, lines: list[list[int]]) -> None:
+        """Inverse of `_lines`."""
+        size = self.SIZE
+        if direction == "left":
+            self.grid = [line[:] for line in lines]
+        elif direction == "right":
+            self.grid = [line[::-1] for line in lines]
+        elif direction == "up":
+            for c in range(size):
+                for r in range(size):
+                    self.grid[r][c] = lines[c][r]
+        else:
+            for c in range(size):
+                for r in range(size):
+                    self.grid[size - 1 - r][c] = lines[c][r]
+
+    @staticmethod
+    def _merge_left(line: list[int]) -> tuple[list[int], int]:
+        """Compress and merge one line towards index 0."""
+        tiles = [value for value in line if value]
+        merged: list[int] = []
+        gained = 0
+        i = 0
+        while i < len(tiles):
+            if i + 1 < len(tiles) and tiles[i] == tiles[i + 1]:
+                merged.append(tiles[i] * 2)
+                gained += tiles[i] * 2
+                i += 2
+            else:
+                merged.append(tiles[i])
+                i += 1
+        return merged + [0] * (len(line) - len(merged)), gained
+
+
+class Game2048Screen(ModalScreen[None]):
+    """Playable 2048 modal. The agent keeps running underneath."""
+
+    BINDINGS = [
+        Binding("up", "move('up')", "↑"),
+        Binding("down", "move('down')", "↓"),
+        Binding("left", "move('left')", "←"),
+        Binding("right", "move('right')", "→"),
+        Binding("w", "move('up')", show=False),
+        Binding("s", "move('down')", show=False),
+        Binding("a", "move('left')", show=False),
+        Binding("d", "move('right')", show=False),
+        Binding("r", "restart", "重开"),
+        Binding("q", "close", "关闭"),
+        Binding("escape", "close", "关闭"),
+    ]
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+        self.game = Game2048()
+
+    def compose(self) -> ComposeResult:
+        with Vertical(id="game-container"):
+            yield Static("", id="game-header", markup=False)
+            with Grid(id="game-grid"):
+                for r in range(self.game.SIZE):
+                    for c in range(self.game.SIZE):
+                        yield Static("", classes="tile", id=f"cell-{r}-{c}")
+            yield Static(
+                "方向键 / WASD 移动 · R 重开 · Q/Esc 关闭",
+                id="game-footer",
+                markup=False,
+            )
+
+    def on_mount(self) -> None:
+        self._refresh()
+
+    # -- actions --------------------------------------------------------------
+
+    def action_move(self, direction: Direction) -> None:
+        if self.game.over:
+            return
+        if self.game.move(direction):
+            self._refresh()
+
+    def action_restart(self) -> None:
+        self.game = Game2048()
+        self._refresh()
+
+    def action_close(self) -> None:
+        self.dismiss()
+
+    # -- rendering --------------------------------------------------------------
+
+    def _refresh(self) -> None:
+        header = self.query_one("#game-header", Static)
+        status = ""
+        if self.game.over:
+            status = " · 游戏结束，按 R 重开"
+        elif self.game.won:
+            status = " · 🎉 达成 2048！可继续挑战"
+        header.update(f"2048 · 分数 {self.game.score}{status}")
+
+        for r in range(self.game.SIZE):
+            for c in range(self.game.SIZE):
+                value = self.game.grid[r][c]
+                cell = self.query_one(f"#cell-{r}-{c}", Static)
+                cell.update(str(value) if value else "")
+                if value:
+                    tile_class = (
+                        f"tile-{value}" if value <= TARGET else "tile-super"
+                    )
+                    cell.set_classes(f"tile {tile_class}")
+                else:
+                    cell.set_classes("tile")