module Vagrant
  module Boxen
    class Provisioner < Vagrant::Provisioners::Base
      class Config < Vagrant::Config::Base
        def memcached!
          @memcached = true
        end

        def memcached?
          @memcached
        end
      end

      def self.config_class
        Config
      end

      def initialize(env, config, provisioner = nil)
        super(env, config)
        @logger        = Log4r::Logger.new("vagrant::provisioners::boxen")
        @manifests_dir = "/tmp/vagrant-boxen-#{env['vm'].uuid}"
        @puppet_provisioner = provisioner ? provisioner : setup_puppet_provisioner
      end

      def prepare
        Dir.mkdir @manifests_dir unless File.directory? @manifests_dir
        if config.memcached?
          File.open("#{@manifests_dir}/site.pp", 'w') { |f| f.print "class { 'memcached': }" }
        else
          File.open("#{@manifests_dir}/site.pp", 'w') { |f| f.print "" }
        end
        @puppet_provisioner.prepare
      end

      def provision!
        @puppet_provisioner.provision!
      end

      private

      def setup_puppet_provisioner
        config = Vagrant::Provisioners::Puppet::Config.new
        config.manifests_path = @manifests_dir
        config.manifest_file  = "site.pp"
        config.module_path    = File.join(File.expand_path('../../../', __FILE__), 'puppet/modules')
        # The root path to be used on the guest machine, changed to avoid
        # collision with the default path for puppet provisioner
        config.pp_path = '/tmp/vagrant-boxen-puppet'
        config.options << [ '--verbose', '--debug '] if ENV['DEBUG'] == '1'
        Vagrant::Provisioners::Puppet.new(env, config)
      end
    end
  end
end