index.js

var path = require('path')
var fsPath = require('fs-path')
var Xvfb = require('xvfb')

var serve = require('./serve.js')
var render = require('./render.js')

var xvfb = new Xvfb()

function StaticSiteGenerator (outputPath, routes, elementToWaitFor) {
    this.outputPath = outputPath
    this.routes = routes
    this.elementToWaitFor = elementToWaitFor || 'body'
}

StaticSiteGenerator.prototype.apply = function (compiler) {
    var self = this
    compiler.plugin('after-emit', function (compilation, done) {

        if (process.platform === 'linux'){
            xvfb.startSync()
        }
        var server = serve(self.outputPath)
        var port = server.address().port
        var outputFiles = render(port, self.routes, self.elementToWaitFor)

        outputFiles.then(files => {
            for (var i = 0; i < files.length; i++) {
                var outputFilePath = path.join(self.outputPath, self.routes[i])
                var outputFileName = path.join(outputFilePath, 'index.html')
                fsPath.writeFile(outputFileName, files[i])
            }
            server.close(function () {
                xvfb.stopSync()
                done()
            })
        }).catch(err => {
            setTimeout(function () {throw err})
            server.close(function () {
                xvfb.stopSync()
                done()
            })
        })

    })
}

module.exports = StaticSiteGenerator