Expose stringify and parse

Author: typicode

README.md

@@ -69,6 +69,10 @@ low.mixin({
 var song = db('songs').second().value()
 ```
 
+__low.stringify(obj)__ and __low.parse(str)__
+
+Overwrite these methods to customize JSON stringifying and parsing.
+
 __db.object__
 
 Database object. Useful if you want to access the content of your JSON file and don't want to go through Lo-Dash methods.

package.json

@@ -7,7 +7,6 @@
     "file",
     "database",
     "JSON",
-    "JSONDatabase",
     "lo-dash",
     "lodash",
     "underscore",

src/index.js

@@ -1,5 +1,6 @@
+var fs = require('fs')
 var _ = require('lodash')
-var Store = require('./store')
+var Writer = require('./writer')
 
 // Compose every function with fn
 function composeAll(obj, fn) {
@@ -11,10 +12,14 @@ function composeAll(obj, fn) {
 }
 
 function Low(filename) {
-  // Create a new JSON file and get object
   if (filename) {
-    var store = new Store(filename)
-    var object = store.object
+    if (fs.existsSync(filename)) {
+      var object = low.parse(fs.readFileSync(filename))
+    } else {
+      fs.writeFileSync(filename, '{}')
+      var object = {}
+    }
+    var writer = new Writer(filename)
   } else {
     var object = {}
   }
@@ -23,18 +28,19 @@ function Low(filename) {
     // Create empty array if it doesn't exist
     object[name] = object[name] || []
 
-    if (filename) store.save()
+    // Save it
+    if (filename) chain.save()
 
-    // Chain array
+    // Create a Lo-Dash chained array
     var chainedArray = _.chain(object[name])
 
     // Hack, wrap every Lo-Dash function to call save.
     // Save is however throttled and only happens if database object
     // has changed.
-    // With Node 0.12 and Object.observe support, this bit of code should
+    // With Node 0.12 and Object.observe support, this bit of code should be
     // removed :)
     if (filename) composeAll(chainedArray, function(arg) {
-      store.save()
+      chain.save()
       return arg
     })
 
@@ -45,9 +51,9 @@ function Low(filename) {
   chain.object = object
 
   // Call it to manually save database
-  chain.save = function() {
-    if (store) store.save()
-  }
+  chain.save = _.throttle(function() {
+    if (filename) writer.write(low.stringify(object))
+  }, 10)
 
   return chain
 }
@@ -60,4 +66,12 @@ low.mixin = function(source) {
   _.mixin(source)
 }
 
-module.exports = low
+low.stringify = function(obj) {
+  return JSON.stringify(obj, null, 2)
+}
+
+low.parse = function(str) {
+  return JSON.parse(str)
+}
+
+module.exports = low

src/store.js

@@ -66,7 +66,6 @@ describe('LowDB', function() {
     beforeEach(function() {
       fs.writeFileSync(dbPath, JSON.stringify({ foo: { a: 1 } }))
       db = low(dbPath)
-      db('foo').push({ a: 1 })
     })
 
     it('loads automatically file', function() {
@@ -75,7 +74,6 @@ describe('LowDB', function() {
 
   })
 
-
   describe('In-memory', function() {
 
     beforeEach(function() {
@@ -112,17 +110,15 @@ describe('LowDB', function() {
 
     beforeEach(function() {
       db = low(dbPath)
-    })
-
-    it('adds functions', function(done) {
       low.mixin({
         hello: function(array, word) {
           array.push('hello ' + word)
         }
       })
+    })
 
+    it('adds functions', function(done) {
       db('foo').hello('world')
-
       setTimeout(function() {
         assert.deepEqual(JSON.parse(fs.readFileSync(dbPath)), { foo: [ 'hello world' ] })
         done()
@@ -131,6 +127,34 @@ describe('LowDB', function() {
 
   })
 
+  describe('stringify and parse', function() {
+
+    var stringify = low.stringify
+    var parse = low.parse
+
+    beforeEach(function() {
+      low.stringify = function() { return '{ foo: [] }' }
+      low.parse = function() { return { bar: [] } }
+      fs.writeFileSync(dbPath, '{}')
+      db = low(dbPath)
+    })
+
+    afterEach(function() {
+      low.stringify = stringify
+      low.parse = parse
+    })
+
+    it('can be overriden', function(done) {
+      assert.deepEqual(db.object, { bar: [] })
+      db.save() // will stringify object
+      setTimeout(function() {
+        assert.equal(fs.readFileSync(dbPath, 'utf-8'), '{ foo: [] }')
+        done()
+      }, 100)
+    })
+
+  })
+
   describe('underscore-db', function() {
 
     beforeEach(function() {
@@ -145,4 +169,3 @@ describe('LowDB', function() {
 
   })
 })
-