const path = require('path')
const Application = require('spectron').Application
const fs = require('fs-extra')

const TEST_DATA_DIR = path.join(__dirname, 'tempTestData')

module.exports = {
  TEST_DATA_DIR,
  createApp,
  endTest,
  screenshotCreateOrCompare,
  waitForLoad,
  wait
}

// Runs WebTorrent Desktop.
// Returns a promise that resolves to a Spectron Application once the app has loaded.
// Takes a Tape test. Makes some basic assertions to verify that the app loaded correctly.
function createApp (t) {
  return new Application({
    path: path.join(__dirname, '..', 'node_modules', '.bin',
      'electron' + (process.platform === 'win32' ? '.cmd' : '')),
    args: [path.join(__dirname, '..')],
    env: {NODE_ENV: 'test'}
  })
}

// Starts the app, waits for it to load, returns a promise
function waitForLoad (app, t) {
  return app.start().then(function () {
    return app.client.windowByIndex(1)
  }).then(function () {
    return app.client.waitUntilWindowLoaded()
  }).then(function () {
    return app.webContents.getTitle()
  }).then(function (title) {
    t.equal(title, 'WebTorrent Desktop', 'app title')
  })
}

function wait (ms) {
  if (ms === undefined) ms = 500 // Default: wait long enough for the UI to update
  return new Promise(function (resolve, reject) {
    setTimeout(resolve, ms)
  })
}

function endTest (app, t, err) {
  return app.stop().then(function () {
    t.end(err)
  })
}

function screenshotCreateOrCompare (app, t, name) {
  const ssPath = path.join(__dirname, 'screenshots', process.platform, name + '.png')
  console.log('Capturing ' + ssPath)
  fs.ensureFileSync(ssPath)
  const ssBuf = fs.readFileSync(ssPath)
  return app.browserWindow.capturePage().then(function (buffer) {
    if (ssBuf.length === 0) {
      console.log('Saving screenshot ' + ssPath)
      fs.writeFileSync(ssPath, buffer)
    } else if (Buffer.compare(buffer, ssBuf) !== 0) {
      return Promise.reject('Screenshot didn\'t match: ' + ssPath)
    } else {
      return Promise.resolve()
    }
  })
}