wikijs/server/tasks/workers/render-page.js

const _ = require('lodash')
const cheerio = require('cheerio')

module.exports = async ({ payload }) => {
  WIKI.logger.info(`Rendering page ${payload.id}...`)

  try {
    await WIKI.ensureDb()

    const page = await WIKI.db.pages.getPageFromDb(payload.id)
    if (!page) {
      throw new Error('Invalid Page Id')
    }

    const site = await WIKI.db.sites.query().findById(page.siteId)

    await WIKI.db.renderers.fetchDefinitions()

    const pipeline = await WIKI.db.renderers.getRenderingPipeline(page.contentType)

    let output = page.content

    if (_.isEmpty(page.content)) {
      WIKI.logger.warn(`Failed to render page ID ${payload.id} because content was empty: [ FAILED ]`)
    }

    for (let core of pipeline) {
      const renderer = require(`../../modules/rendering/${core.key}/renderer.js`)
      output = await renderer.render.call({
        config: core.config,
        children: core.children,
        page,
        site,
        input: output
      })
    }

    // Parse TOC
    const $ = cheerio.load(output)
    let isStrict = $('h1').length > 0 // <- Allows for documents using H2 as top level
    let toc = { root: [] }

    $('h1,h2,h3,h4,h5,h6').each((idx, el) => {
      const depth = _.toSafeInteger(el.name.substring(1)) - (isStrict ? 1 : 2)
      let leafPathError = false

      const leafPath = _.reduce(_.times(depth), (curPath, curIdx) => {
        if (_.has(toc, curPath)) {
          const lastLeafIdx = _.get(toc, curPath).length - 1
          if (lastLeafIdx >= 0) {
            curPath = `${curPath}[${lastLeafIdx}].children`
          } else {
            leafPathError = true
          }
        }
        return curPath
      }, 'root')

      if (leafPathError) { return }

      const leafSlug = $('.toc-anchor', el).first().attr('href')
      $('.toc-anchor', el).remove()

      _.get(toc, leafPath).push({
        title: _.trim($(el).text()),
        anchor: leafSlug,
        children: []
      })
    })

    // Save to DB
    await WIKI.db.pages.query()
      .patch({
        render: output,
        toc: JSON.stringify(toc.root)
      })
      .where('id', payload.id)

    // Save to cache
    // await WIKI.db.pages.savePageToCache({
    //   ...page,
    //   render: output,
    //   toc: JSON.stringify(toc.root)
    // })

    WIKI.logger.info(`Rendered page ${payload.id}: [ COMPLETED ]`)
  } catch (err) {
    WIKI.logger.error(`Rendering page ${payload.id}: [ FAILED ]`)
    WIKI.logger.error(err.message)
    throw err
  }
}