tui-td 0.1.0

data/lib/tui_td/screenshot.rb

@@ -0,0 +1,271 @@
+# frozen_string_literal: true
+
+require "chunky_png"
+
+module TUITD
+  class Screenshot
+    CELL_W = 8
+    CELL_H = 16
+
+    FONT = [
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  # space (32)
+      0x00, 0x00, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x00, 0x18, 0x18, 0x00, 0x00, 0x00, 0x00,  # ! (33)
+      0x00, 0x66, 0x66, 0x66, 0x66, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  # " (34)
+      0x00, 0x00, 0x6c, 0x6c, 0xfe, 0x6c, 0x6c, 0x6c, 0x6c, 0xfe, 0x6c, 0x6c, 0x00, 0x00, 0x00, 0x00,  # # (35)
+      0x00, 0x10, 0x7e, 0xd0, 0xd0, 0xd0, 0x7c, 0x16, 0x16, 0x16, 0x16, 0xfc, 0x10, 0x00, 0x00, 0x00,  # $ (36)
+      0x00, 0x00, 0x06, 0x66, 0x6c, 0x0c, 0x18, 0x18, 0x30, 0x36, 0x66, 0x60, 0x00, 0x00, 0x00, 0x00,  # % (37)
+      0x00, 0x00, 0x38, 0x6c, 0x6c, 0x6c, 0x38, 0x70, 0xda, 0xcc, 0xcc, 0x7a, 0x00, 0x00, 0x00, 0x00,  # & (38)
+      0x00, 0x18, 0x18, 0x18, 0x18, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  # ' (39)
+      0x00, 0x0e, 0x18, 0x30, 0x30, 0x60, 0x60, 0x60, 0x60, 0x30, 0x30, 0x18, 0x0e, 0x00, 0x00, 0x00,  # ( (40)
+      0x00, 0x70, 0x18, 0x0c, 0x0c, 0x06, 0x06, 0x06, 0x06, 0x0c, 0x0c, 0x18, 0x70, 0x00, 0x00, 0x00,  # ) (41)
+      0x00, 0x00, 0x00, 0x00, 0x66, 0x3c, 0x18, 0xff, 0x18, 0x3c, 0x66, 0x00, 0x00, 0x00, 0x00, 0x00,  # * (42)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x18, 0x18, 0x7e, 0x18, 0x18, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  # + (43)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x18, 0x18, 0x30, 0x00, 0x00, 0x00,  # , (44)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  # - (45)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x18, 0x18, 0x00, 0x00, 0x00, 0x00,  # . (46)
+      0x00, 0x06, 0x06, 0x0c, 0x0c, 0x18, 0x18, 0x30, 0x30, 0x60, 0x60, 0xc0, 0xc0, 0x00, 0x00, 0x00,  # / (47)
+      0x00, 0x00, 0x7c, 0xc6, 0xc6, 0xce, 0xde, 0xf6, 0xe6, 0xc6, 0xc6, 0x7c, 0x00, 0x00, 0x00, 0x00,  # 0 (48)
+      0x00, 0x00, 0x18, 0x38, 0x78, 0x58, 0x18, 0x18, 0x18, 0x18, 0x18, 0x7e, 0x00, 0x00, 0x00, 0x00,  # 1 (49)
+      0x00, 0x00, 0x7c, 0xc6, 0x06, 0x06, 0x0c, 0x18, 0x30, 0x60, 0xc6, 0xfe, 0x00, 0x00, 0x00, 0x00,  # 2 (50)
+      0x00, 0x00, 0x7c, 0xc6, 0x06, 0x06, 0x3c, 0x06, 0x06, 0x06, 0xc6, 0x7c, 0x00, 0x00, 0x00, 0x00,  # 3 (51)
+      0x00, 0x00, 0xc0, 0xc0, 0xcc, 0xcc, 0xcc, 0xcc, 0xfe, 0x0c, 0x0c, 0x0c, 0x00, 0x00, 0x00, 0x00,  # 4 (52)
+      0x00, 0x00, 0xfe, 0xc6, 0xc0, 0xc0, 0xfc, 0x06, 0x06, 0x06, 0xc6, 0x7c, 0x00, 0x00, 0x00, 0x00,  # 5 (53)
+      0x00, 0x00, 0x7c, 0xc6, 0xc0, 0xc0, 0xfc, 0xc6, 0xc6, 0xc6, 0xc6, 0x7c, 0x00, 0x00, 0x00, 0x00,  # 6 (54)
+      0x00, 0x00, 0xfe, 0xc6, 0x06, 0x06, 0x0c, 0x18, 0x30, 0x30, 0x30, 0x30, 0x00, 0x00, 0x00, 0x00,  # 7 (55)
+      0x00, 0x00, 0x7c, 0xc6, 0xc6, 0xc6, 0x7c, 0xc6, 0xc6, 0xc6, 0xc6, 0x7c, 0x00, 0x00, 0x00, 0x00,  # 8 (56)
+      0x00, 0x00, 0x7c, 0xc6, 0xc6, 0xc6, 0xc6, 0x7e, 0x06, 0x06, 0xc6, 0x7c, 0x00, 0x00, 0x00, 0x00,  # 9 (57)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x18, 0x18, 0x00, 0x00, 0x00, 0x18, 0x18, 0x00, 0x00, 0x00, 0x00,  # : (58)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x18, 0x18, 0x00, 0x00, 0x00, 0x18, 0x18, 0x30, 0x00, 0x00, 0x00,  # ; (59)
+      0x00, 0x00, 0x06, 0x0c, 0x18, 0x30, 0x60, 0x60, 0x30, 0x18, 0x0c, 0x06, 0x00, 0x00, 0x00, 0x00,  # < (60)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x7e, 0x00, 0x00, 0x7e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  # = (61)
+      0x00, 0x00, 0x60, 0x30, 0x18, 0x0c, 0x06, 0x06, 0x0c, 0x18, 0x30, 0x60, 0x00, 0x00, 0x00, 0x00,  # > (62)
+      0x00, 0x00, 0x7c, 0xc6, 0x06, 0x0c, 0x18, 0x30, 0x30, 0x00, 0x30, 0x30, 0x00, 0x00, 0x00, 0x00,  # ? (63)
+      0x00, 0x00, 0x00, 0x7c, 0xc2, 0xda, 0xda, 0xda, 0xda, 0xde, 0xc0, 0x7c, 0x00, 0x00, 0x00, 0x00,  # @ (64)
+      0x00, 0x00, 0x7c, 0xc6, 0xc6, 0xc6, 0xfe, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # A (65)
+      0x00, 0x00, 0xfc, 0xc6, 0xc6, 0xc6, 0xfc, 0xc6, 0xc6, 0xc6, 0xc6, 0xfc, 0x00, 0x00, 0x00, 0x00,  # B (66)
+      0x00, 0x00, 0x7e, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0x7e, 0x00, 0x00, 0x00, 0x00,  # C (67)
+      0x00, 0x00, 0xfc, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xfc, 0x00, 0x00, 0x00, 0x00,  # D (68)
+      0x00, 0x00, 0x7e, 0xc0, 0xc0, 0xc0, 0xf8, 0xc0, 0xc0, 0xc0, 0xc0, 0x7e, 0x00, 0x00, 0x00, 0x00,  # E (69)
+      0x00, 0x00, 0x7e, 0xc0, 0xc0, 0xc0, 0xf8, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0x00, 0x00, 0x00, 0x00,  # F (70)
+      0x00, 0x00, 0x7e, 0xc0, 0xc0, 0xc0, 0xde, 0xc6, 0xc6, 0xc6, 0xc6, 0x7e, 0x00, 0x00, 0x00, 0x00,  # G (71)
+      0x00, 0x00, 0xc6, 0xc6, 0xc6, 0xc6, 0xfe, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # H (72)
+      0x00, 0x00, 0x7e, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x7e, 0x00, 0x00, 0x00, 0x00,  # I (73)
+      0x00, 0x00, 0x7e, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0xf0, 0x00, 0x00, 0x00, 0x00,  # J (74)
+      0x00, 0x00, 0xc6, 0xc6, 0xc6, 0xcc, 0xf8, 0xcc, 0xc6, 0xc6, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # K (75)
+      0x00, 0x00, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0x7e, 0x00, 0x00, 0x00, 0x00,  # L (76)
+      0x00, 0x00, 0xc6, 0xee, 0xfe, 0xd6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # M (77)
+      0x00, 0x00, 0xc6, 0xc6, 0xe6, 0xe6, 0xd6, 0xd6, 0xce, 0xce, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # N (78)
+      0x00, 0x00, 0x7c, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x7c, 0x00, 0x00, 0x00, 0x00,  # O (79)
+      0x00, 0x00, 0xfc, 0xc6, 0xc6, 0xc6, 0xfc, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0x00, 0x00, 0x00, 0x00,  # P (80)
+      0x00, 0x00, 0x7c, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xd6, 0xd6, 0x7c, 0x18, 0x0c, 0x00, 0x00,  # Q (81)
+      0x00, 0x00, 0xfc, 0xc6, 0xc6, 0xc6, 0xfc, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # R (82)
+      0x00, 0x00, 0x7e, 0xc0, 0xc0, 0xc0, 0x7c, 0x06, 0x06, 0x06, 0x06, 0xfc, 0x00, 0x00, 0x00, 0x00,  # S (83)
+      0x00, 0x00, 0xff, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x00, 0x00, 0x00, 0x00,  # T (84)
+      0x00, 0x00, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x7e, 0x00, 0x00, 0x00, 0x00,  # U (85)
+      0x00, 0x00, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x6c, 0x38, 0x10, 0x00, 0x00, 0x00, 0x00,  # V (86)
+      0x00, 0x00, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xd6, 0xfe, 0xee, 0xc6, 0x00, 0x00, 0x00, 0x00,  # W (87)
+      0x00, 0x00, 0xc6, 0xc6, 0xc6, 0x6c, 0x38, 0x6c, 0xc6, 0xc6, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # X (88)
+      0x00, 0x00, 0xc6, 0xc6, 0xc6, 0xc6, 0x7e, 0x06, 0x06, 0x06, 0x06, 0xfc, 0x00, 0x00, 0x00, 0x00,  # Y (89)
+      0x00, 0x00, 0xfe, 0x06, 0x06, 0x0c, 0x18, 0x30, 0x60, 0xc0, 0xc0, 0xfe, 0x00, 0x00, 0x00, 0x00,  # Z (90)
+      0x00, 0x3e, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x3e, 0x00, 0x00, 0x00,  # [ (91)
+      0x00, 0xc0, 0xc0, 0x60, 0x60, 0x30, 0x30, 0x18, 0x18, 0x0c, 0x0c, 0x06, 0x06, 0x00, 0x00, 0x00,  # \ (92)
+      0x00, 0x7c, 0x0c, 0x0c, 0x0c, 0x0c, 0x0c, 0x0c, 0x0c, 0x0c, 0x0c, 0x0c, 0x7c, 0x00, 0x00, 0x00,  # ] (93)
+      0x00, 0x10, 0x38, 0x6c, 0xc6, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  # ^ (94)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xfe, 0x00,  # _ (95)
+      0x00, 0x30, 0x18, 0x0c, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  # ` (96)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x7c, 0x06, 0x7e, 0xc6, 0xc6, 0xc6, 0x7e, 0x00, 0x00, 0x00, 0x00,  # a (97)
+      0x00, 0x00, 0xc0, 0xc0, 0xc0, 0xfc, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xfc, 0x00, 0x00, 0x00, 0x00,  # b (98)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x7e, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0x7e, 0x00, 0x00, 0x00, 0x00,  # c (99)
+      0x00, 0x00, 0x06, 0x06, 0x06, 0x7e, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x7e, 0x00, 0x00, 0x00, 0x00,  # d (100)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x7e, 0xc6, 0xc6, 0xfe, 0xc0, 0xc0, 0x7e, 0x00, 0x00, 0x00, 0x00,  # e (101)
+      0x00, 0x00, 0x1e, 0x30, 0x30, 0x30, 0x7c, 0x30, 0x30, 0x30, 0x30, 0x30, 0x00, 0x00, 0x00, 0x00,  # f (102)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x7e, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x7c, 0x06, 0x06, 0xfc, 0x00,  # g (103)
+      0x00, 0x00, 0xc0, 0xc0, 0xc0, 0xfc, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # h (104)
+      0x00, 0x00, 0x18, 0x18, 0x00, 0x38, 0x18, 0x18, 0x18, 0x18, 0x18, 0x1c, 0x00, 0x00, 0x00, 0x00,  # i (105)
+      0x00, 0x00, 0x18, 0x18, 0x00, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x70, 0x00,  # j (106)
+      0x00, 0x00, 0xc0, 0xc0, 0xc0, 0xcc, 0xd8, 0xf0, 0xf0, 0xd8, 0xcc, 0xc6, 0x00, 0x00, 0x00, 0x00,  # k (107)
+      0x00, 0x00, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x1c, 0x00, 0x00, 0x00, 0x00,  # l (108)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0xec, 0xd6, 0xd6, 0xd6, 0xd6, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # m (109)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0xfc, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # n (110)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x7c, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x7c, 0x00, 0x00, 0x00, 0x00,  # o (111)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0xfc, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xfc, 0xc0, 0xc0, 0xc0, 0x00,  # p (112)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x7e, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x7e, 0x06, 0x06, 0x06, 0x00,  # q (113)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x7e, 0xc6, 0xc0, 0xc0, 0xc0, 0xc0, 0xc0, 0x00, 0x00, 0x00, 0x00,  # r (114)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x7e, 0xc0, 0xc0, 0x7c, 0x06, 0x06, 0xfc, 0x00, 0x00, 0x00, 0x00,  # s (115)
+      0x00, 0x00, 0x30, 0x30, 0x30, 0x7c, 0x30, 0x30, 0x30, 0x30, 0x30, 0x1e, 0x00, 0x00, 0x00, 0x00,  # t (116)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x7e, 0x00, 0x00, 0x00, 0x00,  # u (117)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0xc6, 0xc6, 0xc6, 0xc6, 0x6c, 0x38, 0x10, 0x00, 0x00, 0x00, 0x00,  # v (118)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0xc6, 0xc6, 0xd6, 0xd6, 0xd6, 0xd6, 0x6e, 0x00, 0x00, 0x00, 0x00,  # w (119)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0xc6, 0x6c, 0x38, 0x38, 0x6c, 0xc6, 0xc6, 0x00, 0x00, 0x00, 0x00,  # x (120)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0xc6, 0x7e, 0x06, 0x06, 0xfc, 0x00,  # y (121)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0xfe, 0x06, 0x0c, 0x18, 0x30, 0x60, 0xfe, 0x00, 0x00, 0x00, 0x00,  # z (122)
+      0x00, 0x0e, 0x18, 0x18, 0x18, 0x18, 0x70, 0x70, 0x18, 0x18, 0x18, 0x18, 0x0e, 0x00, 0x00, 0x00,  # { (123)
+      0x00, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x18, 0x00, 0x00, 0x00,  # | (124)
+      0x00, 0x70, 0x18, 0x18, 0x18, 0x18, 0x0e, 0x0e, 0x18, 0x18, 0x18, 0x18, 0x70, 0x00, 0x00, 0x00,  # } (125)
+      0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x32, 0x7e, 0x4c, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  # ~ (126)
+    ].freeze
+    private_constant :FONT
+
+    ANSI_RGB = {
+      "black"         => [0x00, 0x00, 0x00],
+      "red"           => [0xAA, 0x00, 0x00],
+      "green"         => [0x00, 0xAA, 0x00],
+      "yellow"        => [0xAA, 0x55, 0x00],
+      "blue"          => [0x00, 0x00, 0xAA],
+      "magenta"       => [0xAA, 0x00, 0xAA],
+      "cyan"          => [0x00, 0xAA, 0xAA],
+      "white"         => [0xAA, 0xAA, 0xAA],
+      "bright_black"  => [0x55, 0x55, 0x55],
+      "bright_red"    => [0xFF, 0x55, 0x55],
+      "bright_green"  => [0x55, 0xFF, 0x55],
+      "bright_yellow" => [0xFF, 0xFF, 0x55],
+      "bright_blue"   => [0x55, 0x55, 0xFF],
+      "bright_magenta"=> [0xFF, 0x55, 0xFF],
+      "bright_cyan"   => [0x55, 0xFF, 0xFF],
+      "bright_white"  => [0xFF, 0xFF, 0xFF],
+    }.freeze
+    private_constant :ANSI_RGB
+
+    CUBE = [0x00, 0x5F, 0x87, 0xAF, 0xD7, 0xFF].freeze
+    private_constant :CUBE
+
+    ANSI_INDEX = %w[
+      black red green yellow blue magenta cyan white
+      bright_black bright_red bright_green bright_yellow
+      bright_blue bright_magenta bright_cyan bright_white
+    ].freeze
+    private_constant :ANSI_INDEX
+
+    DEFAULT_FG = [0xC0, 0xC0, 0xC0].freeze
+    DEFAULT_BG = [0x00, 0x00, 0x00].freeze
+
+    def initialize(state)
+      @state = state
+      @rows = _dig(state, :size, :rows) || 40
+      @cols = _dig(state, :size, :cols) || 120
+      @grid = state[:rows] || state["rows"] || []
+    end
+
+    def render(output_path)
+      width = @cols * CELL_W
+      height = @rows * CELL_H
+      image = ChunkyPNG::Image.new(width, height, ChunkyPNG::Color::BLACK)
+
+      @grid.each_with_index do |row, ri|
+        next unless row
+        row.each_with_index do |cell, ci|
+          next unless cell
+          render_cell(image, ri, ci, cell)
+        end
+      end
+
+      image.save(output_path)
+      output_path
+    end
+
+    private
+
+    def render_cell(image, ri, ci, cell)
+      char = cell[:char] || cell["char"] || " "
+      fg = cell[:fg] || cell["fg"] || "default"
+      bg = cell[:bg] || cell["bg"] || "default"
+      bold = cell[:bold] || cell["bold"] || false
+      italic = cell[:italic] || cell["italic"] || false
+      underline = cell[:underline] || cell["underline"] || false
+
+      fg_rgb = resolve_color(fg, DEFAULT_FG)
+      bg_rgb = resolve_color(bg, DEFAULT_BG)
+
+      px = ci * CELL_W
+      py = ri * CELL_H
+
+      fill_rect(image, px, py, CELL_W, CELL_H, bg_rgb)
+
+      return if char == " " || char.ord < 32 || char.ord > 126
+
+      rows_data = glyph_rows(char)
+      return unless rows_data
+
+      draw_glyph(image, px, py, rows_data, fg_rgb, bold: bold, italic: italic)
+
+      draw_underline(image, px, py, CELL_W, fg_rgb) if underline
+    end
+
+    def resolve_color(name, fallback)
+      case name
+      when "default"
+        fallback
+      when /^#([0-9a-fA-F]{6})$/
+        [$1[0..1].to_i(16), $1[2..3].to_i(16), $1[4..5].to_i(16)]
+      when /\Acolor(\d+)\z/
+        xterm_256($1.to_i)
+      when /\Abright_(.+)\z/
+        ANSI_RGB[name] || fallback
+      else
+        ANSI_RGB[name] || fallback
+      end
+    end
+
+    def xterm_256(index)
+      if index < 16
+        name = ANSI_INDEX[index]
+        ANSI_RGB[name] || DEFAULT_FG
+      elsif index < 232
+        r = CUBE[((index - 16) / 36) % 6]
+        g = CUBE[((index - 16) / 6) % 6]
+        b = CUBE[(index - 16) % 6]
+        [r, g, b]
+      else
+        v = 8 + (index - 232) * 10
+        [v, v, v]
+      end
+    end
+
+    def fill_rect(image, x, y, w, h, rgb)
+      color = ChunkyPNG::Color.rgb(*rgb)
+      h.times do |dy|
+        w.times do |dx|
+          image[x + dx, y + dy] = color
+        end
+      end
+    end
+
+    def glyph_rows(char)
+      idx = (char.ord - 32) * 16
+      return nil if idx < 0 || idx + 15 >= FONT.length
+
+      FONT[idx, 16]
+    end
+
+    def draw_glyph(image, px, py, rows, fg_rgb, bold:, italic:)
+      color = ChunkyPNG::Color.rgb(*fg_rgb)
+
+      rows.each_with_index do |byte, dy|
+        next if byte == 0
+
+        slant = italic ? dy / 8 : 0
+
+        8.times do |dx|
+          next unless (byte >> (7 - dx)) & 1 == 1
+
+          image[px + dx + slant, py + dy] = color
+          image[px + dx + slant + 1, py + dy] = color if bold
+        end
+      end
+    end
+
+    def draw_underline(image, px, py, w, fg_rgb)
+      color = ChunkyPNG::Color.rgb(*fg_rgb)
+      y = py + CELL_H - 2
+      w.times { |dx| image[px + dx, y] = color }
+    end
+
+    def _dig(hash, *keys)
+      keys.each do |k|
+        return nil unless hash
+        hash = hash[k] || hash[k.to_s]
+      end
+      hash
+    end
+  end
+end

