elasticdns 0.0.1

data/Gemfile

@@ -0,0 +1,4 @@
+source :rubygems
+gem 'rake'
+gem 'aws-sdk'
+gem 'fileutils'

data/Gemfile.lock

@@ -0,0 +1,28 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    aws-sdk (1.7.0)
+      httparty (~> 0.7)
+      json (~> 1.4)
+      nokogiri (>= 1.4.4)
+      uuidtools (~> 2.1)
+    fileutils (0.7)
+      rmagick (>= 2.13.1)
+    httparty (0.9.0)
+      multi_json (~> 1.0)
+      multi_xml
+    json (1.7.5)
+    multi_json (1.3.7)
+    multi_xml (0.5.1)
+    nokogiri (1.5.5)
+    rake (0.9.2.2)
+    rmagick (2.13.1)
+    uuidtools (2.1.3)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  aws-sdk
+  fileutils
+  rake

data/bin/elasticdns

@@ -0,0 +1,104 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'optparse'
+require 'yaml'
+
+require_relative '../lib/elasticdns'
+
+file = __FILE__
+
+options = {}
+
+opt_parser = OptionParser.new do |opt|
+  opt.banner = "Usage: #{file} [OPTIONS]"
+  opt.separator  ""
+  opt.separator  "Options"
+
+  options[:config_file] = ENV['config_file']
+  opt.on("-c","--config_file=elasticdns.yml","Use specific config file") do |config_file|
+    options[:config_file] = config_file
+  end
+
+  options[:debug] = ENV['debug']
+  opt.on("-d","--debug=1","Use to debug") do |debug|
+    options[:debug] = debug
+  end
+
+  options[:ec2_access_key_id] = ENV['ec2_access_key_id']
+  opt.on("--ec2_access_key_id","--ec2_access_key_id=ec2_key_id","Use specific ec2_key_id") do |ec2_access_key_id|
+    options[:ec2_access_key_id] = ec2_access_key_id
+  end
+
+  options[:ec2_secret_access_key] = ENV['ec2_secret_access_key']
+  opt.on("--ec2_secret_access_key","--ec2_secret_access_key=secret_access_key","Use specific secret_access_key") do |ec2_secret_access_key|
+    options[:ec2_secret_access_key] = ec2_secret_access_key
+  end
+
+  options[:ec2_tag_attribute] = ENV['ec2_tag_attribute'] || "Name"
+  opt.on("--ec2_tag_attribute","--ec2_tag_attribute=Name","Use to specifiy which tags attribute read to detect master servers (default: Name)") do |ec2_tag_attribute|
+    options[:ec2_tag_attribute] = ec2_tag_attribute
+  end
+
+  options[:ec2_ip_attribute] = ENV['ec2_ip_attribute'] || "private_ip_address"
+  opt.on("--ec2_ip_attribute","--ec2_ip_attribute=Name","Use to specifiy which ip attribute read public_ip_address or private_ip_address (default: private_ip_address)") do |ec2_ip_attribute|
+    options[:ec2_ip_attribute] = ec2_ip_attribute
+  end
+
+
+  options[:bind9_notify_file] = ENV['bind9_notify_file']
+  opt.on("--bind9_notify_file","--bind9_notify_file=/etc/bind/also_notify.conf","") do |bind9_notify_file|
+    options[:bind9_notify_file] = bind9_notify_file
+  end
+
+  options[:bind9_acl_masters_file] = ENV['bind9_acl_masters_file']
+  opt.on("--bind9_acl_masters_file","--bind9_acl_masters_file=/etc/bind/acl_masters_file.conf","") do |bind9_acl_masters_file|
+    options[:bind9_acl_masters_file] = bind9_acl_masters_file
+  end
+
+  options[:bind9_named_conf_file] = ENV['bind9_named_conf_file']
+  opt.on("--bind9_named_conf_file","--bind9_named_conf_file=/data/bind/etc/named-master.conf","") do |bind9_named_conf_file|
+    options[:bind9_named_conf_file] = bind9_named_conf_file
+  end
+
+  options[:bind9_zone_files] = ENV['bind9_zone_files']
+  opt.on("--bind9_zone_files","--bind9_zone_files=[/data/bind/var/zone1.priv.hosts,/data/bind/var/zone2.priv.hosts]","") do |bind9_zone_files|
+    options[:bind9_zone_files] = bind9_zone_files
+  end
+
+  options[:bind9_checkconf_path] = ENV['bind9_checkconf_path']
+  opt.on("--bind9_checkconf_path","--bind9_checkconf_path=named-checkconf","") do |bind9_checkconf_path|
+    options[:bind9_checkconf_path] = bind9_checkconf_path
+  end
+
+  options[:bind9_checkzone_path] = ENV['bind9_checkzone_path']
+  opt.on("--bind9_checkzone_path","--bind9_checkzone_path=named-checkzone","") do |bind9_checkzone_path|
+    options[:bind9_checkzone_path] = bind9_checkzone_path
+  end
+
+  options[:bind9_init_cmd] = ENV['bind9_init_cmd']
+  opt.on("--bind9_init_file","--bind9_init_file='sudo /etc/init.d/bind9 reload'","") do |bind9_init_cmd|
+    options[:bind9_init_cmd] = bind9_init_cmd
+  end
+
+  options[:bind9_masters] = ENV['bind9_masters']
+  opt.on("--bind9_masters","--bind9_masters=[master1,maser2]","") do |bind9_masters|
+    options[:bind9_masters] = bind9_masters
+  end
+
+  options[:test] = ENV['test']
+  opt.on("-t","--test","Use to test dns propagation") do |test|
+    options[:test] = test
+  end
+
+  options[:slave] = ENV['slave'] || false
+  opt.on("-s","--slave=false","Use if host is slave (default: false)") do |slave|
+    options[:slave] = slave
+  end
+
+  opt.on("-h","--help","help")
+end
+
+opt_parser.parse!
+
+Elasticdns::Generator.new(options).run

