index.js

const webExt = require('web-ext').default
const inquirer = require('inquirer')

const pluginName = 'web-ext-plugin'

class WebExtWebpackPlugin {
  constructor (CmdRunParams = {}, CmdRunOptions = {}, PluginOptions = {}) {
    this.logger = console.log.bind(console)
    this.CmdRunParams = { ...CmdRunParams, noReload: true, noInput: true }
    this.CmdRunOptions = { ...CmdRunOptions, shouldExitProgram: false }
    this.verbose = PluginOptions.verbose || false
    this.timeout = PluginOptions.timeout || 5000
  }

  androidSetup () {
    let shouldTimeoutReject = true
    this.androidSetupPromise = new Promise((resolve, reject) => {
      setTimeout(() => shouldTimeoutReject && reject(new Error('No android device found.')), this.timeout)
      webExt.util.logger.consoleStream.write = (...args) => {
        const raw = args[0].msg
        const message = (raw || '').trim().split('\n')
        if (message[0] !== 'Android devices found:') { return this.verbose && this.logger(raw) }
        shouldTimeoutReject = false
        const devices = message.slice(1).map(s => s.substr(3))
        if (devices.length === 1) {
          this.logger('Selecting the only device ' + devices[0])
          this.CmdRunParams.adbDevice = devices[0]
          return resolve()
        }
        inquirer.prompt({ type: 'list', choices: devices, name: 'device', message: 'Select one device to continue' }).then(answer => {
          webExt.util.logger.consoleStream.write = e => this.verbose && this.logger(e)
          this.CmdRunParams.adbDevice = answer.device
        }).then(resolve)
      }
    })
  }

  async androidQueryDevice () {
    if (this.CmdRunParams.adbDevice) return
    await webExt.cmd.run(this.CmdRunParams, this.CmdRunOptions).catch(() => {})
    await this.androidSetupPromise.catch(e => {
      this.error(e)
      throw e
    })
  }

  async watchRun () {
    this.watchMode = true
  }

  async afterEmit () {
    if (!this.watchMode) return

    if (this.runner) {
      this.logger('Webpack emitted, reloading...')
      let reloadOK = await Promise.race([this.runner.reloadAllExtensions().then(() => true).catch(this.error), new Promise(resolve => setTimeout(resolve, this.timeout, false))])
      if (reloadOK) {
        this.logger('Extension reloaded.')
        return
      }
    }

    const runner = await webExt.cmd.run(this.CmdRunParams, this.CmdRunOptions).catch(this.error)
    this.runner = runner

    // Fix web-ext's bad
    process.stdin.setRawMode(false)

    runner.registerCleanup(() => {
      this.runner = null
    })
  }

  apply (compiler) {
    const logger = compiler.getInfrastructureLogger(pluginName)
    this.logger = logger.info.bind(logger)
    this.error = logger.error.bind(logger)

    if (this.CmdRunParams.target === 'firefox-android' && !this.CmdRunParams.adbDevice) {
      this.androidSetup.bind(this)()
      this.androidQueryDeviceResult = this.androidQueryDevice.bind(this)()
      compiler.hooks.beforeCompile.tapPromise({ name: pluginName }, () => this.androidQueryDeviceResult)
    }

    compiler.hooks.done.tapPromise({ name: pluginName }, this.afterEmit.bind(this))
    compiler.hooks.watchRun.tapPromise({ name: pluginName }, this.watchRun.bind(this))
  }
}

module.exports = WebExtWebpackPlugin