#!/usr/bin/python3
# This consolidates client XMLs
import os, shutil, traceback, subprocess, time

MYSELF="clientdata/"
EVOLVED="../pre-renewal/client-data/"
MOUBOO="../client-data/"
REVOLT="../evol/client-data/"

# items.xml and their sprites in graphics/
# Remember to regenerate weapons.xml after merging

os.chdir("..")
item_db=[]
paths=[]

def fix_id(l, offset):
    if "id=" in l:
        tmp=l.split("\"")
        try:
            if (int(tmp[1]) > 0):
                tmp[1]=str(int(tmp[1]) + offset)
        except:
            traceback.print_exc()
        l="\"".join(tmp)
    return l

def hp(l):
    global paths
    if "<include" in l:
        return
    if ".png" in l or ".ogg" in l or ".xml" in l:
        tmpi=l.split("\"")
        for e in tmpi:
            if e.endswith(".png"):
                paths.append(e)
            elif e.endswith(".ogg"):
                paths.append(e)
            elif e.endswith(".xml"):
                ## TODO: We must also open the XML and extract paths from it
                paths.append(e)
    return

def unify(YESELF):
    global paths
    for f in paths:
        i="/".join(f.split("/")[:-1])
        ## Raw Copy
        try:
            os.makedirs(os.path.abspath("%s/%s" % (MYSELF, i)), exist_ok=True)
            shutil.copy2("%s/%s" % (YESELF, f),
                         "%s/%s" % (MYSELF, f))
            continue
        except FileNotFoundError:
            pass
        except:
            traceback.print_exc()

        ## Raw2 Copy
        try:
            os.makedirs(os.path.abspath("%s/graphics/%s" % (MYSELF, i)), exist_ok=True)
            shutil.copy2("%s/graphics/%s" % (YESELF, f),
                         "%s/graphics/%s" % (MYSELF, f))
            continue
        except FileNotFoundError:
            pass
        except:
            traceback.print_exc()

        ## Item Icon Copy
        try:
            os.makedirs(os.path.abspath("%s/graphics/items/%s" % (MYSELF, i)), exist_ok=True)
            shutil.copy2("%s/graphics/items/%s" % (YESELF, f),
                         "%s/graphics/items/%s" % (MYSELF, f))
            continue
        except FileNotFoundError:
            pass
        except:
            traceback.print_exc()

        ## Spritesheet Copy
        try:
            os.makedirs(os.path.abspath("%s/graphics/sprites/%s" % (YESELF, i)), exist_ok=True)
            shutil.copy2("%s/graphics/sprites/%s" % (YESELF, f),
                         "%s/graphics/sprites/%s" % (MYSELF, f))
            continue
        except FileNotFoundError:
            print("\033[1mFailed to copy: %s\033[0m" % f)
            pass
        except:
            traceback.print_exc()

    paths=[]
    return

#################################################################################
## Moubootaur Legends is a raw copy
with open("%s/items.xml" % MOUBOO, "r") as f:
    for l in f:
        ## FIXME: RACESPRITE
        #l=l.replace("    ", "\t")
        ## Add headers
        if "Authors:" in l:
            item_db.append("<!-- This file is generated automatically. -->\n")
            item_db.append("<!-- Please do not edit it directly. -->\n")
            item_db.append("<!-- ************************************************************************ -->\n")
        ## Follow includes instead of saving them
        if "<include" in l:
            tmp=l.split("\"")
            if not "name" in tmp[0]:
                print("\033[31mIllegal include: %s" % l)
                continue
            with open("%s/%s" % (MOUBOO, tmp[1]), "r") as k:
                save=False
                for j in k:
                    if not save:
                        if "<items>" in j:
                            save=True
                        continue
                    if "</items>" in j:
                        break
                    hp(j)
                    item_db.append(j)
            continue

        ## Save the line
        hp(l)
        item_db.append(l)
    while not "</items>" in item_db[-1]:
        item_db=item_db[:-1]
item_db=item_db[:-1]
print("\033[32;1mMoubootaur Legends OK\033[0m")


## Handle copy pastes
unify(MOUBOO)

################################################################################
## Legacy/Evolved has an offset of 10k
with open("%s/items.xml" % EVOLVED, "r") as f:
    saving=False
    for l in f:
        ## FIXME: RACESPRITE
        #l=l.replace("    ", "\t")
        ## Fix ID offset
        l=fix_id(l, 10000)
        ## Follow includes instead of saving them
        if "<include" in l:
            tmp=l.split("\"")
            if not "name" in tmp[0]:
                print("\033[31mIllegal include: %s" % l)
                continue
            with open("%s/%s" % (EVOLVED, tmp[1]), "r") as k:
                save=False
                for j in k:
                    if not save:
                        if "<items>" in j:
                            save=True
                        continue
                    if "</items>" in j:
                        break
                    j=fix_id(j, 10000)
                    item_db.append(j)
                    hp(j)
            continue

        ## Save the line
        if not saving:
            if "<items>" in l:
                saving=True
            continue
        item_db.append(l)
        hp(l)
    while not "</items>" in item_db[-1]:
        item_db=item_db[:-1]
item_db=item_db[:-1]
print("\033[32;1mLegacy/Evolved OK\033[0m")



## Handle copy pastes
unify(EVOLVED)

################################################################################
## rEvolt has an offset of 20k
with open("%s/items.xml" % REVOLT, "r") as f:
    saving=False
    for l in f:
        ## FIXME: RACESPRITE
        #l=l.replace("    ", "\t")
        ## Fix ID offset
        l=fix_id(l, 10000)
        ## Follow includes instead of saving them
        if "<include" in l:
            tmp=l.split("\"")
            if not "name" in tmp[0]:
                print("\033[31mIllegal include: %s" % l)
                continue
            with open("%s/%s" % (REVOLT, tmp[1]), "r") as k:
                save=False
                for j in k:
                    if not save:
                        if "<items>" in j:
                            save=True
                        continue
                    if "</items>" in j:
                        break
                    j=fix_id(j, 10000)
                    item_db.append(j)
                    hp(j)
            continue

        ## Save the line
        if not saving:
            if "<items>" in l:
                saving=True
            continue
        item_db.append(l)
        hp(l)
    while not "</items>" in item_db[-1]:
        item_db=item_db[:-1]
#item_db=item_db[:-1]
print("\033[32;1mrEvolt OK\033[0m")



## Handle copy pastes
unify(REVOLT)





################################################################################
## Save the final item database
with open(MYSELF+"items.xml", "w") as f:
    for l in item_db:
        f.write(l)

print("\033[33;1mSaved! Databases merged!\033[0m")

## Clean empty dirs
time.sleep(1)
print("Now deleting empty folders:")
time.sleep(4)
os.chdir(MYSELF)
subprocess.call("""find . -type d -empty -not -iwholename "./.*" -print -delete""", shell=True)