reseed 0.0.3 → 0.0.4

data/lib/reseed/reseed_task.rb

@@ -10,75 +10,107 @@ OpenURI::Buffer.const_set 'StringMax', 0
 
 class ReseedTask
 
-  def execute seeds
+  def execute name, seeds
+    puts "=== #{name} ==="
+
     seeds.each do |s|
+      files_to_reseed = []
+
       if s[:file]
-        reseed_file s
+        files_to_reseed = [s]
       elsif s[:files]
-        reseed_files s
+        files = Dir.glob s.delete :files
+        files_to_reseed = files.map { |f| { :file => f }.merge(s) }
+      end
+
+      if s[:tfs]
+        puts "\r\nTFS checkout latest..."
+        tfs_checkout files
       end
-    end
-  end
 
-  def reseed_file args
-    if args[:latest_dir]
-      reseed_from_latest_dir args[:latest_dir], args[:file], args[:tfs]
-    elsif args[:dir]
-      reseed_from_dir args[:dir], args[:file], args[:tfs]
-    elsif args[:web]
-      reseed_from_web args[:web], args[:file], args[:tfs]
+      puts "\r\n#{files_to_reseed.count} " + (files_to_reseed.count == 1 ? "file..." : "files...") 
+      reseed_files files_to_reseed
     end
-  end
 
-  def reseed_files args
-    files = args.delete :files
+    puts "\r\n"
+  end
 
-    Dir.glob(files).each do |f|
-      reseed_file({ :file => f }.merge(args))
+  def reseed_files files
+    files.each do |args|
+      if args[:latest_dir]
+        reseed_from_latest_dir args[:latest_dir], args[:file]
+      elsif args[:dir]
+        reseed_from_dir args[:dir], args[:file]
+      elsif args[:web]
+        reseed_from_web args[:web], args[:file]
+      end
     end
   end
 
-  def reseed_from_latest_dir base_dir, file, checkout
+  def reseed_from_latest_dir base_dir, file
     source = File.join base_dir, Dir.entries(base_dir).sort.reverse.take(1).first
-    reseed_from_dir source, file, checkout
+    reseed_from_dir source, file
   end
 
-  def reseed_from_dir dir, file, checkout
+  def reseed_from_dir dir, file
     source = File.join(dir, File.basename(file))
-    reseed source, file, checkout 
+    reseed source, file 
   end
 
-  def reseed_from_web url, file, checkout
+  def reseed_from_web url, file
+    if /\.zip$/i.match url
+      extracted_to = download_and_extract url
+      reseed_from_dir extracted_to, file
+      return
+    elsif /(\\|\/)$/.match url
+      url += File.basename file
+    end
+
     open url do |x|
-      reseed x.path, file, checkout
+      reseed x.path, file
     end
   end
 
-  def reseed source, dest, checkout
+  def reseed source, dest
     base_name = File.basename dest
-
     if File.exist? source
-
-      if checkout 
-        tfs_checkout dest
-      end
-      
       begin
        FileUtils.cp source, dest
-       puts "      #{base_name}"
       rescue
-       puts "    ! #{base_name} (Unable to copy)"
+       puts "   #{base_name} (Unable to copy)"
       end
    else
-     puts "    ! #{base_name} (Doesn't exist)"
+     puts "   #{base_name} (Doesn't exist)"
    end
+  end
 
+  def tfs_checkout paths
+    checked_out = 0
+    paths.each_slice(10) do |s|
+      to_checkout = s.map{ |x| "\"#{x}\"" }.join " "
+      `"#{TFS_PATH}" get #{to_checkout} /force /noprompt >nul 2>&1`
+      `"#{TFS_PATH}" checkout #{to_checkout} >nul 2>&1`
+      checked_out += s.count
+      print "\r   #{checked_out} of #{paths.count}"
+    end
+    print "\r\n"
   end
 
-  def tfs_checkout path
-   return false unless system TFS_PATH, "get", path, " /force /noprompt"
-   return false unless system TFS_PATH, "checkout", path
-   return true
+  def download_and_extract url_to_zip
+    temp_dir = Dir.mktmpdir
+
+    open url_to_zip do |f|
+      zip_path = f.path
+      
+      Zip::ZipFile.open(zip_path) do |zip|
+        zip.each do |z|
+          dest = File.join temp_dir, z.name
+          FileUtils.mkdir_p(File.dirname(dest))
+          zip.extract z, dest
+        end
+      end
+    end
+  return temp_dir
  end
 
 end

data/lib/reseed/version.rb

@@ -1,3 +1,3 @@
 module Reseed
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/lib/reseed.rb

@@ -13,7 +13,7 @@ module Reseed
           options = []
           configblock.call options
           task = ReseedTask.new
-          task.execute options
+          task.execute name, options
         end
       end
     }

data/reseed.gemspec

@@ -22,6 +22,7 @@ Gem::Specification.new do |s|
   s.add_development_dependency "rspec"
   s.add_development_dependency "mocha"
   s.add_development_dependency "fakeweb"
+
   
-  # s.add_runtime_dependency "rest-client"
+  s.add_runtime_dependency "zip"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: reseed
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-04-18 00:00:00.000000000 Z
+date: 2012-04-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -59,6 +59,22 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: zip
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Reseeds dependencies and automatically checks out files in TFS if necessary.
 email:
 - joseph.andaverde@softekinc.com