knife-dns-update 0.0.1

checksums.yaml

@@ -0,0 +1,15 @@
+---
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    NjJiMjE0YjgyYWQzMTY0OWQzODM0OTVmMzQ3MTc0OGU5ZGEzM2IzYQ==
+  data.tar.gz: !binary |-
+    NTg4NmFjMDI4OTllZDlkZjIxYTEwNGEyNDNmZmNmZTkzNjhmYzE4ZA==
+!binary "U0hBNTEy":
+  metadata.gz: !binary |-
+    NWU1OWFlMTg5YjU2NzE0NmY1NzJhNDdmNTQwYzYyN2MzODU3MmY5NDIxMGI4
+    MjUwNmJhYzNmYTY1MzNhNDE3NjM4ODlkZjM0YmUxMGJiNzQ0MDI2MjIxN2I5
+    YWRhMGIyZWMzZTIyZjBjYjIzMmIyMTM0ZWE5ZWVmYjZkODcwNjg=
+  data.tar.gz: !binary |-
+    MjI1Y2IyNjFmOTEzMTc4ZDI4NjA1MmY4ZjQ5NWI5YTA0Mzc5NjNkOWZhMzNj
+    NDlhNmM0M2ZjYzI5MTlmOTY4NjcxOWYyZGUyZDY3YzdiMmI5NDg3YTNjZmRi
+    ZWY3NWI5ZDI3YTFkYzk3YTlmNTRkMDQ0YzVhOTgzZDVjNGYwODI=

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+*~
+.*.swp
+.DS_Store
+/.bundle
+/.config
+/.yardoc
+/Gemfile.lock
+/InstalledFiles
+/coverage
+/doc/public
+/lib/bundler/man
+/pkg
+/spec/reports
+/tmp
+/vendor/cache

data/.travis.yml

@@ -0,0 +1,13 @@
+---
+bundler_args: --without developer_workstation
+script: bundle exec thor spec:ci
+rvm:
+- 1.9.3
+- 2.0.0
+notifications:
+  hipchat:
+    secure: ! 'XggFlX0Al0nEMLl0X3Fsyr8X5NSdXwfyhBhKg3bskYoKzbIcexA8eDJBiuSa
+
+      ych7knx0SI/Dzhvs+WnQjWyBhu5VwHBnQvBxMGqVZ71WLn7IbzAME1W8q0b0
+
+      mRzkvCIRWuMaPz3m24LuTyga9w0XB1AFN5+tlP6fCuYXglCOYws='

data/CHANGELOG.md

@@ -0,0 +1,6 @@
+# Changes
+
+## 0.0.1
+
+ * Initial release
+ * Created on Friday, 2013-06-21

data/CONTRIBUTING.md

