#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""CLAUDE.md の運用ルール1〜5のうち、機械的に判定できる項目を検査する。

使い方:
    python .claude/skills/doc-lint/doc-lint.py [ファイル...]
引数なしならリポジトリ直下の *.md、journal/*.md、backlog/**/*.md が対象。
明示したパスは実行ディレクトリから解決する。
ERROR が1件でもあれば終了コード 1。
"""
import re
import sys
from pathlib import Path
from urllib.parse import unquote, urlsplit

TAG      = re.compile(r"\[推測:(P-\d+)\]")
ENTRY    = re.compile(r"^###\s+(P-\d+)\b")
H2       = re.compile(r"^##\s+(.*?)\s*$")
MDLINK   = re.compile(r"\]\(([^)]+\.md[^)]*)\)")
JOURNALFN= re.compile(r"^\d{4}-\d{2}-\d{2}-.+\.md$")
PROMOTE  = re.compile(r"^\s*[-*]\s*格上げ条件\s*[:：]\s*\S")
COMPLETE = re.compile(r"^\s*[-*]\s*完了条件\s*[:：]\s*\S")
BACKLOG_ENTRY = re.compile(r"^###\s+([A-Za-z][A-Za-z0-9_-]*-P-\d+)\s+\S")
BACKLOG_LINK = re.compile(r"^- \[[^\]]+の保留課題\]\(([^)]+)\)$")
BACKLOG_FIELDS = ("関連仕様", "未決事項", "保留できる理由", "再開条件", "完了条件")
MANAGEMENT_SECTIONS = ("保留課題", "検討済み・非採用", "未確認事項")
ROOT = Path(__file__).resolve().parents[3]

# ルールブック自身は、アンチパターンを例として引用するため散文ヒューリスティックの対象外
PROSE_EXEMPT = {"CLAUDE.md"}

issues = []
def add(sev, path, line, msg):
    issues.append((sev, path, line, msg))

def load(path):
    """コードフェンス内を空行に潰した (行番号, 本文) の列を返す。"""
    raw = Path(path).read_text(encoding="utf-8").splitlines()
    out, fence = [], False
    for i, l in enumerate(raw, 1):
        if l.lstrip().startswith("```"):
            fence = not fence
            out.append((i, ""))
            continue
        out.append((i, "" if fence else l))
    return out

def section(lines, title):
    """## <title> 配下の (行番号, 本文) を返す。無ければ None。"""
    start = None
    for idx, (n, l) in enumerate(lines):
        m = H2.match(l)
        if m and m.group(1) == title:
            start = idx + 1
            break
    if start is None:
        return None
    body = []
    for n, l in lines[start:]:
        if H2.match(l):
            break
        body.append((n, l))
    return body

def local_target(path, target):
    """相対リンクをリンク元基準で解決する。外部URLは対象外。"""
    parsed = urlsplit(target)
    if parsed.scheme or parsed.netloc:
        return None, ""
    linked = (path.parent / unquote(parsed.path)).resolve() if parsed.path else path
    return linked, unquote(parsed.fragment)


def check_backlog(path, rel, lines, seen_ids):
    blocks = []
    current = None
    for n, line in lines:
        if H2.match(line):
            current = None
        elif re.match(r"^###\s", line):
            match = BACKLOG_ENTRY.match(line)
            current = None
            if not match:
                add("ERROR", rel, n, "保留課題の見出しは ### <区分>-P-<番号> <題名> とする")
                continue
            pid = match.group(1)
            if pid in seen_ids:
                add("ERROR", rel, n, "保留課題のID %s が重複している(初出: %s)" % (pid, seen_ids[pid]))
            seen_ids[pid] = "%s:%d" % (rel, n)
            current = (pid, n, [])
            blocks.append(current)
        elif current is not None:
            current[2].append(line)

    for pid, n, body in blocks:
        for field in BACKLOG_FIELDS:
            pattern = re.compile(r"^[-*]\s+" + re.escape(field) + r"\s*[:：]\s*\S")
            if not any(pattern.match(line) for line in body):
                add("ERROR", rel, n, "%s に「- %s:」の内容がない" % (pid, field))

    for n, line in lines:
        if TAG.search(line):
            add("ERROR", rel, n, "本文の前提となる推測タグはbacklogではなく仕様書に置く")
        for target in MDLINK.findall(line):
            linked, _ = local_target(path, target)
            if linked is not None and not linked.is_file():
                add("ERROR", rel, n, "backlogの参照先ファイルがない: %s" % target)


