From d04935e340d14a4d930df39b59f9997541727745 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ionic=C4=83=20Biz=C4=83u?= <bizauionica@gmail.com>
Date: Sun, 13 Sep 2015 17:44:40 +0300
Subject: [PATCH] Handle the input data in record method

---
 lib/index.js | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/lib/index.js b/lib/index.js
index 3e48d8b..1f65cf0 100644
--- a/lib/index.js
+++ b/lib/index.js
@@ -109,8 +109,7 @@ GitStats.prototype.record = function (data, callback) {
         delete data.url;
     }
 
-    // Get stats
-    self.get(function (err, stats) {
+    function modify (err, stats) {
 
         var commits = stats.commits
           , day = data.date.format(DATE_FORMAT)
@@ -119,8 +118,22 @@ GitStats.prototype.record = function (data, callback) {
 
         today[data.hash] = 1;
 
-        self.save(stats, callback);
-    });
+        if (data.save === false) {
+            callback(null, stats);
+        } else {
+            self.save(stats, callback);
+        }
+
+        return stats;
+    }
+
+    // Check if we have input data
+    if (data._data) {
+        return modify(null, data._data);
+    } else {
+        // Get stats
+        self.get(modify);
+    }
 
     return self;
 };