148 lines
4.3 KiB
Ruby
148 lines
4.3 KiB
Ruby
require 'log4r'
|
|
|
|
require "vagrant"
|
|
require 'vagrant/util/platform'
|
|
|
|
module VagrantPlugins
|
|
module HostBSD
|
|
# Represents a BSD host, such as FreeBSD and Darwin (Mac OS X).
|
|
class Host < Vagrant.plugin("2", :host)
|
|
include Vagrant::Util
|
|
include Vagrant::Util::Retryable
|
|
|
|
def self.match?
|
|
Vagrant::Util::Platform.darwin? || Vagrant::Util::Platform.bsd?
|
|
end
|
|
|
|
def self.precedence
|
|
# Set a lower precedence because this is a generic OS. We
|
|
# want specific distros to match first.
|
|
2
|
|
end
|
|
|
|
def initialize(*args)
|
|
super
|
|
|
|
@logger = Log4r::Logger.new("vagrant::hosts::bsd")
|
|
@nfs_restart_command = "sudo nfsd restart"
|
|
@nfs_exports_template = "nfs/exports"
|
|
end
|
|
|
|
def nfs?
|
|
retryable(:tries => 10, :on => TypeError) do
|
|
system("which nfsd > /dev/null 2>&1")
|
|
end
|
|
end
|
|
|
|
def nfs_export(id, ip, folders)
|
|
# We need to build up mapping of directories that are enclosed
|
|
# within each other because the exports file has to have subdirectories
|
|
# of an exported directory on the same line. e.g.:
|
|
#
|
|
# "/foo" "/foo/bar" ...
|
|
# "/bar"
|
|
#
|
|
# We build up this mapping within the following hash.
|
|
@logger.debug("Compiling map of sub-directories for NFS exports...")
|
|
dirmap = {}
|
|
folders.each do |_, opts|
|
|
hostpath = opts[:hostpath]
|
|
|
|
found = false
|
|
dirmap.each do |dirs, diropts|
|
|
dirs.each do |dir|
|
|
if dir.start_with?(hostpath) || hostpath.start_with?(dir)
|
|
# TODO: verify opts and diropts are _identical_, raise an error
|
|
# if not. NFS mandates subdirectories have identical options.
|
|
dirs << hostpath
|
|
found = true
|
|
break
|
|
end
|
|
end
|
|
|
|
break if found
|
|
end
|
|
|
|
if !found
|
|
dirmap[[hostpath]] = opts.dup
|
|
end
|
|
end
|
|
|
|
# Sort all the keys by length so that the directory closest to
|
|
# the root is exported first.
|
|
dirmap.each do |dirs, _|
|
|
dirs.sort_by! { |d| d.length }
|
|
end
|
|
|
|
@logger.info("Exporting the following for NFS...")
|
|
dirmap.each do |dirs, opts|
|
|
@logger.info("NFS DIR: #{dirs.inspect}")
|
|
@logger.info("NFS OPTS: #{opts.inspect}")
|
|
end
|
|
|
|
output = TemplateRenderer.render(@nfs_exports_template,
|
|
:uuid => id,
|
|
:ip => ip,
|
|
:folders => dirmap)
|
|
|
|
# The sleep ensures that the output is truly flushed before any `sudo`
|
|
# commands are issued.
|
|
@ui.info I18n.t("vagrant.hosts.bsd.nfs_export")
|
|
sleep 0.5
|
|
|
|
# First, clean up the old entry
|
|
nfs_cleanup(id)
|
|
|
|
# Output the rendered template into the exports
|
|
output.split("\n").each do |line|
|
|
line = line.gsub('"', '\"')
|
|
system(%Q[sudo su root -c "echo '#{line}' >> /etc/exports"])
|
|
end
|
|
|
|
# We run restart here instead of "update" just in case nfsd
|
|
# is not starting
|
|
system(@nfs_restart_command)
|
|
end
|
|
|
|
def nfs_prune(valid_ids)
|
|
return if !File.exist?("/etc/exports")
|
|
|
|
@logger.info("Pruning invalid NFS entries...")
|
|
|
|
output = false
|
|
|
|
File.read("/etc/exports").lines.each do |line|
|
|
if id = line[/^# VAGRANT-BEGIN: (.+?)$/, 1]
|
|
if valid_ids.include?(id)
|
|
@logger.debug("Valid ID: #{id}")
|
|
else
|
|
if !output
|
|
# We want to warn the user but we only want to output once
|
|
@ui.info I18n.t("vagrant.hosts.bsd.nfs_prune")
|
|
output = true
|
|
end
|
|
|
|
@logger.info("Invalid ID, pruning: #{id}")
|
|
nfs_cleanup(id)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
protected
|
|
|
|
def nfs_cleanup(id)
|
|
return if !File.exist?("/etc/exports")
|
|
|
|
# Escape sed-sensitive characters:
|
|
id = id.gsub("/", "\\/")
|
|
id = id.gsub(".", "\\.")
|
|
|
|
# Use sed to just strip out the block of code which was inserted
|
|
# by Vagrant, and restart NFS.
|
|
system("sudo sed -e '/^# VAGRANT-BEGIN: #{id}/,/^# VAGRANT-END: #{id}/ d' -ibak /etc/exports")
|
|
end
|
|
end
|
|
end
|
|
end
|