import os, shutil
from distutils.dir_util import copy_tree
from . import config, constants


def clone_public_data():
    if os.path.isdir(constants.PUBLIC_DATA_DIR):
        initial = os.getcwd()
        go_to_data()
        exit_code = os.system("git pull")
        os.chdir(initial)
        return exit_code

    exit_code = os.system("git clone -b master --single-branch {} {}"
        .format(config.get('public_repo'),
        constants.PUBLIC_DATA_DIR))
    return exit_code


def clone_private_data():
    exit_code = 0
    if os.path.isdir(constants.PRIVATE_DATA_DIR):
        initial = os.getcwd()
        os.chdir(constants.PRIVATE_DATA_DIR)
        exit_code = os.system("git pull")
        os.chdir(initial)
    else:
        exit_code = os.system("git clone -b master --single-branch {} {}"
            .format(config.get('private_repo'), constants.PRIVATE_DATA_DIR))
    if exit_code != 0:
        return exit_code

    copy_tree(
        constants.PRIVATE_DATA_DIR,
        constants.PUBLIC_DATA_DIR
    )
    return exit_code


def has_data(data):
    public_path = os.path.join(constants.PUBLIC_DATA_DIR, data)
    private_path = os.path.join(constants.PRIVATE_DATA_DIR, data)
    return os.path.isdir(public_path) or os.path.isdir(private_path)


def go_to_data(subdir=''):
    path = os.path.join(constants.PUBLIC_DATA_DIR, subdir)
    os.chdir(path)
    return path