neft/lib/testing/env/chrome.coffee

Universal Platform

'use strict'

fs = require 'fs-extra'
childProcess = require 'child_process'
which = require 'which'
utils = require 'src/utils'
signal = require 'src/signal'
logger = require '../logger'
config = require '../cli/config'
processLogs = require '../cli/processLogs'

{log} = Neft

USER_DATA_DIR = "tmp/chrome-#{utils.uid()}"

CHROME_PATHS = [
    'chromium',
    'google-chrome',
    'chromium-browser',
    '/Applications/Chromium.app/Contents/MacOS/Chromium',
    '/Applications/Google Chrome.app/Contents/MacOS/Google Chrome'
]

CHROME_ARGS = [
    '--enable-logging=stderr',
    '--v=1',
    '--no-first-run',
    "--user-data-dir=#{USER_DATA_DIR}",
    '--no-sandbox' # TODO: run chrome on travis in sandbox mode
]

CHROME_LOG_RE = ///
    # declaration
    ^\[
        # pid, date, severity
        (?:[0-9A-Z/]*:){4}
        # type
        (.*?)
        \([0-9]*\)
    \]
    (?:.*)
    # content
    "(.*)"
    (?:.*)$
///

CHROME_CONSOLE_LOG_TYPE = 'CONSOLE'

###
Returns path to the Chrome or Chromium process available on the machine
###
getChromePath = (callback) ->
    utils.async.forEach CHROME_PATHS, (path, index, array, next) ->
        which path, (err, resolvedPath) ->
            if err
                next()
            else
                callback null, resolvedPath
    , ->
        callback new Error 'Cannot resolve Chrome path'

###
Returns JavaScript console logs from Chrome logs
###
getConsoleLogs = (stdout) ->
    lines = String(stdout).split '\n'
    logs = []
    for line in lines
        parsedLine = CHROME_LOG_RE.exec line
        unless parsedLine?
            continue
        [_, type, content] = parsedLine
        if type is CHROME_CONSOLE_LOG_TYPE
            logs.push content
    logs

###
Runs the given URI in Chrome
###
runOnPathWithUri = (path, uri, callback) ->
    args = utils.clone CHROME_ARGS
    args.push "--app=#{uri}"

    command = "#{path} #{args.join(' ')}"
    chrome = childProcess.exec command, env: config.getProcessEnv()

    chrome.on 'exit', ->
        fs.removeSync USER_DATA_DIR
        callback()

    chrome.stderr.on 'data', (data) ->
        for logMsg in getConsoleLogs(data)
            onLog.emit logMsg
        return

    process.once 'SIGINT', ->
        chrome.kill()

    onLog: onLog = signal.create()
    kill: ->
        chrome?.kill()

runAndLog = (path, callback) ->
    log.info "running #{path}"
    logsReader = new processLogs.LogsReader
    url = config.getBrowserHttpServerUrl()
    chromeProcess = runOnPathWithUri path, url, ->
        log.info "#{path} terminated"
        callback logsReader.error
    chromeProcess.onLog (msg) ->
        logsReader.log msg
        if logsReader.terminated
            chromeProcess.kill()

exports.run = (env, callback) ->
    if env.path
        return runAndLog env.path, callback
    getChromePath (err, path) ->
        if err
            return callback err
        runAndLog path, callback