/* jQuery treeTable Plugin 2.1 - http://ludo.cubicphuse.nl/jquery-plugins/treeTable/ */
(function($) {
	// Helps to make options available to all functions
	// TODO: This gives problems when there are both expandable and non-expandable
	// trees on a page. The options shouldn't be global to all these instances!
	var options;

	$.fn.treeTable = function(opts) {
		options = $.extend({}, $.fn.treeTable.defaults, opts);

		return this.each(function() {
			$(this).addClass("treeTable").find("tbody tr").each(function() {
				// Initialize root nodes only whenever possible
				if(!options.expandable || $(this)[0].className.search("child-of-") == -1) {
					initialize($(this));
				}
			});
		});
	};

	$.fn.treeTable.defaults = {
		childPrefix: "child-of-",
		expandable: true,
		indent: 19,
		initialState: "collapsed",
		treeColumn: 0
	};

	// Recursively hide all node's children in a tree
	$.fn.collapse = function() {
		$(this).removeClass('expanded').addClass("collapsed");

		childrenOf($(this)).each(function() {
			if(!$(this).hasClass("collapsed")) {
				$(this).collapse();
			}

			$(this).hide();
		});

		return this;
	};

	// Recursively show all node's children in a tree
	$.fn.expand = function() {
		$(this).removeClass("collapsed").addClass("expanded");

		childrenOf($(this)).each(function() {
			initialize($(this));

			if($(this).is(".expanded.parent")) {
				$(this).expand();
			}

			$(this).show();
		});

		return this;
	};

	$.fn.ensureVisible = function() {
		$.map(ancestorsOf($(this)).reverse(), function(a){ $("#" + a.id).expand(); });
	};

	// Add reverse() function from JS Arrays
	$.fn.reverse = function() {
	  return this.pushStack(this.get().reverse(), arguments);
	};

	// Toggle an entire branch
	$.fn.toggleBranch = function() {
		if($(this).hasClass("collapsed")) {
			$(this).expand();
		}	else {
			$(this).removeClass("expanded").collapse();
		}

		return this;
	};

	// === Private functions

	function ancestorsOf(node) {
		var ancestors = [];
		while(node = parentOf(node)) {
			ancestors[ancestors.length] = node[0];
		}
		return ancestors;
	};

	function childrenOf(node) {
		return $("table.treeTable tbody tr." + options.childPrefix + node[0].id);
	};

	function indent(node, value) {
		var cell = $(node.children("td")[options.treeColumn]);
		var padding = parseInt(cell.css("padding-left"), 10) + value;

		cell.css("padding-left", + padding + "px");

		childrenOf(node).each(function() {
			indent($(this), value);
		});
	};

	function initialize(node) {
		if(!node.hasClass("initialized")) {
			node.addClass("initialized");

			var childNodes = childrenOf(node);

			if(!node.hasClass("parent") && childNodes.length > 0) {
				node.addClass("parent");
			}

			if(node.hasClass("parent")) {
				var cell = $(node.children("td")[options.treeColumn]);
				var padding = (ancestorsOf(node).length + 1) * options.indent;

				childNodes.each(function() {
					$($(this).children("td")[options.treeColumn]).find(".div_tree").css("padding-left", padding + "px");
				});

				if(options.expandable) {
					var expander = $("div .div_tree span:eq(0)", cell);
					expander.addClass("expander");
					expander.click(function() { node.toggleBranch(); return false; });
					$("div .div_tree", cell).prepend(expander);

					// Check for a class set explicitly by the user, otherwise set the default class
					if(!(node.hasClass("expanded") || node.hasClass("collapsed"))) {
					  node.addClass(options.initialState);
					}

					if(node.hasClass("collapsed")) {
						node.collapse();
					} else if (node.hasClass("expanded")) {
						node.expand();
					}
				}
			}
		}
	};

	function move(node, destination) {
		node.insertAfter(destination);
		childrenOf(node).reverse().each(function() { move($(this), node[0]); });
	};

	function parentOf(node) {
		var classNames = node[0].className.split(' ');

		for(key in classNames) {
			if(classNames[key].match("child-of-")) {
				return $("#" + classNames[key].substring(9));
			}
		}
	};
})(jQuery);