pmsrb 0.1.0 → 0.2.0

data/Manifest.txt

@@ -9,6 +9,7 @@ lib/pmsrb.rb
 lib/pmsrb/platform.rb
 lib/pmsrb/progressbar.rb
 lib/pmsrb/version.rb
+lib/pmsrb/memory_profile.rb
 log/debug.log
 script/destroy
 script/generate
@@ -20,4 +21,4 @@ spec/spec_helper.rb
 tasks/deployment.rake
 tasks/environment.rake
 tasks/rspec.rake
-tasks/website.rake
+tasks/website.rake

data/lib/pmsrb.rb

@@ -23,7 +23,13 @@ module PMS
 
   class Vendor
     def run(&block)
+      @cleanup_tasks = []
       instance_eval(&block)
+      @cleanup_tasks.each {|x| x.call}
+    end
+    
+    def cleanup(&block)
+      @cleanup_tasks << block
     end
     
     def download(url, storage_dir=nil)
@@ -67,8 +73,41 @@ module PMS
     
     def extract
       puts "extracting #{@current_file}"
-      extract_tgz if /.tar.gz$/ === @current_file || /.tgz$/ === @current_file
-      extract_zip if /.zip$/ === @current_file 
+      file = @current_file.split("/").pop  
+      case(file)
+        when /tar.gz$/, /tgz$/: extract_tgz
+        when /zip$/: extract_zip
+        when /dmg$/: extract_dmg
+        when /tar.bz2$/, /tbz2$/: extract_tbz2
+        else raise Exception.new("Unknown file type '#{extension}' for '#{@current_file}' in extraction")
+      end
+    end
+    
+    def extract_tbz2
+      pwd = File.expand_path(FileUtils.pwd)
+      FileUtils.cd @working_directory
+      
+      `tar xvjpf #{@current_file[(@working_directory.length+1)..-1]}`
+    
+      FileUtils.cd pwd
+    end
+    
+    def extract_dmg
+      fname=@current_file
+      `hdiutil attach '#{fname}' -mountpoint '#{@working_directory}/dmg_tmp_mount'`
+      
+      Dir["#{@working_directory}/dmg_tmp_mount/**/*"].each do |file|
+        next if File.symlink?(file)
+        next if File.directory?(file)
+        dest_path = file.sub("/dmg_tmp_mount","").split("/")[0...-1].join("/")
+        FileUtils.mkdir_p(dest_path)
+        FileUtils.cp(file, dest_path)
+      end
+      
+      
+      cleanup do
+        `hdiutil detach '#{@working_directory}/dmg_tmp_mount' -force`
+      end
     end
     
     def extract_tgz
@@ -118,16 +157,23 @@ module PMS
     def cp(pattern, newdir)
       newdir = "#{@working_directory}/#{newdir}"
       files = Dir["#{@working_directory}/#{pattern}"]
+      puts "'#{@working_directory}/#{pattern}' => '#{newdir}'"
      
       if files.length == 1
         file = files[0]
         FileUtils.mkdir_p File.dirname(newdir)
-        FileUtils.cp_r file, newdir unless FileTest.directory? file
+        unless FileTest.directory?(file) || File.symlink?(file)
+          puts "'#{file}' => '#{newdir}'"
+          FileUtils.cp_r file, newdir 
+        end
       else        
         files.each do |file|
           dest = newdir + (File.dirname(file))[@working_directory.length..-1]
           FileUtils.mkdir_p dest
-          FileUtils.cp file, dest unless FileTest.directory? file
+          unless FileTest.directory?(file) || File.symlink?(file)
+            puts "'#{file}' => '#{dest}'"
+            FileUtils.cp file, dest 
+          end
         end
       end
     end
@@ -171,13 +217,11 @@ module PMS
     
     def archive(filelist, archive)
       filelist = expand_filelist filelist