data/config.yml

@@ -0,0 +1,22 @@
+debug: 1
+
+ec2:
+    access_key_id: "AKIAITTVP5VOZ2UP3SCA"
+    secret_access_key: "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"
+    region: 'us-west-2'
+    tag_attribute: 'Name'
+    ip_attribute: 'private_ip_address'
+
+
+bind9:
+    notify_file: "/etc/bind/also_notify.conf"
+    acl_masters_file: "/etc/bind/acl_masters_file.conf"
+    named_conf_file: "/data/bind/etc/named-master.conf"
+    zone_files:
+        - "/data/bind/var/zone1.priv.hosts"
+    init_cmd: "sudo /etc/init.d/bind9 reload"
+    checkconf_path: 'named-checkconf'
+    checkzone_path: 'named-checkzone'
+    masters:
+        - admin-01
+        - admin-02

data/elasticdns.gemspec

@@ -0,0 +1,15 @@
+Gem::Specification.new do |s|
+  s.name        = 'elasticdns'
+  s.version     = '0.0.1'
+  s.date        = '2013-06-23'
+  s.executables = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.summary     = "Elasticdns, a simple bind9 configurator"
+  s.description = "Elasticdns, a simple bind9 configurator"
+  s.authors     = ["Gabriel Klein"]
+  s.email       = 'gabriel.klein.fr@gmail.com'
+  s.files       = `git ls-files`.split("\n")
+  s.require_paths = ["lib"]
+  s.homepage    = 'http://github.com/GabKlein/elasticdns'
+  s.add_dependency('fileutils')
+  s.add_dependency('aws-sdk')
+end

data/lib/config.rb

@@ -0,0 +1,47 @@
+module Elasticdns
+  class Config
+    attr_accessor  :ec2_access_key_id, :ec2_secret_access_key, :ec2_attribute, :bind9_notify_file, :bind9_acl_masters_file, :bind9_named_conf_file, :bind9_zone_files, :bind9_checkconf_path, :bind9_checkzone_path, :bind9_init_cmd, :bind9_masters
+
+    def initialize(options={})
+      @config_from_file = from_file(options[:config_file])
+
+      @test = options[:test]
+      @slave =options[:slave]
+
+      @debug = options[:debug] || (@config_from_file['debug'] if @config_from_file)
+
+      @ec2_access_key_id = options[:ec2_access_key_id] || (@config_from_file['ec2']['access_key_id'] if @config_from_file)
+      @ec2_secret_access_key = options[:ec2_secret_access_key] || (@config_from_file['ec2']['secret_access_key'] if @config_from_file)
+      @ec2_region = options[:ec2_region] || (@config_from_file['ec2']['region'] if @config_from_file)
+      @ec2_tag_attribute = options[:ec2_tag_attribute] || (@config_from_file['ec2']['tag_attribute'] if @config_from_file)
+      @ec2_ip_attribute = options[:ec2_ip_attribute] || (@config_from_file['ec2']['ip_attribute'] if @config_from_file)
+
+      @bind9_notify_file = options[:bind9_notify_file] || (@config_from_file['bind9']['notify_file'] if @config_from_file)
+      @bind9_acl_masters_file = options[:bind9_acl_masters_file] || (@config_from_file['bind9']['acl_masters_file'] if @config_from_file)
+      @bind9_named_conf_file = options[:bind9_named_conf_file] || (@config_from_file['bind9']['named_conf_file'] if @config_from_file)
+      @bind9_zone_files = split(options[:bind9_zone_files]) || (@config_from_file['bind9']['zone_files'] if @config_from_file)
+      @bind9_checkconf_path = options[:bind9_checkconf_path] || (@config_from_file['bind9']['checkconf_path'] if @config_from_file)
+      @bind9_checkzone_path = options[:bind9_checkzone_path] || (@config_from_file['bind9']['checkzone_path'] if @config_from_file)
+      @bind9_init_cmd = options[:bind9_init_cmd] || (@config_from_file['bind9']['init_cmd'] if @config_from_file)
+      @bind9_masters = split(options[:bind9_masters]) || (@config_from_file['bind9']['masters'] if @config_from_file)
+    end
+
+    def split(param)
+      if param
+        param.split(',')
+      end
+    end
+
+    def to_hash
+      hash = {}
+      self.instance_variables.each {|var| hash[var.to_s.delete("@").to_sym] = self.instance_variable_get(var) }
+      return hash
+    end
+
+    def from_file(config_file)
+      if config_file && File.exists?(config_file)
+       YAML.load_file(config_file)
+      end      
+    end
+  end
+end

