#!/usr/bin/env python3

import json
import pathlib
import shutil

BUILD_DIR = (pathlib.Path(__file__).parent / "build").resolve()
IMAGE_DIR = (
    pathlib.Path(__file__).parent.parent.parent / "doc" / "main" / "cad" / "output"
).resolve()
HUBS = ["move", "city", "technic", "prime", "essential", "inventor"]

package_json = {
    "name": "@pybricks/images",
    "version": "1.1.0",
    "description": "Distribution of Pybricks images.",
    "license": "MIT",
    "repository": {
        "type": "git",
        "url": "https://github.com/pybricks/pybricks-api",
        "directory": "npm/images",
    },
    "publishConfig": {"registry": "https://registry.npmjs.org", "access": "public"},
}

# ensure empty build directory so we don't end up with stale files
shutil.rmtree(BUILD_DIR, True)
BUILD_DIR.mkdir()

# copy the hub images
for h in HUBS:
    shutil.copyfile(IMAGE_DIR / f"hub-{h}.png", BUILD_DIR / f"hub-{h}.png")

# generate package.json file

# create "exports" item for hub images.
package_json["exports"] = {f"./hub-{h}.png": f"./hub-{h}.png" for h in HUBS}

with open(BUILD_DIR / "package.json", "w") as f:
    json.dump(package_json, f, indent=2)

# copy additional files

ROOT_DIR = (pathlib.Path(__file__).parent).resolve()

for file in "README.md", "CHANGELOG.md", "LICENSE":
    shutil.copy(ROOT_DIR / file, BUILD_DIR / file)