file_tree_profiler 0.0.1 → 0.0.2

data/lib/file_tree_profiler.rb

@@ -4,10 +4,29 @@ require 'file_tree_profiler/file'
 require 'file_tree_profiler/data_file'
 require 'file_tree_profiler/dir_file'
 require 'file_tree_profiler/profile'
+require 'file_tree_profiler/merge/pairing'
+require 'file_tree_profiler/merge'
 require 'file_tree_profiler/export/csv'
 require 'file_tree_profiler/export/sql'
 
 module FileTreeProfiler
+  def self.monitor_report(*args)
+    @@monitor.report(*args) if monitoring?
+  end
+
+  def self.monitoring?
+    defined?(@@monitoring) && @@monitoring == true
+  end
+
+  def self.with_monitoring(monitor)
+    @@monitor = monitor
+    @@monitoring = true
+    yield
+  ensure
+    @@monitor = false
+    @@monitoring = false
+  end
+
   def self.profile path
     Profile.new path
   end

data/lib/file_tree_profiler/data_file.rb

@@ -12,6 +12,9 @@ module FileTreeProfiler
       end.to_s
     rescue Errno::ELOOP => e
       puts "path=#{path} is a circular ref"
+    rescue Errno::ENOENT => e
+      puts "got deleted while profiling!!!"
+      @checksum = EMPTY_CHECKSUM
     end
 
     def size

data/lib/file_tree_profiler/dir_file.rb

@@ -11,10 +11,11 @@ module FileTreeProfiler
     # collects all children as DirFile or DataFile objects
     # and is invoked on each collected DirFile object
     def walk
+      FileTreeProfiler.monitor_report(:profile, :dir, path)
       @children = []
-      Dir.foreach(self.path) do |entry|
+      Dir.foreach(path) do |entry|
         next if (entry == '..' || entry == '.')
-        full_path = ::File.join(self.path, entry)
+        full_path = ::File.join(path, entry)
         if ::File.directory?(full_path)
           children.push DirFile.new(self, entry)
         else

data/lib/file_tree_profiler/merge.rb

@@ -0,0 +1,48 @@
+module FileTreeProfiler
+  class Merge
+    attr_reader :source_profile, :target_profile, :pairings
+
+    def initialize source_profile, target_profile
+      @source_profile = source_profile
+      @target_profile = target_profile
+      @pairings = {}
+      merge :source, source_profile.root
+      merge :target, target_profile.root
+      
+      pairings.each do |relative_path, f|
+        FileTreeProfiler.monitor_report :merge_finalize, relative_path
+        if parent_relative_path = f.parent_relative_path
+          if parent = self[parent_relative_path]
+            f.status_leaf = parent.status != f.status
+          end
+        end
+      end
+    end
+
+    def add scope, file
+      key = file.relative_path
+      pairings[key] ||= Pairing.new
+      pairings[key].add scope, file
+    end
+
+    def merge scope, dir_file
+      FileTreeProfiler.monitor_report :merge_map, :dir, dir_file.path
+      add scope, dir_file
+      dir_file.children.each do |child|
+        if child.class == DirFile
+          merge scope, child
+        else
+          add scope, child
+        end
+      end
+    end
+
+    def [](relative_path)
+      pairings[relative_path]
+    end
+
+    def size
+      pairings.size
+    end
+  end
+end

data/lib/file_tree_profiler/merge/pairing.rb

@@ -0,0 +1,66 @@
+module FileTreeProfiler
+  class Merge
+    class Pairing
+      EQUAL       = 1
+      DIFFERENT   = 2
+      ONLY_SOURCE = 3
+      ONLY_TARGET = 4
+
+      attr_accessor :status_leaf
+
+      def initialize
+        @status_leaf = false
+      end
+
+      def add scope, file
+        @files ||= {}
+        @files[scope] = file
+      end
+
+      def source
+        @files[:source]
+      end
+
+      def target
+        @files[:target]
+      end
+
+      def name
+        any.name
+      end
+
+      def relative_path
+        any.relative_path
+      end
+
+      def parent_relative_path
+        if relative_path == '/'
+          nil
+        else
+          ::File.dirname(relative_path)
+        end
+      end
+
+      def any
+        (source || target)
+      end
+
+      def status
+        @status ||= begin
+          if source && target
+            if source.checksum == target.checksum
+              EQUAL
+            else
+              DIFFERENT
+            end
+          elsif source
+            ONLY_SOURCE
+          else
+            ONLY_TARGET
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/file_tree_profiler/version.rb

