toolchain-import.py

import shutil
import os
import os.path
import sys

from distutils.dir_util import copy_tree
from subprocess import call
import platform


def get_python():
    if platform.system() == "Windows":
        return "python"
    else:
        return "python3"
        

def copytree(src, dst, symlinks=False, ignore=None):
    if not os.path.exists(src) or os.path.isfile(src):
        raise Exception()
    for item in os.listdir(src):
        s = os.path.join(src, item)
        d = os.path.join(dst, item)
        if os.path.exists(d):
            continue
        if os.path.isdir(s):
            shutil.copytree(s, d, symlinks, ignore)
        else:
            shutil.copy2(s, d)

def backup_project(directory):
    print("backing up project to project.back")
    backup_path = os.path.join(directory, "project.back")
    if os.path.exists(backup_path):
        print("project.back already exists, skipping")
        return

    files = os.listdir(directory)

    os.makedirs(backup_path)

    for f in files:
        exceptions = [
            ".git", 
            "toolchain-setup.py",
            "toolchain.zip"
        ]
        if f in exceptions: continue
        entry = os.path.join(directory, f)

        if(os.path.isfile(entry)):
            shutil.copyfile(entry, os.path.join(backup_path, f))
            os.remove(entry)

        elif(os.path.isdir(entry)):
            shutil.copytree(entry, os.path.join(backup_path, f))
            shutil.rmtree(entry)

def download_and_extract_toolchain(directory):
    import urllib.request
    import zipfile
    archive = os.path.join(directory, 'toolchain.zip')

    if not os.path.exists(archive):
        url = "https://codeload.github.com/zheka2304/innercore-mod-toolchain/zip/master"
        print("downloading toolchain archive from " + url)
        urllib.request.urlretrieve(url, archive)
    else: 
        print("toolchain archive already exists in " + directory)

    print("extracting toolchain to " + directory)

    with zipfile.ZipFile(archive, 'r') as zip_ref:
        zip_ref.extractall(directory)

    try:
        copytree(os.path.join(directory, "innercore-mod-toolchain-master/toolchain-mod"), directory)
        shutil.rmtree(os.path.join(directory, "innercore-mod-toolchain-master"))
    except Exception as ex: 
        pass
    finally:
        if not os.path.exists(os.path.join(directory, "toolchain")):
            print("an error occured while extracting toolchain archive, please, retry the operation")
            os.remove(archive)
            exit()
    
    

    
if(len(sys.argv) > 1):
    directory = sys.argv[1]
else: 
    directory = '.'

backup_project(directory)
download_and_extract_toolchain(directory)

import_script = os.path.join(directory, "toolchain", "python", "import.py")
call(get_python() + " " + import_script + " " + directory + " " + os.path.join(directory, "project.back"), shell=True)