#!/usr/bin/env python3
import os
import subprocess
import glob

KRA_DIR = "_tmp/drop_shadow_kra_set"
PNG_DIR = "_tmp/drop_shadow_png_kra_set"

def main():
    kra_files = glob.glob(os.path.join(KRA_DIR, "*.kra"))
    if not kra_files:
        print(f"No .kra files found in {KRA_DIR}. Run generate_kra_set.py first.")
        return

    os.makedirs(PNG_DIR, exist_ok=True)
    print(f"Found {len(kra_files)} KRA files. Beginning headless export via Krita CLI...")

    # Check if Krita exists
    try:
        subprocess.run(["krita", "--version"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
    except FileNotFoundError:
        print("Error: 'krita' executable not found in your PATH.")
        print("Please install Krita or run this script in an environment where Krita is available.")
        return

    count = 0
    for kra_path in kra_files:
        filename = os.path.basename(kra_path)
        png_name = filename.replace(".kra", ".png")
        png_path = os.path.join(PNG_DIR, png_name)

        if os.path.exists(png_path):
            continue

        print(f"Exporting {filename} -> {png_name}...")
        # Krita headless export command
        cmd = ["krita", kra_path, "--export", "--export-filename", png_path]
        res = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        if res.returncode == 0:
            count += 1
        else:
            print(f"  FAILED to export {filename}: {res.stderr.decode()}")

    print(f"Done. Successfully exported {count} files to {PNG_DIR}.")

if __name__ == "__main__":
    main()