SearchableDocumentFragment

SearchableDocumentFragment.js

"use strict";

var createSearchableDocumentFragment = (function() {
	
	var SearchableDocumentFragmentMethods = {
		
		getElementsByTagName(tagName) {
			tagName = tagName.toUpperCase();
			var elementsWithTagName = [];
			
			for(let i=0; this.children.length > i; i++) {
				let child = this.children[i];
				if(child.tagName == tagName) {
					elementsWithTagName.push(child);
				}
				
				elementsWithTagName = [].slice.call(child.getElementsByTagName(tagName)).concat(elementsWithTagName);
			}
			
			return elementsWithTagName;
		},
		
		getElementsByClassName(classNames) {
			var elementsWithClassName = [];
			
			var classes = classNames.split(/\s+/);
			
			for(let i=0; this.children.length > i; i++) {
				let child = this.children[i];
				let childContainsAllClasses = true;
				for(let n=0; classes.length > n; n++) {
					let className = classes[n];
					if(!child.classList.contains(className)) {
						childContainsAllClasses = false;
						break;
					}
					
					if(childContainsAllClasses) {
						elementsWithClassName.push(child);
					}
				}
				
				elementsWithClassName = [].slice.call(child.getElementsByClassName(classNames)).concat(elementsWithClassName);
			}
			
			return elementsWithClassName;
		}
		
	};
	
	return function() {
		var docFragment = document.createDocumentFragment();
		for(var methodName in SearchableDocumentFragmentMethods) {
			docFragment[methodName] = SearchableDocumentFragmentMethods[methodName].bind(docFragment);
		}
		
		return docFragment;
	};

}());