sanetize-mongo.ts

import { isArray, isObject, mapValues } from 'lodash-es'
import type { Document } from 'mongodb'

const mapValuesDeep = <T>(obj: T, cb: any): unknown => {
  if (isArray(obj)) {
    return obj.map(innerObj => mapValuesDeep(innerObj, cb))
  } else if (isObject(obj)) {
    if (!('toHexString' in obj || 'getDay' in obj)) {
      return mapValues(obj, val => mapValuesDeep(val, cb))
    } else {
      return cb(obj)
    }
  } else {
    return cb(obj)
  }
}

const sanetizeMongo = <T extends Document | Document[] | null>(
  document: T,
): T => {
  return mapValuesDeep(document, (e: any) =>
    e.toHexString ? e.toString() : e,
  ) as T
}