#!/usr/bin/env python3
"""Generate iOS App Store screenshots for Roomfit Studio.

iPhone 6.5" Display: 1242 x 2688 portrait
iPad 13": 2064 x 2752 portrait
"""
from PIL import Image, ImageDraw, ImageFont
from pathlib import Path
import sys

sys.path.insert(0, str(Path(__file__).parent))
import importlib.util
spec = importlib.util.spec_from_file_location("mkmod", Path(__file__).parent / "make-screenshots.py")
mkmod = importlib.util.module_from_spec(spec)
spec.loader.exec_module(mkmod)

OUT = Path(__file__).parent

SIZES = {
    "ios_iphone65": (1242, 2688),
    "ios_ipad13": (2064, 2752),
}


def main():
    for device, size in SIZES.items():
        for i, data in enumerate(mkmod.SCREENS, 1):
            img = mkmod.draw_screen(size, data)
            out = OUT / f"screenshot_{device}_{i}.png"
            # iOS requires no alpha channel
            if img.mode == "RGBA":
                img = img.convert("RGB")
            img.save(out, "PNG", optimize=True)
            print(f"Saved: {out} ({size[0]}x{size[1]})")


if __name__ == "__main__":
    main()