markup.js

'use strict'

let marked = require('marked')
let hljs = require('highlight.js')
let cheerio = require('cheerio')

marked.setOptions({
  highlight: (code, lang) => {
    return hljs.highlightAuto(code).value
  }
})

marked.setOptions({
  highlight: (code, lang) => {
    return hljs.highlightAuto(code).value
  }
})

module.exports = (doc) => {
  doc.data.forEach(document => {
    console.log(`Generating markup for ${document.id}`)

    let description = document.attributes.description

    if (description) {
      document.attributes.description = highlight(description)
    }

    replaceDescriptionFor(document.attributes.methods)
    replaceDescriptionFor(document.attributes.properties)
    replaceDescriptionFor(document.attributes.events)
  })

  return doc
}

function replaceDescriptionFor (items) {
  if (items) {
    items.forEach(item => {
      let itemDescription = item.description
      if (itemDescription) {
        item.description = highlight(itemDescription)
      }
    })
  }
}

function highlight (description) {
  let markedup = removeHLJSPrefix(marked(description))
  let $ = cheerio.load(markedup)

  let codeBlocks = $('pre code')

  codeBlocks.each((i, el) => {
    let element = $(el)
    let klass = element.attr('class')
    let lang = ''
    let tableHeader = ''
    if (klass) {
      let type = klass.split('-').pop()
      if (isFile(type)) {
        tableHeader = `
          <thead>
            <tr>
              <td colspan="2">${type}</td>
            </tr>
          </thead>`
      }
      lang = determineLanguage(type)
    }
    let lines = element.html().split('\n')

    // get rid of empty blank line
    if (lines[lines.length - 1].trim() === '') {
      lines.pop()
    }

    let wrappedLines = `<pre>${lines.join('\n')}</pre>`
    let lineNumbers = lines.map((_, i) => (i + 1) + '\n').join('')

    element.parent().after(`<div class="highlight ${lang}">
      <div class="ribbon"></div>
      <div class="scroller">
        <table class="CodeRay">${tableHeader}
          <tbody>
            <tr>
              <td class="line-numbers"><pre>${lineNumbers}</pre></td>
              <td class="code">${wrappedLines}</td>
            </tr>
          </tbody>
        </table>
      </div>
    </div>
    `)

    element.parent().remove()
  })

  return $.html()
}

function determineLanguage (lang) {
  switch (lang) {
    case 'js':
    case 'javascript':
      return 'javascript'
    default:
      return lang
  }
}

function removeHLJSPrefix (string) {
  return string.replace(/hljs-/g, '')
}

function isFile (string) {
  return /\./.test(string)
}