"""白名单生成AstrBot可安装包，复用已校验的共享模块打包路径。""" import argparse import ast import hashlib import json import runpy import tempfile from pathlib import Path from zipfile import ZIP_DEFLATED, ZipFile, ZipInfo from build_zhenxun import build as build_shared from build_zhenxun import read_version, source_identity from help_image import build_help_image from format_generated import format_generated from verify_astrbot import MANIFEST, ROOT_NAME, verify_files, verify_package from verify_zhenxun import read_package as read_shared def host_logging(content): """仅替换日志入口，保留共享业务源码和注释；未知用法交给包校验拒绝。""" lines = content.decode("utf-8").splitlines(keepends=True) edits = [] for node in ast.parse(content).body: if isinstance(node, ast.Import) and any( alias.name == "logging" for alias in node.names ): if len(node.names) != 1 or node.names[0].asname: raise ValueError("Unsupported logging import") edits.append((node.lineno, node.end_lineno, "")) if ( isinstance(node, ast.Assign) and len(node.targets) == 1 and isinstance(node.targets[0], ast.Name) and isinstance(node.value, ast.Call) and isinstance(node.value.func, ast.Attribute) and isinstance(node.value.func.value, ast.Name) and node.value.func.value.id == "logging" and node.value.func.attr == "getLogger" ): alias = node.targets[0].id statement = "from astrbot.api import logger" if alias != "logger": statement += " as " + alias edits.append((node.lineno, node.end_lineno, statement + "\n")) for start, end, replacement in sorted(edits, reverse=True): lines[start - 1 : end] = [replacement] return "".join(lines).encode("utf-8") def render_readme(root): template = (root / "readme/astrbot.md").read_text(encoding="utf-8") marker = "" if marker not in template: if "./images/help-astrbot.png" not in template: raise ValueError("AstrBot README requires its help menu image") return template.encode("utf-8") if template.count(marker) != 1: raise ValueError("AstrBot README requires one command marker") definitions = runpy.run_path(str(root / "src/farm_core/commands.py")) lines = ["| 分类 | 指令 | 说明 |", "| --- | --- | --- |"] specs = definitions["command_specs"]("astrbot") for group, rows in definitions["help_groups"](specs): for usage, description in rows: values = (group, f"`/{usage}`", description) lines.append( "| " + " | ".join( value.replace("|", r"\|").replace("\n", " ") for value in values ) + " |" ) return template.replace(marker, "\n".join(lines)).encode("utf-8") def build(root, output, *, help_image=None): root, output = Path(root).resolve(), Path(output).resolve() version = read_version(root) if output.exists(): raise FileExistsError("Use a new output directory") if output.is_relative_to(root) and not output.is_relative_to(root / "dist"): raise ValueError("In-project output must be inside dist") with tempfile.TemporaryDirectory() as temporary: shared = read_shared(build_shared(root, Path(temporary) / "shared")) files = { name: content for name, content in shared.items() if name.startswith("_vendor/") or name in ( "_version.py", "version.txt", "LICENSE", ) if not name.endswith("/__main__.py") } source = root / "integrations/astrbot" for name in ( "main.py", "avatars.py", "_conf_schema.json", "requirements.txt", "CHANGELOG.md", ): files[name] = (source / name).read_bytes() # AstrBot识别固定文件名logo.png；README单独使用高清原图。 files["logo.png"] = (root / "logo-astrbot.png").read_bytes() files["logo-readme.png"] = (root / "logo.png").read_bytes() files["images/farm-preview.png"] = shared["images/farm-preview.png"] files["images/help-astrbot.png"] = build_help_image( root, "astrbot", help_image, ) files["README.md"] = render_readme(root) files["runtime.py"] = (root / "integrations/shared/runtime.py").read_bytes() files["runtime_logging.py"] = (source / "runtime_logging.py").read_bytes() files["verify_package.py"] = ( root / "scripts/verify_astrbot.py" ).read_bytes() # JSON是YAML的子集；标准库即可生成AstrBot可读的metadata.yaml。 files["metadata.yaml"] = ( json.dumps( { "name": ROOT_NAME, "display_name": "真寻农场", "short_desc": "开通农场，体验种植、收获、偷菜与土地开垦。", "desc": "农场种植小游戏，支持种子商店、分类仓库、作物交易、" "除虫、偷菜、土地开垦和动态图片帮助；使用本地存档。", "version": version, "author": "Shu-Ying", "repo": "https://github.com/Shu-Ying/astrbot_plugin_farm", "social_link": "https://github.com/Shu-Ying", "tags": ["农场", "种植", "小游戏"], "support_platforms": ["aiocqhttp"], }, ensure_ascii=False, indent=2, ) + "\n" ).encode() for name, content in files.items(): if name.endswith(".py"): files[name] = host_logging(content) format_generated(root, files) for name, content in files.items(): if name.endswith(".py"): ast.parse(content, filename=name, feature_version=(3, 10)) manifest = { "schema": 1, "target": "astrbot", "plugin": ROOT_NAME, "version": version, "source": source_identity(root), "files": { name: hashlib.sha256(content).hexdigest() for name, content in sorted(files.items()) }, } files[MANIFEST] = ( json.dumps( manifest, ensure_ascii=False, sort_keys=True, indent=2, ) + "\n" ).encode() verify_files(files) output.mkdir(parents=True) for name, content in sorted(files.items()): target = output / "repository" / name target.parent.mkdir(parents=True, exist_ok=True) target.write_bytes(content) archive_path = output / f"{ROOT_NAME}-{version}.zip" with ZipFile(archive_path, "w", compression=ZIP_DEFLATED) as archive: for name, content in sorted(files.items()): item = ZipInfo(f"{ROOT_NAME}/{name}", (2020, 1, 1, 0, 0, 0)) item.create_system = 3 item.compress_type = ZIP_DEFLATED item.external_attr = 0o100644 << 16 archive.writestr(item, content) verify_package(archive_path) if archive_path.stat().st_size > 16_000_000: raise ValueError("AstrBot plugin ZIP exceeds 16 MB") digest = hashlib.sha256(archive_path.read_bytes()).hexdigest() archive_path.with_suffix(".zip.sha256").write_text( f"{digest} {archive_path.name}\n", encoding="ascii", ) return archive_path def main(): parser = argparse.ArgumentParser(description=__doc__) parser.add_argument("--output", type=Path, required=True) parser.add_argument("--help-image", type=Path) args = parser.parse_args() print( build( Path(__file__).resolve().parents[1], args.output, help_image=args.help_image, ) ) if __name__ == "__main__": main()