data/lib/elasticdns.rb

@@ -0,0 +1,166 @@
+require 'yaml'
+require 'aws-sdk'
+require 'tempfile'
+require 'fileutils'
+require 'digest/md5'
+require 'pp'
+require_relative 'config'
+
+module Elasticdns
+  class Generator
+    def initialize(options={})
+      @config = Elasticdns::Config.new(options).to_hash
+    end
+
+    def run
+      pp @config if @config[:debug]
+      if @config[:slave] == 'true'
+        update_acl_masters_file
+      else
+        update_notify_file
+      end
+    end
+
+    def ec2init
+      AWS.memoize do
+        AWS::EC2.new(:access_key_id => @config[:ec2_access_key_id], :secret_access_key => @config[:ec2_secret_access_key], :region => @config[:ec2_region])
+      end
+    end
+
+    def closefile(file)
+      file.close
+      file.unlink
+    end
+
+    def md5(string)
+      md5 = Digest::MD5.hexdigest(string)
+    end
+
+    def md5chk(orig_file, new_md5)
+      File.open(orig_file, 'r') do |file|
+        file.each_line do |line|
+          if line =~ /### MD5:/
+            old_md5 = line.split(':')[1]
+            if old_md5 =~ /#{new_md5}/
+              return new_md5
+            end
+          end
+        end
+      end if File.exists?(orig_file)
+      return false
+    end
+
+    def updatefile(ips, new_md5, temp_file, orig_file, prefix)
+      puts "Updating #{orig_file}"
+      md5 = "### MD5:#{new_md5}\n"
+      line = "#{prefix} { #{ips.to_s.gsub(/\[|\]|"/, '').gsub(/,/, ';')}; };\n"
+      temp_file.write(md5)
+      temp_file.write(line)
+      temp_file.rewind
+      FileUtils.mv(temp_file.path, orig_file)
+      return line
+    end
+
+    def bind_checkconf
+      `#{@config[:bind9_checkconf_path]} #{@config[:bind9_named_conf_file]}`
+    end
+
+    def bind_checkzone
+      @config[:bind9_zone_files].each do |zone|
+        `#{@config[:bind9_checkzone_path]} #{zone}`
+      end
+    end
+
+    def bind_init_cmd
+      if bind_checkconf && bind_checkzone
+        `#{@config[:bind9_init_cmd]}`
+      end
+    end
+
+    def update_hosts_file
+      orig_file = '/etc/hosts'
+      temp_file = Tempfile.new('hosts')
+      begin
+        ec2 = ec2init
+        File.open(orig_file, 'r') do |file|
+          file.each_line do |line|
+            unless line =~ /### EC2 HOSTS/
+              temp_file.write("#{line}")
+            end
+          end
+        end
+        ec2.instances.each do |i|
+          if i.status == :running
+            if i.tags[@config[:ec2_tag_attribute]]
+              temp_file.write("#{i.private_ip_address}\t#{i.tags[@config[:ec2_tag_attribute]]}\t### EC2 HOSTS\n")
+            end
+          end
+        end
+        temp_file.rewind
+        FileUtils.mv(temp_file.path, orig_file)
+      ensure
+        temp_file.close
+        temp_file.unlink
+      end
+    end
+
+    def update_notify_file
+      # File dedicated to master servers
+      orig_file = @config[:bind9_notify_file]
+      temp_file = Tempfile.new('also-notify')
+      prefix = "also-notify"
+      begin
+        ec2 = ec2init
+        ips = []
+        ec2.instances.each do |i|
+          if i.status == :running
+            ips << i.send(@config[:ec2_ip_attribute])
+            #unless @config[:bind9_masters].include?(i.tags[@config[:ec2_tag_attribute]])
+          end
+        end
+        write_to_file(ips, orig_file, temp_file, prefix)
+      ensure
+        temp_file.close
+        temp_file.unlink
+      end
+    end
+
+    def update_acl_masters_file
+      # File dedicated to slave servers
+      orig_file = @config[:bind9_acl_masters_file]
+      temp_file = Tempfile.new('acl_masters_file')
+      prefix = "acl 'masters'"
+      begin
+        ec2 = ec2init
+        ips = []
+        ec2.instances.each do |i|
+          meta_name = i.tags[@config[:ec2_tag_attribute]]
+          if i.status == :running
+            @config[:bind9_masters].each do |master|
+              if meta_name == master
+                ips << i.send(@config[:ec2_ip_attribute])
+              end
+            end
+          end
+        end
+        write_to_file(ips, orig_file, temp_file, prefix)
+      ensure
+        closefile(temp_file)
+      end
+    end
+
+    def write_to_file(ips, orig_file, temp_file, prefix)
+      if ips[0]
+        unless md5chk(orig_file, md5(ips.to_s))
+          update = updatefile(ips, md5(ips.to_s), temp_file, orig_file, prefix)
+          bind_init_cmd
+          puts "#{orig_file} updated with: " + update
+        else
+          puts "nothing changed in #{orig_file}: " + md5(ips.to_s)
+        end
+      else
+        puts 'No master found'
+      end
+    end
+  end
+end