data/lib/tui_td/state.rb

@@ -0,0 +1,111 @@
+# frozen_string_literal: true
+
+module TUITD
+  # Represents the parsed state of a terminal screen.
+  # Provides high-level query methods for AI consumption.
+  class State
+    attr_reader :rows, :cols, :grid, :cursor
+
+    def initialize(data)
+      @rows = data[:size][:rows]
+      @cols = data[:size][:cols]
+      @grid = data[:rows]
+      @cursor = data[:cursor]
+    end
+
+    # Get plain text of the entire terminal (no ANSI)
+    def plain_text
+      @grid.map { |row| row.map { |c| c[:char] }.join.rstrip }.join("\n")
+    end
+
+    # Get text at a specific position
+    def text_at(row, col, length = @cols - col)
+      return "" if row >= @rows || col >= @cols
+      @grid[row][col, length].map { |c| c[:char] }.join
+    end
+
+    # Search for text across the entire terminal
+    def find_text(pattern)
+      results = []
+      @grid.each_with_index do |row, ri|
+        text = row.map { |c| c[:char] }.join
+        pos = 0
+        while (match = text.index(pattern, pos))
+          results << { row: ri, col: match, text: pattern, full_line: text }
+          pos = match + 1
+        end
+      end
+      results
+    end
+
+    # Get the color at a specific cell
+    def foreground_at(row, col)
+      return nil if row >= @rows || col >= @cols
+      @grid[row][col][:fg]
+    end
+
+    def background_at(row, col)
+      return nil if row >= @rows || col >= @cols
+      @grid[row][col][:bg]
+    end
+
+    def style_at(row, col)
+      return nil if row >= @rows || col >= @cols
+      cell = @grid[row][col]
+      { bold: cell[:bold], italic: cell[:italic], underline: cell[:underline] }
+    end
+
+    def to_ai_json
+      h = extract_highlights
+      cursor_info = @cursor.is_a?(Hash) ? @cursor : {}
+      r = cursor_info[:row] || cursor_info["row"] || 0
+      c = cursor_info[:col] || cursor_info["col"] || 0
+      styled_count = h.count { |hl| hl[:bold] || hl[:italic] || hl[:underline] || hl[:fg] || hl[:bg] }
+
+      summary = +"Cursor at [#{r},#{c}]. "
+      summary << "#{styled_count} styled row#{styled_count == 1 ? '' : 's'}"
+      fgs = h.flat_map { |hl| hl[:fg] }.compact.uniq
+      bgs = h.flat_map { |hl| hl[:bg] }.compact.uniq
+      summary << ", colors: fg=#{fgs.sort.join(',')}" unless fgs.empty?
+      summary << ", bg=#{bgs.sort.join(',')}" unless bgs.empty?
+      summary << "."
+
+      {
+        size: { rows: @rows, cols: @cols },
+        cursor: cursor_info,
+        text: plain_text,
+        highlights: h,
+        summary: summary,
+      }
+    end
+
+    private
+
+    def extract_highlights
+      highlights = []
+      @grid.each_with_index do |row, ri|
+        row_text = row.map { |c| c[:char] }.join
+        next if row_text.strip.empty?
+
+        fgs = row.map { |c| c[:fg] || c["fg"] || "default" }
+                 .uniq.reject { |c| c == "default" }
+        bgs = row.map { |c| c[:bg] || c["bg"] || "default" }
+                 .uniq.reject { |c| c == "default" }
+        bold = row.any? { |c| c[:bold] || c["bold"] }
+        italic = row.any? { |c| c[:italic] || c["italic"] }
+        underline = row.any? { |c| c[:underline] || c["underline"] }
+
+        next if fgs.empty? && bgs.empty? && !bold && !italic && !underline
+
+        h = { row: ri, text: row_text }
+        h[:bold] = true if bold
+        h[:italic] = true if italic
+        h[:underline] = true if underline
+        h[:fg] = fgs.size == 1 ? fgs.first : fgs unless fgs.empty?
+        h[:bg] = bgs.size == 1 ? bgs.first : bgs unless bgs.empty?
+        highlights << h
+      end
+      highlights
+    end
+  end
+end

