################################################################################
#
#      Author: Stephen Nelson-Smith <stephen@atalanta-systems.com>
#      Author: Zachary Patten <zachary@jovelabs.com>
#   Copyright: Copyright (c) 2011-2012 Atalanta Systems Ltd
#     License: Apache License, Version 2.0
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.
#
################################################################################

module Cucumber::Chef::Helpers::ChefClient

################################################################################

  # call this in a Before hook
  def chef_set_client_config(config={})
    @chef_client_config = (@chef_client_config || {
      :environment => "_default",
      :log_level => :debug,
      :log_location => "/var/log/chef/client.log",
      :chef_server_url => "https://api.opscode.com/organizations/#{config[:orgname]}",
      :validation_client_name => "#{config[:orgname]}-validator"
    }).merge(config)
  end

################################################################################

  # call this before chef_run_client
  def chef_set_client_attributes(name, attributes={})
    @chef_client_attributes = (@chef_client_attributes || {}).merge(attributes) { |k,o,n| (k = (o + n)) }
  end

################################################################################

  def chef_run_client(name,*args)
    chef_config_client(name)
    output = command_run_remote(name, ["/usr/bin/chef-client -j /etc/chef/attributes.json -N #{name}", args].join(" "))
    log("chef-client", "ran on node '#{name}'")
    output
  end

################################################################################

  def chef_config_client(name)
    # make sure our configuration location is there
    client_rb = File.join("/", container_root(name), "etc/chef/client.rb")
    FileUtils.mkdir_p(File.dirname(client_rb))

    File.open(client_rb, 'w') do |f|
      f.puts(Cucumber::Chef.generate_do_not_edit_warning("Chef Client Configuration"))
      f.puts
      f.puts("log_level               :#{@chef_client_config[:log_level]}")
      f.puts("log_location            \"#{@chef_client_config[:log_location]}\"")
      f.puts("chef_server_url         \"#{@chef_client_config[:chef_server_url]}\"")
      f.puts("ssl_verify_mode         :verify_none")
      f.puts("validation_client_name  \"#{@chef_client_config[:validation_client_name]}\"")
      f.puts("node_name               \"#{name}\"")
      f.puts("environment             \"#{@chef_client_config[:environment]}\"")
      f.puts
      f.puts("Mixlib::Log::Formatter.show_time = true")
    end

    attributes_json = File.join("/", container_root(name), "etc", "chef", "attributes.json")
    FileUtils.mkdir_p(File.dirname(attributes_json))
    File.open(attributes_json, 'w') do |f|
      f.puts((@chef_client_attributes || {}).to_json)
    end

    # make sure our log location is there
    log_location = File.join("/", container_root(name), @chef_client_config[:log_location])
    FileUtils.mkdir_p(File.dirname(log_location))

    command_run_local("cp /etc/chef/validation.pem #{container_root(name)}/etc/chef/ 2>&1")
  end

################################################################################

end

################################################################################