import os
import shutil
import subprocess
from pathlib import Path


COMMON_FILES = ['.git', 'README.md', __file__.split('/')[-1]]


def remove_old_files():
    filenames = os.listdir('./')
    filenames = [f for f in filenames if f not in COMMON_FILES]
    for file_path in filenames:
        p = Path(file_path)
        if p.exists():
            if p.is_file():
                p.unlink()
            elif p.is_dir():
                shutil.rmtree(p)


def clone_repository():
    repo_url = 'https://github.com/KonradSzafer/hugging-face-qa-bot.git'
    subprocess.run(['git', 'clone', repo_url])


def copy_files():
    src = './hugging-face-qa-bot'
    for item in COMMON_FILES:
        full_path = os.path.join(src, item)
        if os.path.isfile(full_path):
            os.remove(full_path)
        elif os.path.isdir(full_path):
            shutil.rmtree(full_path)
    for item in Path(src).iterdir():
        shutil.move(str(item), '.')
    shutil.rmtree(src)


if __name__ == '__main__':
    remove_old_files()
    clone_repository()
    copy_files()