Extending the events array with #files and #modified_files.

lib/fsevents.rb

@@ -4,6 +4,7 @@
 require 'osx/foundation'
 OSX.require_framework '/System/Library/Frameworks/CoreServices.framework/Frameworks/CarbonCore.framework'
 
+require 'fsevents/ext'
 require 'fsevents/stream'
 
 module FSEvents

lib/fsevents/ext.rb

@@ -0,0 +1 @@
+require 'fsevents/ext/array'

lib/fsevents/ext/array.rb

@@ -0,0 +1,9 @@
+module EventArray
+  def files
+    collect { |x|  x.files }.flatten
+  end
+
+  def modified_files
+    collect { |x|  x.modified_files }.flatten
+  end
+end

lib/fsevents/stream.rb

@@ -34,6 +34,7 @@ def stream_callback
         paths.regard_as('*')
 
         events = []
+        events.extend(EventArray)
         event_count.times { |i|  events << Event.new(event_IDs[i], paths[i], self) }
 
         callback.call(events)

spec/event_array_spec.rb

@@ -0,0 +1,42 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+
+describe EventArray do
+  before :each do
+    @event_array = []
+    @event_array.extend(EventArray)
+  end
+
+  it 'should return files' do
+    @event_array.should respond_to(:files)
+  end
+
+  describe 'returning files' do
+    it 'should collect files from its events' do
+      events = Array.new(3) { stub('event', :files => Array.new(3) { stub('file') }) }
+      files = []
+      events.each do |event|
+        @event_array << event
+        files += event.files
+      end
+
+      @event_array.files.should == files
+    end
+  end
+
+  it 'should return modified files' do
+    @event_array.should respond_to(:modified_files)
+  end
+
+  describe 'returning modified files' do
+    it 'should collect modified files from its events' do
+      events = Array.new(3) { stub('event', :modified_files => Array.new(3) { stub('file') }) }
+      files = []
+      events.each do |event|
+        @event_array << event
+        files += event.modified_files
+      end
+
+      @event_array.modified_files.should == files
+    end
+  end
+end

spec/stream_spec.rb

@@ -270,6 +270,12 @@
         @proc.call(*@args)
       end
 
+      it 'should extend the event array' do
+        @args = @args_hash.values_at(*@callback_arg_order)
+        @callback.expects(:call).with { |events|  events.is_a?(EventArray) }
+        @proc.call(*@args)
+      end
+
       it "should update the stream's last event" do
         @stream.expects(:update_last_event)
         @proc.call(*@args)