build_mira_ui.py

import subprocess
from os import mkdir, getuid, getgid
from os.path import join, exists, abspath
from shutil import copytree, rmtree

from colored import fg, attr
from lib.utils import config_path, config_dict, prompt, write_json, target_folder

tmp_folder = './web/build'

if exists(tmp_folder):
    print(fg(111) + 'clean tmp folder' + attr('reset'))
    rmtree(tmp_folder, ignore_errors=True)
mkdir(tmp_folder)

web_folder = join(target_folder, 'web')


def ask_env():
    chrome_extension_id = prompt('Enter CHROME_EXTENSION_ID: ')
    firefox_extension_id = prompt('Enter FIREFOX_EXTENSION_ID: ')
    firefox_extension_url = prompt('Enter FIREFOX_EXTENSION_URL: ')
    # ga = prompt('Enter GA: ')
    site_title = prompt('Enter SITE_TITLE: ')
    http_proxy = prompt('Enter http proxy for docker container, if you need to access github via proxy: ')
    https_proxy = prompt('Enter https proxy for docker container, press Enter to use the same with http proxy: ')
    if not https_proxy:
        https_proxy = http_proxy
    config_dict['web']['CHROME_EXTENSION_ID'] = chrome_extension_id
    config_dict['web']['FIREFOX_EXTENSION_ID'] = firefox_extension_id
    config_dict['web']['FIREFOX_EXTENSION_URL'] = firefox_extension_url
    # config_dict['web']['GA'] = ga
    config_dict['web']['SITE_TITLE'] = site_title
    config_dict['web']['HTTP_PROXY'] = http_proxy
    config_dict['web']['HTTPS_PROXY'] = https_proxy

    write_json(config_path, config_dict)


if config_dict.get('web') is None:
    config_dict['web'] = {}
    ask_env()
else:
    print('There is a previous saved config:')
    for key in config_dict['web']:
        print(fg(116) + key + '=' + config_dict['web'][key] + attr('reset'))
    use_saved = prompt('Do you want to use these configuration? y for yes, n for no: ')
    if use_saved == 'n':
        ask_env()

use_tag = None
while use_tag != 'y' and use_tag != 'n':
    use_tag = prompt('Do you want to use tag for building? y for yes, n for no')

tag_to_checkout = None
if use_tag == 'y':
    tag_to_checkout = prompt('You choose use tag, please specify the tag you want to use (e.g. v4.2.0): ')
else:
    print(fg(154) + 'You choose not use tag, will checkout master branch' + attr('reset'))

uid = getuid()
gid = getgid()

cmd_base = 'docker run --rm -v {0}:/build -u {1}:{2}'.format(abspath(tmp_folder), uid, gid)
if config_dict.get('web') is not None:
    env_list = ['--env HOME=/build']
    for key in config_dict.get('web'):
        env_list.append('--env {0}={1}'.format(key, config_dict['web'][key]))
    cmd_base = cmd_base + ' ' + ' '.join(env_list)

cmd = cmd_base + ' node:16 bash -c \'cd /build && git clone https://github.com/irohalab/mira-ui.git mira-ui && cd mira-ui && '
if tag_to_checkout is not None:
    cmd = cmd + 'git checkout tags/{0} -b {0}-branch && '.format(tag_to_checkout)

if config_dict['web'].get('HTTP_PROXY') is not None:
    cmd = cmd + 'npm config set proxy {0} && '.format(config_dict['web']['HTTP_PROXY'])

cmd = cmd + 'npm install && npm run build\''
return_code = subprocess.call(cmd, shell=True, cwd=target_folder)

if return_code != 0:
    print(fg(9) + 'build failed!' + attr('reset'))
    exit(-1)

if exists(web_folder):
    rmtree(web_folder)
copytree(join(tmp_folder, 'mira-ui', 'dist', 'mira-ui'), web_folder)

print(fg(119) + 'All done! built files is copied to ' + web_folder + attr('reset'))