✨ feat(tsl-api-reference): index class and unit APIs
Migrate reference pages to declaration-line storage and rebuild the 13-column index. Add qualified member lookup plus regression and end-to-end coverage.
This commit is contained in:
@@ -5,11 +5,66 @@ import re
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
ENTRY_RE = re.compile(r"^(#{2,3})(?!#)\s+`(.+?)`\s*$")
|
||||
TOP_LEVEL_RE = re.compile(r"^##(?!#)\s+`(.+?)`\s*$")
|
||||
PLAIN_H2_RE = re.compile(r"^##(?!#)\s+")
|
||||
BARE_ENTRY_RE = re.compile(r"^(#{3,4})(?!#)\s+`(.+?)`\s*$")
|
||||
HEADING_RE = re.compile(r"^(#{1,6})(?!#)\s+")
|
||||
H3_RE = re.compile(r"^###(?!#)\s+")
|
||||
DECLARATION_RE = re.compile(r"^声明:(.*?)\s*$")
|
||||
TOP_LEVEL_TYPES = {"function", "class", "unit"}
|
||||
FENCE_RE = re.compile(r"^\s*(```|~~~)")
|
||||
CLASS_MEMBER_LABELS = {
|
||||
"function",
|
||||
"class function",
|
||||
"property",
|
||||
"field",
|
||||
"static field",
|
||||
"const",
|
||||
"static const",
|
||||
}
|
||||
UNIT_DIRECT_LABELS = {"function", "var", "const", "class"}
|
||||
SKILL_ROOT = Path(__file__).resolve().parents[1]
|
||||
DEFAULT_TSV = SKILL_ROOT / "data" / "function_index.tsv"
|
||||
DEFAULT_CODEGEN_ROOT = SKILL_ROOT / "references" / "codegen"
|
||||
REQUIRED_COLUMNS = (
|
||||
"name",
|
||||
"scope",
|
||||
"module",
|
||||
"signature",
|
||||
"page",
|
||||
"anchor",
|
||||
"tags",
|
||||
"summary",
|
||||
)
|
||||
OPTIONAL_COLUMNS = (
|
||||
"kind",
|
||||
"binding",
|
||||
"visibility",
|
||||
"owner",
|
||||
"qualified_name",
|
||||
)
|
||||
|
||||
HELP_EPILOG = """\
|
||||
检索分两步,--kw 的摘要不能直接作为 API 事实:
|
||||
|
||||
1. 先按关键词取候选(只有摘要表格,不含条目正文)
|
||||
lookup.py --kw 数组 去重
|
||||
2. 再按候选里的 qualified_name 取回条目正文
|
||||
lookup.py --name arrDropDuplicate
|
||||
|
||||
已知确切名称时可以直接用 --name,跳过第 1 步。
|
||||
|
||||
退出码:
|
||||
0 取回成功;--name 无匹配也是 0(打印提示,不算错误)
|
||||
1 function_index.tsv 缺失、格式错误,或候选指向的条目正文找不到
|
||||
2 参数不合法(缺少动作、空值、--limit 小于 1、使用短选项或缩写)
|
||||
"""
|
||||
|
||||
|
||||
def non_empty(value):
|
||||
if not value.strip():
|
||||
raise argparse.ArgumentTypeError("must not be empty")
|
||||
return value
|
||||
|
||||
|
||||
def find_tsv(start):
|
||||
@@ -31,19 +86,39 @@ def load_rows(tsv_path):
|
||||
if not lines:
|
||||
return []
|
||||
header = lines[0].split("\t")
|
||||
missing = [column for column in REQUIRED_COLUMNS if column not in header]
|
||||
if missing:
|
||||
raise ValueError(f"missing required columns: {', '.join(missing)}")
|
||||
if len(header) != len(set(header)):
|
||||
raise ValueError("duplicate column names")
|
||||
rows = []
|
||||
for line in lines[1:]:
|
||||
for line_number, line in enumerate(lines[1:], start=2):
|
||||
if not line.strip():
|
||||
continue
|
||||
values = line.split("\t")
|
||||
values += [""] * (len(header) - len(values))
|
||||
rows.append(dict(zip(header, values)))
|
||||
if len(values) != len(header):
|
||||
raise ValueError(
|
||||
f"line {line_number} has {len(values)} columns; "
|
||||
f"expected {len(header)}"
|
||||
)
|
||||
row = dict(zip(header, values))
|
||||
for column in OPTIONAL_COLUMNS:
|
||||
row.setdefault(column, "")
|
||||
rows.append(row)
|
||||
return rows
|
||||
|
||||
|
||||
def search_exact(rows, name):
|
||||
key = name.lower()
|
||||
return [row for row in rows if row.get("name", "").lower() == key]
|
||||
key = name.casefold()
|
||||
return [
|
||||
row
|
||||
for row in rows
|
||||
if key
|
||||
in {
|
||||
row.get("name", "").casefold(),
|
||||
row.get("qualified_name", "").casefold(),
|
||||
}
|
||||
]
|
||||
|
||||
|
||||
def search_keyword(rows, terms):
|
||||
@@ -57,34 +132,156 @@ def search_keyword(rows, terms):
|
||||
row.get("module", ""),
|
||||
row.get("tags", ""),
|
||||
row.get("summary", ""),
|
||||
row.get("kind", ""),
|
||||
row.get("binding", ""),
|
||||
row.get("visibility", ""),
|
||||
row.get("owner", ""),
|
||||
row.get("qualified_name", ""),
|
||||
]
|
||||
).lower()
|
||||
).casefold()
|
||||
if all(term in haystack for term in lowered_terms):
|
||||
matches.append(row)
|
||||
return matches
|
||||
|
||||
|
||||
def slice_entry(codegen_root, page, signature):
|
||||
def slug(text):
|
||||
return re.sub(r"[^a-z0-9_]", "", text.casefold())
|
||||
|
||||
|
||||
def fence_flags(lines):
|
||||
flags = []
|
||||
in_fence = False
|
||||
marker = ""
|
||||
for line in lines:
|
||||
stripped = line.lstrip()
|
||||
if not in_fence and FENCE_RE.match(line):
|
||||
marker = stripped[:3]
|
||||
flags.append(True)
|
||||
in_fence = True
|
||||
continue
|
||||
flags.append(in_fence)
|
||||
if in_fence and stripped.startswith(marker):
|
||||
in_fence = False
|
||||
marker = ""
|
||||
return flags
|
||||
|
||||
|
||||
def top_level_header_end(lines, start, flags):
|
||||
for index in range(start + 1, len(lines)):
|
||||
if flags[index]:
|
||||
continue
|
||||
if HEADING_RE.match(lines[index]):
|
||||
return index
|
||||
return len(lines)
|
||||
|
||||
|
||||
def declaration_type(lines, start, end, flags):
|
||||
for index in range(start + 1, end):
|
||||
if flags[index]:
|
||||
continue
|
||||
stripped = lines[index].strip()
|
||||
if not stripped:
|
||||
continue
|
||||
match = DECLARATION_RE.fullmatch(stripped)
|
||||
return match.group(1).strip().casefold() if match else ""
|
||||
return ""
|
||||
|
||||
|
||||
def markdown_heading(
|
||||
lines, index, root_kind, unit_class_open, flags
|
||||
):
|
||||
"""Return level, visible title, signature, and anchor base for one API."""
|
||||
bare = BARE_ENTRY_RE.match(lines[index])
|
||||
if not bare:
|
||||
return None
|
||||
level = len(bare.group(1))
|
||||
signature = bare.group(2)
|
||||
end = top_level_header_end(lines, index, flags)
|
||||
declaration = declaration_type(lines, index, end, flags)
|
||||
if declaration == "static function":
|
||||
return None
|
||||
if root_kind == "class":
|
||||
if level != 3 or declaration not in CLASS_MEMBER_LABELS:
|
||||
return None
|
||||
elif root_kind == "unit":
|
||||
if level == 3 and declaration not in UNIT_DIRECT_LABELS:
|
||||
return None
|
||||
if level == 4 and (
|
||||
not unit_class_open or declaration not in CLASS_MEMBER_LABELS
|
||||
):
|
||||
return None
|
||||
if level not in {3, 4}:
|
||||
return None
|
||||
else:
|
||||
return None
|
||||
return level, signature, signature, slug(signature)
|
||||
|
||||
|
||||
def iter_markdown_api_headings(lines):
|
||||
flags = fence_flags(lines)
|
||||
root_kind = ""
|
||||
unit_class_open = False
|
||||
for index, line in enumerate(lines):
|
||||
if flags[index]:
|
||||
continue
|
||||
if PLAIN_H2_RE.match(line):
|
||||
root_kind = ""
|
||||
unit_class_open = False
|
||||
top_level = TOP_LEVEL_RE.match(line)
|
||||
if not top_level:
|
||||
continue
|
||||
signature = top_level.group(1)
|
||||
end = top_level_header_end(lines, index, flags)
|
||||
root_kind = declaration_type(lines, index, end, flags)
|
||||
if root_kind:
|
||||
yield index, (
|
||||
2,
|
||||
signature,
|
||||
signature,
|
||||
slug(signature.split("(", 1)[0]),
|
||||
)
|
||||
continue
|
||||
if not root_kind:
|
||||
continue
|
||||
if root_kind == "unit" and H3_RE.match(line):
|
||||
unit_class_open = False
|
||||
bare = BARE_ENTRY_RE.match(line)
|
||||
if root_kind == "unit" and bare and len(bare.group(1)) == 3:
|
||||
end = top_level_header_end(lines, index, flags)
|
||||
unit_class_open = (
|
||||
declaration_type(lines, index, end, flags) == "class"
|
||||
)
|
||||
heading = markdown_heading(
|
||||
lines, index, root_kind, unit_class_open, flags
|
||||
)
|
||||
if heading:
|
||||
yield index, heading
|
||||
|
||||
|
||||
def slice_entry(codegen_root, page, signature, anchor):
|
||||
md = codegen_root / page
|
||||
if not md.is_file():
|
||||
return ""
|
||||
lines = md.read_text(encoding="utf-8").splitlines()
|
||||
start = None
|
||||
for idx, line in enumerate(lines):
|
||||
match = ENTRY_RE.match(line)
|
||||
if match and match.group(2) == signature:
|
||||
seen_anchors = {}
|
||||
for idx, heading in iter_markdown_api_headings(lines):
|
||||
level, _, entry_signature, base_anchor = heading
|
||||
occurrence = seen_anchors.get(base_anchor, 0)
|
||||
seen_anchors[base_anchor] = occurrence + 1
|
||||
entry_anchor = base_anchor if occurrence == 0 else f"{base_anchor}-{occurrence}"
|
||||
if entry_anchor == anchor and entry_signature == signature:
|
||||
start = idx
|
||||
start_level = len(match.group(1))
|
||||
start_level = level
|
||||
break
|
||||
if start is None:
|
||||
return ""
|
||||
|
||||
end = len(lines)
|
||||
flags = fence_flags(lines)
|
||||
for idx in range(start + 1, len(lines)):
|
||||
entry_match = ENTRY_RE.match(lines[idx])
|
||||
if entry_match:
|
||||
end = idx
|
||||
break
|
||||
if flags[idx]:
|
||||
continue
|
||||
heading_match = HEADING_RE.match(lines[idx])
|
||||
if heading_match and len(heading_match.group(1)) <= start_level:
|
||||
end = idx
|
||||
@@ -95,10 +292,28 @@ def slice_entry(codegen_root, page, signature):
|
||||
return "\n".join(block)
|
||||
|
||||
|
||||
def declaration_term(row):
|
||||
kind = row.get("kind", "") or "function"
|
||||
binding = row.get("binding", "")
|
||||
mapped = {
|
||||
("method", "instance"): "function",
|
||||
("method", "class"): "class function",
|
||||
("field", "static"): "static field",
|
||||
("constant", "static"): "static const",
|
||||
}.get((kind, binding))
|
||||
if mapped:
|
||||
return mapped
|
||||
return {
|
||||
"constant": "const",
|
||||
"variable": "var",
|
||||
}.get(kind, kind)
|
||||
|
||||
|
||||
def format_row(row):
|
||||
return "\t".join(
|
||||
[
|
||||
row.get("name", ""),
|
||||
row.get("qualified_name", "") or row.get("name", ""),
|
||||
declaration_term(row),
|
||||
row.get("signature", ""),
|
||||
f"{row.get('page', '')}#{row.get('anchor', '')}",
|
||||
row.get("summary", ""),
|
||||
@@ -128,13 +343,56 @@ def default_tsv_path():
|
||||
def main(argv=None):
|
||||
if hasattr(sys.stdout, "reconfigure"):
|
||||
sys.stdout.reconfigure(encoding="utf-8")
|
||||
parser = argparse.ArgumentParser(description="Query TSL API reference facts.")
|
||||
group = parser.add_mutually_exclusive_group(required=True)
|
||||
group.add_argument("--name", help="exact API/function name")
|
||||
group.add_argument("--kw", nargs="+", help="keyword terms (AND)")
|
||||
parser.add_argument("--tsv", help="explicit path to function_index.tsv")
|
||||
parser.add_argument("--limit", type=int, default=50)
|
||||
parser = argparse.ArgumentParser(
|
||||
description="检索 TSL API 参考数据,取回 API 的签名、参数与返回值事实。",
|
||||
epilog=HELP_EPILOG,
|
||||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||||
# 只接受完整拼写的长选项:不提供 -h 短选项,也不接受 --na / --li 这类
|
||||
# 缩写。缩写打错时会静默命中另一个选项并以 rc=0 返回,看不出写错了。
|
||||
add_help=False,
|
||||
allow_abbrev=False,
|
||||
)
|
||||
parser.add_argument(
|
||||
"--help",
|
||||
action="help",
|
||||
help="显示本帮助并退出(不提供 -h 短选项)",
|
||||
)
|
||||
group = parser.add_argument_group("动作(必选其一)").add_mutually_exclusive_group(
|
||||
required=True
|
||||
)
|
||||
group.add_argument(
|
||||
"--name",
|
||||
type=non_empty,
|
||||
help="按 API 名称精确取回条目正文;同时匹配简单名称与 qualified_name,"
|
||||
"大小写不敏感。简单成员名会返回所有 owner 下的同名 API,"
|
||||
"用完整限定名(如 DemoUnit.Document.Save)缩小到指定 class/unit。"
|
||||
"这是唯一的事实来源",
|
||||
)
|
||||
group.add_argument(
|
||||
"--kw",
|
||||
nargs="+",
|
||||
type=non_empty,
|
||||
help="按关键词检索候选清单,多个词是 AND 关系(全部命中才返回);"
|
||||
"逐行搜索 name、signature、module、tags、summary、kind、binding、"
|
||||
"visibility、owner 和 qualified_name 这些列的拼接文本,"
|
||||
"大小写不敏感且按子串匹配。只返回摘要表格,不含条目正文",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--tsv",
|
||||
metavar="PATH",
|
||||
help="显式指定 function_index.tsv;默认使用本 skill 内置的 data/function_index.tsv",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--limit",
|
||||
type=int,
|
||||
default=50,
|
||||
metavar="N",
|
||||
help="--kw 输出的候选条数上限,取值 >= 1(默认 %(default)s);"
|
||||
"--name 取回正文时不受该上限影响",
|
||||
)
|
||||
args = parser.parse_args(argv)
|
||||
if args.limit < 1:
|
||||
parser.error("--limit must be >= 1")
|
||||
|
||||
tsv_path = Path(args.tsv) if args.tsv else default_tsv_path()
|
||||
if not tsv_path or not tsv_path.is_file():
|
||||
@@ -145,20 +403,40 @@ def main(argv=None):
|
||||
)
|
||||
return 1
|
||||
|
||||
rows = load_rows(tsv_path)
|
||||
try:
|
||||
rows = load_rows(tsv_path)
|
||||
except (OSError, UnicodeError, ValueError) as error:
|
||||
print(f"ERROR: failed to load {tsv_path}: {error}", file=sys.stderr)
|
||||
return 1
|
||||
if not rows:
|
||||
print(
|
||||
f"ERROR: {tsv_path} has no entries; reinstall tsl-api-reference "
|
||||
"or pass --tsv PATH",
|
||||
file=sys.stderr,
|
||||
)
|
||||
return 1
|
||||
codegen_root = codegen_root_for_tsv(tsv_path)
|
||||
|
||||
if args.name:
|
||||
if args.name is not None:
|
||||
matches = search_exact(rows, args.name)
|
||||
if not matches:
|
||||
print(f"No TSL API named '{args.name}'. Try: --kw <中文关键词>")
|
||||
return 0
|
||||
entries = []
|
||||
for row in matches:
|
||||
body = slice_entry(codegen_root, row["page"], row["signature"])
|
||||
if body:
|
||||
print(body)
|
||||
else:
|
||||
print(f"### `{row['signature']}`\n{row.get('summary', '')}")
|
||||
body = slice_entry(
|
||||
codegen_root, row["page"], row["signature"], row["anchor"]
|
||||
)
|
||||
if not body:
|
||||
print(
|
||||
f"ERROR: entry body not found: "
|
||||
f"{row['page']}#{row['anchor']} ({row['signature']})",
|
||||
file=sys.stderr,
|
||||
)
|
||||
return 1
|
||||
entries.append((row, body))
|
||||
for row, body in entries:
|
||||
print(body)
|
||||
print(
|
||||
f"\n<!-- {row['scope']}/{row['module']} · "
|
||||
f"{row['page']}#{row['anchor']} -->\n"
|
||||
@@ -167,7 +445,7 @@ def main(argv=None):
|
||||
|
||||
matches = search_keyword(rows, args.kw)
|
||||
print(f"# {len(matches)} matches for: {' '.join(args.kw)}")
|
||||
print("name\tsignature\tpage#anchor\tsummary")
|
||||
print("qualified_name\tdeclaration\tsignature\tpage#anchor\tsummary")
|
||||
for row in matches[: args.limit]:
|
||||
print(format_row(row))
|
||||
if len(matches) > args.limit:
|
||||
|
||||
Reference in New Issue
Block a user