Enhancement: IP/NIC selection

.gitignore

@@ -18,3 +18,4 @@ tags
 
 .vagrant
 .vagrant_dns.json
+.idea

Gemfile

@@ -5,11 +5,12 @@ source 'https://rubygems.org'
 
 gem 'rubydns', '1.0.2'
 gem 'rexec'
-gem 'rake'
+gem 'rake', '~> 10.0'
 
 # Vagrant's special group
 group :plugins do
   gem 'landrush', path: '.'
+  gem 'landrush-ip', '~> 0.2.1'
 end
 
 group :test do

README.md

@@ -36,7 +36,17 @@ Landrush assigns your vm's hostname from either the vagrant config (see the `exa
 
 Every time a VM is started, its IP address is automatically detected and a DNS record is created that maps the hostname to its IP.
 
-If for any reason the auto-detection detects no IP address or the wrong IP address, or you want to override it, you can do like so:
+If for any reason the auto-detection detects no IP address or the wrong IP address you can control how it looks for it.
+
+You can start by excluding interfaces matching an array of regex patterns (the value shown here is the default):
+
+    config.landrush.exclude = [/lo[0-9]*/, /docker[0-9]+/, /tun[0-9]+/]
+
+Or, if you know which interface you need the IP of you can specify that too (default is none):
+
+    config.landrush.interface = 'eth0'
+
+If all else fails, you can override it entirely:
 
     config.landrush.host_ip_address = '1.2.3.4'
 

Rakefile

@@ -1,7 +1,20 @@
+require 'rubygems'
+require 'bundler/setup'
 require 'bundler/gem_tasks'
 require 'rake/testtask'
 require 'rubocop/rake_task'
 
+# Immediately sync all stdout
+$stdout.sync = true
+$stderr.sync = true
+
+# Change to the directory of this file
+Dir.chdir(File.expand_path('../', __FILE__))
+
+# Rubocop
+RuboCop::RakeTask.new
+
+# Tests
 Rake::TestTask.new do |t|
   t.pattern = 'test/**/*_test.rb'
   t.libs << 'test'
@@ -15,5 +28,3 @@ task default: [
 task :generate_diagrams do
   sh 'cd doc; seqdiag --fontmap=support/seqdiag.fontmap -Tsvg vagrant_dns_without_landrush.diag'
 end
-
-RuboCop::RakeTask.new

Vagrantfile

@@ -0,0 +1,27 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+
+Vagrant.configure('2') do |config|
+  config.vm.define 'landrush-test-debian' do |machine|
+    machine.vm.box = 'debian/jessie64'
+
+    # Add a DHCP network so we don't know its IP :P
+    machine.vm.network 'private_network', type: 'dhcp'
+
+    machine.vm.provider :virtualbox do |provider, _|
+      provider.memory = 512
+      provider.cpus = 2
+    end
+
+    machine.landrush_ip.override = true
+
+    machine.vm.hostname = 'landrush-dev'
+    machine.vm.network 'private_network', type: 'dhcp'
+
+    # Landrush (DNS)
+    machine.landrush.enabled = true
+    machine.landrush.tld = 'landrush'
+    machine.landrush.interface = 'eth1'
+    machine.landrush.exclude = [/lo[0-9]*/, /docker[0-9]+/, /tun[0-9]+/]
+  end
+end

lib/landrush/cap/all/read_host_visible_ip_address.rb

@@ -0,0 +1,49 @@
+module Landrush
+  module Cap
+    module All
+      module ReadHostVisibleIpAddress
+        def self.filter_addresses(addresses)
+          unless @machine.config.landrush.exclude.nil?
+            re = Regexp.union(@machine.config.landrush.exclude)
+
+            addresses = addresses.select do |addr|
+              !addr['name'].match(re)
+            end
+          end
+
+          addresses
+        end
+
+        def self.read_host_visible_ip_address(machine)
+          @machine = machine
+
+          addr      = nil
+          addresses = machine.guest.capability(:landrush_ip_get)
+
+          # Short circuit this one first: if an explicit interface is defined, look for it and return it if found.
+          # Technically, we could do a single loop, but execution time is not vital here.
+          # This allows us to be more accurate, especially with logging what's going on.
+          unless machine.config.landrush.interface.nil?
+            addr = addresses.detect { |a| a['name'] == machine.config.landrush.interface }
+
+            machine.env.ui.warn "[landrush] Unable to find interface #{machine.config.landrush.interface}" if addr.nil?
+          end
+
+          if addr.nil?
+            addresses = filter_addresses addresses
+
+            raise 'No addresses found' if addresses.empty?
+
+            addr = addresses.last
+          end
+
+          ip = IPAddr.new(addr['ipv4'])
+
+          machine.env.ui.info "[landrush] Using #{addr['name']} (#{addr['ipv4']})"
+
+          ip.to_s
+        end
+      end
+    end
+  end
+end

lib/landrush/config.rb

@@ -16,7 +16,7 @@ class Config < Vagrant.plugin('2', :config)
       :host_ip_address => nil,
       :guest_redirect_dns => true,
       :interface => nil,
-      :exclude => %w(lo[0-9]* docker[0-9]+ tun[0-9]+)
+      :exclude => [/lo[0-9]*/, /docker[0-9]+/, /tun[0-9]+/]
     }
 
     def initialize