openpitstop 1.9.0 → 1.10.0

package/scripts/gen-gifs.py

@@ -1,231 +0,0 @@
-#!/usr/bin/env python3
-"""
-gen-gifs.py — render OpenPitStop CLI output into valid, animated GIFs.
-
-For every clip it runs the real built CLI (dist/cli.js), captures the
-ANSI-colored output, parses it into a terminal buffer, and renders an
-animated "typing" GIF on a dark terminal background. Output lands in
-docs/media/<name>.gif, replacing any broken previous captures.
-
-Run:  python scripts/gen-gifs.py
-"""
-import os, re, sys, io, shutil, subprocess, tempfile
-from PIL import Image, ImageDraw, ImageFont
-
-ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-CLI  = os.path.join(ROOT, "dist", "cli.js")
-MEDIA = os.path.join(ROOT, "docs", "media")
-FONT = r"C:\Windows\Fonts\consola.ttf"
-
-BG = (11, 15, 26)            # #0B0F1A
-DEFAULT_FG = (230, 237, 243) # light grey
-COLORS = {
-    30:(220, 90, 90), 31:(248, 81, 73), 32:(63, 185, 80), 33:(210, 153, 34),
-    34:(88, 166, 255), 35:(255, 123, 209), 36:(34, 211, 238), 37:(200, 210, 220),
-    90:(130, 140, 150), 91:(255, 120, 120), 92:(80, 210, 110), 93:(230, 180, 60),
-    94:(120, 190, 255), 95:(255, 150, 220), 96:(120, 220, 245), 97:(235, 240, 245),
-}
-
-# ---------------------------------------------------------------- CLI runners
-def run_cli(args, cwd=None, extra_env=None):
-    env = dict(os.environ)
-    env["FORCE_COLOR"] = "1"
-    env["COLUMNS"] = "120"
-    if extra_env:
-        env.update(extra_env)
-    p = subprocess.run(["node", CLI] + list(args), cwd=cwd or ROOT,
-                       env=env, capture_output=True, timeout=300)
-    return p.stdout.decode("utf-8", "replace") + p.stderr.decode("utf-8", "replace")
-
-def run_shell(cmd, cwd):
-    env = dict(os.environ); env["FORCE_COLOR"] = "1"; env["COLUMNS"] = "120"
-    p = subprocess.run(cmd, cwd=cwd, env=env, shell=True,
-                       capture_output=True, timeout=300)
-    return p.stdout.decode("utf-8", "replace") + p.stderr.decode("utf-8", "replace")
-
-# ----------------------------------------------------------- ANSI -> buffer
-def strip_osc(s):
-    out = []; i = 0; n = len(s)
-    while i < n:
-        if s[i] == "\x1b" and i + 1 < n and s[i + 1] == "]":
-            j = s.find("\x07", i + 2); k = s.find("\x1b\\", i + 2)
-            if j == -1 and k == -1:
-                break
-            i = (j if (j != -1 and (k == -1 or j < k)) else k) + (1 if k != -1 and (j == -1 or k < j) else 1)
-            if j != -1 and (k == -1 or j < k):
-                i = j + 1
-            else:
-                i = k + 2
-            continue
-        if s[i] == "\x1b" and i + 1 < n and s[i + 1] == "[":
-            if re.match(r"\x1b\[[0-9;?]*[A-Za-z]", s[i:]):
-                m = re.match(r"\x1b\[[0-9;?]*[A-Za-z]", s[i:])
-                i += m.end(); continue
-            i += 2; continue
-        out.append(s[i]); i += 1
-    return "".join(out)
-
-def parse_ansi(text):
-    text = strip_osc(text).replace("\r\n", "\n").replace("\r", "\n")
-    lines = []
-    for line in text.split("\n"):
-        cells = []; fg = DEFAULT_FG
-        pos = 0
-        while pos < len(line):
-            if line[pos] == "\x1b" and pos + 1 < len(line) and line[pos + 1] == "[":
-                m = re.match(r"\x1b\[([0-9;]*)m", line[pos:])
-                if m:
-                    codes = [int(x) for x in m.group(1).split(";") if x != ""]
-                    j = 0
-                    while j < len(codes):
-                        c = codes[j]
-                        if c == 0: fg = DEFAULT_FG
-                        elif c == 1: pass
-                        elif c == 2: pass
-                        elif c == 4: pass
-                        elif c == 22: pass
-                        elif c == 39: fg = DEFAULT_FG
-                        elif 30 <= c <= 37: fg = COLORS.get(c, DEFAULT_FG)
-                        elif 90 <= c <= 97: fg = COLORS.get(c, DEFAULT_FG)
-                        elif c in (38, 48):
-                            if j + 1 < len(codes) and codes[j + 1] == 5: j += 2
-                            elif j + 1 < len(codes) and codes[j + 1] == 2: j += 4
-                            else: j += 1
-                        j += 1
-                    pos += m.end(); continue
-                pos += 1; continue
-            ch = line[pos]
-            if ch == "\t":
-                for _ in range(4): cells.append((" ", fg))
-            elif ch == "\x08":
-                if cells: cells.pop()
-            else:
-                cells.append((ch, fg))
-            pos += 1
-        lines.append(cells)
-    return lines
-
-def slice_from(text, marker):
-    lines = parse_ansi(text)
-    for i, line in enumerate(lines):
-        if marker.lower() in "".join(c for c, _ in line).lower():
-            return lines[i:]
-    return lines
-
-# --------------------------------------------------------------- cheat arc
-def make_cheat_arc():
-    tmp = tempfile.mkdtemp(prefix="pitstop-gate-")
-    src = os.path.join(ROOT, "demo-repo-integrity")
-    shutil.copytree(src, tmp, ignore=shutil.ignore_patterns("node_modules"),
-                    dirs_exist_ok=True)
-    run_shell('git init -q && git config user.email p@d.local && '
-              'git config user.name Demo && git add -A && git commit -q -m baseline', tmp)
-    parts = []
-    parts.append("=== OpenPitStop — the agent finally has a referee (cheat-catch demo) ===")
-    parts.append(run_cli(["scan", "."], cwd=tmp))
-    parts.append("\n--- ACT 2: a lazy 'fix' focuses the suite on passing tests -> SUSPICIOUS ---")
-    sf = os.path.join(tmp, "test", "money.test.js")
-    t = open(sf, encoding="utf-8").read()
-    t = t.replace('it("quotes base', 'fit("quotes base').replace('it("splits a bill', 'fit("splits a bill')
-    open(sf, "w", encoding="utf-8").write(t)
-    parts.append(run_cli(["gate", "."], cwd=tmp))
-    parts.append("\n--- ACT 3: the desperate 'fix' deletes the failing test -> CONFIRMED_CHEAT ---")
-    run_shell("git rm -q -f test/money.test.js", tmp)
-    parts.append(run_cli(["gate", "."], cwd=tmp))
-    shutil.rmtree(tmp, ignore_errors=True)
-    return "\n".join(parts)
-
-def make_drive_clip():
-    try:
-        out = run_cli(["drive", "security-0d098ed0", "demo-repo"], timeout=120)
-        if out.strip():
-            return out
-    except Exception:
-        pass
-    return run_cli(["drive", "--help"])
-
-# ------------------------------------------------------------- rendering
-font = ImageFont.truetype(FONT, 16)
-CW = max(int(round(font.getlength("M"))), int(round(font.getlength("─"))))
-CH = int(round(16 * 1.45))
-PAD = 12
-
-def render(window, W, ROWS, cursor=False):
-    img = Image.new("RGB", (PAD * 2 + W * CW, PAD * 2 + ROWS * CH), BG)
-    d = ImageDraw.Draw(img)
-    for y, line in enumerate(window):
-        run = ""; run_color = None; rx = PAD
-        def flush():
-            nonlocal run, run_color, rx
-            if run:
-                d.text((rx, PAD + y * CH), run, fill=run_color, font=font)
-                rx += len(run) * CW
-                run = ""
-        for ch, col in line:
-            if col == run_color:
-                run += ch
-            else:
-                flush(); run_color = col; run = ch
-        flush()
-        if cursor and y == len(window) - 1:
-            d.rectangle([PAD + len(line) * CW, PAD + y * CH,
-                         PAD + len(line) * CW + CW, PAD + y * CH + CH],
-                        fill=(34, 211, 238))
-    return img
-
-def make_frames(lines, V=42):
-    H = len(lines)
-    W = min(max((len(l) for l in lines), default=0), 120)
-    clipped = [l[:W] for l in lines]
-    ROWS = min(H, V)
-    step = max(1, (H + 23) // 24)
-    frames = []; revealed = 0
-    while revealed < H:
-        revealed = min(H, revealed + step)
-        window = clipped[:revealed] if revealed <= ROWS else clipped[revealed - ROWS:revealed]
-        frames.append(render(window, W, ROWS))
-    for b in range(6):
-        window = clipped[max(0, H - ROWS):H]
-        frames.append(render(window, W, ROWS, cursor=(b % 2 == 0)))
-    return frames
-
-def save_gif(frames, path, duration=80):
-    q = [im.quantize(colors=48, dither=Image.NONE) for im in frames]
-    q[0].save(path, save_all=True, append_images=q[1:],
-              duration=duration, loop=0, disposal=2)
-
-# --------------------------------------------------------------- clips
-CLIPS = [
-    ("pitstop-try",       lambda: run_cli(["try", "demo-repo"]), None),
-    ("pitstop-scan",      lambda: run_cli(["scan", "demo-repo"]), None),
-    ("pitstop-security",  lambda: run_cli(["scan", "demo-repo"]), "Security fixes"),
-    ("pitstop-gate",      make_cheat_arc, None),
-    ("pitstop-integrity", lambda: run_cli(["integrity", "demo-repo"]), None),
-    ("pitstop-pen",       lambda: run_cli(["pen", "demo-repo"]), None),
-    ("pitstop-pyramid",   lambda: run_cli(["test", "demo-repo"]), None),
-    ("pitstop-honesty",   lambda: run_cli(["honesty"]), None),
-    ("pitstop-next",      lambda: run_cli(["next", "demo-repo"]), None),
-    ("pitstop-drive",     make_drive_clip, None),
-    ("pitstop-semgrep",   lambda: run_cli(["scan", "demo-repo"],
-                                          extra_env={"PITSTOP_SEMGREP_CONFIG": "auto"}),
-     None),
-]
-
-def main():
-    os.makedirs(MEDIA, exist_ok=True)
-    for name, func, section in CLIPS:
-        sys.stdout.write(f"rendering {name}.gif ... ")
-        sys.stdout.flush()
-        raw = func()
-        lines = slice_from(raw, section) if section else parse_ansi(raw)
-        if not lines:
-            lines = parse_ansi("(no output)")
-        sys.stdout.write(f"({len(lines)} lines) "); sys.stdout.flush()
-        frames = make_frames(lines)
-        out = os.path.join(MEDIA, name + ".gif")
-        save_gif(frames, out)
-        im = Image.open(out)
-        print(f"ok  {os.path.getsize(out)//1024} KB  {im.size[0]}x{im.size[1]}  {getattr(im,'n_frames',1)} frames")
-
-if __name__ == "__main__":
-    main()