data/rakefile.rb

@@ -0,0 +1,90 @@
+require './lib/ec2_dns.rb'
+
+task :update_hosts_file do
+  orig_file = '/etc/hosts'
+  temp_file = Tempfile.new('hosts')
+  begin
+    ec2 = ec2init
+    File.open(orig_file, 'r') do |file|
+      file.each_line do |line|
+        unless line =~ /### EC2 HOSTS/
+          temp_file.write("#{line}")
+        end
+      end
+    end
+    ec2.instances.each do |i|
+      if i.status == :running
+        if i.tags[:meta_name]
+          temp_file.write("#{i.private_ip_address}\t#{i.tags[:meta_name]}\t### EC2 HOSTS\n")
+        end
+      end
+    end
+    temp_file.rewind
+    FileUtils.mv(temp_file.path, orig_file)
+  ensure
+    temp_file.close
+    temp_file.unlink
+  end
+end
+
+task :update_notify_file do  
+  orig_file = @cnf['bind9']['notify_file']
+  temp_file = Tempfile.new('also-notify')
+  prefix = "also-notify"
+  begin
+    ec2 = ec2init
+    ips = []
+    ec2.instances.each do |i|
+      if i.status == :running
+        ips << i.private_ip_address
+      end
+    end
+    if ips[0]
+      unless md5chk(orig_file, md5(ips.to_s))
+        update = updatefile(ips, md5(ips.to_s), temp_file, orig_file, prefix)
+        bind_reload
+        puts 'update with: ' + update
+      else
+        puts 'nothing changed: ' + md5(ips.to_s)
+      end
+    else
+      puts 'no master'
+    end
+  ensure
+    temp_file.close
+    temp_file.unlink
+  end
+end
+
+task :update_acl_masters_file do  
+  orig_file = @cnf['bind9']['acl_masters_file']
+  temp_file = Tempfile.new('acl_masters_file')
+  prefix = "acl 'masters'"
+  begin
+    ec2 = ec2init
+    ips = []
+    ec2.instances.each do |i|
+      meta_name = i.tags[@cnf['ec2']['meta_tag_name']]
+      if i.status == :running
+        @cnf['bind9']['masters'].each do |master|
+          if meta_name == master
+            ips << i.private_ip_address
+          end
+        end
+      end
+    end
+    if ips[0]
+      unless md5chk(orig_file, md5(ips.to_s))
+        update = updatefile(ips, md5(ips.to_s), temp_file, orig_file, prefix)
+        bind_reload
+        puts 'update with: ' + update
+      else
+        puts 'nothing changed: ' + md5(ips.to_s)
+      end
+    else
+      puts 'no master'
+    end
+  ensure
+    closefile(temp_file)
+  end
+end

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification 
+name: elasticdns
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Gabriel Klein
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2013-06-23 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: fileutils
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: aws-sdk
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+description: Elasticdns, a simple bind9 configurator
+email: gabriel.klein.fr@gmail.com
+executables: 
+- elasticdns
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .DS_Store
+- Gemfile
+- Gemfile.lock
+- bin/elasticdns
+- config.yml
+- elasticdns.gemspec
+- lib/config.rb
+- lib/elasticdns.rb
+- rakefile.rb
+has_rdoc: true
+homepage: http://github.com/GabKlein/elasticdns
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Elasticdns, a simple bind9 configurator
+test_files: []
+