data/lib/tui_td/test_runner.rb

@@ -0,0 +1,178 @@
+# frozen_string_literal: true
+
+require "json"
+
+module TUITD
+  # Executes TUI tests defined in JSON format.
+  #
+  #   plan = File.read("test/hello.json")
+  #   results = TUITD::TestRunner.new(plan).run
+  #   puts results[:passed]  # => true
+  #
+  # JSON format:
+  #   {
+  #     "name": "My test",
+  #     "rows": 24, "cols": 80, "timeout": 10,
+  #     "steps": [
+  #       {"start": "my_tui"},
+  #       {"wait_for_text": "> "},
+  #       {"send": "hello\n"},
+  #       {"assert_text": "hello"},
+  #       {"assert_fg": [0, 0], "is": "cyan"},
+  #       {"close": true}
+  #     ]
+  #   }
+  #
+  class TestRunner
+    Result = Struct.new(:step, :passed, :message, keyword_init: true)
+
+    def initialize(source)
+      raw = source.is_a?(String) ? JSON.parse(source) : source
+      @plan = raw.transform_keys(&:to_sym)
+      @plan[:steps] = @plan[:steps].map { |s| s.transform_keys(&:to_sym) }
+    end
+
+    def run
+      results = []
+      all_passed = true
+      driver = nil
+      rows = @plan[:rows] || 40
+      cols = @plan[:cols] || 120
+      timeout = @plan[:timeout] || 30
+      chdir = @plan[:chdir]
+
+      @plan[:steps].each do |step|
+        action = step.keys.first.to_s
+        value = step.values.first
+        start_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+
+        begin
+          r = case action
+              when "start"
+                driver&.close
+                driver = Driver.new(value.to_s, rows: rows, cols: cols, timeout: timeout, chdir: chdir)
+                driver.start
+                Result.new(step: action, passed: true, message: "Started: #{value}")
+
+              when "send"
+                ensure_driver!(driver)
+                driver.send(value.to_s)
+                Result.new(step: action, passed: true, message: "Sent #{value.to_s.length} characters")
+
+              when "send_key"
+                ensure_driver!(driver)
+                driver.send_keys(value.to_s.to_sym)
+                Result.new(step: action, passed: true, message: "Sent key: #{value}")
+
+              when "wait_for_text"
+                ensure_driver!(driver)
+                driver.wait_for_text(value.to_s)
+                Result.new(step: action, passed: true, message: "Found: #{value}")
+
+              when "wait_for_stable"
+                ensure_driver!(driver)
+                driver.wait_for_stable
+                Result.new(step: action, passed: true, message: "Stable")
+
+              when "assert_text"
+                ensure_driver!(driver)
+                state = State.new(driver.state_data)
+                if state.find_text(value.to_s).any?
+                  Result.new(step: action, passed: true, message: "Text found: #{value}")
+                else
+                  Result.new(step: action, passed: false, message: "Text NOT found: #{value}")
+                end
+
+              when "assert_fg"
+                ensure_driver!(driver)
+                row, col = coords(step)
+                expected = step[:is] || step["is"]
+                state = State.new(driver.state_data)
+                actual = state.foreground_at(row, col)
+                if actual == expected
+                  Result.new(step: action, passed: true, message: "FG at [#{row},#{col}] is #{expected}")
+                else
+                  Result.new(step: action, passed: false, message: "FG at [#{row},#{col}] is #{actual}, expected #{expected}")
+                end
+
+              when "assert_bg"
+                ensure_driver!(driver)
+                row, col = coords(step)
+                expected = step[:is] || step["is"]
+                state = State.new(driver.state_data)
+                actual = state.background_at(row, col)
+                if actual == expected
+                  Result.new(step: action, passed: true, message: "BG at [#{row},#{col}] is #{expected}")
+                else
+                  Result.new(step: action, passed: false, message: "BG at [#{row},#{col}] is #{actual}, expected #{expected}")
+                end
+
+              when "assert_style"
+                ensure_driver!(driver)
+                row, col = coords(step)
+                state = State.new(driver.state_data)
+                actual = state.style_at(row, col)
+                expected = {}
+                expected[:bold] = step[:bold] unless step[:bold].nil?
+                expected[:italic] = step[:italic] unless step[:italic].nil?
+                expected[:underline] = step[:underline] unless step[:underline].nil?
+                match = expected.all? { |k, v| actual[k] == v }
+                if match
+                  Result.new(step: action, passed: true, message: "Style at [#{row},#{col}] matches #{expected}")
+                else
+                  Result.new(step: action, passed: false, message: "Style at [#{row},#{col}] is #{actual}, expected #{expected}")
+                end
+
+              when "screenshot"
+                ensure_driver!(driver)
+                path = value.is_a?(String) ? value : "/tmp/tui_td_#{Time.now.to_i}.png"
+                driver.screenshot(path)
+                Result.new(step: action, passed: true, message: "Saved: #{path}")
+
+              when "html"
+                ensure_driver!(driver)
+                path = value.is_a?(String) ? value : "/tmp/tui_td_#{Time.now.to_i}.html"
+                HtmlRenderer.new(driver.state_data).render(path)
+                Result.new(step: action, passed: true, message: "Saved: #{path}")
+
+              when "close"
+                driver&.close
+                driver = nil
+                Result.new(step: action, passed: true, message: "Closed")
+
+              else
+                Result.new(step: action, passed: false, message: "Unknown action: #{action}")
+              end
+
+        rescue StandardError => e
+          r = Result.new(step: action, passed: false, message: "#{e.class}: #{e.message}")
+        end
+
+        results << r
+        all_passed &&= r.passed
+      end
+
+      driver&.close
+
+      {
+        name: @plan[:name] || "(unnamed)",
+        passed: all_passed,
+        results: results.map(&:to_h)
+      }
+    end
+
+    private
+
+    def ensure_driver!(driver)
+      raise Error, "No session. Add a 'start' step first." if driver.nil?
+    end
+
+    def coords(step)
+      pos = step[:assert_fg] || step[:assert_bg] || step[:assert_style]
+      pos = value if pos.nil? && (value = step.values.first).is_a?(Array)
+      row = pos.is_a?(Array) ? pos[0] : (pos[:row] || pos["row"] || 0)
+      col = pos.is_a?(Array) ? pos[1] : (pos[:col] || pos["col"] || 0)
+      [row, col]
+    end
+  end
+end

data/lib/tui_td/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module TUITD
+  VERSION = "0.1.0"
+end

data/lib/tui_td.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module TUITD
+  class Error < StandardError; end
+end
+
+require_relative "tui_td/version"
+require_relative "tui_td/driver"
+require_relative "tui_td/ansi_parser"
+require_relative "tui_td/state"
+require_relative "tui_td/screenshot"
+require_relative "tui_td/html_renderer"
+require_relative "tui_td/test_runner"
+require_relative "tui_td/mcp/server"
+require_relative "tui_td/cli"
+
+module TUITD
+
+  # Convenience method: start a TUI driver, capture initial state
+  def self.drive(command, **opts)
+    driver = Driver.new(command, **opts)
+    driver.start
+    driver
+  end
+end