server.js

'use strict'

// The server where the site is exposed through a static file server
// while developing locally.

const fs = require('fs')
const http = require('http')
const path = require('path')
const chokidar = require('chokidar')
const junk = require('junk')
const st = require('st')
const build = require('./build')

const mount = st({
  path: path.join(__dirname, 'build'),
  cache: false,
  index: 'index.html',
  passthrough: true
})

const port = process.env.PORT || 8080
const selectedLocales = process.env.DEFAULT_LOCALE ? process.env.DEFAULT_LOCALE.toLowerCase().split(',') : process.env.DEFAULT_LOCALE
const preserveLocale = process.argv.includes('--preserveLocale')
const serveOnly = process.argv.includes('--serve-only')

// Watches for file changes in the locale, layout and static directories, and
// rebuilds the modified one.
const opts = {
  ignoreInitial: true,
  usePolling: true
}
const locales = chokidar.watch(path.join(__dirname, 'locale'), opts)
const css = chokidar.watch(path.join(__dirname, 'layouts/css/**/*.scss'), opts)
const layouts = chokidar.watch(path.join(__dirname, 'layouts/**/*.hbs'), opts)
const staticFiles = chokidar.watch(path.join(__dirname, 'static'), opts)

// Gets the locale name by path.
function getLocale (filePath) {
  const pre = path.join(__dirname, 'locale')
  return filePath.slice(pre.length + 1, filePath.indexOf(path.sep, pre.length + 1))
}

// This function has two meanings:
// 1. Build for the specific language.
// 2. Choose what languages for the menu.
function dynamicallyBuildOnLanguages (source, locale) {
  if (!selectedLocales || selectedLocales.length === 0) {
    fs.readdir(path.join(__dirname, 'locale'), (err, locales) => {
      if (err) {
        throw err
      }

      const filteredLocales = locales.filter(file => junk.not(file))
      const localesData = build.generateLocalesData(filteredLocales)
      build.buildLocale(source, locale, { preserveLocale, localesData })
    })
  } else {
    const localesData = build.generateLocalesData(selectedLocales)
    build.buildLocale(source, locale, { preserveLocale, localesData })
  }
}

build.getSource((err, source) => {
  if (err) {
    throw err
  }

  locales.on('change', (filePath) => {
    const locale = getLocale(filePath)

    if (!selectedLocales || selectedLocales.includes(locale)) {
      console.log(`The language ${locale} is changed, '${filePath}' is modified.`)
      dynamicallyBuildOnLanguages(source, locale)
    }
  })

  locales.on('add', (filePath) => {
    const locale = getLocale(filePath)

    if (!selectedLocales || selectedLocales.includes(locale)) {
      console.log(`The language ${locale} is changed, '${filePath}' is added.`)
      dynamicallyBuildOnLanguages(source, locale)
      locales.add(filePath)
    }
  })
})

css.on('change', () => build.buildCSS())
css.on('add', (filePath) => {
  css.add(filePath)
  build.buildCSS()
})

layouts.on('change', () => build.fullBuild({ selectedLocales, preserveLocale }))
layouts.on('add', (filePath) => {
  layouts.add(filePath)
  build.fullBuild({ selectedLocales, preserveLocale })
})

staticFiles.on('change', build.copyStatic)
staticFiles.on('add', (filePath) => {
  staticFiles.add(filePath)
  build.copyStatic()
})

const mainLocale = (selectedLocales && selectedLocales[0]) || 'en'

// Initializes the server and mounts it in the generated build directory.
http.createServer((req, res) => {
  // If we are accessing the root, it should be redirected to the default language,
  // We shouldn't get a 404 page.

  if (req.url === '/') {
    req.url = `/${mainLocale}`
  }
  mount(req, res)
}).listen(port, () => {
  console.log(`\x1B[32mServer running at http://localhost:${port}/${mainLocale}/\x1B[39m`)
})

if (!serveOnly) {
  // Start the initial build of static HTML pages
  build.copyStatic()
  build.buildCSS()
  build.fullBuild({ selectedLocales, preserveLocale })
}