Contents

class TestLab

  class Provisioner

    # HostsFile Provisioner Error Class
    class HostsFileError < ProvisionerError; end

    # HostsFile Provisioner Class
    #
    # @author Zachary Patten <zachary AT jovelabs DOT com>
    class HostsFile

      def initialize(config={}, ui=nil)
        @config  = (config || Hash.new)
        @ui      = (ui     || TestLab.ui)
        @command = ZTK::Command.new(:ui => @ui, :silence => true, :ignore_exit_status => true)

        @ui.logger.debug { "config(#{@config.inspect})" }
      end

      # HostsFile: Container Provision
      def on_container_callback(container)
        remove_hosts
        add_hosts(container)

        true
      end
      alias :on_container_create :on_container_callback
      alias :on_container_up :on_container_callback
      alias :on_container_provision :on_container_callback

      alias :on_container_deprovision:on_container_callback
      alias :on_container_down :on_container_callback
      alias :on_container_destroy :on_container_callback

    private

      def add_hosts(container)
        @command.exec(<<-EOF)
set -x
cat <<EOI | #{sudo} tee -a /etc/hosts
#{def_tag}
#{hosts_blob(container)}
#{end_tag}
EOI
        EOF
      end

      def remove_hosts
        @command.exec(sed_hostsfile)
      end

      def hosts_blob(container)
        blob = Array.new
        container.node.containers.each do |con|
          blob << "#{con.primary_interface.ip}\t#{con.id} #{con.fqdn}"
        end

        blob.join("\n")
      end

      def sed_hostsfile
        case RUBY_PLATFORM
        when /darwin/ then
          %(#{sudo} sed -i '' '/#{def_tag}/,/#{end_tag}/d' /etc/hosts)
        when /linux/ then
          %(#{sudo} sed -i '/#{def_tag}/,/#{end_tag}/d' /etc/hosts)
        end
      end

      # NFS Exports Start Definition Tag
      def def_tag
        "#TESTLAB-HOSTSFILE"
      end

      # NFS Exports End Definition Tag
      def end_tag
        "#TESTLAB-HOSTSFILE"
      end

    end

  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
testlab-1.11.1	lib/testlab/provisioners/hosts_file.rb
testlab-1.11.0	lib/testlab/provisioners/hosts_file.rb
testlab-1.10.0	lib/testlab/provisioners/hosts_file.rb
testlab-1.9.2	lib/testlab/provisioners/hosts_file.rb
testlab-1.9.1	lib/testlab/provisioners/hosts_file.rb
testlab-1.9.0	lib/testlab/provisioners/hosts_file.rb