class Pantry::Chef::UploadCookbook
Given a cookbook, upload it to the server. A given cookbook is stored in two locations. The cookbook tarball the Server receives is stored in Pantry.root/chef/cookbook-cache/.tgz to keep it available for sending back down to Clients who need the cookbook. To facilitate using the Chef
library for dependency checks, the Server also unpacks the cookbook locally to Pantry.root/chef/cookbooks/.
Attributes
Public Class Methods
# File lib/pantry/chef/upload_cookbook.rb, line 18 def initialize(cookbook_path = nil) @cookbook_path = cookbook_path end
Public Instance Methods
Server receives request message for a new Cookbook Upload. Checks that the upload is valid Fires off an upload receiver and returns the UUID for the client to use
# File lib/pantry/chef/upload_cookbook.rb, line 63 def perform(message) cookbook_name = message[:cookbook_name] cookbook_size = message[:cookbook_size] cookbook_checksum = message[:cookbook_checksum] cookbook_cache = Pantry.root.join("chef", "cookbook-cache") cookbook_home = Pantry.root.join("chef", "cookbooks") FileUtils.mkdir_p(cookbook_cache) FileUtils.mkdir_p(cookbook_home) save_tar_path = cookbook_cache.join("#{cookbook_name}.tgz") uploader_info = server.receive_file(cookbook_size, cookbook_checksum) uploader_info.on_complete do # Store the tarball into the cookbook cache FileUtils.mv uploader_info.uploaded_path, save_tar_path # Unpack the cookbook itself stdout, stderr = Open3.capture2e( "tar", "-xzC", cookbook_home.to_s, "-f", save_tar_path.to_s ) Pantry.logger.debug("[Upload Cookbook] Unpack cookbook #{stdout.inspect}, #{stderr.inspect}") end [true, uploader_info.receiver_uuid, uploader_info.file_uuid] rescue => ex [false, ex.message] end
Multi-step prepratory step here:
-
Find the cookbook in question
-
Figure out if it’s a valid cookbook (do some checks
Chef
doesn’t itself do) -
Tar up the cookbook
-
Figure out size and a checksum
-
Package all this information into the message to send to the server
# File lib/pantry/chef/upload_cookbook.rb, line 30 def prepare_message(options) require 'chef/cookbook_loader' cookbook_name = File.basename(@cookbook_path) cookbooks_dir = File.dirname(@cookbook_path) loader = ::Chef::CookbookLoader.new([cookbooks_dir]) cookbook = loader.load_cookbooks[cookbook_name] raise UnknownCookbook, "Unable to find cookbook at #{@cookbook_path}" unless cookbook raise MissingMetadata, "No metadata.rb found for cookbook at #{@cookbook_path}" unless File.exist?(File.join(@cookbook_path, "metadata.rb")) tempfile = Tempfile.new(cookbook_name) @cookbook_tarball = "#{tempfile.path}.tgz" tempfile.unlink # TODO Handle if this fails? Dir.chdir(cookbooks_dir) do Open3.capture2("tar", "czf", @cookbook_tarball, cookbook_name) end Pantry.ui.say("Uploading cookbook #{cookbook_name}...") message = super message[:cookbook_name] = cookbook.metadata.name message[:cookbook_size] = File.size(@cookbook_tarball) message[:cookbook_checksum] = Pantry.file_checksum(@cookbook_tarball) message end
# File lib/pantry/chef/upload_cookbook.rb, line 92 def receive_server_response(response) upload_allowed = response.body[0] Pantry.logger.debug("[Upload Cookbook] #{response.inspect}") if upload_allowed == "true" send_info = client.send_file(@cookbook_tarball, response.body[1], response.body[2]) send_info.wait_for_finish else Pantry.ui.say("ERROR: #{response.body[1]}") end end