vagrant-dns-server 0.1

data/.gitignore

@@ -0,0 +1 @@
+pkg

data/.rvmrc

@@ -0,0 +1 @@
+rvm use 1.9.3@dns --create

data/Gemfile

@@ -0,0 +1,6 @@
+# A sample Gemfile
+source "https://rubygems.org"
+
+gem 'rubydns'
+gem 'zmq'
+gem 'thor'

data/Gemfile.lock

@@ -0,0 +1,20 @@
+GEM
+  remote: https://rubygems.org/
+  specs:
+    eventmachine (1.0.0)
+    rainbow (1.1.4)
+    rexec (1.5.1)
+      rainbow
+    rubydns (0.5.3)
+      eventmachine (~> 1.0.0)
+      rexec (~> 1.5.1)
+    thor (0.16.0)
+    zmq (2.1.4)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  rubydns
+  thor
+  zmq

data/README.md

@@ -0,0 +1,17 @@
+# Intro
+vagrant-dns enables easy dns management of multiple local Vagrant machines.
+
+It composed from a Vagrant plugin and standalone dns server, the plugin auto registers the host and ip of the VM once it boots up making it available for lookup on both the host and guest machines.
+
+# Usage
+
+Add the plugin to the project Gemfile, use vagrant_dns server to boot the local dns server
+
+
+# Alternatives
+
+There are two existing project that aim to provide similar functionality:
+
+ * BerlinVagran [vagrant-dns](https://github.com/BerlinVagrant/vagrant-dns) which seems to be able to manage only a single machine and is OSX only.
+ 
+ * [vagrant-hostname](https://github.com/mosaicxm/vagrant-hostmaster) which manipulates /etc/hosts, this requires to enter sudo password for each machine (not an option when starting numerous machines).

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/vagrant_dns

@@ -0,0 +1,20 @@
+#!/usr/bin/env ruby
+
+$:.push 'lib'
+require 'rubygems'
+require 'thor'
+require 'server'
+
+class Main < Thor
+
+  desc 'server','Starts the dns server that will accept dns changes from vagrant machines'
+  def server
+    @server = VagrantDns::Server.new   
+    @server.process 
+    trap("INT") { @server.shutdown;exit 0}
+  end
+end
+
+
+
+Main.start(ARGV)

data/lib/constants.rb

@@ -0,0 +1,3 @@
+
+URL = 'tcp://127.0.0.1:7005'
+CHANNEL = 'vagrant'

data/lib/dns.rb

@@ -0,0 +1,47 @@
+require 'rubydns'
+
+HOSTS = Hash[%w(jenkins artifactory x2.takadu.com).zip(15..17)]
+IN = Resolv::DNS::Resource::IN
+UPSTREAM = RubyDNS::Resolver.new([[:tcp, "8.8.8.8", 53]])
+
+module VagrantDns
+
+  class Registry 
+    def initialize
+	@reg = {}
+    end
+    def register(host,ip)
+	@reg[host] = ip	
+    end
+
+    def unregister(host)
+	@reg[host] = nil
+    end
+
+    def read(host)
+     @reg[host]	
+    end
+  end
+
+  class DnsServer
+    def initialize
+	RubyDNS::run_server(:listen => [[:tcp, "localhost", 53]]) do
+	  otherwise do |transaction|
+	    ip = REG.read(transaction.name)
+	    if(ip)
+		transaction.respond!(ip)
+	    else
+		puts "not found moving to UPSTREAM"
+		begin
+		  transaction.passthrough!(UPSTREAM)
+		rescue Exception => e
+		  puts e
+		end
+	    end
+	  end
+	end
+    end
+  end
+end
+
+REG = VagrantDns::Registry.new

data/lib/middleware/network_binder.rb

@@ -0,0 +1,23 @@
+module VagrantDns
+  class NetworkBinder
+    def bind(vm)
+	networks = vm.config.vm.networks.find_all do |type,network_parameters|
+	    type == :hostonly 
+	end
+      ips = networks.map {|type,params| params.first}
+	host = vm.config.vm.host_name
+	Status.new.up(host,ips.first)
+    end		   
+  end
+  class Status
+    # if socket is created on middleware initialization vagrant will get stuck
+    def up(host,ip)
+	context = ZMQ::Context.new
+	pub = context.socket ZMQ::PUB
+	puts "connecting to #{URL}"
+	pub.connect URL
+	pub.send "#{CHANNEL} #{host} #{ip} up"
+	pub.close
+    end
+  end
+end

data/lib/middleware/update.rb

@@ -0,0 +1,20 @@
+
+module VagrantDns
+  module Middleware
+    class Update
+	def initialize(app, env)
+	  @app = app
+	end
+
+	def call(env)
+	  @app.call(env)
+	  update env[:vm] if env["vm"].state == :running
+	end
+
+	protected
+	def update(vm)
+        VagrantDns::NetworkBinder.new.bind(vm)
+	end
+    end
+  end
+end

data/lib/pub.ruby

@@ -0,0 +1,15 @@
+# pub.rb
+require 'zmq'
+
+context = ZMQ::Context.new
+chan    = ARGV[0]
+user    = ARGV[1]
+pub     = context.socket ZMQ::PUB
+# pub.bind 'ipc:///tmp/2'
+pub.connect 'tcp://127.0.0.1:5004'
+
+while msg = STDIN.gets
+  msg.strip!
+  pub.send "#{chan} #{user} #{msg}"
+end
+

data/lib/server.rb

@@ -0,0 +1,40 @@
+require 'zmq'
+require 'constants'
+require 'dns'
+require 'eventmachine'
+
+module VagrantDns
+  class Server
+    def initialize
+	  context = ZMQ::Context.new
+	  @sub = context.socket ZMQ::SUB
+	  puts "binding to #{URL}"
+	  @sub.bind URL
+	  @sub.setsockopt ZMQ::SUBSCRIBE, CHANNEL
+    end
+
+    def process
+	Thread.new {
+	  @dns = VagrantDns::DnsServer.new
+	}
+	puts "processing "
+	begin
+	  while line = @sub.recv
+	    chan, host, ip, msg = line.split ' ', 4
+	    puts "#{chan} [#{ip}->#{host}]: is #{msg}"
+	    REG.register(host,ip)
+	  end
+	catch Execption  => e
+	  $stderr.puts e
+	rescue 
+	  shutdown
+	end
+    end
+
+    def shutdown
+	$stderr.puts "Cleaning up server"
+	@sub.close
+	EM::stop()
+    end
+  end
+end

data/lib/vagrant_init.rb

@@ -0,0 +1,9 @@
+require 'vagrant'
+require 'constants'
+require 'middleware/update'
+require 'middleware/network_binder'
+require 'zmq'
+
+Vagrant.actions[:provision].insert_after(Vagrant::Action::VM::Provision, VagrantDns::Middleware::Update)
+Vagrant.actions[:start].insert_after(Vagrant::Action::VM::Provision, VagrantDns::Middleware::Update)
+

data/lib/version.rb

@@ -0,0 +1,3 @@
+module VagrantDns
+   VERSION=0.1
+end

data/vagrant-dns-server.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Ronen Narkis"]
+  gem.email         = ["narkisr@gmail.com"]
+  gem.description   = %q{A vagrant plugin to automanage dns registration on local machine}
+  gem.summary       = %q{A vagrant plugin that sets up a local dns server}
+  gem.homepage      = "https://github.com/narkisr/vagrant-dns"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "vagrant-dns-server"
+  gem.require_paths = ["lib"]
+  gem.version       = VagrantDns::VERSION
+
+  gem.add_dependency  'vagrant', '~>1.0.3'
+  gem.add_dependency  'rubydns'
+  gem.add_dependency  'zmq'
+  gem.add_development_dependency  'bundler'
+  gem.add_development_dependency  'mocha', '~>0.12.3'
+  gem.add_development_dependency  'rake'
+
+end

metadata

@@ -0,0 +1,158 @@
+--- !ruby/object:Gem::Specification
+name: vagrant-dns-server
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- Ronen Narkis
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-01-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: vagrant
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.3
+- !ruby/object:Gem::Dependency
+  name: rubydns
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: zmq
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.12.3
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.12.3
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A vagrant plugin to automanage dns registration on local machine
+email:
+- narkisr@gmail.com
+executables:
+- vagrant_dns
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rvmrc
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- bin/vagrant_dns
+- lib/constants.rb
+- lib/dns.rb
+- lib/middleware/network_binder.rb
+- lib/middleware/update.rb
+- lib/pub.ruby
+- lib/server.rb
+- lib/vagrant_init.rb
+- lib/version.rb
+- vagrant-dns-server.gemspec
+homepage: https://github.com/narkisr/vagrant-dns
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: A vagrant plugin that sets up a local dns server
+test_files: []