"""Package the application image for the platform's existing firmware upload flow."""
import argparse
import hashlib
import json
from pathlib import Path
import re
import shutil
import struct

project = Path(__file__).resolve().parents[1]
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('--model', required=True, help='Exact inventory model, e.g. ECHOEAR-S3')
parser.add_argument('--app', type=Path, default=project / 'build/VolcRTCDemo.bin')
parser.add_argument('--out', type=Path, required=True)
args = parser.parse_args()
version = re.search(r'^#define DEVICE_FIRMWARE_VERSION "([^"]+)"', (project / 'main/firmware_version.h').read_text(), re.M).group(1)
image = args.app.read_bytes()
if not args.model or len(args.model.encode()) > 120:
    parser.error('Model must match the device inventory and fit 120 bytes')
if not 1024 <= len(image) <= 4 * 1024 * 1024:
    parser.error('The application must fit the 4 MiB OTA slot')
if image[0] != 0xE9 or struct.unpack_from('<H', image, 12)[0] != 9:
    parser.error('Expected an ESP32-S3 application image')
if version.encode() + b'\0' not in image:
    parser.error('Build is stale: compile the current firmware version first')
args.out.mkdir(parents=True, exist_ok=True)
output = args.out / f'echoear-{version}.bin'
shutil.copyfile(args.app, output)
manifest = {'model': args.model, 'version': version, 'size': len(image), 'sha256': hashlib.sha256(image).hexdigest(), 'file': output.name, 'max_ota_size': 4 * 1024 * 1024, 'upload_kind': 'application_only'}
(args.out / 'manifest.json').write_text(json.dumps(manifest, ensure_ascii=False, indent=2) + '\n', encoding='utf-8')
print(json.dumps(manifest, ensure_ascii=False))