Installed underscore library and changed the findCode function to return full snippit object and not just snippits

Author: smeidan

lib/controllers/files.js

@@ -7,6 +7,7 @@ var codeParser = require('../../codeSnippitExtractor.js');
 var async = require('async');
 var esprima = require('esprima');
 var textSearch = require('mongoose-text-search');
+var uu = require('underscore');
 
 var extractDependencies = function(content) {
 
@@ -75,17 +76,27 @@ exports.findCode = function(req, res, next) {
     var snippIterator = function (doc, callback) {
       var docContent = doc.obj.contents;
       var snippit = codeParser(docContent, libFunction);
-      callback(null, snippit);
+      var resultsArr = [];
+      snippit.forEach(function(snippit) {
+        console.log(snippit.length);
+        var snippitObj = {
+          snippit: snippit,
+          docContent: docContent,
+          repoUrl: '(placeholder URL)'
+        };
+        resultsArr.push(snippitObj);
+      })
+      callback(null, resultsArr);
     };
 
     async.map(files.results, snippIterator, function(err, snippitsArr) {
       var returnArr = [];
-
-      snippitsArr.forEach(function(snippitsArr) {
-        if ( snippitsArr ) {
-          snippitsArr.forEach(function(snippit) {
-            returnArr.push(snippit);
-          });
+      snippitsArr = uu.flatten(snippitsArr);
+      snippitsArr.forEach(function(snippitObj) {
+        if ( snippitObj.snippit ) {
+          // snippitsArr.forEach(function(snippit) {
+            returnArr.push(snippitObj);
+          // });
         }
       });
 

package.json

@@ -4,6 +4,7 @@
   "dependencies": {
     "async": "^0.9.0",
     "body-parser": "~1.0.0",
+    "cheerio": "~0.17.0",
     "compression": "~1.0.1",
     "connect-mongo": "~0.4.1",
     "cookie-parser": "~1.0.1",
@@ -21,7 +22,7 @@
     "passport-local": "~0.1.6",
     "request": "^2.36.0",
     "static-favicon": "~1.0.1",
-    "cheerio": "~0.17.0"
+    "underscore": "^1.6.0"
   },
   "devDependencies": {
     "grunt": "~0.4.1",