def check_backlog_section(path, rel, lines):
    positions = []
    for title in MANAGEMENT_SECTIONS:
        headings = [n for n, line in lines if H2.match(line) and H2.match(line).group(1) == title]
        if len(headings) > 1:
            add("ERROR", rel, headings[1], "## %s が重複している" % title)
        if headings:
            positions.append(headings[0])
    if positions != sorted(positions):
        add("ERROR", rel, min(positions), "末尾の管理セクションは保留課題 → 検討済み・非採用 → 未確認事項の順に置く")

    deferred = section(lines, "保留課題")
    if deferred is None:
        return
    contents = [(n, line) for n, line in deferred if line.strip()]
    if not contents:
        add("ERROR", rel, 1, "空の ## 保留課題 はセクションごと削除する")
    for n, line in contents:
        match = BACKLOG_LINK.fullmatch(line)
        if not match:
            add("ERROR", rel, n, "保留課題には - [<区分>の保留課題](backlogのファイル#見出し) だけを書く")
            continue
        target = match.group(1)
        linked, anchor = local_target(path, target)
        if (linked is None or not linked.is_relative_to(ROOT / "backlog")
                or linked.suffix != ".md" or not anchor):
            add("ERROR", rel, n, "保留課題はbacklog内のMarkdownファイルの見出しへリンクする: %s" % target)
            continue
        if not linked.is_file():
            add("ERROR", rel, n, "保留課題の参照先ファイルがない: %s" % target)
            continue
        headings = set()
        for _, linked_line in load(linked):
            heading = re.match(r"^#{1,6}\s+(.+?)\s*#*\s*$", linked_line)
            if heading:
                slug = re.sub(r"[^\w\s-]", "", heading.group(1)).lower().replace(" ", "-")
                headings.add(slug)
        if anchor not in headings:
            add("ERROR", rel, n, "保留課題の参照先見出しがない: %s" % target)


