knife-cleanup 0.0.1 → 0.0.2

data/lib/chef/knife/cleanup_versions.rb

@@ -19,7 +19,10 @@ module ServerCleanup
   class CleanupVersions < Chef::Knife
 
     deps do
+      require 'fileutils'
       require 'chef/api_client'
+      require 'chef/cookbook_loader'
+      require 'chef/knife/cookbook_download'
     end
 
     banner "knife cleanup versions (options)"
@@ -77,6 +80,8 @@ module ServerCleanup
         cbv[cb].each do |cb_ver|
           print "#{cb_ver} "
           if config[:delete]
+            dir = ".cleanup/#{cb}/"
+            backup_cookbook(cb,cb_ver,dir)
             delete_cookbook(cb, cb_ver)
           end
         end
@@ -90,8 +95,24 @@ module ServerCleanup
     end
     
     def delete_cookbook(cb, cb_ver)
+      ui.msg "Deleting cookbook #{cb} version #{cb_ver}"
       rest.delete_rest("cookbooks/#{cb}/#{cb_ver}")
       print ". "
     end
+
+    def backup_cookbook(cb,cb_ver,dir)
+      ui.msg "\nFist backing up cookbook #{cb} version #{cb_ver}"
+      FileUtils.mkdir_p(dir)
+      dld = Chef::Knife::CookbookDownload.new
+      dld.name_args = [cb, cb_ver]
+      dld.config[:download_directory] = dir
+      dld.config[:force] = true
+      begin
+        dld.run
+      rescue
+        ui.msg "Failed to download cookbook #{cb} version #{cb_ver}... Skipping"
+        FileUtils.rm_r(File.join(dir, cb + "-" + cb_ver))
+      end
+    end
   end
 end

data/lib/knife-cleanup/version.rb

@@ -1,5 +1,5 @@
 module Knife
   module Cleanup
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: knife-cleanup
 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-02-24 00:00:00.000000000 Z
+date: 2013-02-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: chef
@@ -53,6 +53,7 @@ extra_rdoc_files: []
 files:
 - .gitignore
 - Gemfile
+- README.md
 - Rakefile
 - knife-cleanup.gemspec
 - lib/chef/knife/cleanup_versions.rb