lib/maily/mailer.rb in maily-0.7.2 vs lib/maily/mailer.rb in maily-0.8.0

@@ -38,11 +38,11 @@
     def total_emails
       emails.size
     end
 
     def register_hook(email_name, *args)
-      email = find_email(email_name)
+      email = find_email(email_name) || add_email(email_name)
       email && email.register_hook(args)
     end
 
     def hide_email(*email_names)
       email_names.each do |email_name|
@@ -53,11 +53,16 @@
     private
 
     def parse_emails
       _emails = klass.send(:public_instance_methods, false)
 
-      _emails.map(&:to_s).each do |email|
-        self.emails[email] = Maily::Email.new(email, self)
+      _emails.each do |email|
+        add_email(email)
       end
+    end
+
+    def add_email(email_name)
+      email = Maily::Email.new(email_name.to_s, self)
+      self.emails[email.name] = email
     end
   end
 end
\ No newline at end of file