hifi_qt.py

import hifi_utils
import hifi_android
import hashlib
import os
import platform
import re
import shutil
import tempfile
import json
import xml.etree.ElementTree as ET
import functools

print = functools.partial(print, flush=True)

# Encapsulates the vcpkg system 
class QtDownloader:
    CMAKE_TEMPLATE = """
# this file auto-generated by hifi_qt.py
get_filename_component(QT_CMAKE_PREFIX_PATH "{}" ABSOLUTE CACHE)
get_filename_component(QT_CMAKE_PREFIX_PATH_UNCACHED "{}" ABSOLUTE)

# If the cached cmake toolchain path is different from the computed one, exit
if(NOT (QT_CMAKE_PREFIX_PATH_UNCACHED STREQUAL QT_CMAKE_PREFIX_PATH))
    message("QT_CMAKE_PREFIX_PATH: ${{QT_CMAKE_PREFIX_PATH}}")
    message("QT_CMAKE_PREFIX_PATH_UNCACHED: ${{QT_CMAKE_PREFIX_PATH_UNCACHED}}")
    message(FATAL_ERROR "QT_CMAKE_PREFIX_PATH has changed, please wipe the build directory and rerun cmake")
endif()
"""
    def __init__(self, args):
        self.args = args
        self.configFilePath = os.path.join(args.build_root, 'qt.cmake')
        self.version = '5.15.2'

        defaultBasePath = os.path.expanduser('~/tivoli/qt')
        self.basePath = os.getenv('TIVOLI_QT_BASE', defaultBasePath)
        if not os.path.isdir(self.basePath):
            os.makedirs(self.basePath)
        self.path = os.path.join(self.basePath, self.version)
        self.fullPath = os.path.join(self.path, 'qt5-install')

        self.hasQt = False
        if os.getenv("QT_CMAKE_PREFIX_PATH"):
            self.hasQt = True
            self.cmakePath = os.getenv("QT_CMAKE_PREFIX_PATH")
        else:
            self.cmakePath = os.path.join(self.fullPath, 'lib/cmake')

        print("Using qt path {}".format(self.path))
        lockDir, lockName = os.path.split(self.path)
        lockName += '.lock'
        if not os.path.isdir(lockDir):
            os.makedirs(lockDir)

        self.lockFile = os.path.join(lockDir, lockName)

        if self.hasQt:
            return
            
        system = platform.system()
        arch = platform.machine().lower()

        baseQtUrl = 'https://cdn.tivolicloud.com/dependencies/vcpkg/'

        if arch == "x86_64" or arch == "amd64":
            if system == 'Windows':
                self.qtUrl = baseQtUrl + 'tivoli-qt5-install-5.15.2-windows.tar.gz'
            elif system == 'Darwin':
                self.qtUrl = baseQtUrl + 'tivoli-qt5-install-5.15.2-macos.tar.gz'
            elif system == 'Linux':
                issue = open("/etc/issue", "r").read()
                if issue.startswith("Ubuntu 16.04"):
                    self.qtUrl = baseQtUrl + 'tivoli-qt5-install-5.15.2-ubuntu-16.04.tar.gz'
                elif issue.startswith("Ubuntu 18.04"):
                    self.qtUrl = baseQtUrl + 'tivoli-qt5-install-5.15.2-ubuntu-18.04.tar.gz'
                elif issue.startswith("Ubuntu 20.04"):
                    self.qtUrl = baseQtUrl + 'tivoli-qt5-install-5.15.2-ubuntu-20.04.tar.gz'
                elif issue.startswith("Arch Linux"):
                    self.qtUrl = baseQtUrl + 'tivoli-qt5-install-5.15.2-arch-linux.tar.gz'
                else:
                    raise Exception('Unsupported Linux version')
            else:
                raise Exception('Unsupported operating system')
        
        elif arch == "aarch64" or arch == "arm64":
            if system == 'Darwin':
                # we dont have qt5 with webengine for arm64 yet so we're
                # compiling under rosetta until we do
                self.qtUrl = baseQtUrl + 'tivoli-qt5-install-5.15.2-macos.tar.gz'
            elif system == 'Linux':
                issue = open("/etc/issue", "r").read()
                if issue.startswith("Ubuntu 18.04"): 
                    self.qtUrl = baseQtUrl + 'tivoli-qt5-install-5.15.2-ubuntu-18.04-arm64v8.tar.gz'
                elif issue.startswith("Ubuntu 20.04"): 
                    self.qtUrl = baseQtUrl + 'tivoli-qt5-install-5.15.2-ubuntu-20.04-arm64v8.tar.gz'
                else:
                    raise Exception('Unsupported arm64 Linux version')
            else:
                raise Exception('Unsupported arm64 operating system')
        
        elif arch == "i386":
            raise Exception('32-bit not supported')

        else:
            raise Exception('Unknown CPU architecture')

    def writeConfig(self):
        print("Writing cmake config to {}".format(self.configFilePath))
        # Write out the configuration for use by CMake
        cmakeConfig = QtDownloader.CMAKE_TEMPLATE.format(self.cmakePath, self.cmakePath).replace('\\', '/')
        with open(self.configFilePath, 'w') as f:
            f.write(cmakeConfig)

    def installQt(self):
        if not os.path.isdir(self.fullPath) and not self.hasQt:
            print ('Downloading Qt from CDN')
            print('Extracting ' + self.qtUrl + ' to ' + self.path)
            hifi_utils.downloadAndExtract(self.qtUrl, self.path)
        else:
            print ('Qt has already been downloaded')