import requests
import os
import subprocess
from rich.progress import Progress
from rich import print
import ctypes
import sys
import shutil

REPO_URL = "https://alexisdelhaie.ovh/dlcenter/chronos-repo/installer/"
VERSION_FILE = "version"
COMPANY_NAME = "alexlegarnd"
SOFTWARE_NAME = "chronos-installer"
CACHE_FOLDER = "cache"
INSTALL_FOLDER = "{}\\{}\\{}".format(os.getenv('APPDATA'), COMPANY_NAME, SOFTWARE_NAME)
CACHE_FOLDER_PATH = "{}\\{}".format(INSTALL_FOLDER, CACHE_FOLDER)
INSTALLED_VERSION_PATH = "{}\\{}".format(INSTALL_FOLDER, VERSION_FILE)
EXECUTABLE = "Install.exe"
EXECUTABLE_PATH = "{}\\{}".format(INSTALL_FOLDER, EXECUTABLE)
FILES = ["libeay32.dll", "ssleay32.dll", "7z.dll", EXECUTABLE]


def get_installed_version():
    if os.path.exists(INSTALLED_VERSION_PATH):
        file = open(INSTALLED_VERSION_PATH, "r")
        version = file.read()
        file.close()
        return version
    return "0.0.0"


def is_admin():
    try:
        return ctypes.windll.shell32.IsUserAnAdmin()
    except:
        return False


def get_version_from_repo():
    url = "{}{}".format(REPO_URL, VERSION_FILE)
    r = requests.get(url, allow_redirects=True)
    return r.text


def download_file_from_repo(filename):
    url = "{}{}".format(REPO_URL, filename)
    with open("{}\\{}".format(INSTALL_FOLDER, filename), 'wb') as f:
        response = requests.get(url, stream=True)
        total_length = response.headers.get('content-length')

        if total_length is None:  # no content length header
            f.write(response.content)
        else:
            with Progress() as progress:
                task = progress.add_task("Downloading {}".format(filename), total=int(total_length))
                for data in response.iter_content(chunk_size=4096):
                    f.write(data)
                    progress.update(task, advance=len(data))


def create_install_folder():
    company_directory = "{}\\{}".format(os.getenv('APPDATA'), COMPANY_NAME)
    if not os.path.exists(company_directory):
        os.mkdir(company_directory)
    if not os.path.exists(INSTALL_FOLDER):
        os.mkdir(INSTALL_FOLDER)


def clean_folder(path):
    if os.path.exists(path):
        for filename in os.listdir(path):
            if filename != CACHE_FOLDER:
                file_path = os.path.join(path, filename)
                try:
                    if os.path.isfile(file_path) or os.path.islink(file_path):
                        os.unlink(file_path)
                    elif os.path.isdir(file_path):
                        shutil.rmtree(file_path)
                except Exception as e:
                    print('[bold red]ERROR[/] Failed to delete {}. Reason: {}'.format(file_path, e))


def clean():
    if '--clean' in sys.argv:
        clean_folder(INSTALL_FOLDER)
        clean_folder(CACHE_FOLDER_PATH)
    if '--clean-cache' in sys.argv:
        clean_folder(CACHE_FOLDER_PATH)
    if '--clean-installer' in sys.argv:
        clean_folder(INSTALL_FOLDER)


def main():
    if not is_admin():
        print('[bold red]ERROR[/] This app need to be launched eleveted')
        input('-- Press enter key to quit --')
        sys.exit(0)
    print('Chronos installer bootstrap\n\n@alexlegarnd -- https://alexisdelhaie.ovh/')
    print('Github: https://github.com/alexlegarnd/chronos-installer-bootstrap\n')
    version = get_version_from_repo()
    create_install_folder()
    clean()
    if get_installed_version() != version:
        print('[bold green]INFO[/] Downloading latest version of the software installer')
        clean_folder(INSTALL_FOLDER)
        for file in FILES:
            download_file_from_repo(file)
        if os.path.exists(INSTALLED_VERSION_PATH):
            os.remove(INSTALLED_VERSION_PATH)
        file = open(INSTALLED_VERSION_PATH, "w")
        file.write(version)
        file.close()
    if os.path.exists(EXECUTABLE_PATH):
        subprocess.Popen([EXECUTABLE_PATH], cwd=INSTALL_FOLDER)
    else:
        print('[bold red]ERROR[/] Executable not found')
        input('-- Press enter key to quit --')


main()