vagrant-dnsdock-hostupdater 0.0.10

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 9c1d494bc348ad4bf14e33a49b68878e9089ad25
+  data.tar.gz: f58089650e77004454d1c56d7beec701dc534b96
+SHA512:
+  metadata.gz: 716cdb7352cd498a5636712a6e488e713ee80ef8ba6dacad2a7e23b0c86a8641f1b5da568d73b48260511a80f63821e496a0018e46b59c64f4323b0be8ffc087
+  data.tar.gz: 150e266e511a23b431fced083a0c22d5fcd1659fba4f780146aadfd7aa57abe8ba04110b2f53ae0509322d88325f9599a0a010b1ed34889eee1feb50a190ebfa

data/Rakefile

@@ -0,0 +1,3 @@
+require 'rubygems'
+require 'bundler/setup'
+Bundler::GemHelper.install_tasks

data/lib/client.rb

@@ -0,0 +1,5 @@
+require_relative 'host-manager'
+
+HostManager::Client
+    .create('10.10.10.1', 2991)
+    .send_action('delete', 'the1234.test.host', '127.0.0.1', 'containerid')

data/lib/event-watcher.rb

@@ -0,0 +1,35 @@
+require_relative 'host-manager'
+require 'yaml'
+
+loop {
+  begin
+    Docker::Event.stream { |event|
+      HostManager.log.info 'Connected to docker socket. Listening for events.'
+      # respond to all container events
+      if event.type == 'container'
+        container_id = event.actor.id.chomp
+
+        container = Docker::Container.get(container_id)
+        ip = container.info['NetworkSettings']['IPAddress']
+        container_name = container.info['Name'].sub!(/^\//, '')
+
+        client = HostManager::Client.create('10.10.10.1', 2991)
+        client.host_suffix '.local.signal.sh'
+
+        if event.action == 'start'
+          client.send_action('create', container_name, ip, container_id)
+        elsif event.action == 'die'
+          client.send_action('delete', container_name, ip, container_id)
+        end
+
+        client.close
+      end
+
+      break
+    }
+
+  rescue Docker::Error::TimeoutError
+    HostManager.log.warn 'Docker socket connection timed out whilst listening for events. Reconnecting...'
+  end
+}
+

data/lib/host-manager.rb

@@ -0,0 +1,193 @@
+require 'socket' # Get sockets from stdlib
+require 'json'
+require_relative 'hosts/hosts'
+require 'docker'
+require 'logger'
+require 'log4r'
+
+module OS
+  def OS.windows?
+    (/cygwin|mswin|mingw|bccwin|wince|emx/ =~ RUBY_PLATFORM) != nil
+  end
+
+  def OS.mac?
+    (/darwin/ =~ RUBY_PLATFORM) != nil
+  end
+
+  def OS.unix?
+    !OS.windows?
+  end
+
+  def OS.linux?
+    OS.unix? and not OS.mac?
+  end
+end
+
+module HostManager
+
+  def self.log
+    unless @log
+      @log = Log4r::Logger.new('main')
+      # @log.outputters << Log4r::Outputter.stdout
+      @log.outputters << Log4r::FileOutputter.new('logmain', :filename => 'host-manager.log')
+    end
+
+    @log
+  end
+
+  def self.comment_prefix
+    'Managed by Vagrant plugin - do not remove.'
+  end
+
+  def self.host_file_path
+    OS.windows? ? `echo %SystemRoot%\\System32\\drivers\\etc\\hosts`.chomp : '/etc/hosts'
+  end
+
+  class Base
+
+    def log
+      HostManager.log
+    end
+
+    def host_name(container_name)
+      if container_name.include?('.')
+        container_name
+      else
+        container_name + host_suffix
+      end
+    end
+
+    def host_suffix(suffix = nil)
+      if suffix != nil
+        @domain_suffix = suffix
+      else
+        @domain_suffix
+      end
+    end
+  end
+
+  class Server < Base
+    def run(ip, port)
+      HostManager.log.info "Running on host manager service on #{ip}:#{port}"
+      @tcp_server = TCPServer.new ip, port
+
+      loop {
+
+        client = @tcp_server.accept
+        content = client.gets
+
+        log.debug("Recieved data: #{content.to_s.strip}")
+
+        begin
+          unless content.to_s.chomp.strip.empty?
+            data = JSON.parse(content)
+            required_keys = %w(action hostname ip containerId)
+            required_keys.each do |key|
+              unless data.key?(key)
+                raise "Data for key #{key} not provided."
+              end
+            end
+            execute(data['action'], data['hostname'], data['ip'], data['containerId'])
+          end
+        rescue => e
+          log.error("Caught exception attempting to execute with data #{content}. #{e}")
+        end
+
+        client.close
+      }
+
+    end
+
+    def get_hosts
+      Hosts::File.read(HostManager.host_file_path)
+    end
+
+    def execute(action, hostname, ip, container_id = nil)
+      hosts = get_hosts
+
+      if action == 'create'
+        hosts.elements.each do |element|
+          if valid_entry?(element)
+            if hostname == element.name && ip == element.address
+              raise "Entry already exists! (#{ip} #{hostname})"
+            end
+          end
+        end
+
+        comment = HostManager.comment_prefix + (container_id ? " Container ID: #{container_id}" : '')
+
+        hosts.elements << Hosts::Entry.new(ip, hostname, :comment => comment)
+        hosts.write
+
+      elsif action == 'delete'
+        hosts.elements.delete_if do |element|
+          if valid_entry?(element)
+            if hostname == element.name
+              log.info 'Removing entry for ' + element.name
+              true
+            end
+          end
+        end
+
+        hosts.write
+      else
+        raise 'No valid action specified.'
+      end
+    end
+
+    def self.create(ip, port)
+      server = self.new
+      server.run(ip, port)
+    end
+
+    private
+
+    # Determine if entry is managed by this utility or not
+    def valid_entry?(element)
+      # check it's a managed host first
+      if element.respond_to?(:comment) && element.comment.to_s.start_with?(HostManager.comment_prefix)
+        # check we have hostname and ip
+        if element.respond_to?(:name) && element.respond_to?(:address)
+          return true
+        end
+      end
+    end
+
+  end
+
+  class Client < Base
+
+    def initialize(ip, port)
+      @socket = TCPSocket.new(ip, port)
+    end
+
+    def close
+      log.info('Closing client connection.')
+      @socket.close
+    end
+
+    def send_action(action, hostname, ip, container_id = nil)
+      data = {
+          :action => action,
+          :hostname => host_name(hostname),
+          :ip => ip,
+      }
+
+      if container_id
+        data[:containerId] = container_id
+      end
+
+      send(data)
+    end
+
+    def send(data)
+      log.info("Sending data via client: #{data.to_json}")
+      @socket.write data.to_json + "\n"
+    end
+
+    def self.create(ip, port)
+      self.new(ip, port)
+    end
+  end
+
+end

data/lib/hosts/aef/hosts.rb

@@ -0,0 +1,51 @@
+# encoding: UTF-8
+=begin
+Copyright Alexander E. Fischer <aef@raxys.net>, 2012
+
+This file is part of Hosts.
+
+Permission to use, copy, modify, and/or distribute this software for any
+purpose with or without fee is hereby granted, provided that the above
+copyright notice and this permission notice appear in all copies.
+
+THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
+FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+PERFORMANCE OF THIS SOFTWARE.
+=end
+
+require 'set'
+require 'pathname'
+require_relative 'linebreak/linebreak'
+
+# Namespace for projects of Alexander E. Fischer <aef@raxys.net>.
+#
+# If you want to be able to simply type Example instead of Aef::Example to
+# address classes in this namespace simply write the following before using the
+# classes.
+#
+# @example Including the namespace
+#   include Aef
+# @author Alexander E. Fischer
+module Aef
+
+  # Namespace for the hosts library
+  module Hosts
+
+    # An exception for errors happening while parsing a hosts file
+    class ParserError < RuntimeError; end
+  end
+end
+
+require_relative 'hosts/version'
+require_relative 'hosts/helpers'
+require_relative 'hosts/file'
+require_relative 'hosts/element'
+require_relative 'hosts/section'
+require_relative 'hosts/entry'
+require_relative 'hosts/comment'
+require_relative 'hosts/empty_element'
+

data/lib/hosts/aef/hosts/comment.rb

@@ -0,0 +1,73 @@
+# encoding: UTF-8
+=begin
+Copyright Alexander E. Fischer <aef@raxys.net>, 2012
+
+This file is part of Hosts.
+
+Permission to use, copy, modify, and/or distribute this software for any
+purpose with or without fee is hereby granted, provided that the above
+copyright notice and this permission notice appear in all copies.
+
+THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
+FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+PERFORMANCE OF THIS SOFTWARE.
+=end
+
+require_relative '../hosts'
+
+module Aef
+  module Hosts
+
+    # Represents a comment-only line as element of a hosts file
+    class Comment < Element
+      
+      # The comment
+      #
+      # @return [String]
+      attr_reader :comment
+
+      # Initializes a comment
+      #
+      # @param comment [String] the comment
+      # @param options [Hash]
+      # @option options [String] :cache sets a cached String representation
+      def initialize(comment, options = {})
+        validate_options(options, :cache)
+        
+        raise ArgumentError, 'Comment cannot be empty' unless comment
+
+        @comment = comment.to_s
+        @cache   = options[:cache].to_s unless options[:cache].nil?
+      end
+
+      # Sets the comment
+      def comment=(comment)
+        set_if_changed(:comment, comment.to_s) do
+          invalidate_cache!
+        end
+      end
+
+      # A String representation for debugging purposes
+      #
+      # @return [String]
+      def inspect
+        generate_inspect(self, :comment, :cache)
+      end
+
+      protected
+
+      # Defines the algorithm to generate a String representation from scratch.
+      #
+      # @return [String] a generated String representation
+      def generate_string(options = {})
+        "##{comment}\n"
+      end
+    end
+
+  end
+end
+

data/lib/hosts/aef/hosts/element.rb

@@ -0,0 +1,108 @@
+# encoding: UTF-8
+=begin
+Copyright Alexander E. Fischer <aef@raxys.net>, 2012
+
+This file is part of Hosts.
+
+Permission to use, copy, modify, and/or distribute this software for any
+purpose with or without fee is hereby granted, provided that the above
+copyright notice and this permission notice appear in all copies.
+
+THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
+FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+PERFORMANCE OF THIS SOFTWARE.
+=end
+
+require_relative '../hosts'
+
+module Aef
+  module Hosts
+
+    # The base class for elements which are aggregated by the Aef::Hosts::File
+    # class.
+    #
+    # @abstract This class is not supposed to be instantiated.
+    class Element
+
+      include Helpers
+
+      # Cached String representation
+      #
+      # @return [String, Hash, nil]
+      attr_reader :cache
+
+      # Deletes the cached String representation
+      #
+      # @return [Aef::Hosts::Element] a self reference
+      def invalidate_cache!
+        @cache = nil
+
+        self
+      end
+
+      # Tells if a String representation is cached or not
+      #
+      # @return [true, false] true if cache is not empty
+      def cache_filled?
+        @cache ? true : false
+      end
+
+      # A String representation for debugging purposes
+      #
+      # @return [String]
+      def inspect
+        generate_inspect(self, :cache)
+      end
+
+      # Provides a String representation of the element
+      #
+      # @param [Hash] options
+      # @option options [true, false] :force_generation if set to true, the
+      #   cache won't be used, even if it not empty
+      # @option options [:unix, :windows, :mac] :linebreak_encoding the
+      #   linebreak encoding of the result. If nothing is specified the result
+      #   will be encoded as if :unix was specified.
+      # @see Aef::Linebreak#encode
+      def to_s(options = {})
+        validate_options(options, :force_generation, :linebreak_encoding)
+
+        string = ''
+
+        if !cache_filled? || options[:force_generation]
+          string << generate_string(options)
+        else
+          string << cache_string(options)
+        end
+
+        if options[:linebreak_encoding]
+          string = Aef::Linebreak.encode(string, options[:linebreak_encoding])
+        end
+
+        string
+      end
+
+      protected
+
+      # Defines the algorithm to generate a String representation from scratch.
+      #
+      # @abstract This method needs to be implemented in descendant classes.
+      # @return [String] a generated String representation
+      def generate_string(options = {})
+        raise NotImplementedError
+      end
+
+      # Defines the algorithm to construct the String representation from cache
+      #
+      # @return [String] the cached String representation
+      def cache_string(options = {})
+        @cache.dup
+      end
+    end
+
+  end
+end
+