require 'fileutils'
require 'systemu'

module RSpecSystem
  # A NodeSet implementation for Vagrant.
  class NodeSet::Vagrant < RSpecSystem::NodeSet::Base
    include RSpecSystem::Log

    ENV_TYPE = 'vagrant'

    def initialize(setname, config)
      super
      @vagrant_path = File.expand_path(File.join(RSpec.configuration.system_tmp, 'vagrant_projects', setname))
    end

    # Setup the NodeSet by starting all nodes.
    def setup
      log.info "[Vagrant#setup] Begin setting up vagrant"
      create_vagrantfile

      log.info "[Vagrant#setup] Running 'vagrant destroy'"
      vagrant("destroy --force")

      log.info "[Vagrant#setup] Running 'vagrant up'"
      vagrant("up")
    end

    # Shutdown the NodeSet by shutting down or pausing all nodes.
    def teardown
      log.info "[Vagrant#teardown] Running 'vagrant destroy'"
      vagrant("destroy --force")
    end

    # Run a command on a host in the NodeSet.
    #
    # @param opts [Hash] options
    def run(opts)
      #log.debug("[Vagrant#run] called with #{opts.inspect}")

      dest = opts[:n].name
      cmd = opts[:c]

      r = nil
      Dir.chdir(@vagrant_path) do
        cmd = "vagrant ssh #{dest} --command \"cd /tmp && sudo #{cmd}\""
        log.debug("[vagrant#run] Running command: #{cmd}")
        r = systemu cmd
        log.debug("[Vagrant#run] Finished running command: #{cmd}.")
      end

      {
        :exit_code => r[0].exitstatus,
        :stdout => r[1],
        :stderr => r[2]
      }
    end

    # Transfer files to a host in the NodeSet.
    #
    # @param opts [Hash] options
    def rcp(opts)
      #log.debug("[Vagrant@rcp] called with #{opts.inspect}")

      dest = opts[:d].name
      source = opts[:sp]
      dest_path = opts[:dp]

      # TODO: This is damn ugly, because we ssh in as vagrant, we copy to a
      # temp path then move later. This pattern at the moment only really works
      # on dirs.
      log.info("[Vagrant#rcp] Transferring files from #{source} to #{dest}:#{dest_path}")

      # TODO: The static temp path here is definately insecure
      cmd = "scp -r -F '#{ssh_config}' '#{source}' #{dest}:/tmp/tmpxfer"
      log.debug("[Vagrant#rcp] Running command: #{cmd}")
      systemu cmd

      # Now we move the file into place
      run(:n => opts[:d], :c => "mv /tmp/tmpxfer #{dest_path}")
    end

    # Create the Vagrantfile for the NodeSet.
    #
    # @api private
    def create_vagrantfile
      log.info "[Vagrant#create_vagrantfile] Creating vagrant file here: #{@vagrant_path}"
      FileUtils.mkdir_p(@vagrant_path)
      File.open(File.expand_path(File.join(@vagrant_path, "Vagrantfile")), 'w') do |f|
        f.write('Vagrant::Config.run do |c|')
        nodes.each do |k,v|
          log.debug "Filling in content for #{k}"
          f.write(<<-EOS)
  c.vm.define '#{k}' do |vmconf|
    vmconf.vm.host_name = "#{k}"
#{template_node(v.provider_specifics['vagrant'])}
  end
          EOS
        end
        f.write('end')
      end
      log.debug "[Vagrant#create_vagrantfile] Finished creating vagrant file"
    end

    # Here we get vagrant to drop the ssh_config its using so we can monopolize
    # it for transfers and custom stuff. We drop it into a single file, and
    # since its indexed based on our own node names its quite ideal.
    #
    # @api private
    def ssh_config
      ssh_config_path = File.expand_path(File.join(@vagrant_path, "ssh_config"))
      begin
        File.unlink(ssh_config_path)
      rescue Errno::ENOENT
      end
      self.nodes.each do |k,v|
        Dir.chdir(@vagrant_path) do
          result = systemu("vagrant ssh-config #{k} >> #{ssh_config_path}")
          puts result.inspect
        end
      end
      ssh_config_path
    end

    # Provide Vagrantfile templates from node definition.
    #
    # @api private
    # @param settings [Hash] provider specific settings for vagrant
    def template_node(settings)
      template = <<-EOS
    vmconf.vm.box = '#{settings['box']}'
    vmconf.vm.box_url = '#{settings['box_url']}'
      EOS
    end

    # Execute vagrant command in vagrant_path
    #
    # @api private
    # @param args [String] args to vagrant
    # @todo This seems a little too specific these days, might want to
    #   generalize. It doesn't use systemu, because we want to see the output
    #   immediately, but still - maybe we can make systemu do that.
    def vagrant(args)
      Dir.chdir(@vagrant_path) do
        system("vagrant #{args}")
      end
      nil
    end
  end
end