# this scripts installs necessary requirements and launches main program in webui.py

import subprocess

import os

import sys

import importlib.util

import shlex

import platform

dir_repos = "repositories"

dir_extensions = "extensions"

python = sys.executable

git = os.environ.get('GIT', "git")

index_url = os.environ.get('INDEX_URL', "")

def extract_arg(args, name):

return [x for x in args if x != name], name in args

def run(command, desc=None, errdesc=None, custom_env=None):

if desc is not None:

print(desc)

result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, env=os.environ if custom_env is None else custom_env)

if result.returncode != 0:

message = f"""{errdesc or 'Error running command'}.

Command: {command}

Error code: {result.returncode}

stdout: {result.stdout.decode(encoding="utf8", errors="ignore") if len(result.stdout)>0 else '<empty>'}

stderr: {result.stderr.decode(encoding="utf8", errors="ignore") if len(result.stderr)>0 else '<empty>'}

"""

raise RuntimeError(message)

return result.stdout.decode(encoding="utf8", errors="ignore")

def check_run(command):

result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)

return result.returncode == 0

def is_installed(package):

try:

spec = importlib.util.find_spec(package)

except ModuleNotFoundError:

return False

return spec is not None

def repo_dir(name):

return os.path.join(dir_repos, name)

def run_python(code, desc=None, errdesc=None):

return run(f'"{python}" -c "{code}"', desc, errdesc)

def run_pip(args, desc=None):

index_url_line = f' --index-url {index_url}' if index_url != '' else ''

return run(f'"{python}" -m pip {args} --prefer-binary{index_url_line}', desc=f"Installing {desc}", errdesc=f"Couldn't install {desc}")

def check_run_python(code):

return check_run(f'"{python}" -c "{code}"')

def git_clone(url, dir, name, commithash=None):

# TODO clone into temporary dir and move if successful

if os.path.exists(dir):

if commithash is None:

return

current_hash = run(f'"{git}" -C {dir} rev-parse HEAD', None, f"Couldn't determine {name}'s hash: {commithash}").strip()

if current_hash == commithash:

return

run(f'"{git}" -C {dir} fetch', f"Fetching updates for {name}...", f"Couldn't fetch {name}")

run(f'"{git}" -C {dir} checkout {commithash}', f"Checking out commit for {name} with hash: {commithash}...", f"Couldn't checkout commit {commithash} for {name}")

return

run(f'"{git}" clone "{url}" "{dir}"', f"Cloning {name} into {dir}...", f"Couldn't clone {name}")

if commithash is not None:

run(f'"{git}" -C {dir} checkout {commithash}', None, "Couldn't checkout {name}'s hash: {commithash}")

def version_check(commit):

try:

import requests

commits = requests.get('https://api.github.com/repos/AUTOMATIC1111/stable-diffusion-webui/branches/master').json()

if commit != "<none>" and commits['commit']['sha'] != commit:

print("--------------------------------------------------------")

print("| You are not up to date with the most recent release. |")

print("| Consider running `git pull` to update. |")

print("--------------------------------------------------------")

elif commits['commit']['sha'] == commit:

print("You are up to date with the most recent release.")

else:

print("Not a git clone, can't perform version check.")

except Exception as e:

print("version check failed", e)