rehype-urls/index.js

Rehype plugin to rewrite URLs of `href` and `src` attributes

var has = require('hast-util-has-property')
var url = require('url')
var opt = require('stdopt')
var visit = require('unist-util-visit')

module.exports = function transform (options) {
  if (typeof options === 'function') {
    options = { transform: options }
  }
  options.transform = options.transform || function () {}

  return function transformer (tree) {
    visit(tree, 'element', function (node) {
      modify(node, 'href')
      modify(node, 'src')
    })
  }

  function modify (node, prop) {
    if (has(node, prop)) {
      var obj = url.parse(node.properties[prop])
      var res = opt(options.transform(obj, node)).or(obj).value()
      node.properties[prop] = url.format(res)
    }
  }
}