main_plugin_farm/scripts/build_astrbot.py

"""白名单生成AstrBot可安装包，复用已校验的共享模块打包路径。"""

import argparse
import ast
import hashlib
import json
import runpy
from pathlib import Path
import tempfile
from zipfile import ZIP_DEFLATED, ZipFile, ZipInfo

from build_zhenxun import build as build_shared, read_version, source_identity
from verify_astrbot import MANIFEST, ROOT_NAME, verify_files, verify_package
from verify_zhenxun import read_package as read_shared
from help_image import build_help_image


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 = '<!-- FARM_COMMANDS -->'
    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'):
            content = host_logging(content)
            files[name] = content
            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()