knife-tar 1.3.0

data/lib/chef/knife/data_bag_tar_download.rb

@@ -0,0 +1,44 @@
+require 'chef/knife'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+require 'chef/data_bag'
+require 'chef/data_bag_item'
+
+class Chef
+  class Knife
+    class DataBagTarDownload < Chef::Knife
+
+      banner "knife data bag tar download tarPath [options]"
+      category "data bag tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args.first, true)
+        DataBagTarDownload.download_data_bags tar_file
+        tar_file.save
+
+      end
+      
+      def self.download_data_bags(tar_file)
+        dir = tar_file.data_bags_path
+        Chef::DataBag.list.each do |bag_name, url|
+          system("mkdir -p #{File.join(dir, bag_name)}")
+          Chef::DataBag.load(bag_name).each do |item_name, itemUrl|
+            Chef::Log.info("Backing up data bag #{bag_name} item #{item_name}")
+            item = Chef::DataBagItem.load(bag_name, item_name)
+            File.open(File.join(dir, bag_name, "#{item_name}.json"), "w") do |dbag_file|
+              dbag_file.print(item.raw_data.to_json)
+            end
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/data_bag_tar_upload.rb

@@ -0,0 +1,53 @@
+require 'chef/knife'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+
+class Chef
+  class Knife
+    class DataBagTarUpload < Chef::Knife
+
+      banner "knife data bag tar upload tarPath [options]"
+      category "data bag tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args.first)
+        DataBagTarUpload.upload_data_bags tar_file
+        
+      end
+      
+      def self.upload_data_bags(tar_file)
+        data_bag_from_file = Chef::Knife::DataBagFromFile.new
+        
+        tar_file.data_bags.each do |databag_path|
+          
+          #TODO: May want to consider moving this logic into TarFile.
+          
+          databag = File.basename(File.expand_path("..", databag_path))
+          
+          #In order to upload a data bag value the data bag itself must exist
+          #so attempt to create it now
+          
+          #We make the assumption here that the parent directory of each data bag
+          #file is the data bag name. 
+          
+          databag_create = Chef::Knife::DataBagCreate.new
+          databag_create.name_args = [databag]
+          databag_create.run
+          
+          #To upload a data bag we must know the data bag name and the path to the file
+          
+          data_bag_from_file.name_args = [databag, databag_path]
+          data_bag_from_file.run
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/environment_tar_download.rb

@@ -0,0 +1,44 @@
+require 'chef/knife'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+require 'chef/environment'
+
+class Chef
+  class Knife
+    class EnvironmentTarDownload < Chef::Knife
+
+      @@DEFAULT_ENVIRONMENT = "_default"
+
+      banner "knife environment tar download tarPath [options]"
+      category "environment tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args.first, true)
+        EnvironmentTarDownload.download_environments tar_file
+        tar_file.save
+        
+      end
+      
+      def self.download_environments(tar_file)
+        dir = tar_file.environments_path
+        Chef::Environment.list.each do |component_name, url|
+          if component_name != @@DEFAULT_ENVIRONMENT
+            Chef::Log.info("Backing up environment #{component_name}")
+            component_obj = Chef::Environment.load(component_name)
+            File.open(File.join(dir, "#{component_name}.json"), "w") do |component_file|
+              component_file.print(component_obj.to_json)
+            end
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/environment_tar_upload.rb

@@ -0,0 +1,36 @@
+require 'chef/knife'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+
+class Chef
+  class Knife
+    class EnvironmentTarUpload < Chef::Knife
+
+      banner "knife environment tar upload tarPath [options]"
+      category "environment tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args.first)
+        EnvironmentTarUpload.upload_environments tar_file
+        
+      end
+      
+      def self.upload_environments(tar_file)
+        environment_from_file = Chef::Knife::EnvironmentFromFile.new
+        
+        tar_file.environments.each do |environment_path|
+          environment_from_file.name_args = [environment_path]
+          environment_from_file.run
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/node_tar_download.rb

@@ -0,0 +1,40 @@
+require 'chef/knife'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+require 'chef/node'
+
+class Chef
+  class Knife
+    class NodeTarDownload < Chef::Knife
+
+      banner "knife node tar download tarPath [options]"
+      category "node tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args.first, true)
+        NodeTarDownload.download_nodes tar_file
+        tar_file.save
+        
+      end
+      
+      def self.download_nodes(tar_file)
+        dir = tar_file.nodes_path
+        Chef::Node.list.each do |component_name, url|
+          Chef::Log.info("Backing up node #{component_name}")
+          component_obj = Chef::Node.load(component_name)
+          File.open(File.join(dir, "#{component_name}.json"), "w") do |component_file|
+            component_file.print(component_obj.to_json)
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/node_tar_upload.rb

@@ -0,0 +1,37 @@
+require 'chef/knife'
+require 'chef/node'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+
+class Chef
+  class Knife
+    class NodeTarUpload < Chef::Knife
+
+      banner "knife node tar upload tarPath [options]"
+      category "node tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args.first)
+        NodeTarUpload.upload_nodes tar_file
+
+      end
+      
+      def self.upload_nodes(tar_file)
+        node_from_file = Chef::Knife::NodeFromFile.new
+        
+        tar_file.nodes.each do |nodes_path|
+          node_from_file.name_args = [nodes_path]
+          node_from_file.run
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/role_tar_download.rb

