52 lines
1.4 KiB
Ruby
52 lines
1.4 KiB
Ruby
require_relative 'bucket'
|
|
|
|
module VagrantPlugins
|
|
module Cachier
|
|
module ProvisionExt
|
|
def self.included(base)
|
|
base.class_eval do
|
|
def cachier_debug(msg)
|
|
@logger.debug "[CACHIER] #{msg}"
|
|
end
|
|
|
|
alias :old_call :call
|
|
def call(env)
|
|
@env = env
|
|
|
|
return old_call(env) unless env[:machine].config.cache.enabled?
|
|
|
|
old_call(env)
|
|
|
|
configure_cache_buckets
|
|
end
|
|
|
|
alias :old_run_provisioner :run_provisioner
|
|
def run_provisioner(*args)
|
|
configure_cache_buckets
|
|
old_run_provisioner(*args)
|
|
end
|
|
|
|
def configure_cache_buckets
|
|
return unless @env[:machine].config.cache.enabled?
|
|
|
|
if @env[:machine].config.cache.auto_detect
|
|
Bucket.auto_detect(@env)
|
|
end
|
|
|
|
return unless @env[:machine].config.cache.buckets.any?
|
|
|
|
@env[:ui].info 'Configuring cache buckets...'
|
|
cache_config = @env[:machine].config.cache
|
|
cache_config.buckets.each do |bucket_name, configs|
|
|
cachier_debug "Installing #{bucket_name} with configs #{configs.inspect}"
|
|
Bucket.install(bucket_name, @env, configs)
|
|
end
|
|
|
|
data_file = @env[:machine].data_dir.join('cache_dirs')
|
|
data_file.open('w') { |f| f.print @env[:cache_dirs].uniq.join("\n") }
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|