Contents

class TestLab

  class Provisioner

    # Resolv Provisioner Error Class
    class ResolvError < ProvisionerError; end

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

      def initialize(config={}, ui=nil)
        @config = (config || Hash.new)
        @ui     = (ui     || TestLab.ui)

        @config[:resolv] ||= Hash.new

        @config[:resolv][:servers] ||= Array.new
        @config[:resolv][:servers].concat([TestLab::Network.ips])

        @config[:resolv][:search] ||= Array.new
        @config[:resolv][:search].concat([TestLab::Container.domains])

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

      # Resolv: Node Provision
      #
      # @param [TestLab::Node] node The node which we want to provision.
      # @return [Boolean] True if successful.
      def on_node_provision(node)
        @ui.logger.debug { "RESOLV Provisioner: Node #{node.id}" }

        @config[:resolv][:servers] = %w(127.0.0.1)

        render_resolv_conf(node)

        true
      end

      # Resolv: Container Provision
      #
      # @param [TestLab::Container] container The container which we want to
      #   provision.
      # @return [Boolean] True if successful.
      def on_container_provision(container)
        @ui.logger.debug { "RESOLV Provisioner: Container #{container.id}" }

        render_resolv_conf(container)

        true
      end

      def render_resolv_conf(object)
        object.file(:target => %(/etc/resolv.conf), :chown => "root:root") do |file|
          file.puts(ZTK::Template.do_not_edit_notice(:message => "TestLab v#{TestLab::VERSION} RESOLVER Configuration"))
          file.puts(ZTK::Template.render(resolv_conf_template, @config))
        end
      end

    private

      def resolv_conf_template
        File.join(TestLab::Provisioner.template_dir, 'resolv', 'resolv.conf.erb')
      end

    end

  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
testlab-1.2.1	lib/testlab/provisioners/resolv.rb
testlab-1.2.0	lib/testlab/provisioners/resolv.rb
testlab-1.1.0	lib/testlab/provisioners/resolv.rb