Add column name transform

porsager · porsager · commit 346bf3ff694f · 2019-11-25T09:21:18.000+01:00
diff --git a/lib/index.js b/lib/index.js
@@ -8,10 +8,19 @@ import {
   arraySerializer,
   arrayParser,
   inferType,
+  toPascal,
+  toCamel,
+  toKebab,
   errors,
   types
 } from './types.js'
 
+Object.assign(Postgres, {
+  toPascal,
+  toCamel,
+  toKebab
+})
+
 export default function Postgres(url, options) {
   options = parseOptions(url, options)
 
diff --git a/lib/types.js b/lib/types.js
@@ -145,6 +145,22 @@ function arrayParserLoop(s, x, parser) {
   return xs
 }
 
+export const toCamel = x => {
+  let str = x[0]
+  for (let i = 1; i < x.length; i++)
+    str += x[i] === '_' ? x[++i].toUpperCase() : x[i]
+  return str
+}
+
+export const toPascal = x => {
+  let str = x[0].toUpperCase()
+  for (let i = 1; i < x.length; i++)
+    str += x[i] === '_' ? x[++i].toUpperCase() : x[i]
+  return str
+}
+
+export const toKebab = x => x.replace(/_/g, '-')
+
 export const errors = {
   connection: (x, options) => Object.assign(
     new Error('write CONNECTION_' + x + ' ' + options.path || (options.host + ':' + options.port)),
