electron/script/bootstrap.py

#!/usr/bin/env python

import argparse
import os
import subprocess
import sys

from lib.config import LIBCHROMIUMCONTENT_COMMIT, BASE_URL
from lib.util import scoped_cwd


SOURCE_ROOT = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
VENDOR_DIR = os.path.join(SOURCE_ROOT, 'vendor')
PYTHON_26_URL = 'https://chromium.googlesource.com/chromium/deps/python_26'


def main():
  os.chdir(SOURCE_ROOT)

  args = parse_args()
  update_submodules()
  update_apm()
  update_node_modules()
  bootstrap_brightray(args.url)
  if sys.platform == 'cygwin':
    update_win32_python()

  touch_config_gypi()
  update_atom_shell()


def parse_args():
  parser = argparse.ArgumentParser(description='Bootstrap this project')
  parser.add_argument('-u', '--url',
                      help='The base URL from which to download '
                      'libchromiumcontent (i.e., the URL you passed to '
                      'libchromiumcontent\'s script/upload script',
                      default=BASE_URL,
                      required=False)
  return parser.parse_args()


def update_submodules():
  subprocess.check_call(['git', 'submodule', 'sync', '--quiet'])
  subprocess.check_call(['git', 'submodule', 'update', '--init',
                         '--recursive'])


def bootstrap_brightray(url):
  bootstrap = os.path.join(VENDOR_DIR, 'brightray', 'script', 'bootstrap')
  subprocess.check_call([sys.executable, bootstrap, '--commit',
                         LIBCHROMIUMCONTENT_COMMIT, url])


def update_apm():
  ## NB: Without this, subprocess incorrectly searches for npm.exe
  npm_cmd = 'npm'
  if sys.platform in ['win32', 'cygwin']:
    npm_cmd += '.cmd'

  with scoped_cwd(os.path.join('vendor', 'apm')):
    subprocess.check_call([npm_cmd, 'install', '.'])


def update_node_modules():
  for dirname in ['atom/browser/default_app', '.']:
    update_node_modules_for_dir(dirname)


def update_node_modules_for_dir(dirname):
  with scoped_cwd(dirname):
    apm = os.path.join(SOURCE_ROOT, 'vendor', 'apm', 'bin', 'apm')
    subprocess.check_call(['node', os.path.relpath(apm), 'install'])


def update_win32_python():
  with scoped_cwd(VENDOR_DIR):
    if not os.path.exists('python_26'):
      subprocess.check_call(['git', 'clone', PYTHON_26_URL])
    else:
      with scoped_cwd('python_26'):
        subprocess.check_call(['git', 'pull', '--rebase', 'origin', 'master'])


def touch_config_gypi():
  config_gypi = os.path.join(SOURCE_ROOT, 'vendor', 'node', 'config.gypi')
  with open(config_gypi, 'w+') as f:
    content = '\n{}'
    if f.read() != content:
      f.write(content)


def update_atom_shell():
  update = os.path.join(SOURCE_ROOT, 'script', 'update.py')
  subprocess.check_call([sys.executable, update])


if __name__ == '__main__':
  sys.exit(main())