#!/usr/bin/env python3
"""CLI entrypoint to export GeoData assets for Unity."""
from __future__ import annotations

import argparse
import os
import sys
from typing import Iterable

from geodata_pipeline.config import Config, DEFAULT_CONFIG_PATH, ensure_default_config
from geodata_pipeline.heightmaps import export_heightmaps
from geodata_pipeline.orthophotos import export_orthophotos
from geodata_pipeline.setup_helpers import ensure_directories, materialize_archives


def parse_args(argv: Iterable[str] | None = None) -> argparse.Namespace:
    parser = argparse.ArgumentParser(description="Export GeoData assets for Unity.")
    parser.add_argument(
        "--config",
        default=DEFAULT_CONFIG_PATH,
        help="Path to config JSON (created on --setup if missing).",
    )
    parser.add_argument(
        "--export",
        choices=["heightmap", "textures", "all"],
        default="all",
        help="Which assets to export.",
    )
    parser.add_argument("--raw-dgm1-path", dest="raw_dgm1_path", help="Override raw DGM1 directory.")
    parser.add_argument("--raw-dop20-path", dest="raw_dop20_path", help="Override raw DOP20 JP2 directory.")
    parser.add_argument(
        "--build-from-archive",
        action="store_true",
        help="Populate raw inputs from archives (unzips zips, leaves dop20 filelist in archive).",
    )
    parser.add_argument(
        "--setup",
        action="store_true",
        help="Create default directory structure and config if missing; does not export.",
    )
    parser.add_argument(
        "--force-vrt",
        action="store_true",
        help="Rebuild VRTs even if present (useful after moving raw data).",
    )
    return parser.parse_args(argv)


def load_config(args: argparse.Namespace) -> Config:
    if args.setup:
        cfg = Config.default()
        cfg.save(args.config)
        return cfg.with_overrides(raw_dgm1_path=args.raw_dgm1_path, raw_dop20_path=args.raw_dop20_path)
    if os.path.exists(args.config):
        cfg = Config.load(args.config)
    else:
        cfg = Config.default()
        cfg.save(args.config)
    return cfg.with_overrides(raw_dgm1_path=args.raw_dgm1_path, raw_dop20_path=args.raw_dop20_path)


def main(argv: Iterable[str] | None = None) -> int:
    args = parse_args(argv)
    cfg = load_config(args)

    if args.setup:
        ensure_directories(cfg)
        print(f"Directories ensured. Config at {args.config}.")
        if args.export is None:
            return 0

    if args.build_from_archive:
        materialize_archives(cfg)

    exit_codes = []
    if args.export in ("heightmap", "all"):
        exit_codes.append(export_heightmaps(cfg, force_vrt=args.force_vrt))
    if args.export in ("textures", "all"):
        exit_codes.append(export_orthophotos(cfg, force_vrt=args.force_vrt))

    return max(exit_codes) if exit_codes else 0


if __name__ == "__main__":
    sys.exit(main())