hexo-theme-hyonline/scripts/lib/helpers/toc.js

Futuristic online store theme for Hexo.

'use strict'

const { tocObj, escapeHTML, encodeURL } = require('hexo-util')

module.exports = getToc

function getToc(str, options = {}) {
  options = Object.assign(
    {
      min_depth: 1,
      max_depth: 6,
      class: 'toc',
      list_number: false
    },
    options
  )

  const data = tocObj(str, {
    min_depth: options.min_depth,
    max_depth: options.max_depth
  })

  if (!data.length) return ''

  const className = escapeHTML(options.class)
  const listNumber = options.list_number

  let result = `<ol class="${className}">`

  const lastNumber = [0, 0, 0, 0, 0, 0]
  let firstLevel = 0
  let lastLevel = 0

  for (let i = 0, len = data.length; i < len; i++) {
    const el = data[i]
    const { level, id, text } = el
    const href = id ? `#${encodeURL(id)}` : null

    lastNumber[level - 1]++

    for (let i = level; i <= 5; i++) {
      lastNumber[i] = 0
    }

    if (firstLevel) {
      for (let i = level; i < lastLevel; i++) {
        result += '</li></ol>'
      }

      if (level > lastLevel) {
        result += `<ol class="${className}-child">`
      } else {
        result += '</li>'
      }
    } else {
      firstLevel = level
    }

    result += `<li class="${className}-item ${className}-level-${level}">`
    if (href) {
      result += `<a class="${className}-link" href="${href}">`
    } else {
      result += `<a class="${className}-link">`
    }

    if (listNumber) {
      result += `<span class="${className}-number">`

      for (let i = firstLevel - 1; i < level; i++) {
        result += `${lastNumber[i]}.`
      }

      result += '</span> '
    }

    result += `<span class="${className}-text">${text}</span></a>`

    lastLevel = level
  }

  for (let i = firstLevel - 1; i < lastLevel; i++) {
    result += '</li></ol>'
  }

  return result
}

module.exports = getToc