Added frame support for multi sites

Author: rafacesar

lib/simplehar.js

@@ -29,9 +29,10 @@ var fs = require('fs'),
 								}
 
 								if(args.return) {
-									if(ilen > 1 && total.push(html) === ilen) {
-										args.callback(total);
-									}
+									if(ilen > 1 && total.push(html) === ilen)
+										args.callback(
+											simplehar.multiFrame(total, args.frame, true)
+										);
 									else if(ilen < 2)
 										args.callback(html);
 								}
@@ -98,7 +99,7 @@ var fs = require('fs'),
 					fs.writeFileSync(filename, html.html || html);
 			}
 			if(args.return)
-				return htmls;
+				return simplehar.multiFrame(htmls, args.frame, ilen > 1);
 
 		}
 
@@ -360,6 +361,32 @@ simplehar.makeFrame = function(html, args) {
 
 };
 
+//Return multiple sites frames
+simplehar.multiFrame = function(htmls, frame, multi) {
+	'use strict';
+	
+	var i = 0,
+		ilen = htmls.length,
+		result;
+	
+	if((htmls[0].js || htmls[0].css) && frame && multi) {
+		result = {html:[]};
+		for(;i<ilen;i++)
+			result.html.push(htmls[i].html);
+		
+		if(htmls[0].js)
+			result.js = htmls[0].js;
+		
+		if(htmls[0].css)
+			result.css = htmls[0].css;
+		
+		return result;
+	}
+	
+	return htmls;
+	
+};
+
 //Copy the source to simpleharSrc folder
 simplehar.copySrc = function(dir) {
 	'use strict';