107 add transformfirst option

.gitignore

@@ -1,2 +1,3 @@
 node_modules
-.idea
+.idea
+*.iml

README.md

@@ -161,7 +161,7 @@ flatten({
 
 ### transformKey
 
-Transform each part of a flat key before and after flattening.
+Transform each part of a flat key before and after flattening. 
 
 ```javascript
 var flatten = require('flat')
@@ -208,6 +208,59 @@ unflatten({
 // }
 ```
 
+### transformFirst
+
+Transforms the top-level key before and after flattening. It has two arguments:
+
+1. The key to transform
+1. An optional delimiter. This is not required, but allows the transformation to
+add a new key to the transformation.
+
+```javascript
+var flatten = require('flat')
+var unflatten = require('flat').unflatten
+
+flatten({
+    key1: {
+        keyA: 'valueI'
+    },
+    key2: {
+        keyB: 'valueII'
+    },
+    key3: { a: { b: { c: 2 } } }
+}, {
+    transformFirst: function(key, delim){
+      return 'foo' + delim + key;
+    }
+})
+
+// {
+//   'foo.key1.keyA': 'valueI',
+//   'foo.key2.keyB': 'valueII',
+//   'foo.key3.a.b.c': 2
+// }
+
+unflatten({
+      'foo.key1.keyA': 'valueI',
+      'foo.key2.keyB': 'valueII',
+      'foo.key3.a.b.c': 2
+}, {
+    transformFirst: function(key, delim){
+      return key.substring((key+delim).length)
+    }
+})
+
+// {
+//     key1: {
+//         keyA: 'valueI'
+//     },
+//     key2: {
+//         keyB: 'valueII'
+//     },
+//     key3: { a: { b: { c: 2 } } }
+// }
+```
+
 ## Command Line Usage
 
 `flat` is also available as a command line tool. You can run it with 

index.js

@@ -14,6 +14,7 @@ function flatten (target, opts) {
   const delimiter = opts.delimiter || '.'
   const maxDepth = opts.maxDepth
   const transformKey = opts.transformKey || keyIdentity
+  const transformFirst = opts.transformFirst || keyIdentity
   const output = {}
 
   function step (object, prev, currentDepth) {
@@ -30,18 +31,18 @@ function flatten (target, opts) {
 
       const newKey = prev
         ? prev + delimiter + transformKey(key)
-        : transformKey(key)
+        : transformFirst(transformKey(key), delimiter)
 
       if (!isarray && !isbuffer && isobject && Object.keys(value).length &&
         (!opts.maxDepth || currentDepth < maxDepth)) {
-        return step(value, newKey, currentDepth + 1)
+        return step(value, newKey, currentDepth + 1, false)
       }
 
       output[newKey] = value
     })
   }
 
-  step(target)
+  step(target, false, 1)
 
   return output
 }
@@ -52,6 +53,7 @@ function unflatten (target, opts) {
   const delimiter = opts.delimiter || '.'
   const overwrite = opts.overwrite || false
   const transformKey = opts.transformKey || keyIdentity
+  const transformFirst = opts.transformFirst || keyIdentity
   const result = {}
 
   const isbuffer = isBuffer(target)
@@ -110,7 +112,7 @@ function unflatten (target, opts) {
   }, {})
 
   Object.keys(target).forEach(function (key) {
-    const split = key.split(delimiter).map(transformKey)
+    const split = transformFirst(key, delimiter).split(delimiter).map(transformKey)
     let key1 = getkey(split.shift())
     let key2 = getkey(split[0])
     let recipient = result