lib/under_os/ui/collection/delegate.rb in under-os-1.0.0 vs lib/under_os/ui/collection/delegate.rb in under-os-1.1.0

@@ -57,12 +57,14 @@
   ################################################################
   # DataSource API
   ################################################################
 
   def collectionView(collection, didSelectItemAtIndexPath: indexPath)
-    @collection.emit(:select, index: indexPath.row, section: indexPath.section)
+    item = collection.cellForItemAtIndexPath(indexPath).uos_view_for(@collection)
+    @collection.emit(:select, item: item, index: indexPath.row, section: indexPath.section)
   end
 
   def collectionView(collection, didDeselectItemAtIndexPath: indexPath)
-    @collection.emit(:unselect, index: indexPath.row, section: indexPath.section)
+    item = collection.cellForItemAtIndexPath(indexPath).uos_view_for(@collection)
+    @collection.emit(:unselect, item: item, index: indexPath.row, section: indexPath.section)
   end
 end