OpenStreetBrowser/src/twigFunctions.js

var OverpassLayer = require('overpass-layer')
var OpeningHours = require('opening_hours')
var colorInterpolate = require('color-interpolate')
var osmParseDate = require('openstreetmap-date-parser')
var osmFormatDate = require('openstreetmap-date-format')
const natsort = require('natsort').default
const md5 = require('md5')
const yaml = require('js-yaml')

var md5cache = {}

OverpassLayer.twig.extendFunction('tagsPrefix', function (tags, prefix) {
  var ret = {}
  var count = 0

  for (var k in tags) {
    if (k.substr(0, prefix.length) === prefix) {
      ret[k.substr(prefix.length)] = k
      count++
    }
  }

  if (count === 0) {
    return null
  }

  return ret
})

OverpassLayer.twig.extendFunction('openingHoursState', function (openingHours) {
  try {
    var oh = new OpeningHours(openingHours)
  } catch (err) {
    console.log('Error in opening_hours: ' + err)
    return 'unknown'
  }

  return oh.getStateString(new Date(), true)
})
OverpassLayer.twig.extendFilter('websiteUrl', function (value) {
  if (value.match(/^https?:\/\//)) {
    return value
  }

  return 'http://' + value
})
OverpassLayer.twig.extendFilter('matches', function (value, param) {
  if (value === null || typeof value === 'undefined') {
    return false
  }

  if (!param.length) {
    throw new Error("Filter 'matches' needs a parameter!")
  }

  const r = new RegExp(...param)
  return value.toString().match(r)
})
OverpassLayer.twig.extendFilter('natsort', function (values, options) {
  return values.sort(natsort(options))
})
OverpassLayer.twig.extendFilter('unique', function (values, options) {
  // source: https://stackoverflow.com/a/14438954
  function onlyUnique (value, index, self) {
    return self.indexOf(value) === index
  }
  return values.filter(onlyUnique)
})
OverpassLayer.twig.extendFunction('colorInterpolate', function (map, value) {
  var colormap = colorInterpolate(map)
  return colormap(value)
})
OverpassLayer.twig.extendFilter('osmParseDate', function (value) {
  return osmParseDate(value)
})
OverpassLayer.twig.extendFilter('osmFormatDate', function (value, param) {
  return osmFormatDate(value, param.length ? param[0] : {})
})
OverpassLayer.twig.extendFilter('md5', function (value) {
  if (!(value in md5cache)) {
    md5cache[value] = md5(value)
  }

  return md5cache[value]
})
OverpassLayer.twig.extendFunction('evaluate', function (tags) {
  const ob = {
    id: 'x0',
    meta: {},
    tags: tags,
    type: 'special'
  }

  return global.currentCategory.layer.mainlayer.evaluate(ob)
})
function enumerate (list) {
  if (!list) {
    return ''
  }

  if (typeof list === 'string') {
    list = list.split(/;/g)
  }

  if (list.length > 2) {
    let result = lang_str.enumerate_start.replace('{0}', list[0]).replace('{1}', list[1])

    for (let i = 2; i < list.length - 1; i++) {
      result = lang_str.enumerate_middle.replace('{0}', result).replace('{1}', list[i])
    }

    return lang_str.enumerate_end.replace('{0}', result).replace('{1}', list[list.length - 1])
  }
  else if (list.length == 2) {
    return lang_str.enumerate_2.replace('{0}', list[0]).replace('{1}', list[1])
  }
  else if (list.length > 0) {
    return list[0]
  }

  return ''
}
OverpassLayer.twig.extendFunction('enumerate', (list) => enumerate(list))
OverpassLayer.twig.extendFilter('enumerate', (list) => enumerate(list))
OverpassLayer.twig.extendFilter('ksort', (list) => {
  if (Array.isArray(list)) {
    return list
  }

  let keys = list._keys || Object.keys(list)
  keys.sort()
  let result = Object.assign({}, list)
  result._keys = keys
  return result
})
OverpassLayer.twig.extendFunction('debug', function () {
  console.log.apply(null, arguments)
})
OverpassLayer.twig.extendFilter('debug', function (value, param) {
  if (param) {
    console.log.apply(null, [value, ...param])
  } else {
    console.log(value)
  }
  return value
})
OverpassLayer.twig.extendFilter('json_pp', function (value, param) {
  const options = param[0] || {}

  if (value === 'undefined') {
    return 'null'
  }
  throw new Error('foo')

  value = twigClear(value)

  return JSON.stringify(value, null, 'indent' in options ? ' '.repeat(options.indent) : '  ')
})
OverpassLayer.twig.extendFilter('yaml', function (value, param) {
  const options = param[0] || {}

  value = twigClear(value)

  return yaml.dump(value, options)
})

function twigClear (value) {
  if (value === null || typeof value !== 'object') {
    return value
  }

  const v = {}
  for (let k in value) {
    if (k !== '_keys') {
      v[k] = value[k]
    }
  }

  return v
}