'use strict'

var jsYAML = require('js-yaml')

module.exports = yamlConfig

var origin = 'remark-yaml-config:invalid-options'

// Modify remark to read configuration from comments.
function yamlConfig() {
  var Parser = this.Parser
  var Compiler = this.Compiler
  var parser = Parser && Parser.prototype.blockTokenizers
  var compiler = Compiler && Compiler.prototype.visitors

  if (parser && parser.yamlFrontMatter) {
    parser.yamlFrontMatter = factory(parser.yamlFrontMatter)
  }

  if (compiler && compiler.yaml) {
    compiler.yaml = factory(compiler.yaml)
  }
}

// Wrapper factory.
function factory(original) {
  replacement.locator = original.locator

  return replacement

  // Replacer for tokeniser or visitor.
  function replacement(node) {
    var self = this
    var result = original.apply(self, arguments)
    var marker = result && result.type ? result : node
    var data

    try {
      data = jsYAML.safeLoad(marker.value)
      data = data && data.remark

      if (data) {
        self.setOptions(data)
      }
    } catch (error) {
      self.file.fail(error.message, marker, origin)
    }

    return result
  }
}

'use strict'

var yaml = require('js-yaml')

module.exports = yamlConfig

var origin = 'remark-yaml-config:invalid-options'

// Modify remark to read configuration from comments.
function yamlConfig() {
  var Parser = this.Parser
  var Compiler = this.Compiler
  var parser = Parser && Parser.prototype.blockTokenizers
  var compiler = Compiler && Compiler.prototype.visitors

  if (parser && parser.yamlFrontMatter) {
    parser.yamlFrontMatter = factory(parser.yamlFrontMatter)
  }

  if (compiler && compiler.yaml) {
    compiler.yaml = factory(compiler.yaml)
  }
}

// Wrapper factory.
function factory(original) {
  replacement.locator = original.locator

  return replacement

  // Replacer for tokeniser or visitor.
  function replacement(node) {
    var self = this
    var result = original.apply(self, arguments)
    var marker = result && result.type ? result : node
    var data

    try {
      data = yaml.safeLoad(marker.value)
      data = data && data.remark

      if (data) {
        self.setOptions(data)
      }
    } catch (error) {
      self.file.fail(error.message, marker, origin)
    }

    return result
  }
}

Refactor code-style