optimize-pages-tree.js

#!/usr/bin/env node

// TODO: this is currently just a helper script included from `fix-mkdocs-pages.sh`

const fs = require('fs')
const yaml = require('js-yaml')

function firstObjectKey(obj) {
  return Object.keys(obj)[0]
}

function dfs(node) {
  if (node instanceof Array) {
    return node.map(x => dfs(x)) // go deeper
  }

  if (node instanceof Object) {
    const key = firstObjectKey(node)
    const entry = node[key]

    if (entry instanceof Array) {
      if (entry.length === 1) {
        return entry[0] // the actual transformation happens here
      } else {
        node[key] = dfs(entry) // go deeper
      }
    }
  }

  return node
}

const data = fs.readFileSync('/dev/stdin', 'utf8')
const yamldoc = yaml.safeLoad(data)
const newdoc = dfs(yamldoc)

console.log(yaml.safeDump(newdoc))