@@ -1,3 +1,3 @@
 module FileTreeProfiler
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/spec/example_folders/differents/profile_a/foo2/bar.txt

@@ -0,0 +1 @@
+bar.txt

data/spec/example_folders/differents/profile_a/foo2/baz/foo.txt

@@ -0,0 +1 @@
+foo.txt

data/spec/example_folders/differents/profile_a/foo2/foo.txt

@@ -0,0 +1 @@
+foo.txt

data/spec/example_folders/differents/profile_c/foo2/baz.txt

@@ -0,0 +1 @@
+baz.txt

data/spec/example_folders/differents/profile_c/foo2/baz/foo.txt

@@ -0,0 +1 @@
+foo.txt

data/spec/example_folders/differents/profile_c/foo2/foo.txt

@@ -0,0 +1 @@
+different

data/spec/lib/file_tree_profiler/merge/files_spec.rb

@@ -0,0 +1,56 @@
+require 'spec_helper'
+
+describe FileTreeProfiler::Merge::Pairing do
+  let(:source) do
+    mock(:source).tap do |source|
+      source.stub(:checksum).and_return('a1')
+      source.stub(:name).and_return('foo')
+    end
+  end
+
+  let(:target) do
+    mock(:target).tap do |source|
+      source.stub(:checksum).and_return('a1')
+      source.stub(:name).and_return('foo')
+    end
+  end
+
+  context :equal do
+    before do
+      subject.add :source, source
+      subject.add :target, target
+    end
+
+    its(:name) { should == 'foo' }
+    its(:status) { should == described_class::EQUAL }
+  end
+
+  context :different do
+    before do
+      source.stub(:checksum).and_return('b2')
+      subject.add :source, source
+      subject.add :target, target
+    end
+
+    its(:name) { should == 'foo' }
+    its(:status) { should == described_class::DIFFERENT }
+  end
+
+  context :only_target do
+    before do
+      subject.add :target, target
+    end
+
+    its(:name) { should == 'foo' }
+    its(:status) { should == described_class::ONLY_TARGET }
+  end
+
+  context :only_source do
+    before do
+      subject.add :source, source
+    end
+
+    its(:name) { should == 'foo' }
+    its(:status) { should == described_class::ONLY_SOURCE }
+  end
+end

data/spec/lib/file_tree_profiler/profile_spec.rb

@@ -35,6 +35,45 @@ describe FileTreeProfiler::Profile do
     context :partially do
       specify { profile_a.root.children.first.name.should == profile_c.root.children.first.name }
       specify { profile_a.root.children.first.checksum.should == profile_c.root.children.first.checksum }
+
+      context :merge do
+        let(:merge) { FileTreeProfiler::Merge.new(profile_a, profile_c) }
+        subject { merge }
+        its(:size) { should == 11 }
+
+        specify do
+          subject.pairings.map{|k,v| k}.should == \
+            ["/", "/foo", "/foo/bar.txt", "/foo/foo.txt", "/foo2", "/foo2/bar.txt", "/foo2/baz", "/foo2/baz/foo.txt", "/foo2/foo.txt", "/foo2/baz.txt", "/partially_different.txt"]
+        end
+
+        shared_examples_for :pairing do |relative_path, expected_status, expected_status_leaf|
+          expected_status = {
+            :equal => FileTreeProfiler::Merge::Pairing::EQUAL,
+            :different => FileTreeProfiler::Merge::Pairing::DIFFERENT,
+            :only_target => FileTreeProfiler::Merge::Pairing::ONLY_TARGET,
+            :only_source => FileTreeProfiler::Merge::Pairing::ONLY_SOURCE
+          }[expected_status] || raise("unknown status = '#{expected_status}'")
+          context relative_path do
+            subject { merge[relative_path] }
+            its(:relative_path) { should == relative_path }
+            its(:parent_relative_path) { should == ::File.dirname(relative_path) }
+            its(:name) { should == ::File.basename(relative_path) }
+            its(:class) { should == FileTreeProfiler::Merge::Pairing }
+            its(:status) { should == expected_status }
+            its(:status_leaf) { should == expected_status_leaf }
+          end
+        end
+
+        include_examples :pairing, '/foo', :equal, true
+        include_examples :pairing, '/foo/bar.txt', :equal, false
+        include_examples :pairing, '/foo/foo.txt', :equal, false
+        include_examples :pairing, '/partially_different.txt', :only_target, true
+        include_examples :pairing, '/foo2/baz', :equal, true
+        include_examples :pairing, '/foo2/baz/foo.txt', :equal, false
+        include_examples :pairing, '/foo2/bar.txt', :only_source, true
+        include_examples :pairing, '/foo2/baz.txt', :only_target, true
+        include_examples :pairing, '/foo2/foo.txt', :different, false
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: file_tree_profiler
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-01-05 00:00:00.000000000 Z
+date: 2013-01-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sequel
@@ -82,6 +82,8 @@ files:
 - lib/file_tree_profiler/export/csv.rb
 - lib/file_tree_profiler/export/sql.rb
 - lib/file_tree_profiler/file.rb
