gridsome-transformer-docutils/lib/index.js

Docutils transformer for Gridsome

const docutils = require('docutils')
const LRU = require('lru-cache')
const memoize = require('fast-memoize')

const { extractDocumentTitle } = require('./parserPlugins')

class DocutilsTransformer {
  static mimeTypes () {
    return ['application/xml']
  }

  constructor (options = {}, { localOptions = {} } = {}) {
    this.plugins = (options.plugins || []).concat(localOptions.plugins || [])

    this.getDocumentData = memoize(this.process.bind(this), {
      cache: {
        create: () => new LRU({ max: 1000 })
      }
    })
  }

  parse (content) {
    const { title } = this.getDocumentData(content)

    return {
      title,
      content
    }
  }

  extendNodeType ({ graphql: { GraphQLJSON } }) {
    return {
      root: {
        type: GraphQLJSON,
        resolve: node => this.getDocumentData(node.content).document
      }
    }
  }

  process (content) {
    let title = ''

    const document = docutils.parse(content, [
      extractDocumentTitle(documentTitle => {
        title = documentTitle
      }),
      ...this.plugins
    ])

    return {
      title,
      document
    }
  }
}

module.exports = DocutilsTransformer