petems/vagrant-adam

require 'vagrant'

module VagrantPlugins
  module Adam
    # Config definition to validate given configuration values
    class Config < Vagrant.plugin('2', :config)
      attr_accessor :provision_url

      def initialize
        @provision_url = UNSET_VALUE
        @logger = Log4r::Logger.new('vagrantplugins::adam::config')
      end

      def finalize!
        @provision_url = nil if @provision_url == UNSET_VALUE
      end

      def validate!(_machine)
        finalize!
        errors = []

        unless @provision_url.nil?
          unless (valid_uri? @provision_url) || (valid_file? @provision_url)
            unless (valid_uri? @provision_url)
              msg = <<-EOH
  '#{ @provision_url }' is not a valid URL
              EOH
              errors << msg
            end
            unless (valid_file? @provision_url)
              msg = <<-EOH
  File '#{ @provision_url }' could not be found.
              EOH
              errors << msg
            end
          end
        end

        if errors.any?
          rendered_errors = Vagrant::Util::TemplateRenderer.render(
                              'config/validation_failed',
                              errors: { 'vagrant-adam' => errors }
                            )
          fail Vagrant::Errors::ConfigInvalid, errors: rendered_errors
        end
      end

      private

      def valid_uri?(value)
        uri = URI.parse value
        uri.is_a?(URI::HTTP) || uri.is_a?(URI::HTTPS)
      rescue URI::InvalidURIError
        false
      end

      def valid_file?(value)
        File.file?(value)
      end
    end
  end
end