kml-utils/src/parse/folder.js

This lib is modified from @mapbox/togeojson and @mapbox/tokml, enhanced with folder capability

var FOLDER_KEY_NAME = require('../constants').FOLDER_KEY_NAME
var domUtils = require('../utils/dom')
var parent = domUtils.parent
var nodeVal = domUtils.nodeVal
var get = domUtils.get
var get1 = domUtils.get1
var getChild1 = domUtils.getChild1
var attr = domUtils.attr
var setAttr = domUtils.setAttr

/**
 * returns folder tree: [{ key, parent, name, children }]
 * and set Folder element with key attribute
 */
function parse (kmlDocument, options) {
  var folderEls = []
  options.folderElements.forEach(function (a) {
    folderEls = folderEls.concat(Array.from(get(kmlDocument, a)))
  })

  if (folderEls.length === 0) {
    return []
  }

  var defaultFolderName = options.defaultFolderName
  var parentEl = parent(folderEls[0])

  var n = 0
  var process = function (folderEl) {
    var parentEl = parent(folderEl)
    var nameEl = getChild1(folderEl, 'name')
    var descEl = getChild1(folderEl, 'desc')
    var valueEl = getChild1(folderEl, 'value')

    var key = (n++) + ''
    setAttr(folderEl, FOLDER_KEY_NAME, key)

    var result = {
      key: key,
      parent: options.folderElements.some(function(a){ return a === parentEl.tagName }) ? attr(parentEl, FOLDER_KEY_NAME) : null,
      name: nameEl ? nodeVal(nameEl) : defaultFolderName,
      children: folderEls.filter(function (e) {
        return parent(e) === folderEl
      }).map(process)
    }
    if (descEl) {
      result.desc = nodeVal(descEl)
    }
    if (valueEl) {
      result.value = nodeVal(valueEl)
    }
    return result
  }

  return folderEls.filter(function (e) {
    return parent(e) === parentEl
  }).map(process)
}

exports.parse = parse