added queryselectorAll fix

zackify · Dec 23, 2014 · 7c0133a · 7c0133a
1 parent aed30be
commit 7c0133a
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 11 deletions.
diff --git a/index.html b/index.html
@@ -2,7 +2,7 @@
 <style src="test.css"></style>
 <script type="text/javascript">
 window.onload = function(){
-	console.log(_('div').parent().children(1).css('background-color','black').attr('lol','hi'))
+	console.log(_('div').css('background-color','black').attr('lol','hi'))
 	_('div').css('color','white')
 
 	_('div').on('click',function(event){
@@ -13,5 +13,7 @@
 
 <body>
 <div> hi</div>
+<br />
+<div> two</div>
 
 </body>
diff --git a/legit.js b/legit.js
@@ -4,13 +4,16 @@ function _(selector){
 	if(typeof selector === 'object'){
 		self.element = self.selector
 	}
-	else self.element = document.querySelector(self.selector);
-
+	else{
+		var allElements = document.querySelectorAll(self.selector);
+		if(allElements.length == 1) self.element = document.querySelector(self.selector);
+		else self.elements = allElements
+	}
 	/* BASIC FUNCTIONS */
 
 	//grab the html of an element mz('selector').html()
 	self.html = function(){
-		return self.element;
+		return self.elements || self.element;
 	}
 	//get the height of an element
 	self.height = function(){
@@ -21,6 +24,10 @@ function _(selector){
 		return self.element.offsetWidth;
 	}
 	//get parent element
+	self.first = function(){
+		if(self.elements) self.element = self.elements[0]
+		return self;
+	}
 	self.parent = function(){
 		self.element = self.element.parentNode;
 		return self;
@@ -34,16 +41,27 @@ function _(selector){
 	//get and set attribute values
 	self.attr = function(name,value){
 		//if no value set, return the current value
-		if(!value) return self.element.getAttribute(name);
-
-		self.element.setAttribute(name,value);
+	  if(!value) return self.element.getAttribute(name);
+		else if(self.elements){
+			for (var i=0; i < self.elements.length; i++){
+				self.elements[i].setAttribute(name,value);
+			}
+		}
+		else{
+			self.element.setAttribute(name,value);
+		}
 		return self;
 	}
 
 	/* Event FUNCTIONS */
 
 	self.on = function(type,callback){
-		self.element['on' + type] = callback;
+		if(self.elements){
+			for (var i=0; i < self.elements.length; i++){
+				self.elements[i]['on' + type] = callback;
+			}
+		}
+		else self.element['on' + type] = callback;
 		return self
 	}
 
@@ -72,9 +90,15 @@ function _(selector){
 	//add css rule inline
 	self.css = function(name,value){
 		//if no value set, return the current value
-		if(!value) return self.element.style[name];
-
-		self.element.style[name] = value;
+		if(self.elements){
+			for (var i=0; i < self.elements.length; i++){
+				self.elements[i].style[name] = value;
+			}
+		}
+		else{
+			if(!value) return self.element.style[name];
+			self.element.style[name] = value;
+		}
 		return self;
 	}