www

commit cd432a1e2a103c98a4966c9f3e359e63ab1f3d8a
parent 97c3f13112c5eca1b69951c7b1989cac4905ca5c
Author: Dan Stillman <dstillman@zotero.org>
Date:   Tue, 29 Aug 2006 11:06:54 +0000

Add 'recursive' flag to Scholar.getCollections() to retrieve all collections in a flat ordered array


Diffstat:
M
chrome/chromeFiles/content/scholar/xpcom/data_access.js
 | 
31
+++++++++++++++++++++++++++++--

1 file changed, 29 insertions(+), 2 deletions(-)
diff --git a/chrome/chromeFiles/content/scholar/xpcom/data_access.js b/chrome/chromeFiles/content/scholar/xpcom/data_access.js
@@ -2642,8 +2642,22 @@ Scholar.Collection.prototype.getDescendents = function(nested, type){
 		+ ' UNION SELECT itemID AS id, 1 AS type, NULL AS collectionName '
 		+ 'FROM collectionItems WHERE collectionID=' + this._id);
 	
+	if (type){
+		switch (type){
+			case 'item':
+			case 'collection':
+				break;
+			default:
+				throw ("Invalid type '" + type + "' in Collection.getDescendents()");
+		}
+	}
+	
 	for(var i=0, len=children.length; i<len; i++){
-		switch (children[i]['type']){
+		// This seems to not work without parseInt() even though
+		// typeof children[i]['type'] == 'number' and
+		// children[i]['type'] === parseInt(children[i]['type']),
+		// which sure seems like a bug to me
+		switch (parseInt(children[i]['type'])){
 			case 0:
 				if (!type || type=='collection'){
 					toReturn.push({
@@ -3374,7 +3388,7 @@ Scholar.ItemFields = new function(){
  * Takes parent collectionID as optional parameter;
  * by default, returns root collections
  */
-Scholar.getCollections = function(parent){
+Scholar.getCollections = function(parent, recursive){
 	var toReturn = new Array();
 	
 	if (!parent){
@@ -3400,6 +3414,19 @@ Scholar.getCollections = function(parent){
 		}
 		
 		toReturn.push(obj);
+		
+		// If recursive, get descendents
+		if (recursive){
+			var desc = obj.getDescendents(false, 'collection');
+			for (var j in desc){
+				var obj2 = Scholar.Collections.get(desc[j]['id']);
+				if (!obj2){
+					throw ('Collection ' + desc[j] + ' not found');
+				}
+				
+				toReturn.push(obj2);
+			}
+		}
 	}
 	
 	return toReturn;