################################################################################
#
#      Author: Stephen Nelson-Smith <stephen@atalanta-systems.com>
#      Author: Zachary Patten <zachary@jovelabs.com>
#   Copyright: Copyright (c) 2011-2013 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 || {
      :log_level => :info,
      :log_location => "/var/log/chef/client.log",
      :chef_server_url => "https://api.opscode.com/organizations/#{config[:orgname]}",
      :validation_client_name => "#{config[:orgname]}-validator",
      :ssl_verify_mode => :verify_none,
      :environment => nil # use default; i.e. set no value
    }).merge(config)
    logger.info { "Setting chef client config '#{@chef_client_config.inspect}'." }

    true
  end

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

  # call this before chef_run_client
  def chef_set_client_attributes(name, attributes={})
    @containers[name] ||= Hash.new
    @containers[name][:chef_client] = (@containers[name][:chef_client] || {}).merge(attributes) { |k,o,n| (k = (o + n).uniq) }
    logger.info { "Setting chef client attributes to '#{@containers[name][:chef_client].inspect}' for container '#{name}'." }

    true
  end

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

  def chef_run_client(name, *args)
    chef_config_client(name)

    logger.info { "Removing artifacts #{Cucumber::Chef::Config[:artifacts].values.collect{|z| "'#{z}'" }.join(' ')}." }
    (command_run_chroot(name, "/bin/rm -fv #{Cucumber::Chef::Config[:artifacts].values.join(' ')}") rescue nil)

    logger.info { "Running chef client on container '#{name}'." }

    arguments = {
      "--json-attributes" => File.join("/etc", "chef", "attributes.json").to_s,
      "--log_level" => (ENV['LOG_LEVEL'] || "INFO").downcase
    }.reject{ |k,v| v.nil? }.sort

    output = nil
    bm = ::Benchmark.realtime do
      output = command_run_chroot(name, ["/usr/bin/chef-client", arguments, args].flatten.join(" "))
    end
    logger.info { "Chef client run on container '#{name}' took %0.4f seconds." % bm }

    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))

    max_key_size = @chef_client_config.keys.collect{ |z| z.to_s.size }.max

    File.open(client_rb, 'w') do |f|
      f.puts(Cucumber::Chef.generate_do_not_edit_warning("Chef Client Configuration"))
      f.puts
      @chef_client_config.merge(:node_name => name).each do |(key,value)|
        next if value.nil?
        f.puts("%-#{max_key_size}s  %s" % [key, value.inspect])
      end
      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((@containers[name][:chef_client] || {}).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")

    true
  end

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

  def chef_client_artifacts(name)
    # this is messy and needs to be refactored into a more configurable
    # solution; but for now this should do the trick

    ssh = $test_lab.proxy_ssh(name)

    feature_file = $scenario.file_colon_line.split(":").first
    feature_line = $scenario.file_colon_line.split(":").last
    scenario_tag = $scenario.name.gsub(" ", "_")

    feature_name = File.basename(feature_file, ".feature")
    feature_dir = feature_file.split("/")[-2]

    Cucumber::Chef::Config[:artifacts].each do |label, remote_path|
      result = ssh.exec("/bin/bash -c '[[ -f #{remote_path} ]] ; echo $?'", :silence => true)
      if (result.output =~ /0/)
        logger.info { "Retrieving artifact '#{remote_path}' from container '#{name}'." }

        local_path = File.join(Cucumber::Chef.artifacts_dir, feature_dir, "#{feature_name}.txt")
        tmp_path = File.join("/tmp", label)

        FileUtils.mkdir_p(File.dirname(local_path))
        ssh.download(remote_path, tmp_path)
        data = IO.read(tmp_path).chomp

        message = "#{$scenario.name} (#{File.basename(feature_file)}:#{feature_line}:#{label})"
        header = ("-" * message.length)

        f = File.open(local_path, "a")
        f.write("#{header}\n")
        f.write("#{message}\n")
        f.write("#{header}\n")
        f.write("#{data}\n")

        File.chmod(0644, local_path)
      end
    end

    true
  end

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

end

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