+- lib/file_tree_profiler/merge.rb
+- lib/file_tree_profiler/merge/pairing.rb
 - lib/file_tree_profiler/profile.rb
 - lib/file_tree_profiler/version.rb
 - spec/example_folders/a/a.txt
@@ -91,9 +93,15 @@ files:
 - spec/example_folders/b/bb.txt
 - spec/example_folders/differents/profile_a/foo/bar.txt
 - spec/example_folders/differents/profile_a/foo/foo.txt
+- spec/example_folders/differents/profile_a/foo2/bar.txt
+- spec/example_folders/differents/profile_a/foo2/baz/foo.txt
+- spec/example_folders/differents/profile_a/foo2/foo.txt
 - spec/example_folders/differents/profile_b/foo/bar.txt
 - spec/example_folders/differents/profile_c/foo/bar.txt
 - spec/example_folders/differents/profile_c/foo/foo.txt
+- spec/example_folders/differents/profile_c/foo2/baz.txt
+- spec/example_folders/differents/profile_c/foo2/baz/foo.txt
+- spec/example_folders/differents/profile_c/foo2/foo.txt
 - spec/example_folders/differents/profile_c/partially_different.txt
 - spec/example_folders/equals/profile_a/bar.txt
 - spec/example_folders/equals/profile_a/foo/foo.txt
@@ -106,6 +114,7 @@ files:
 - spec/lib/file_tree_profiler/data_file_spec.rb
 - spec/lib/file_tree_profiler/dir_file_spec.rb
 - spec/lib/file_tree_profiler/export/sql_export_spec.rb
+- spec/lib/file_tree_profiler/merge/files_spec.rb
 - spec/lib/file_tree_profiler/profile_spec.rb
 - spec/lib/file_tree_profiler_spec.rb
 - spec/spec_helper.rb
@@ -143,9 +152,15 @@ test_files:
 - spec/example_folders/b/bb.txt
 - spec/example_folders/differents/profile_a/foo/bar.txt
 - spec/example_folders/differents/profile_a/foo/foo.txt
+- spec/example_folders/differents/profile_a/foo2/bar.txt
+- spec/example_folders/differents/profile_a/foo2/baz/foo.txt
+- spec/example_folders/differents/profile_a/foo2/foo.txt
 - spec/example_folders/differents/profile_b/foo/bar.txt
 - spec/example_folders/differents/profile_c/foo/bar.txt
 - spec/example_folders/differents/profile_c/foo/foo.txt
+- spec/example_folders/differents/profile_c/foo2/baz.txt
+- spec/example_folders/differents/profile_c/foo2/baz/foo.txt
+- spec/example_folders/differents/profile_c/foo2/foo.txt
 - spec/example_folders/differents/profile_c/partially_different.txt
 - spec/example_folders/equals/profile_a/bar.txt
 - spec/example_folders/equals/profile_a/foo/foo.txt
@@ -158,6 +173,7 @@ test_files:
 - spec/lib/file_tree_profiler/data_file_spec.rb
 - spec/lib/file_tree_profiler/dir_file_spec.rb
 - spec/lib/file_tree_profiler/export/sql_export_spec.rb
+- spec/lib/file_tree_profiler/merge/files_spec.rb
 - spec/lib/file_tree_profiler/profile_spec.rb
 - spec/lib/file_tree_profiler_spec.rb
 - spec/spec_helper.rb