def check(path, seen_backlog_ids=None):
    path = Path(path).resolve()
    rel = path.relative_to(ROOT).as_posix() if path.is_relative_to(ROOT) else path.as_posix()
    base = path.name
    is_journal = path.is_relative_to(ROOT / "journal")
    is_backlog = path.is_relative_to(ROOT / "backlog")
    lines = load(path)

    if is_backlog:
        check_backlog(path, rel, lines, seen_backlog_ids if seen_backlog_ids is not None else {})
    elif not is_journal:
        check_backlog_section(path, rel, lines)

    unconf = section(lines, "未確認事項")
    rejected = section(lines, "検討済み・非採用")
    unconf_span = set(n for n, _ in unconf) if unconf else set()
    tagged = {pid for n, line in lines if n not in unconf_span for pid in TAG.findall(line)}

    # --- ルール4: 推測タグと未確認事項の対応 ---
    entries = set()
    if unconf is not None:
        blocks = []          # [(pid, 行番号, [本文行])] 重複IDも個別に検査する
        for n, l in unconf:
            m = ENTRY.match(l)
            if m:
                pid = m.group(1)
                if pid in entries:
                    add("ERROR", rel, n, "未確認事項のID %s が重複している" % pid)
                entries.add(pid)
                blocks.append((pid, n, []))
            elif blocks:
                blocks[-1][2].append(l)
        for pid, ln, body in blocks:
            has_promote = any(PROMOTE.match(x) for x in body)
            has_complete = any(COMPLETE.match(x) for x in body)
            if pid in tagged and not has_promote:
                add("ERROR", rel, ln,
                    "%s に「- 格上げ条件:」の行がない(何を確認すれば事実に格上げできるか)" % pid)
            elif not has_promote and not has_complete:
                add("ERROR", rel, ln,
                    "%s に「- 格上げ条件:」または「- 完了条件:」の内容がない" % pid)

    for n, l in lines:
        if n in unconf_span:
            continue
        for pid in TAG.findall(l):
            if unconf is None or pid not in entries:
                add("ERROR", rel, n,
                    "本文の [推測:%s] に対応する項目が ## 未確認事項 にない" % pid)
    # 逆向き(本文タグのない未確認事項)は「未解決の問い」として正当なのでエラーにしない

    # --- ルール3: journal への/からのリンク禁止 ---
    for n, l in lines:
        for target in MDLINK.findall(l):
            linked, _ = local_target(path, target)
            if linked is not None and linked.is_relative_to(ROOT / "journal"):
                add("ERROR", rel, n, "journal へのリンクが張られている: %s" % target)
            elif is_journal and linked is not None:
                add("ERROR", rel, n, "journal から他ファイルへのリンクが張られている: %s" % target)

    # --- ルール1: 検討済み・非採用 ---
    if rejected is not None:
        items = [(n, l) for n, l in rejected if l.strip().startswith(("-", "*"))]
        for n, l in items:
            if not re.search(r"\(journal/[^)]+\.md\)", l):
                add("ERROR", rel, n,
                    "検討済み・非採用の項目が journal ファイル名を参照していない")
        if len(items) > 5:
            add("WARN", rel, items[5][0],
                "検討済み・非採用が %d 件(目安5件)。古いものから落とす" % len(items))

    # --- journal 固有 ---
    if is_journal:
        if not JOURNALFN.match(base):
            add("ERROR", rel, 1, "journal のファイル名が YYYY-MM-DD-<topic>.md 形式でない")
        head = " ".join(l for n, l in lines[:20])
        if "結論" not in head:
            add("WARN", rel, 1, "冒頭20行に結論が見当たらない(結論を先頭に置く)")

    # --- ルール1: 調査・比較の混入(要目視確認) ---
    if not is_journal and not is_backlog and base not in PROSE_EXEMPT:
        # 検討済み・非採用 は却下案を列挙する場所なので対象外
        skip = set(n for n, _ in rejected) if rejected else set()
        for n, l in lines:
            if n in skip:
                continue
            if "メリット" in l and "デメリット" in l:
                add("WARN", rel, n, "比較表らしき記述(メリット/デメリット)。journal 行きでは?")
            elif re.search(r"却下|不採用|見送っ|も検討したが|案[A-Z]|[A-Z]案", l):
                add("WARN", rel, n, "却下案の検討過程らしき記述。制約文に書き換えるか journal へ")
            elif re.search(r"優先度", l) and l.lstrip().startswith("#"):
                add("WARN", rel, n, "優先度評価は journal 行きでは?")

def main(argv):
    issues.clear()
    targets = [Path(arg).resolve() for arg in argv[1:]]
    if not targets:
        targets = (sorted(ROOT.glob("*.md")) + sorted((ROOT / "journal").glob("*.md"))
                   + sorted((ROOT / "backlog").rglob("*.md")))
    targets = list(dict.fromkeys(targets))
    if not targets:
        print("対象ファイルなし")
        return 0
    seen_backlog_ids = {}
    for t in targets:
        if t.is_file():
            check(t, seen_backlog_ids)
        else:
            add("ERROR", t.as_posix(), 1, "対象ファイルが見つからない")

    errs  = [i for i in issues if i[0] == "ERROR"]
    warns = [i for i in issues if i[0] == "WARN"]
    for sev, path, line, msg in errs + warns:
        print("%-5s %s:%d  %s" % (sev, path, line, msg))
    print("\n%d files / ERROR %d / WARN %d" % (len(targets), len(errs), len(warns)))
    return 1 if errs else 0

if __name__ == "__main__":
    if hasattr(sys.stdout, "reconfigure"):
        sys.stdout.reconfigure(encoding="utf-8")
    sys.exit(main(sys.argv))