add ability to set data attributes via .data(obj)

Author: enure

README.md

@@ -183,4 +183,4 @@ Detailed test output is logged in the JavaScript console of your browser.
   [issues]: https://github.com/madrobby/zepto/issues
   [mkd]: http://github.github.com/github-flavored-markdown/
   [evidence.js]: https://github.com/tobie/Evidence
-  [optional]: http://mislav.uniqpath.com/2010/05/semicolons/
+  [optional]: http://mislav.uniqpath.com/2010/05/semicolons/

src/data.js

@@ -22,9 +22,18 @@
     return store
   }
 
+  function setDataAttributes(node,obj) {
+    node.each(function() {
+      for (var k in obj) {
+        setData(this, k, obj[k]);
+      }
+    });
+  };
+
   $.fn.data = function(name, value) {
     return value === undefined ?
-      this.length == 0 ? undefined : getData(this[0], name) :
+      this.length == 0 ? undefined : 
+      $.isObject(name) ? setDataAttributes(this, name) : getData(this[0], name) :
       this.each(function(idx){
         setData(this, name, $.isFunction(value) ?
                 value.call(this, idx, getData(this, name)) : value)

test/data.html

@@ -18,10 +18,15 @@ <h1>Zepto data() tests</h1>
   <div id="fixtures">
   <div id="data_attr" data-one="uno" data-two="due"></div>
   <div id="data_full" data-mode="awesome"></div>
+  <div id="data_obj" data-mode="awesome"></div>
   <ol id="data_list">
     <li data-category="arts"></li>
     <li data-category="science"></li>
   </ol>
+  <ol id="data_list2">
+    <li></li>
+    <li></li>
+  </ol>
   </div>
 
   <script>
@@ -89,7 +94,45 @@ <h1>Zepto data() tests</h1>
         t.assertEqual('ni', all.two)
         t.assertEqual('Kurosawa', all.person.name)
         t.assertUndefined(all.mode)
+      },
+
+      testSettingDataWithObj: function(t){
+          var el = $('#data_obj')
+
+          el.data({
+              'foo': 'bar',
+              'answer': 42,
+              'color': 'blue'
+          })
+
+          var all = el.data()
+
+          t.assertEqual(all.answer, 42)
+          t.assertEqual(all.color, 'blue')
+          t.assertEqual(all.foo, 'bar')
+
+          el.data('foo', 'baz')
+
+          t.assertEqual(all.foo, 'baz')
+          t.assertEqual(all.answer, 42)
+      },
+
+      testSettingDataWithObjOnManyElements: function(t){
+          var items = $('#data_list2 li')
+
+          items.data({
+              'foo': 'bar',
+              'answer': 42,
+              'color': 'purple'
+          })
+
+          var values = items.map(function(){ return $(this).data('foo') })
+          t.assertEqual('bar, bar', values.join(', '))
+
+          var values2 = items.map(function(){ return $(this).data('answer') })
+          t.assertEqual('42, 42', values2.join(', '))
       }
+
     })
   </script>
 </body>