#!/usr/bin/python3

import sys
import os
import pathlib
import map_helpers
import shutil

def main(map_folder):
    list_of_files = list()
    for root, directories, filenames in os.walk(map_folder):
        for filename in [f for f in filenames if f.endswith(".dmm")]:
            list_of_files.append(pathlib.Path(root, filename))

    last_dir = ""
    for i in range(0, len(list_of_files)):
        this_dir = list_of_files[i].parent
        if last_dir != this_dir:
            print("--------------------------------")
            last_dir = this_dir
        print("[{}]: {}".format(i, str(list_of_files[i])[len(map_folder):]))

    print("--------------------------------")
    in_list = input("List the maps you want to merge (example: 1,3-5,12):\n")
    in_list = in_list.replace(" ", "")
    in_list = in_list.split(",")

    valid_indices = list()
    for m in in_list:
        index_range = m.split("-")
        if len(index_range) == 1:
            index = string_to_num(index_range[0])
            if index >= 0 and index < len(list_of_files):
                valid_indices.append(index)
        elif len(index_range) == 2:
            index0 = string_to_num(index_range[0])
            index1 = string_to_num(index_range[1])
            if index0 >= 0 and index0 <= index1 and index1 < len(list_of_files):
                valid_indices.extend(range(index0, index1 + 1))

#    if tgm == "1":
#        print("\nMaps will be converted to tgm.")
#        tgm = True
#    else:
#        print("\nMaps will not be converted to tgm.")
#        tgm = False
    tgm = False

    print("\nMerging these maps:")
    for i in valid_indices:
        print(str(list_of_files[i])[len(map_folder):])
    merge = input("\nPress Enter to merge...")
    if merge == "abort":
        print("\nAborted map merge.")
        sys.exit()
    else:
        for i in valid_indices:
            path_str = str(list_of_files[i])
            shutil.copyfile(path_str, path_str + ".before")
            path_str_pretty = path_str[len(map_folder):]
            try:
                if map_helpers.merge_map(path_str, path_str + ".backup", tgm) != 1:
                    print("ERROR MERGING: {}".format(path_str_pretty))
                    os.remove(path_str + ".before")
                    continue
                print("MERGED: {}".format(path_str_pretty))
            except FileNotFoundError:
                print("\nERROR: File not found! Make sure you run 'Prepare Maps.bat' before merging.")
                print(path_str_pretty + " || " + path_str_pretty + ".backup")

    print("\nFinished merging.")
    print("\nNOTICE: A version of the map files from before merging have been created for debug purposes.\nDo not delete these files until it is sure your map edits have no undesirable changes.")

def string_to_num(s):
    try:
        return int(s)
    except ValueError:
        return -1

if __name__ == "__main__":
    main(sys.argv[1])