mirror of
https://github.com/Shu-Ying/main_plugin_farm.git
synced 2026-10-28 21:19:58 +08:00
90 lines
3.7 KiB
Python
90 lines
3.7 KiB
Python
"""白名单生成AstrBot可安装包，复用已校验的共享模块打包路径。"""
|
|
|
|
import argparse
|
|
import ast
|
|
import hashlib
|
|
import json
|
|
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
|
|
|
|
|
|
def build(root, output):
|
|
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',
|
|
)}
|
|
source = root / 'integrations/astrbot'
|
|
for name in ('main.py', 'avatars.py', '_conf_schema.json',
|
|
'requirements.txt'):
|
|
files[name] = (source / name).read_bytes()
|
|
files['README.md'] = (source / 'README.md').read_bytes()
|
|
files['runtime.py'] = (
|
|
root / 'integrations/zhenxun/runtime.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': '农场',
|
|
'desc': '种植、收获、仓库、交易与农场图片，复用Farm共享核心。',
|
|
'version': version, 'author': 'Shu-Ying',
|
|
'repo': 'https://github.com/Shu-Ying/astrbot_plugin_farm',
|
|
}, ensure_ascii=False, indent=2) + '\n').encode()
|
|
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)
|
|
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)
|
|
args = parser.parse_args()
|
|
print(build(Path(__file__).resolve().parents[1], args.output))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|