lib/ProMotion/styling/styling.rb in ProMotion-2.1.0 vs lib/ProMotion/styling/styling.rb in ProMotion-2.2.0

@@ -1,15 +1,16 @@
 module ProMotion
   module Styling
     def set_attributes(element, args = {})
       args = get_attributes_from_symbol(args)
       args.each { |k, v| set_attribute(element, k, v) }
+      element.send(:on_styled) if element.respond_to?(:on_styled)
       element
     end
 
     def set_attribute(element, k, v)
-      return element unless element
+      return unless element
 
       if !element.is_a?(CALayer) && v.is_a?(Hash) && element.respond_to?("#{k}=")
         element.send("#{k}=", v)
       elsif v.is_a?(Hash) && element.respond_to?(k)
         sub_element = element.send(k)
@@ -69,9 +70,10 @@
     def add_to(parent_element, elements, attrs = {})
       attrs = get_attributes_from_symbol(attrs)
       Array(elements).each do |element|
         parent_element.addSubview element
         set_attributes(element, attrs) if attrs && attrs.length > 0
+        element.send(:on_load) if element.respond_to?(:on_load)
       end
       elements
     end
 
     def view_or_self