lib/buildmaster/cotta/cotta_dir.rb in BuildMaster-1.0.6 vs lib/buildmaster/cotta/cotta_dir.rb in BuildMaster-1.0.9

@@ -1,104 +1,149 @@
+require 'rubygems/package'
+
 $:.unshift File.dirname(__FILE__)
 
 require 'cotta_file'
+require 'io_chain'
 
 module BuildMaster
 class CottaDir
+  include IoChain
   attr_reader :path, :system
 
   def initialize(system, path)
     @path = path
     @system = system
     @name = @path.basename.to_s
   end
-  
+
   def cotta
-    return Cotta.new(@system)
+    Cotta.new(@system)
   end
-  
+
   def name
     name = nil
     if root?
       name = path.to_s
     else
       name = @path.basename.to_s
     end
     return name
   end
-  
+
   def root?
     parent.nil?
   end
-  
+
   def exists?
     return @system.dir_exists?(@path)
   end
-  
+
+  def stat
+    @system.dir_stat(@path)
+  end
+
   def parent
     parent_path = @path.cotta_parent
     return nil unless parent_path
     return CottaDir.new(@system, parent_path)
   end
-  
+
+  def relative_path_from(entry)
+    @path.relative_path_from(entry.path)
+  end
+
   def dir(name)
     return CottaDir.new(@system, @path.join(name))
   end
-  
+
   def file(name)
     return CottaFile.new(@system, @path.join(name))
   end
-  
+
   def mkdirs
     if (not exists?)
       parent.mkdirs
       @system.mkdir @path
     end
   end
-  
+
   def delete
-    list.each {|children| children.delete}
+    list.each {|children|
+      children.delete
+    }
     @system.delete_dir(@path)
   end
-  
+
   def move_to(target)
     target.parent.mkdirs
     @system.move_dir(@path, target.path)
   end
-  
+
   def move_to_path(target_path)
     move_to(cotta.dir(target_path))
   end
-  
+
   def copy_to(target)
     target.parent.mkdirs
     @system.copy_dir(@path, target.path)
   end
-  
+
+  def archive(target = nil)
+    unless target
+      target = parent.file("#{name}.tar")
+    end
+    target.write_binary do |io|
+      writer = Gem::Package::TarWriter.new(io) do |tar_io|
+        archive_dir(tar_io, self)
+      end
+    end
+    target
+  end
+
+  def archive_dir(tar_io, dir)
+    dir.list.each do |child|
+      stat = child.stat
+      entry_name = child.relative_path_from(self).to_s
+      mode = stat.mode
+      if (stat.file?)
+        tar_io.add_file(entry_name, mode) do |entry_output|
+          child.read_binary do |input|
+            copy_io(input, entry_output)
+          end
+        end
+      elsif (stat.directory?)
+        tar_io.mkdir(entry_name, mode)
+        archive_dir(tar_io, child)
+      end
+    end
+  end
+  private :archive_dir
+
   def copy_to_path(target_path)
     copy_to(cotta.dir(target_path))
   end
-  
+
   def list
     @system.list(@path).collect do |item|
       candidate = dir(item)
       if (not candidate.exists?)
         candidate = file(item)
       end
       candidate
     end
   end
-  
+
   def ==(other)
     return @path == other.path && @system == other.system
   end
-  
+
   def inspect
-    return "#{self.class}:#{self.object_id}-#{@system.inspect}-#@path"
+    return "#{self.class}:#{self.object_id}-#@path"
   end
-  
+
   def to_s
     @path
   end
-    
+
 end
 end
\ No newline at end of file