Add cleanup_disks action

This commit is contained in:
Brian Cain 2020-01-14 10:00:55 -08:00
parent 5bf8d2363e
commit cc8ed8a616
No known key found for this signature in database
GPG Key ID: 9FC4639B2E4510A0

View File

@ -0,0 +1,49 @@
require "json"
module Vagrant
module Action
module Builtin
class CleanupDisks
# Removes any attached disks no longer defined in a Vagrantfile config
def initialize(app, env)
@app = app
@logger = Log4r::Logger.new("vagrant::action::builtin::disk")
end
def call(env)
machine = env[:machine]
defined_disks = get_disks(machine, env)
# Call into providers machine implementation for disk management
if !defined_disks.empty?
if machine.provider.capability?(:cleanup_disks)
machine.provider.capability(:cleanup_disks, defined_disks)
else
env[:ui].warn(I18n.t("vagrant.actions.disk.provider_unsupported",
provider: machine.provider_name))
end
end
# Continue On
@app.call(env)
end
def read_disk_metadata(machine)
meta_file = machine.data_dir.join("disk_meta")
disk_meta = JSON.parse(meta_file.read)
return disk_meta
end
def get_disks(machine, env)
return @_disks if @_disks
@_disks = []
@_disks = machine.config.vm.disks
@_disks
end
end
end
end
end