chentsulin/koa-snake-res

'use strict'

var snakeCase = require('snake-case')

module.exports = function() {
  return function *(next) {
    yield next

    if (this.response.is('json')) {
      this.body = toSnakeCase(this.body)
    }
  }
}

function toSnakeCase(json) {
  for (var key in json) {
    if (json.hasOwnProperty(key)) {
      if (json[key] && typeof json[key] === 'object') {
        json[key] = toSnakeCase(json[key])
      }
      var _value = json[key]
      delete json[key]
      json[ snakeCase(key) ] = _value
    }
  }

  return json
}