@@ -0,0 +1,43 @@
+# Contributing
+
+## Developing
+
+1. Fork the repository on GitHub
+2. Create your feature branch (`git checkout -b feature/awesomeness`)
+3. Create your changes.
+   * Add test cases in `spec/`. It's best if you first write a failing
+     test case, commit it, and then fix it in next commit - this makes
+     the whole change easier to review.
+   * Document your changes.
+4. Commit your changes (`git commit -am 'Add more awesomeness'`)
+5. Push to the branch (`git push -u origin feature/awesomeness`)
+6. Create new Pull Request on GitHub
+
+## Testing
+
+### Install what's needed
+
+Make sure you have [http://gembundler.com/](Gem Bundler) version 1.3
+or greater installed.  If in doubt, just use [http://rvm.io/](RVM) or
+[http://rbenv.org/](rbenv).
+
+    $ gem install bundler
+
+Clone the project:
+
+    $ git clone git://github.com/3ofcoins/knife-dns-update.git
+
+Then, run:
+
+    $ cd knife-dns-update
+    $ bundle install
+    
+Bundler will install all the needed gems and their dependencies.
+
+### Running tests
+
+    $ bundle exec thor spec
+    
+To generate test coverage report, tell it to Thor
+    
+    $ bundle exec thor spec --coverage

data/Gemfile

@@ -0,0 +1,13 @@
+source 'https://rubygems.org'
+
+gemspec
+
+# Helpers used with development, but not needed in runtime, build
+# time, or for tests.
+group :developer_workstation do
+  gem 'awesome_print'
+  gem 'pry'
+  gem 'pry-debugger'
+  gem 'pry-rescue'
+  gem 'pry-stack_explorer'
+end

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Maciej Pasternacki <maciej@3ofcoins.net>
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,25 @@
+# Knife Dns Update
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'knife-dns-update'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install knife-dns-update
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+See the [CONTRIBUTING.md](CONTRIBUTING.md) file

data/Thorfile

@@ -0,0 +1,60 @@
+$:.push File.expand_path('../lib', __FILE__)
+require 'rubygems'
+
+require 'bundler/setup'
+
+require 'rake/testtask'
+require 'thor/rake_compat'
+
+class Default < Thor
+  class Gem < Thor
+    namespace :gem
+
+    include Thor::RakeCompat
+    Bundler::GemHelper.install_tasks
+
+    desc "build", "Build knife-dns-update-#{KnifeDnsUpdate::VERSION}.gem into the pkg directory"
+    def build
+      Rake::Task["build"].execute
+    end
+
+    desc "release", "Create tag v#{KnifeDnsUpdate::VERSION} and build and push knife-dns-update-#{KnifeDnsUpdate::VERSION}.gem to Rubygems"
+    def release
+      Rake::Task["release"].execute
+    end
+
+    desc "install", "Build and install knife-dns-update-#{KnifeDnsUpdate::VERSION}.gem into system gems"
+    def install
+      Rake::Task["install"].execute
+    end
+  end
+
+  class Test < Thor
+    namespace :test
+    default_command :all
+
+    include Thor::RakeCompat
+
+    Rake::TestTask.new :spec do |task|
+      task.libs << 'spec'
+      task.test_files = FileList['spec/**/*_spec.rb']
+    end
+
+    desc 'spec', 'Run specs'
+    def spec
+      Rake::Task['spec'].execute
+    end
+
+    desc 'all', 'Run all tests'
+    def all
+      invoke(:spec)
+    end
+
+    desc 'ci', 'Run all tests for continuous integration'
+    def ci
+      ENV['CI'] = 'true'
+      invoke(:all)
+    end
+  end
+end
+

data/knife-dns-update.gemspec

@@ -0,0 +1,29 @@
+# -*- mode: ruby; coding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'knife-dns-update/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "knife-dns-update"
+  spec.version       = KnifeDnsUpdate::VERSION
+  spec.authors       = ["Maciej Pasternacki"]
+  spec.email         = ["maciej@pasternacki.net"]
+  spec.description   = "Updates DNS based on Chef database"
+  spec.summary       = "Updates Route 53 DNS entries based on Chef database contents"
+  spec.homepage      = "https://github.com/3ofcoins/knife-dns-update/"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "chef"
+  spec.add_dependency "fog", "~> 1.9"
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "minitest"
+  spec.add_development_dependency "mocha"
+  spec.add_development_dependency 'simplecov'
+  spec.add_development_dependency "thor", "~> 0.18.1"
+  spec.add_development_dependency "wrong", ">= 0.7.0"
+end

data/lib/chef/knife/dns_update.rb

@@ -0,0 +1,127 @@
+require 'chef/knife'
+
+module KnifeDnsUpdate
+  class DnsUpdate < Chef::Knife
+    banner "knife dns update"
+
+    deps do
+      require 'set'
+      require 'chef/node'
+      require 'chef/api_client'
+      require 'chef/search/query'
+      require 'fog'
+      require 'knife-dns-update'
+    end
+
+    option :dry_run,
+      :short => "-n",
+      :long => "--dry-run",
+      :description => "Don't execute updates",
+      :boolean => false
+
+    option :aws_access_key_id,
+      :short => "-A ID",
+      :long => "--aws-access-key-id KEY",
+      :description => "Your AWS Access Key ID",
+      :proc => Proc.new { |key| Chef::Config[:knife][:aws_access_key_id] = key }
+
+    option :aws_secret_access_key,
+      :short => "-K SECRET",
+      :long => "--aws-secret-access-key SECRET",
+      :description => "Your AWS API Secret Access Key",
+      :proc => Proc.new { |key| Chef::Config[:knife][:aws_secret_access_key] = key }
+
+    attr_reader :entries, :cfg
+
+    def entry(name, type, data)
+      @entries ||= {}
+
+      name = nil if name == '@'
+      fqdn = [name, cfg.subdomain, cfg.zone].compact.join('.') + '.'
+      type = type.to_s.upcase
+
+      if entries.key?([fqdn, type])
+        ui.warn("Redefining #{fqdn} IN #{type} #{entries[[name,type]]} --> #{data}")
+      elsif config[:verbosity] >= 2
+        ui.info("Defining: #{fqdn} IN #{type} #{data}")
+      end
+      entries[[fqdn, type]] = data
+    end
+
+    def run
+      @cfg = Config.new('config/dns.rb')
+
+      Chef::Node.list.keys.sort.map(&Chef::Node.method(:load)).each do |node|
+        entry(node.name.gsub(/[^a-z0-9-]+/, '-'), *cfg.record_for_node(node))
+      end
+
+      cfg.entries.each do |name, optses|
+        optses.each do |opts|
+          entry(name, opts[:type], opts[:data]) if opts[:type] && opts[:data]
+          entry(name, :a, opts[:a]) if opts[:a]
+          entry(name, :cname, opts[:cname]) if opts[:cname]
+
+          if opts[:block] && !(opts[:node] || opts[:query])
+            # Bare block
+            entry(name, *opts[:block].call)
+          end
+
+          is_root_record = (name == '@')
+
+          if opts[:node]
+            entry(name, *cfg.record_for_node(
+              Chef::Node.load(opts[:node]), is_root_record, &opts[:block]))
+          end
+
+          if opts[:query]
+            q = Chef::Search::Query.new
+            q.search(:node, opts[:query]).first.each do |node|
+              entry( name.sub('*', cfg.personalization_token(node)),
+                     *cfg.record_for_node(node, is_root_record, &opts[:block]))
+            end
+          end
+        end
+      end
+
+      dns_config = Chef::Config[:dns].dup
+      dns_config[:provider] ||= 'AWS'
+      if dns_config[:provider] == 'AWS'
+        dns_config[:aws_access_key_id] ||= config[:aws_access_key_id]
+        dns_config[:aws_secret_access_key] ||= config[:aws_secret_access_key]
+      end
+      dns = Fog::DNS.new(dns_config)
+      zone = dns.zones.find { |z| z.domain =~ /#{Regexp.quote(cfg.zone)}\.?$/ }
+      raise "No zone found for #{cfg.zone}; available zones: #{dns.zones.map(&:domain).join(', ')}" unless zone
+
+      managed = Set.new
+
+      zone.records.each do |rec|
+        ui.info("Looking at: #{rec.name} IN #{rec.type} #{rec.value}") if config[:verbosity] >= 3
+
+        if cfg.subdomain && rec.name !~ /#{Regexp.quote(cfg.subdomain)}\.#{Regexp.quote(cfg.zone)}\.?$/
+          ui.info("Not in subdomain, skipping: #{rec.name} IN #{rec.type} #{rec.value}") if config[:verbosity] >= 2
+          next
+        end
+
+        value = rec.value.respond_to?(:first) ? rec.value.first : rec.value
+        if entries[[rec.name, rec.type]] == value
+          # Record exists on server and has the same value. We leave it be.
+          ui.info("= #{rec.name} IN #{rec.type} #{value}") if config[:verbosity] >= 1
+          managed.add([rec.name, rec.type])
+        else
+          ui.info "- #{rec.name} IN #{rec.type} #{value}"
+          rec.destroy unless config[:dry_run]
+        end
+
+        # FIXME next if cfg.subdomain && rec.name !~ /#{Regex.quote(cfg.subdomain).
+      end
+
+      entries.each do |resource, value|
+        next if managed.include?(resource)
+        name, type = resource
+        ui.info "+ #{name} IN #{type} #{value}"
+        zone.records.create(:name => name, :type => type, :value => value, :ttl => cfg.ttl) unless config[:dry_run]
+      end
+    end
+  end
+end

data/lib/knife-dns-update/config.rb

@@ -0,0 +1,75 @@
+module KnifeDnsUpdate
+  class Config
+    class << self
+      def option(name, default=nil)
+        ivar = "@#{name}".to_sym
+        define_method name do |*args|
+          if args.length.zero?
+            if instance_variable_defined?(ivar)
+              instance_variable_get(ivar)
+            else
+              default
+            end
+          elsif block_given?
+            instance_variable_set(ivar, yield(*args))
+          elsif args.length == 1
+            instance_variable_set(ivar, args[0])
+          else
+            instance_variable_set(ivar, args)
+          end
+        end
+      end
+    end
+
+    option(:zone) { |domain| domain.gsub(/^\.*|\.*$/, '') }
+    option :subdomain
+    option :ttl, 3600
+    attr_reader :entries
+
+    def initialize(path=nil)
+      @entries = {}
+      instance_eval(File.read(path), path, 1) if path
+    end
+
+    def entry(name, opt={}, &block)
+      raise "Entry #{name} without definition or block" unless opt||block_given?
+
+      case opt
+      when /^\d+\.\d+\.\d+\.\d+$/ then opt = { :a => opt }
+      when /^[a-z0-9.-]+\.$/      then opt = { :cname => opt }
+      when /^[a-z0-9.-]+$/        then opt = { :cname => "#{opt}.#{zone}." }
+      when /^name:[a-z0-9.-]+$/   then opt = { :node => opt[5..-1] }
+      when /:/                    then opt = { :query => opt }
+      end
+
+      opt[:block] = block if block_given?
+
+      if opt[:q]
+        opt[:query] = opt[:q]
+        opt.delete(:q)
+      end
+
+      entries[name] ||= []
+      entries[name] << opt
+    end
+
+    def record_for_node(node, is_root_record=false)
+      case
+      when block_given?
+        yield node
+      when !is_root_record && node['cloud'] && node['cloud']['public_hostname']
+        [ :cname, node['cloud']['public_hostname'] ]
+      when node['cloud'] && node['cloud']['public_ipv4']
+        [ :a, node['cloud']['public_ipv4'] ]
+      when !is_root_record && node['fqdn'] && node['fqdn'] !~ /#{Regexp.escape(zone)}$/
+        [ :cname, node['fqdn'] ]
+      when node['ipaddress']
+        [ :a, node['ipaddress'] ]
+      end
+    end
+
+    def personalization_token(node)
+      node.name.gsub(/[^a-z0-9-]+/, '-')
+    end
+  end
+end

data/lib/knife-dns-update/version.rb

@@ -0,0 +1,3 @@
+module KnifeDnsUpdate
+  VERSION = "0.0.1"
+end

data/lib/knife-dns-update.rb

@@ -0,0 +1,6 @@
+require "knife-dns-update/version"
+require "knife-dns-update/config"
+
+module KnifeDnsUpdate
+  # Your code goes here...
+end

data/spec/knife-dns-update/version_spec.rb

@@ -0,0 +1,11 @@
+require 'spec_helper'
+
+# A smoke test spec to make sure tests actually work
+
+module KnifeDnsUpdate
+  describe VERSION do
+    it 'is equal to itself' do
+      expect { VERSION == KnifeDnsUpdate::VERSION }
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,19 @@
+require 'rubygems'
+require 'bundler/setup'
+
+require 'minitest/autorun'
+require 'minitest/spec'
+require 'mocha/setup'
+require 'wrong'
+require 'wrong/adapters/minitest'
+
+Wrong.config.alias_assert :expect, override: true
+include Wrong
+
+if ENV['COVERAGE']
+  require 'simplecov'
+  SimpleCov.start
+  SimpleCov.command_name 'rake spec'
+end
+
+require "knife-dns-update"

metadata

@@ -0,0 +1,173 @@
+--- !ruby/object:Gem::Specification
+name: knife-dns-update
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Maciej Pasternacki
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-07-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: chef
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: fog
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.9'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.18.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.18.1
+- !ruby/object:Gem::Dependency
+  name: wrong
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.7.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.7.0
+description: Updates DNS based on Chef database
+email:
+- maciej@pasternacki.net
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .travis.yml
+- CHANGELOG.md
+- CONTRIBUTING.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Thorfile
+- knife-dns-update.gemspec
+- lib/chef/knife/dns_update.rb
+- lib/knife-dns-update.rb
+- lib/knife-dns-update/config.rb
+- lib/knife-dns-update/version.rb
+- spec/knife-dns-update/version_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/3ofcoins/knife-dns-update/
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.5
+signing_key: 
+specification_version: 4
+summary: Updates Route 53 DNS entries based on Chef database contents
+test_files:
+- spec/knife-dns-update/version_spec.rb
+- spec/spec_helper.rb