@@ -0,0 +1,40 @@
+require 'chef/knife'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+require 'chef/role'
+
+class Chef
+  class Knife
+    class RoleTarDownload < Chef::Knife
+
+      banner "knife role tar download tarPath [options]"
+      category "role tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args.first, true)
+        RoleTarDownload.download_roles tar_file
+        tar_file.save
+        
+      end
+      
+      def self.download_roles(tar_file)
+        dir = tar_file.roles_path
+        Chef::Role.list.each do |component_name, url|
+          Chef::Log.info("Backing up role #{component_name}")
+          component_obj = Chef::Role.load(component_name)
+          File.open(File.join(dir, "#{component_name}.json"), "w") do |component_file|
+            component_file.print(component_obj.to_json)
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/role_tar_upload.rb

@@ -0,0 +1,36 @@
+# coding: UTF-8
+
+require 'chef/knife'
+require 'chef/role'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+
+class Chef
+  class Knife
+    class RoleTarUpload < Chef::Knife
+
+      banner "knife role tar upload TARPATH [options]"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args[0])
+        RoleTarUpload.upload_roles tar_file
+        
+      end
+      
+      def self.upload_roles tar_file
+        role_from_file = Chef::Knife::RoleFromFile.new 
+        role_from_file.name_args = tar_file.roles
+        role_from_file.config[:print_after] = true
+        role_from_file.run
+      end
+
+    end
+  end
+end

data/lib/chef/knife/tar_download_all.rb

@@ -0,0 +1,36 @@
+require 'chef/knife'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+
+class Chef
+  class Knife
+    class TarDownload < Chef::Knife
+
+      banner "knife tar download tarPath [options]"
+      category "tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args.first, true)
+        
+        ClientTarDownload.download_clients tar_file
+        CookbookTarDownload.download_cookbooks tar_file
+        DataBagTarDownload.download_data_bags tar_file
+        EnvironmentTarDownload.download_environments tar_file
+        NodeTarDownload.download_nodes tar_file
+        RoleTarDownload.download_roles tar_file
+        UserTarDownload.download_users tar_file
+        
+        tar_file.save
+        
+      end
+
+    end
+  end
+end

data/lib/chef/knife/tar_upload_all.rb

@@ -0,0 +1,70 @@
+require 'chef/knife'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+
+class Chef
+  class Knife
+    class TarUpload < Chef::Knife
+
+      banner "knife tar upload tarPath [options]"
+      category "tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new @name_args.first
+        
+        # Attempt to upload all the components in the tar file
+        # If our tar file does not contain a component ignore the error and skip it
+        begin
+          ClientTarUpload.upload_clients tar_file
+        rescue TarFile::MissingChefComponentError => e
+          ui.info("No Client files to upload")
+        end
+        
+        begin
+          CookbookTarUpload.upload_cookbooks tar_file
+        rescue TarFile::MissingChefComponentError => e
+          ui.info("No Cookbooks to upload")
+        end
+        
+        begin
+          DataBagTarUpload.upload_data_bags tar_file
+        rescue TarFile::MissingChefComponentError => e
+          ui.info("No data bag files to upload")
+        end
+        
+        begin
+          EnvironmentTarUpload.upload_environments tar_file
+        rescue TarFile::MissingChefComponentError => e
+          ui.info("No Environment files to upload")
+        end
+        
+        begin
+          NodeTarUpload.upload_nodes tar_file
+        rescue TarFile::MissingChefComponentError => e
+          ui.info("No Node files to upload")
+        end
+        
+        begin
+          RoleTarUpload.upload_roles tar_file
+        rescue TarFile::MissingChefComponentError => e
+          ui.info("No Role files to upload")
+        end
+        
+        begin
+          UserTarUpload.upload_users tar_file
+        rescue TarFile::MissingChefComponentError => e
+          ui.info("No User files to upload")
+        end
+        
+      end
+
+    end
+  end
+end

data/lib/chef/knife/user_tar_download.rb

@@ -0,0 +1,40 @@
+require 'chef/knife'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+require 'chef/webui_user'
+
+class Chef
+  class Knife
+    class UserTarDownload < Chef::Knife
+
+      banner "knife user tar download tarPath [options]"
+      category "user tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args.first, true)
+        UserTarDownload.download_users tar_file
+        tar_file.save
+        
+      end
+      
+      def self.download_users(tar_file)
+        dir = tar_file.web_users_path
+        Chef::WebUIUser.list.each do |component_name, url|
+          Chef::Log.info("Backing up user #{component_name}")
+          component_obj = Chef::WebUIUser.load(component_name)
+          File.open(File.join(dir, "#{component_name}.json"), "w") do |component_file|
+            component_file.print(component_obj.to_json)
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/knife/user_tar_upload.rb

@@ -0,0 +1,49 @@
+require 'chef/knife'
+require 'chef/knife/core/object_loader'
+require 'chef/json_compat'
+require 'chef/webui_user'
+
+class Chef
+  class Knife
+    class UserTarUpload < Chef::Knife
+
+      banner "knife user tar upload tarPath [options]"
+      category "user tar"
+
+      def run
+        #Get Arguments
+        if @name_args.size != 1
+          ui.info("Please specify a tar path")
+          show_usage
+          exit 1
+        end
+        
+        tar_file = Chef::TarFile.new(@name_args.first)
+        UserTarUpload.upload_users tar_file
+        
+      end
+      
+      def self.upload_users(tar_file)
+        current_users = Chef::WebUIUser.list.keys
+        users_loader = Chef::Knife::Core::ObjectLoader.new(Chef::WebUIUser, ui)
+        
+        tar_file.web_users.each do |web_user_path|
+          
+          user = users_loader.load_from("users", web_user_path)
+          
+           # In order to 'update' a user we have to remove it first, so if the user exists destroy it
+          if current_users.include? user.name
+            ui.info("Deleting Chef User [#{user.name}] in order to update it")
+            WebUIUser.load(user.name).destroy
+          end
+          
+          user.save
+          ui.info("Updated User : #{user.name}")
+        end
+      end
+      
+      
+
+    end
+  end
+end