2011-12-03 19:05:50 -08:00

146 lines
4.2 KiB
Ruby

require 'vagrant/config/vm/sub_vm'
require 'vagrant/config/vm/provisioner'
module Vagrant
module Config
class VMConfig < Base
configures :vm
include Util::StackedProcRunner
attr_accessor :name
attr_accessor :auto_port_range
attr_accessor :box
attr_accessor :box_url
attr_accessor :box_ovf
attr_accessor :base_mac
attr_accessor :boot_mode
attr_accessor :host_name
attr_reader :forwarded_ports
attr_reader :shared_folders
attr_reader :network_options
attr_reader :provisioners
attr_accessor :system
def initialize
@forwarded_ports = {}
@shared_folders = {}
@network_options = []
@provisioners = []
end
def forward_port(name, guestport, hostport, options=nil)
options = {
:guestport => guestport,
:hostport => hostport,
:protocol => :tcp,
:adapter => 0,
:auto => false
}.merge(options || {})
forwarded_ports[name] = options
end
def share_folder(name, guestpath, hostpath, opts=nil)
@shared_folders[name] = {
:guestpath => guestpath,
:hostpath => hostpath,
:owner => nil,
:group => nil,
:nfs => false
}.merge(opts || {})
end
def network(ip, options=nil)
options = {
:ip => ip,
:netmask => "255.255.255.0",
:adapter => 1,
:mac => nil,
:name => nil
}.merge(options || {})
@network_options[options[:adapter]] = options
end
def provision(name, options=nil, &block)
@provisioners << Provisioner.new(top, name, options, &block)
end
def customize(&block)
push_proc(&block)
end
def defined_vms
@defined_vms ||= {}
end
# This returns the keys of the sub-vms in the order they were
# defined.
def defined_vm_keys
@defined_vm_keys ||= []
end
def define(name, options=nil, &block)
name = name.to_sym
options ||= {}
# Add the name to the array of VM keys. This array is used to
# preserve the order in which VMs are defined.
defined_vm_keys << name
# Add the SubVM to the hash of defined VMs
defined_vms[name] ||= SubVM.new
defined_vms[name].options.merge!(options)
defined_vms[name].push_proc(&block)
end
def validate(errors)
errors.add(I18n.t("vagrant.config.vm.box_missing")) if !box
errors.add(I18n.t("vagrant.config.vm.box_not_found", :name => box)) if box && !box_url && !env.box
errors.add(I18n.t("vagrant.config.vm.boot_mode_invalid")) if ![:vrdp, :gui].include?(boot_mode.to_sym)
errors.add(I18n.t("vagrant.config.vm.base_mac_invalid")) if env.box && !base_mac
shared_folders.each do |name, options|
if !File.directory?(File.expand_path(options[:hostpath].to_s, env.root_path))
errors.add(I18n.t("vagrant.config.vm.shared_folder_hostpath_missing",
:name => name,
:path => options[:hostpath]))
end
if options[:nfs] && (options[:owner] || options[:group])
# Owner/group don't work with NFS
errors.add(I18n.t("vagrant.config.vm.shared_folder_nfs_owner_group",
:name => name))
end
end
# Validate some basic networking
network_options.each do |options|
next if !options
ip = options[:ip].split(".")
if ip.length != 4
errors.add(I18n.t("vagrant.config.vm.network_ip_invalid",
:ip => options[:ip]))
elsif ip.last == "1"
errors.add(I18n.t("vagrant.config.vm.network_ip_ends_one",
:ip => options[:ip]))
end
end
# Each provisioner can validate itself
provisioners.each do |prov|
# TODO: Remove at some point
if prov.shortcut == :chef_server
errors.add(I18n.t("vagrant.config.vm.provisioner_chef_server_changed"))
else
prov.validate(errors)
end
end
end
end
end
end