aw-spec-cli 0.1.20 → 0.1.21

package/assets/.appweaver/scripts/python/aw-plan-coverage.py

@@ -0,0 +1,423 @@
+#!/usr/bin/env python3
+"""
+aw-plan-coverage — which of a lane plan's elements the generated code realises.
+
+Backs the `plan-coverage` subcommand of aw-spec-actions, called identically by the bash
+and the powershell twin, so the two cannot drift.
+
+WHY IT EXISTS. `implement` hands a whole plan to one generation pass and gets back the list
+of files it wrote. A pass that stops part-way — five use-cases of seven — returns a
+non-empty list, and a non-empty list is all the workflow used to check. The missing
+elements have no failing test either, because nothing was ever written to fail. This
+reads the plan's element list deterministically and answers, per element, whether the
+lane's code names it and whether the generating model reported it done.
+
+TWO KINDS OF ELEMENT.
+  checked      an identifier the code must contain — an entity, value object, domain
+               event or use-case (backend); a component (frontend); an identifier in a
+               delta plan's ADDED table. Searched for in the lane's source path, test
+               files excluded. Entities and value objects must appear as a whole word.
+               Every other checked name may carry a suffix, because stacks add one by
+               convention: `MedicationOrderPlaced` is realised as
+               `MedicationOrderPlacedEvent`, `CancelOrder` as `CancelOrderCommand`.
+  report-only  an element whose plan name rarely survives into code verbatim — an
+               endpoint route (route groups split it), a screen, a state, a MODIFIED row,
+               and the ports and adapters, which the plan names by intent and a framework
+               often realises generically (`IRepository<MedicationOrder>` for
+               `IMedicationOrderRepository`). Only the model's own report can say it was
+               done.
+An element is a GAP when it is checked and not found, or when the model's report does not
+say it was done — including when the report does not mention it at all.
+
+THE REPORT. `--reported` takes the `elements` list a generation pass returned —
+[{"element": "<name as listed>", "status": "done|blocked", "reason": "…"}] — and may be
+given more than once; a later report overrides an earlier one for the same element, which
+is how a completion pass's answers replace the first pass's.
+
+`--list-only` returns the element names without scanning, for the generation prompt.
+
+Always exits 0 with a JSON object; the workflow decides what a gap means.
+
+Usage:
+  aw-plan-coverage.py --repo R --plan P --kind backend|frontend [--mode bootstrap|delta]
+                      [--source-path S] [--test-path T] [--reported JSON]... [--list-only]
+"""
+
+import argparse
+import json
+import os
+import re
+import sys
+
+IDENT = re.compile(r"^[A-Za-z_][A-Za-z0-9_]*(?:\.[A-Za-z_][A-Za-z0-9_]*)*$")
+TICK = re.compile(r"`([^`]+)`")
+
+DONE = {"done", "generated", "implemented", "ok", "exists", "existing", "complete",
+        "completed", "present", "extended", "modified", "created", "updated"}
+
+# Directories that never hold a lane's own source.
+SKIP_DIRS = {".git", ".hg", ".svn", ".appweaver", "node_modules", "bin", "obj", "dist",
+             "build", "out", "target", ".next", ".nuxt", ".svelte-kit", ".angular",
+             ".turbo", "coverage", "__pycache__", ".venv", "venv", ".gradle", ".idea",
+             ".vs", ".vscode", "specs", "guardrails", "knowledge", "foundations"}
+CODE_EXT = {".cs", ".fs", ".vb", ".java", ".kt", ".kts", ".scala", ".groovy", ".py",
+            ".go", ".rs", ".ts", ".tsx", ".js", ".jsx", ".mjs", ".cjs", ".vue",
+            ".svelte", ".astro", ".razor", ".cshtml", ".html", ".rb", ".php", ".swift",
+            ".dart", ".ex", ".exs", ".clj", ".sql", ".proto", ".graphql"}
+TEST_DIRS = {"test", "tests", "__tests__", "__test__", "spec", "specs", "e2e", "testing"}
+TEST_FILE = re.compile(r"(\.(test|spec|e2e)\.[^.]+$)|(^test_.*\.py$)|(_test\.(py|go)$)"
+                       r"|(Tests?\.(cs|fs|vb|java|kt|scala)$)|(Spec\.(java|kt|scala|groovy)$)")
+MAX_FILES = 50000
+MAX_BYTES = 2 * 1024 * 1024
+
+
+def emit(obj):
+    # ONE LINE: the engine parses the last line of a script's stdout.
+    sys.stdout.write(json.dumps(obj, separators=(",", ":")) + "\n")
+
+
+def norm_path(p):
+    p = (p or "").replace("\\", "/").strip()
+    while p.startswith("./"):
+        p = p[2:]
+    return "" if p in ("", ".") else p.rstrip("/") + "/"
+
+
+def placeholder(text):
+    """A template slot the plan author never filled — `[Command/Query]`, `I[External]`."""
+    return "[" in text or "]" in text
+
+
+def split_row(line):
+    cells = line.strip().strip("|").split("|")
+    return [c.strip() for c in cells]
+
+
+def is_separator(line):
+    s = line.strip()
+    return s.startswith("|") and set(s.replace("|", "").replace(":", "").strip()) <= {"-"} \
+        and "-" in s
+
+
+def read_plan(path):
+    """The plan's lines with fenced blocks and HTML comments blanked out."""
+    text = open(path, encoding="utf-8").read()
+    text = re.sub(r"<!--.*?-->", lambda m: "\n" * m.group(0).count("\n"), text, flags=re.S)
+    out, fence = [], False
+    for line in text.splitlines():
+        if line.lstrip().startswith("```"):
+            fence = not fence
+            out.append("")
+            continue
+        out.append("" if fence else line)
+    return out
+
+
+def walk_sections(lines):
+    """Yield (h2, h3, kind, payload) for every heading and table row.
+
+    kind is "h2" / "h3" / "h4" with the heading text, or "row" with ({header: cell}).
+    h2 / h3 carry the enclosing headings' text so a caller can key on section numbers
+    ("## 2." / "### 2.2") or names ("## ADDED").
+    """
+    h2 = h3 = ""
+    headers = None
+    for i, line in enumerate(lines):
+        s = line.strip()
+        if s.startswith("#"):
+            headers = None
+            level = len(s) - len(s.lstrip("#"))
+            title = s[level:].strip()
+            if level == 2:
+                h2, h3 = title, ""
+            elif level == 3:
+                h3 = title
+            yield h2, h3, "h%d" % level, title
+            continue
+        if not s.startswith("|"):
+            headers = None
+            continue
+        if is_separator(s):
+            continue
+        nxt = lines[i + 1].strip() if i + 1 < len(lines) else ""
+        if is_separator(nxt):
+            headers = [c.lower() for c in split_row(s)]
+            continue
+        if headers:
+            cells = split_row(s)
+            yield h2, h3, "row", dict(zip(headers, cells))
+
+
+def section_no(title):
+    m = re.match(r"^(\d+(?:\.\d+)?)[.\s]", title + " ")
+    return m.group(1) if m else ""
+
+
+def col(row, prefix):
+    for k, v in row.items():
+        if k.startswith(prefix):
+            return v
+    return ""
+
+
+def first_tick(text):
+    toks = [t for t in TICK.findall(text) if not placeholder(t)]
+    return toks[0].strip() if toks else ""
+
+
+WHOLE_WORD_KINDS = {"entity", "value object"}
+
+
+def extract(lines, kind, mode):
+    """[(name, element_kind, checked)] in plan order, de-duplicated by name."""
+    found, seen = [], set()
+
+    def add(name, ekind, checked):
+        name = re.sub(r"\s+", " ", (name or "").strip())
+        if not name or placeholder(name) or name in seen:
+            return
+        seen.add(name)
+        found.append((name, ekind, bool(checked and IDENT.match(name))))
+
+    for h2, h3, what, payload in walk_sections(lines):
+        n2, n3 = section_no(h2), section_no(h3)
+        h2u = h2.upper()
+        if mode == "delta":
+            if what != "row":
+                continue
+            added, modified = h2u.startswith("ADDED"), h2u.startswith("MODIFIED")
+            if not (added or modified):
+                continue
+            cell = col(payload, "element")
+            ekind = col(payload, "kind").strip().lower() or ("modified" if modified else "")
+            toks = [t.strip() for t in TICK.findall(cell) if not placeholder(t)]
+            if not toks:
+                plain = cell.strip()
+                if plain and not placeholder(plain):
+                    add(plain, ekind or "element", False)
+                continue
+            # A MODIFIED element already exists, so finding its name proves nothing; routes,
+            # screens and states rarely keep their plan name in code.
+            report_only = modified or ekind in ("endpoint", "binding", "screen", "state")
+            for t in toks:
+                add(t, ekind or "element", not report_only)
+            continue
+        if kind == "backend":
+            if what == "h4" and n3 in ("1.3", "2.2"):
+                add(first_tick(payload), "entity" if n3 == "1.3" else "use-case", True)
+            elif what == "h3" and n2 == "3" and not n3:
+                add(first_tick(payload), "endpoint", False)
+            elif what == "row" and n3 == "1.2":
+                add(first_tick(col(payload, "event")), "event", True)
+            elif what == "row" and n3 == "1.4":
+                add(first_tick(col(payload, "value object")), "value object", True)
+            elif what == "row" and n3 == "2.1":
+                add(first_tick(col(payload, "port")), "port", False)
+            elif what == "row" and n3 == "4.1":
+                add(first_tick(col(payload, "adapter")), "adapter", False)
+        else:
+            if what == "row" and n2 == "1":
+                add(first_tick(col(payload, "screen")), "screen", False)
+            elif what == "row" and n2 == "3":
+                add(first_tick(col(payload, "component")), "component", True)
+    return found
+
+
+HASH_COMMENT_EXT = {".py", ".rb", ".ex", ".exs"}
+BLOCK_COMMENT = re.compile(r"/\*.*?\*/|<!--.*?-->|@\*.*?\*@", re.S)
+LINE_COMMENT = re.compile(r"(?<![:\"'])//[^\n]*")
+HASH_COMMENT = re.compile(r"(?m)^\s*#[^\n]*")
+
+
+def strip_comments(text, ext):
+    """Code without its comments, so a doc comment that MENTIONS an element — "returned by
+    GetOrder" — is not taken for the element itself."""
+    text = BLOCK_COMMENT.sub(" ", text)
+    if ext in HASH_COMMENT_EXT:
+        return HASH_COMMENT.sub("", text)
+    return LINE_COMMENT.sub("", text)
+
+
+def is_test_file(rel):
+    parts = rel.split("/")
+    if any(p.lower() in TEST_DIRS for p in parts[:-1]):
+        return True
+    return bool(TEST_FILE.search(parts[-1]))
+
+
+def scan(repo, source_path, test_path, names):
+    """{name: [relative files naming it]} over the lane's source path, tests excluded.
+
+    `names` holds (name, whole) keys: whole is True when the name must appear as a whole
+    word, False when an identifier that STARTS with it counts.
+    """
+    hits = {k: [] for k in names}
+    if not names:
+        return hits
+    patterns = {(n, whole): re.compile(r"(?<![A-Za-z0-9_])" + re.escape(n) +
+                                       (r"(?![A-Za-z0-9_])" if whole else ""))
+                for n, whole in names}
+    base = os.path.join(repo, source_path.rstrip("/")) if source_path else repo
+    if not os.path.isdir(base):
+        return hits
+    tests = norm_path(test_path)
+    src = norm_path(source_path)
+    exclude_tests_dir = tests and tests != src
+    count = 0
+    for dirpath, dirnames, filenames in os.walk(base):
+        dirnames[:] = [d for d in dirnames if d not in SKIP_DIRS and not d.startswith(".")]
+        for fn in filenames:
+            if os.path.splitext(fn)[1].lower() not in CODE_EXT:
+                continue
+            full = os.path.join(dirpath, fn)
+            rel = os.path.relpath(full, repo).replace("\\", "/")
+            if exclude_tests_dir and (rel + "/").startswith(tests):
+                continue
+            if is_test_file(os.path.relpath(full, base).replace("\\", "/")):
+                continue
+            count += 1
+            if count > MAX_FILES:
+                return hits
+            try:
+                if os.path.getsize(full) > MAX_BYTES:
+                    continue
+                text = open(full, encoding="utf-8", errors="ignore").read()
+            except OSError:
+                continue
+            text = strip_comments(text, os.path.splitext(fn)[1].lower())
+            for key, rx in patterns.items():
+                if len(hits[key]) < 5 and key[0] in text and rx.search(text):
+                    hits[key].append(rel)
+    return hits
+
+
+def norm_name(s):
+    s = (s or "").replace("`", "")
+    s = re.sub(r"\{([A-Za-z_][A-Za-z0-9_]*)(?::[^}]*)?\}", r"{\1}", s)
+    return re.sub(r"\s+", " ", s).strip().lower()
+
+
+def load_reports(raw_list):
+    """{normalised name: (status, reason)} — later reports win. Unreadable input counts as
+    no report, which leaves every report-only element unconfirmed."""
+    out, readable = {}, False
+    for raw in raw_list or []:
+        raw = (raw or "").strip()
+        if raw in ("", "None", "null"):
+            continue
+        try:
+            data = json.loads(raw)
+        except ValueError:
+            continue
+        if isinstance(data, dict):
+            data = data.get("elements") or []
+        if not isinstance(data, list):
+            continue
+        readable = True
+        for item in data:
+            if isinstance(item, str):
+                out[norm_name(item)] = ("done", "")
+                continue
+            if not isinstance(item, dict):
+                continue
+            name = item.get("element") or item.get("name") or ""
+            if not name:
+                continue
+            status = str(item.get("status") or "").strip().lower()
+            out[norm_name(name)] = (status, str(item.get("reason") or "").strip())
+    return out, readable
+
+
+def main():
+    ap = argparse.ArgumentParser(description="plan element coverage of a lane's code")
+    ap.add_argument("--repo", default=".")
+    ap.add_argument("--plan", required=True)
+    ap.add_argument("--kind", choices=["backend", "frontend"], required=True)
+    ap.add_argument("--mode", default="", help="bootstrap | delta (default: read the plan)")
+    ap.add_argument("--source-path", default="")
+    ap.add_argument("--test-path", default="")
+    ap.add_argument("--reported", action="append", default=[])
+    ap.add_argument("--list-only", action="store_true")
+    args = ap.parse_args()
+
+    repo = os.path.abspath(args.repo)
+    plan = args.plan if os.path.isabs(args.plan) else os.path.join(repo, args.plan)
+    base = {"plan": args.plan, "kind": args.kind, "source_path": args.source_path,
+            "test_path": args.test_path}
+
+    if not os.path.isfile(plan):
+        base.update({"ok": False, "plan_exists": False, "mode": args.mode, "total": 0,
+                     "names": [], "names_csv": "", "names_text": "(the plan does not exist)",
+                     "elements": [], "gaps": [], "gap_names": [], "gap_names_csv": "",
+                     "gap_count": 1,
+                     "gaps_text": "the plan %s does not exist" % args.plan,
+                     "message": "plan-coverage: the plan %s does not exist" % args.plan})
+        emit(base)
+        return
+
+    lines = read_plan(plan)
+    mode = args.mode if args.mode in ("bootstrap", "delta") else (
+        "delta" if any(l.strip().upper().startswith("## ADDED") for l in lines) else "bootstrap")
+    elements = extract(lines, args.kind, mode)
+    names = [n for n, _, _ in elements]
+    base.update({"plan_exists": True, "mode": mode, "total": len(elements),
+                 "names": names, "names_csv": ", ".join(names),
+                 # For a prompt: never empty, so the sentence around it still reads.
+                 "names_text": ", ".join(names) if names else
+                 "(none could be read from the plan — report each element you realise by "
+                 "the name the plan gives it)"})
+
+    if args.list_only:
+        base.update({"ok": True, "message": "%d plan element(s) listed" % len(elements)})
+        emit(base)
+        return
+
+    def probe_key(name, ekind):
+        return (name.split(".")[-1], mode == "bootstrap" and ekind in WHOLE_WORD_KINDS)
+
+    checked = [n for n, _, c in elements if c]
+    probe = {probe_key(n, k) for n, k, c in elements if c}
+    hits = scan(repo, norm_path(args.source_path), args.test_path, probe)
+    reports, readable = load_reports(args.reported)
+
+    rows, gaps, gap_names = [], [], []
+    for name, ekind, is_checked in elements:
+        files = hits.get(probe_key(name, ekind), []) if is_checked else []
+        status, reason = reports.get(norm_name(name), ("", ""))
+        done = status in DONE
+        why = []
+        if is_checked and not files:
+            why.append("no code under %s names it" % (args.source_path or "the repo"))
+        if status and not done:
+            why.append("reported %s%s" % (status, (": " + reason) if reason else ""))
+        elif not status and not is_checked:
+            why.append("not reported done")
+        rows.append({"element": name, "kind": ekind, "checked": is_checked,
+                     "found": bool(files) if is_checked else None, "files": files,
+                     "reported": status or "", "gap": bool(why)})
+        if why:
+            gaps.append("%s (%s) — %s" % (name, ekind, "; ".join(why)))
+            gap_names.append(name)
+
+    total_checked = len(checked)
+    found = sum(1 for r in rows if r["checked"] and r["found"])
+    ok = not gaps
+    if not elements:
+        msg = ("no plan element could be read from %s — its layout does not follow the "
+               "plan template, so coverage was not checked" % args.plan)
+    elif ok:
+        msg = "all %d plan element(s) realised (%d found in code, %d reported done)" % (
+            len(elements), found, len(elements) - total_checked)
+    else:
+        msg = "%d of %d plan element(s) not realised: %s" % (
+            len(gaps), len(elements), "; ".join(gaps))
+    base.update({"ok": ok, "reported": readable, "checked_on_disk": total_checked,
+                 "found_on_disk": found, "elements": rows, "gaps": gaps,
+                 "gap_names": gap_names, "gap_names_csv": ", ".join(gap_names),
+                 "gap_count": len(gaps),
+                 "gaps_text": "; ".join(gaps) if gaps else "none", "message": msg})
+    emit(base)
+
+
+if __name__ == "__main__":
+    main()