From 58871853f5c9839f070eb96caa83e0be7f1a0a1e Mon Sep 17 00:00:00 2001 From: Mitchell Hashimoto Date: Fri, 1 Mar 2013 16:31:54 -0800 Subject: [PATCH] Hashes for all the networks --- config/default.rb | 8 +++-- plugins/kernel_v2/config/vm.rb | 63 +++++----------------------------- 2 files changed, 14 insertions(+), 57 deletions(-) diff --git a/config/default.rb b/config/default.rb index ae97fab94..7733fcce6 100644 --- a/config/default.rb +++ b/config/default.rb @@ -18,9 +18,11 @@ Vagrant.configure("2") do |config| config.vm.guest = :linux # Share SSH locally by default - config.vm.network :forwarded_port, 22, 2222, - :id => "ssh", - :auto_correct => true + config.vm.network :forwarded_port, + host: 22, + guest: 2222, + id: "ssh", + auto_correct: true # Share the root folder. This can then be overridden by # other Vagrantfiles, if they wish. diff --git a/plugins/kernel_v2/config/vm.rb b/plugins/kernel_v2/config/vm.rb index e1dd26073..bd1c4759e 100644 --- a/plugins/kernel_v2/config/vm.rb +++ b/plugins/kernel_v2/config/vm.rb @@ -1,4 +1,5 @@ require "pathname" +require "securerandom" require "vagrant" require "vagrant/config/v2/util" @@ -32,51 +33,17 @@ module VagrantPlugins # Internal state @__compiled_provider_configs = {} @__finalized = false - @__networks = [] + @__networks = {} @__providers = {} end # Custom merge method since some keys here are merged differently. def merge(other) super.tap do |result| + result.instance_variable_set(:@__networks, @__networks + other.networks) result.instance_variable_set(:@synced_folders, @synced_folders.merge(other.synced_folders)) result.instance_variable_set(:@provisioners, @provisioners + other.provisioners) - # Merge the networks by prepending any networks we have onto the new - # configuration, except allow overriding by the other configuration - # by `:id`. - other_networks = other.instance_variable_get(:@__networks) - new_networks = @__networks.dup - other_networks.each do |type, args| - opts = {} - opts = args.last if args.last.is_a?(Hash) - - # If an ID is set, then we try to override it if it exists - # in the previously set networks. - if opts[:id] - new_networks.each_index do |index| - _, new_args = new_networks[index] - new_opts = {} - new_opts = new_args.last if new_args.last.is_a?(Hash) - - if new_opts[:id] == opts[:id] - # Merge the options hash - args[args.length - 1] = opts.merge(new_opts) - - # Delete the existing network - new_networks.delete_at(index) - break - end - end - end - - # If the network wasn't found or there was no ID set, then - # we just append it to the array of networks - new_networks << [type, args] - end - - result.instance_variable_set(:@__networks, new_networks) - # Merge the providers by prepending any configuration blocks we # have for providers onto the new configuration. other_providers = other.instance_variable_get(:@__providers) @@ -126,8 +93,12 @@ module VagrantPlugins # * `:public_network` - The machine gets an IP on a shared network. # # @param [Symbol] type Type of network - def network(type, *args) - @__networks << [type, args] + # @param [Hash] options Options for the network. + def network(type, options=nil) + options ||= {} + id = options[:id] || SecureRandom.uuid + @__networks[id] ||= {} + @__networks[id].merge!(options) end # Configures a provider for this VM. @@ -179,22 +150,6 @@ module VagrantPlugins # default VM which just inherits the rest of the configuration. define(DEFAULT_VM_NAME) if defined_vm_keys.empty? - # Do some defaults for networks - @__networks.each do |type, args| - if type == :forwarded_port - options = args.last - - # If the last argument isn't an option hash, add it on. - if !options.is_a?(Hash) - options = {} - args << options - end - - # Set the default name - options[:id] = "#{args[0].to_s(32)}-#{args[1].to_s(32)}" - end - end - # Compile all the provider configurations @__providers.each do |name, blocks| # Find the configuration class for this provider