-      puts "archiving #{filelist}"
-      if /.tgz$/ === archive || /.tar.gz$/ === archive
-        archive_tgz filelist, archive
-      elsif /.zip$/ === archive
-        archive_zip filelist, archive
-      else
-        raise Exception.new("archive type '#{archive}' is not supported")
+      puts "archiving #{filelist.inspect}"
+      case(archive)
+        when /\.tgz$/, /\.tar\.gz$/: archive_tgz filelist, archive
+        when /\.zip$/: archive_zip filelist, archive
+        else raise Exception.new("archive type '#{archive}' is not supported")
       end
     end
     

data/lib/pmsrb/memory_profile.rb

@@ -0,0 +1,155 @@
+include ObjectSpace
+
+MEMORY_PROFILE_BAD_SIZE_METHOD = {FileTest => true, File => true, File::Stat => true}
+
+class Object
+  def memory_profile_size_of_object(seen={})
+    return 0 if seen.has_key? object_id
+    seen[object_id] = true
+    count = 1
+    if kind_of? Hash
+      each_pair do |key,value|
+        count += key.memory_profile_size_of_object(seen)
+        count += value.memory_profile_size_of_object(seen)
+      end
+    elsif kind_of? Array
+      count += size
+      each do |element|
+        count += element.memory_profile_size_of_object(seen)
+      end
+    end
+
+    count += instance_variables.size
+    instance_variables.each do |var|
+      count += instance_variable_get(var.to_sym).memory_profile_size_of_object(seen)
+    end
+
+    count
+  end
+
+  def memory_profile_inspect(seen={},level=0)
+    return object_id.to_s if seen.has_key? object_id
+    seen[object_id] = true
+    result = ' '*level
+    if kind_of? Hash
+      result += "{\n" + ' '*level
+      each_pair do |key,value|
+        result += key.memory_profile_inspect(seen,level+1) + "=>\n"
+        result += value.memory_profile_inspect(seen,level+2) + ",\n" + ' '*level
+      end
+      result += "}\n" + ' '*level
+    elsif kind_of? Array
+      result += "[\n" + ' '*level
+      each do |element|
+        result += element.memory_profile_inspect(seen,level+1) + ",\n" + ' '*level
+      end
+      result += "]\n" + ' '*level
+    elsif kind_of? String
+      result += self
+    elsif kind_of? Numeric
+      result += self.to_s
+    elsif kind_of? Class
+      result += to_s
+    else
+      result += "---"+self.class.to_s + "---\n" + ' '*level
+    end
+
+
+    instance_variables.each do |var|
+      result += var + "=" + instance_variable_get(var.to_sym).memory_profile_inspect(seen,level+1) + "\n" + ' '*level
+    end
+
+    result
+  end
+
+end
+
+module MemoryProfile
+  LOG_FILE = "/tmp/memory_profile.log"
+
+  def MemoryProfile::report
+    Dir.chdir "/tmp"
+    ObjectSpace::garbage_collect
+    sleep 10 # Give the GC thread a chance
+    all = []
+    ObjectSpace.each_object do |obj|
+      next if obj.object_id == all.object_id 
+        
+      all << obj
+    end
+    
+    tally = Hash.new(0)
+    max_obj = nil
+    max_count = 0
+    all.each do |obj|
+      count = obj.memory_profile_size_of_object
+      if max_count < count
+        max_obj = obj
+        max_count = count
+      end
+      
+      tally[obj.class]+=count
+    end
+    
+    open( LOG_FILE, 'a') do |outf|
+      outf.puts '+'*70
+      tally.keys.sort{|a,b| 
+        if tally[a] == tally[b]
+          a.to_s <=> b.to_s
+        else
+          -1*(tally[a]<=>tally[b])
+        end
+      }.each do |klass|
+        outf.puts "#{klass}\t#{tally[klass]}"
+      end
+      
+      outf.puts '-'*70
+      outf.puts "Max obj was #{max_obj.class} at #{max_count}"
+      outf.puts "Maximum object is..."
+      outf.puts max_obj.memory_profile_inspect
+    end
+  end
+
+  def MemoryProfile::simple_count
+    Dir.chdir "/tmp"
+    ObjectSpace::garbage_collect
+    sleep 10 # Give the GC thread a chance
+
+    tally = Hash.new(0)
+    ObjectSpace.each_object do |obj|
+      next if obj.object_id == tally.object_id
+      tally[obj.class]+=1
+    end
+    
+    open( LOG_FILE, 'a') do |outf|
+      outf.puts '='*70
+      outf.puts "MemoryProfile report for #{$0}"
+      outf.puts `cat /proc/#{Process.pid}/status`
+      
+      tally.keys.sort{|a,b| 
+        if tally[a] == tally[b]
+          a.to_s <=> b.to_s
+        else
+          -1*(tally[a]<=>tally[b])
+        end
+      }.each do |klass|
+        outf.puts "#{klass}\t#{tally[klass]}"
+      end
+    end
+  end
+end
+
+if $0 == __FILE__ then
+  File.unlink MemoryProfile::LOG_FILE if FileTest.exist? MemoryProfile::LOG_FILE
+
+  at_exit{ system("cat #{MemoryProfile::LOG_FILE}")}
+end
+
+at_exit{
+  MemoryProfile::simple_count
+  MemoryProfile::report
+}
+
+
+
+

