💣 monkey patch on Builtin::Provision action \o/ (closes GH-60)
This commit is contained in:
parent
aa793367b7
commit
0f533a9b30
3 changed files with 50 additions and 57 deletions
40
lib/vagrant-cachier/action/install_buckets.rb
Normal file
40
lib/vagrant-cachier/action/install_buckets.rb
Normal file
|
@ -0,0 +1,40 @@
|
||||||
|
require_relative '../bucket'
|
||||||
|
|
||||||
|
module VagrantPlugins
|
||||||
|
module Cachier
|
||||||
|
class Action
|
||||||
|
class InstallBuckets
|
||||||
|
def initialize(app, env)
|
||||||
|
@app = app
|
||||||
|
end
|
||||||
|
|
||||||
|
def call(env)
|
||||||
|
@app.call(env)
|
||||||
|
|
||||||
|
@env = env
|
||||||
|
configure_cache_buckets
|
||||||
|
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
|
|
@ -4,11 +4,6 @@ unless Gem::Requirement.new('>= 1.4.0').satisfied_by?(Gem::Version.new(Vagrant::
|
||||||
raise 'vagrant-cachier requires Vagrant >= 1.4.0 in order to work!'
|
raise 'vagrant-cachier requires Vagrant >= 1.4.0 in order to work!'
|
||||||
end
|
end
|
||||||
|
|
||||||
require_relative 'provision_ext'
|
|
||||||
Vagrant::Action::Builtin::Provision.class_eval do
|
|
||||||
include VagrantPlugins::Cachier::ProvisionExt
|
|
||||||
end
|
|
||||||
|
|
||||||
# Add our custom translations to the load path
|
# Add our custom translations to the load path
|
||||||
I18n.load_path << File.expand_path("../../../locales/en.yml", __FILE__)
|
I18n.load_path << File.expand_path("../../../locales/en.yml", __FILE__)
|
||||||
|
|
||||||
|
@ -85,7 +80,17 @@ module VagrantPlugins
|
||||||
|
|
||||||
action_hook ALL_ACTIONS do |hook|
|
action_hook ALL_ACTIONS do |hook|
|
||||||
require_relative 'action/configure_bucket_root'
|
require_relative 'action/configure_bucket_root'
|
||||||
|
require_relative 'action/install_buckets'
|
||||||
|
|
||||||
hook.before Vagrant::Action::Builtin::Provision, Action::ConfigureBucketRoot
|
hook.before Vagrant::Action::Builtin::Provision, Action::ConfigureBucketRoot
|
||||||
|
# This will do the initial buckets installation
|
||||||
|
hook.after Vagrant::Action::Builtin::Provision, Action::InstallBuckets
|
||||||
|
end
|
||||||
|
|
||||||
|
# This ensure buckets are reconfigured after provisioners runs
|
||||||
|
action_hook :provisioner_run do |hook|
|
||||||
|
require_relative 'action/install_buckets'
|
||||||
|
hook.after :run_provisioner, Action::InstallBuckets
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
|
@ -1,52 +0,0 @@
|
||||||
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
|
|
Loading…
Reference in a new issue