class Chef::Knife::CookbookSiteShare

Public Instance Methods

do_upload(cookbook_filename, cookbook_category, user_id, user_secret_filename) click to toggle source
# File lib/chef/knife/cookbook_site_share.rb, line 125
def do_upload(cookbook_filename, cookbook_category, user_id, user_secret_filename)
  uri = "#{config[:supermarket_site]}/api/v1/cookbooks"

  category_string = Chef::JSONCompat.to_json({ "category" => cookbook_category })

  http_resp = Chef::CookbookSiteStreamingUploader.post(uri, user_id, user_secret_filename, {
    :tarball => File.open(cookbook_filename),
    :cookbook => category_string,
  })

  res = Chef::JSONCompat.from_json(http_resp.body)
  if http_resp.code.to_i != 201
    if res["error_messages"]
      if res["error_messages"][0] =~ /Version already exists/
        ui.error "The same version of this cookbook already exists on Supermarket."
        exit(1)
      else
        ui.error "#{res['error_messages'][0]}"
        exit(1)
      end
    else
      ui.error "Unknown error while sharing cookbook"
      ui.error "Server response: #{http_resp.body}"
      exit(1)
    end
  end
  res
end
get_category(cookbook_name) click to toggle source
# File lib/chef/knife/cookbook_site_share.rb, line 115
def get_category(cookbook_name)
  data = noauth_rest.get("#{config[:supermarket_site]}/api/v1/cookbooks/#{@name_args[0]}")
  data["category"]
rescue => e
  return "Other" if e.kind_of?(Net::HTTPServerException) && e.response.code == "404"
  ui.fatal("Unable to reach Supermarket: #{e.message}. Increase log verbosity (-VV) for more information.")
  Chef::Log.trace("\n#{e.backtrace.join("\n")}")
  exit(1)
end
run() click to toggle source
# File lib/chef/knife/cookbook_site_share.rb, line 60
def run
  config[:cookbook_path] ||= Chef::Config[:cookbook_path]

  if @name_args.length < 1
    show_usage
    ui.fatal("You must specify the cookbook name.")
    exit(1)
  elsif @name_args.length < 2
    cookbook_name = @name_args[0]
    category = get_category(cookbook_name)
  else
    cookbook_name = @name_args[0]
    category = @name_args[1]
  end

  cl = Chef::CookbookLoader.new(config[:cookbook_path])
  if cl.cookbook_exists?(cookbook_name)
    cookbook = cl[cookbook_name]
    Chef::CookbookUploader.new(cookbook).validate_cookbooks
    tmp_cookbook_dir = Chef::CookbookSiteStreamingUploader.create_build_dir(cookbook)
    begin
      Chef::Log.trace("Temp cookbook directory is #{tmp_cookbook_dir.inspect}")
      ui.info("Making tarball #{cookbook_name}.tgz")
      shell_out!("#{tar_cmd} -czf #{cookbook_name}.tgz #{cookbook_name}", :cwd => tmp_cookbook_dir)
    rescue => e
      ui.error("Error making tarball #{cookbook_name}.tgz: #{e.message}. Increase log verbosity (-VV) for more information.")
      Chef::Log.trace("\n#{e.backtrace.join("\n")}")
      exit(1)
    end

    if config[:dry_run]
      ui.info("Not uploading #{cookbook_name}.tgz due to --dry-run flag.")
      result = shell_out!("#{tar_cmd} -tzf #{cookbook_name}.tgz", :cwd => tmp_cookbook_dir)
      ui.info(result.stdout)
      FileUtils.rm_rf tmp_cookbook_dir
      return
    end

    begin
      do_upload("#{tmp_cookbook_dir}/#{cookbook_name}.tgz", category, Chef::Config[:node_name], Chef::Config[:client_key])
      ui.info("Upload complete")
      Chef::Log.trace("Removing local staging directory at #{tmp_cookbook_dir}")
      FileUtils.rm_rf tmp_cookbook_dir
    rescue => e
      ui.error("Error uploading cookbook #{cookbook_name} to Supermarket: #{e.message}. Increase log verbosity (-VV) for more information.")
      Chef::Log.trace("\n#{e.backtrace.join("\n")}")
      exit(1)
    end

  else
    ui.error("Could not find cookbook #{cookbook_name} in your cookbook path.")
    exit(1)
  end
end
tar_cmd() click to toggle source
# File lib/chef/knife/cookbook_site_share.rb, line 154
def tar_cmd
  if !@tar_cmd
    @tar_cmd = "tar"
    begin
      # Unix and Mac only - prefer gnutar
      if shell_out("which gnutar").exitstatus.equal?(0)
        @tar_cmd = "gnutar"
      end
    rescue Errno::ENOENT
    end
  end
  @tar_cmd
end