121 lines
4.0 KiB
Python
121 lines
4.0 KiB
Python
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()
|