data/lib/pmsrb/platform.rb

@@ -4,6 +4,18 @@
 # 
 #   Platform.new("mswin").short_name => windows, since we are creating an instance.
 class Platform
+  def self.mac
+    yield if(mac?)
+  end
+  
+  def self.windows
+    yield if(windows?)
+  end
+  
+  def self.linux
+    yield if(linux?)
+  end
+  
 	def self.mac?
 	  PLATFORM =~ /darwin/
 	end

data/lib/pmsrb/version.rb

@@ -1,7 +1,7 @@
 module Pmsrb #:nodoc:
   module VERSION #:nodoc:
     MAJOR = 0
-    MINOR = 1
+    MINOR = 2
     TINY  = 0
 
     STRING = [MAJOR, MINOR, TINY].join('.')

data/spec/pmsrb_spec.rb

@@ -272,4 +272,30 @@ describe "The pmsrb dsl" do
     creds.pass.should == "pass"
     FileUtils.rm_r(ENV["HOME"]+"/.test_pmsrb")
   end
+  
+  it "extracts DMG files" do
+  Platform.mac do
+    vendor do
+      working_directory working_dir
+      download "http://www.vojousoftware.com/download/attime.dmg"
+      extract
+      File.exists?(working_dir+"/dmg_tmp_mount").should == true
+      File.exists?(working_dir+"/ReadMe.rtf").should == true
+    end
+  File.exists?(working_dir+"/dmg_tmp_mount").should == false
+  end
+  end
+  
+  it "fails extracting unknown file types" do
+    lambda { 
+    vendor do
+      @current_file = "random.bad.extension"
+      extract 
+    end
+    }.should raise_error
+  end
+  
+  it "can extract .tar.bz2" do
+    Vendor.new.methods.include?("extract_tbz2").should == true
+  end
 end

metadata

@@ -3,8 +3,8 @@ rubygems_version: 0.9.4
 specification_version: 1
 name: pmsrb
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
-date: 2008-04-19 00:00:00 -06:00
+  version: 0.2.0
+date: 2008-05-26 00:00:00 -06:00
 summary: pmsrb is a binary (p)ackage (m)anagement (s)ystem meant for gracefully handling the differences in binary distribution on different platforms.
 require_paths: 
 - lib
@@ -42,6 +42,7 @@ files:
 - lib/pmsrb/platform.rb
 - lib/pmsrb/progressbar.rb
 - lib/pmsrb/version.rb
+- lib/pmsrb/memory_profile.rb
 - log/debug.log
 - script/destroy
 - script/generate