lib/slide.rb in slidedown-0.1.0 vs lib/slide.rb in slidedown-0.1.1

@@ -1,12 +1,24 @@
 class Slide
-  attr_accessor :text, :classes
-  
+  attr_accessor :text, :classes, :notes
+
   def initialize(text, *classes)
-    @text = text
+    @text    = text
     @classes = classes
+    @notes   = nil
+    
+    extract_notes!
   end
-  
+
   def html
     MakersMark::Generator.new(@text).to_html
   end
-end
\ No newline at end of file
+  
+  private
+  
+  def extract_notes!
+    @text.gsub!(/^!NOTES\s*(.*\n)$/m) do |note|
+      @notes = note.to_s.chomp.gsub('!NOTES', '')
+      ''
+    end
+  end
+end