setup.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Crocrodile Setup.

Setup Crocrodile chess engine.
"""
import ctypes
import glob
import os
import pkgutil
import platform
import shutil
import stat
import sys
import tarfile
import time
import zipfile
from urllib.request import Request, urlopen

try:
    shutil.rmtree("setup-env")
except FileNotFoundError:
    pass
os.mkdir("setup-env")
os.chdir("setup-env")

try:
    is_admin = os.getuid() == 0
except AttributeError:
    is_admin = ctypes.windll.shell32.IsUserAnAdmin() != 0

if not is_admin:
    print(
        "ERROR: Program is not running as root. Please run this program as root.",
        file=sys.stderr,
    )
    time.sleep(10)
    sys.exit(1)


def load_requirements(requirements):
    """Load requirements."""
    to_install = requirements
    installed = list()
    for module in pkgutil.iter_modules():
        if module.name in to_install:
            to_install.remove(module.name)
            installed.append(module.name)
    return to_install, installed


def download(url, destination):
    if not url.startswith("http"):
        raise RuntimeError("Incorrect and possibly insecure protocol in url")
    httprequest = Request(url)
    response = urlopen(httprequest)
    with open(destination, "wb") as destination:
        destination.write(response.read())


def detect_python():
    """Detect Python installation."""
    print("Detecting: Python...", end=" ", flush=True)
    test = os.system("python3 --version > " + os.devnull + " 2> " + os.devnull)
    if test == 0:
        print("Done.")
        return "python3"
    test = os.system("py -3 --version > " + os.devnull + " 2> " + os.devnull)
    if test == 0:
        print("Done.")
        return "py -3"
    print("Done.")
    print("/!\\ Python installation not found !")
    print(
        "    Please verify that Python is installed and launchable with 'python3' command"
    )
    stop()


def detect_pip(python):
    """Detect pip installation."""
    print("Detecting: pip...", end=" ", flush=True)
    test = os.system(python + " -m pip --version > " + os.devnull + " 2> " + os.devnull)
    print("Done.")
    if test == 0:
        return
    print("Downloading: https://bootstrap.pypa.io/get-pip.py...", end=" ", flush=True)
    download("https://bootstrap.pypa.io/get-pip.py", "get-pip.py")
    print("Done.")
    print("Installing: pip...", end=" ", flush=True)
    test = os.system(python + " get-pip.py > " + os.devnull + " 2> " + os.devnull)
    print("Done.")
    if test != 0:
        print("/!\\ Failed to install pip.")
        stop()


def install_requirements(requirements, python):
    """Install missing requirements."""
    for requirement in requirements:
        if requirement.endswith("\n"):
            requirement = requirement[:-1]
        print(
            "Downloading: https://pypi.org/simple/{0}/...".format(requirement),
            end=" ",
            flush=True,
        )
        for retry in range(3):
            test = os.system(
                python
                + " -m pip download --prefer-binary "
                + requirement
                + " > "
                + os.devnull
                + " 2> "
                + os.devnull
            )
            if test == 0:
                break
            if retry == 0:
                print()
            print("Failed to download, retrying (" + str(retry + 1) + "/3)")
        if retry == 2:
            print("\n/!\\ Failed to download", requirement)
            stop()
        print("Done.")
    for wheel in glob.glob("*.whl"):
        print("Installing:", wheel + "...", end=" ", flush=True)
        test = os.system(
            python
            + " -m pip install --no-dependencies "
            + wheel
            + " > "
            + os.devnull
            + " 2> "
            + os.devnull
        )
        if test != 0:
            print("\n/!\\ Failed to install", wheel)
        print("Done.")
    for targz in glob.glob("*.tar.gz"):
        print("Unpacking:", targz + "...", end=" ", flush=True)
        file = tarfile.open(targz)
        file.extractall(path="./setup-{0}/".format(targz[:-7]))
        file.close()
        print("Done.")
        print("Installing:", targz[:-7] + "...", end=" ", flush=True)
        os.chdir("setup-" + targz[:-7])
        os.chdir(targz[:-7])
        os.system(python + " setup.py build > " + os.devnull + " 2> " + os.devnull)
        os.system(python + " setup.py install > " + os.devnull + " 2> " + os.devnull)
        os.chdir("../..")
        print("Done.")


def install_crocrodile(python):
    print(
        "Downloading: https://codeload.github.com/Virinas-code/crocrodile-easy/zip/refs/heads/master...",
        end=" ",
        flush=True,
    )
    download(
        "https://codeload.github.com/Virinas-code/crocrodile-easy/zip/refs/heads/master",
        "crocrodile.zip",
    )
    print("Done.")
    print("Unpacking: crocrodile.zip...", end=" ", flush=True)
    system = platform.system()
    with zipfile.ZipFile("crocrodile.zip", "r") as zip_ref:
        if system == "Linux":
            zip_ref.extractall("/usr/lib/crocrodile/")
        elif system == "Windows":
            zip_ref.extractall("C:/Program Files (x86)/Crocrodile/")
        else:
            print()
            print("/!\\ Unindentified or unsupported OS.")
            stop()
    print("Done.")
    print("Installing: Crocrodile...", end=" ", flush=True)
    if system == "Linux":
        with open("/usr/bin/crocrodile", "w") as file:
            file.write(
                "#!/usr/bin/sh\ncd /usr/lib/crocrodile/Crocrodile-master/\n"
                + python
                + " uci.py"
            )
            st = os.stat("/usr/bin/crocrodile")
            os.chmod("/usr/bin/crocrodile", st.st_mode | stat.S_IEXEC)
    elif system == "Windows":
        with open("C:/Program Files (x86)/Crocrodile/crocrodile.bat", "w") as file:
            file.write(
                "cd C:/Program Files (x86)/Crocrodile/Crocrodile-master/\n"
                + python
                + " uci.py"
            )
    print("Done.")


def install(requirements):
    """Install Crocrodile."""
    python = detect_python()
    detect_pip(python)
    install_requirements(requirements, python)
    install_crocrodile(python)


def stop():
    """Stop installation."""
    shutil.rmtree("setup-env")
    print("Installation canceled.")
    sys.exit(1)


def get_requirements():
    print(
        "Downloading: https://raw.githubusercontent.com/Virinas-code/Crocrodile/master/requirements.txt...",
        end=" ",
    )
    download(
        "https://raw.githubusercontent.com/Virinas-code/Crocrodile/master/requirements.txt",
        "requirements.txt",
    )
    print("Done.")
    required = list()
    for line in open("requirements.txt").readlines():
        required.append(line.split("==")[0])
    return required


requirements = get_requirements()

print("Welcome in Crocrodile setup program !")

print("Loading requirements...", end=" ", flush=True)
to_install, installed = load_requirements(requirements)
print("Done.")

print()
print("Installed packages:")
print("   ", "\t".join(installed))
print("Packages to install:")
print("   ", "\t".join(to_install))
if not sys.argv:
    continuation = input("Continue ? [Y/n] ").lower()
else:
    continuation = ""
print()
if continuation in ("y", "yes", ""):
    install(to_install)
elif continuation in ("n", "no"):
    print("Canceling...")
    stop()
else:
    print("Bad answer. Canceling...")
    stop()