#!/usr/bin/env python

# pmtiles to files
import argparse
import os
import shutil

from pmtiles.convert import mbtiles_to_pmtiles, pmtiles_to_mbtiles, pmtiles_to_dir

parser = argparse.ArgumentParser(
    description="Convert between PMTiles and other archive formats."
)
parser.add_argument("input", help="Input .mbtiles or .pmtiles")
parser.add_argument("output", help="Output .mbtiles, .pmtiles, or directory")
parser.add_argument(
    "--maxzoom", help="the maximum zoom level to include in the output."
)
parser.add_argument(
    "--overwrite", help="Overwrite the existing output.", action="store_true"
)
args = parser.parse_args()

if os.path.exists(args.output) and not args.overwrite:
    print("Output exists, use --overwrite to overwrite the output.")
    exit(1)
if args.overwrite:
    if os.path.isfile(args.output):
        os.remove(args.output)
    elif os.path.isdir(args.output):
        shutil.rmtree(args.output)

if args.input.endswith(".mbtiles") and args.output.endswith(".pmtiles"):
    print("Notice: check out the new PMTiles converter at https://github.com/protomaps/go-pmtiles")
    mbtiles_to_pmtiles(args.input, args.output, args.maxzoom)

elif args.input.endswith(".pmtiles") and args.output.endswith(".mbtiles"):
    pmtiles_to_mbtiles(args.input, args.output)

elif args.input.endswith(".pmtiles"):
    pmtiles_to_dir(args.input, args.output)

else:
    print("Conversion not implemented")
