knife_sous 0.0.1

data/.gitignore

@@ -0,0 +1 @@
+tmp

data/.ruby-version

@@ -0,0 +1 @@
+ruby-1.9.3-p362@knife_sous

data/Gemfile

@@ -0,0 +1,9 @@
+source "http://rubygems.org"
+
+gemspec
+
+gem 'guard'
+gem 'guard-rspec'
+gem 'pry'
+gem 'rb-fsevent'
+

data/Gemfile.lock

@@ -0,0 +1,106 @@
+PATH
+  remote: .
+  specs:
+    knife_sous (0.0.1)
+      knife-solo (~> 0.2.0)
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    archive-tar-minitar (0.5.2)
+    chef (11.4.0)
+      erubis
+      highline (>= 1.6.9)
+      json (>= 1.4.4, <= 1.7.7)
+      mixlib-authentication (>= 1.3.0)
+      mixlib-cli (~> 1.3.0)
+      mixlib-config (>= 1.1.2)
+      mixlib-log (>= 1.3.0)
+      mixlib-shellout
+      net-ssh (~> 2.6)
+      net-ssh-multi (~> 1.1.0)
+      ohai (>= 0.6.0)
+      rest-client (>= 1.0.4, < 1.7.0)
+      yajl-ruby (~> 1.1)
+    coderay (1.0.8)
+    diff-lcs (1.1.3)
+    erubis (2.7.0)
+    guard (1.6.2)
+      listen (>= 0.6.0)
+      lumberjack (>= 1.0.2)
+      pry (>= 0.9.10)
+      terminal-table (>= 1.4.3)
+      thor (>= 0.14.6)
+    guard-rspec (2.4.0)
+      guard (>= 1.1)
+      rspec (~> 2.11)
+    highline (1.6.15)
+    ipaddress (0.8.0)
+    json (1.7.7)
+    knife-solo (0.2.0)
+      chef (>= 10.12)
+      librarian (~> 0.0.20)
+      net-ssh (>= 2.2.2, < 3.0)
+    librarian (0.0.26)
+      archive-tar-minitar (>= 0.5.2)
+      chef (>= 0.10)
+      highline
+      thor (~> 0.15)
+    listen (0.7.2)
+    lumberjack (1.0.2)
+    method_source (0.8.1)
+    mime-types (1.21)
+    mixlib-authentication (1.3.0)
+      mixlib-log
+    mixlib-cli (1.3.0)
+    mixlib-config (1.1.2)
+    mixlib-log (1.4.1)
+    mixlib-shellout (1.1.0)
+    net-ssh (2.6.5)
+    net-ssh-gateway (1.2.0)
+      net-ssh (>= 2.6.5)
+    net-ssh-multi (1.1)
+      net-ssh (>= 2.1.4)
+      net-ssh-gateway (>= 0.99.0)
+    ohai (6.16.0)
+      ipaddress
+      mixlib-cli
+      mixlib-config
+      mixlib-log
+      mixlib-shellout
+      systemu
+      yajl-ruby
+    pry (0.9.11.4)
+      coderay (~> 1.0.5)
+      method_source (~> 0.8)
+      slop (~> 3.4)
+    rake (10.0.3)
+    rb-fsevent (0.9.3)
+    rest-client (1.6.7)
+      mime-types (>= 1.16)
+    rspec (2.12.0)
+      rspec-core (~> 2.12.0)
+      rspec-expectations (~> 2.12.0)
+      rspec-mocks (~> 2.12.0)
+    rspec-core (2.12.2)
+    rspec-expectations (2.12.1)
+      diff-lcs (~> 1.1.3)
+    rspec-mocks (2.12.1)
+    slop (3.4.3)
+    systemu (2.5.2)
+    terminal-table (1.4.5)
+    thor (0.17.0)
+    yajl-ruby (1.1.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 1.1)
+  guard
+  guard-rspec
+  knife_sous!
+  pry
+  rake (~> 10.0.0)
+  rb-fsevent
+  rspec (~> 2.12.0)

data/Guardfile

@@ -0,0 +1,6 @@
+
+guard 'rspec', cli: '--color' do
+  watch(%r{^spec/.+_spec\.rb$})
+  watch(%r{^lib/(.+)\.rb$})     { |m| "spec/lib/#{m[1]}_spec.rb" }
+  watch('spec/spec_helper.rb')  { "spec" }
+end

data/README.md

@@ -0,0 +1,51 @@
+# Knife Sous
+
+A Chef Knife plugin which uses a DSL to configure and manage
+[knife-solo](http://matschaffer.github.com/knife-solo/) nodes.
+
+## Usage
+
+Simply ensure the gem is installed using:
+
+    gem install knife_sous
+
+Or add this to your Gemfile if you use bundler:
+
+    gem 'knife_sous'
+
+Having the gem installed will add Knife subcommands. Run `knife sous` with no
+arguments to see a list of available commands.
+
+
+## Configure
+
+Node configuration is done in `nodes/nodes.rb`. It uses a DSL similar to Rake.
+You can namespace your nodes however you want. Example `nodes.rb`:
+
+```rb
+namespace :production do |prod|
+  prod.namespace :web do |web|
+    web.node :nodetastic,
+      node_config: 'nodes/some_node.json',
+      hostname: 12.34.56.78,
+      user: 'deploy',
+      ssh_port: 1234
+    end
+  end
+  prod.namespace :db do |db|
+    db.node :database_node, node_config: 'nodes/database.json', hostname: 123.456.78
+  end
+end
+
+node :vagrant, node_config:'nodes/some_node.json'
+
+```
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request
+

data/Rakefile

@@ -0,0 +1,16 @@
+#!/usr/bin/env rake
+
+require 'rubygems'
+require 'bundler/setup'
+require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
+
+desc 'Default: run unit tests.'
+task :default => :spec
+
+desc "Run all specs"
+RSpec::Core::RakeTask.new do |t|
+  t.pattern = 'spec/**/*_spec.rb'
+  t.rspec_opts = '--color'
+ end
+

data/knife_sous.gemspec

@@ -0,0 +1,28 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+
+Gem::Specification.new do |s|
+  s.authors     = "Shaun Dern"
+  s.email       = "shaun@substantail.com"
+
+  s.name        = 'knife_sous'
+  s.version     = "0.0.1"
+  s.description = %q{Knife plugin for managing knife-solo nodes}
+  s.summary     = %q{A Chef Knife plugin which uses a DSL to configure and
+manage knife solo nodes. Run knife solo commands on one or multiple nodes.}
+  s.homepage    = %q{http://github.com/substantial/knife_sous}
+  s.license     = "MIT"
+
+  s.rubyforge_project = "knife_sous"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_development_dependency "rspec", "~> 2.12.0"
+  s.add_development_dependency "rake", "~> 10.0.0"
+  s.add_development_dependency "bundler", "~> 1.1"
+  s.add_dependency "knife-solo", "~> 0.2.0"
+end
+

data/lib/chef/knife/sous_bootstrap.rb

@@ -0,0 +1,31 @@
+require 'chef/knife'
+require 'knife_sous/processor_command'
+require 'knife_sous/configure_command'
+require 'knife_sous/node_command'
+
+class Chef
+  class Knife
+    class SousBootstrap < Knife
+      include KnifeSous::ProcessorCommand
+      include KnifeSous::ConfigureCommand
+      include KnifeSous::NodeCommand
+
+      deps do
+        require 'chef/knife/solo_bootstrap'
+      end
+
+      banner "knife sous bootstrap [NAMESPACE] NODE"
+
+      def solo_bootstrap_node(node)
+        solo_bootstrap_command = configure_command(Chef::Knife::SoloBootstrap.new, node)
+        Chef::Knife::SoloBootstrap.load_deps
+        solo_bootstrap_command.name_args << node.hostname
+        solo_bootstrap_command.name_args << node.node_config
+        solo_bootstrap_command.run
+      end
+
+      alias_method :solo_command, :solo_bootstrap_node
+    end
+  end
+end
+

data/lib/chef/knife/sous_clean.rb

@@ -0,0 +1,30 @@
+require 'chef/knife'
+require 'knife_sous/processor_command'
+require 'knife_sous/configure_command'
+require 'knife_sous/node_command'
+
+class Chef
+  class Knife
+    class SousClean < Knife
+      include KnifeSous::ProcessorCommand
+      include KnifeSous::ConfigureCommand
+      include KnifeSous::NodeCommand
+
+      deps do
+        require 'chef/knife/solo_clean'
+      end
+
+      banner "knife sous clean [NAMESPACE] NODE"
+
+      def solo_clean_node(node)
+        solo_cook_command = configure_command(Chef::Knife::SoloClean.new, node)
+        Chef::Knife::SoloClean.load_deps
+        solo_cook_command.name_args << node.hostname
+        solo_cook_command.run
+      end
+
+      alias_method :solo_command, :solo_clean_node
+    end
+  end
+end
+

data/lib/chef/knife/sous_cook.rb

@@ -0,0 +1,31 @@
+require 'chef/knife'
+require 'knife_sous/processor_command'
+require 'knife_sous/configure_command'
+require 'knife_sous/node_command'
+
+class Chef
+  class Knife
+    class SousCook < Knife
+      include KnifeSous::ProcessorCommand
+      include KnifeSous::ConfigureCommand
+      include KnifeSous::NodeCommand
+
+      deps do
+        require 'chef/knife/solo_cook'
+      end
+
+      banner "knife sous cook [NAMESPACE] NODE"
+
+      def solo_cook_node(node)
+        solo_cook_command = configure_command(Chef::Knife::SoloCook.new, node)
+        Chef::Knife::SoloCook.load_deps
+        solo_cook_command.name_args << node.hostname
+        solo_cook_command.name_args << node.node_config
+        solo_cook_command.run
+      end
+
+      alias_method :solo_command, :solo_cook_node
+    end
+  end
+end
+

data/lib/chef/knife/sous_init.rb

@@ -0,0 +1,39 @@
+require 'chef/knife'
+require 'knife_sous/processor_command'
+
+class Chef
+  class Knife
+    class SousInit < Knife
+      include KnifeSous::ProcessorCommand
+
+      banner "knife sous init [DIRECTORY]"
+
+      def node_manifest_example_path
+        File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'templates', 'nodes.example.rb'))
+      end
+
+      def run
+        unless name_args.empty?
+          target = File.expand_path(name_args.first)
+        else
+          target = File.expand_path(Dir.pwd, File.join('nodes'))
+        end
+        copy_template(target)
+      end
+
+      def copy_template(location)
+        FileUtils.mkdir_p(location)
+
+        target_file = File.join(location, 'nodes.example.rb' )
+
+        unless File.exists?(target_file)
+          FileUtils.cp(node_manifest_example_path, target_file)
+          ui.msg "Example nodes.rb copied to #{target_file}"
+        else
+          ui.error "File already exists: #{target_file}"
+        end
+      end
+    end
+  end
+end
+

data/lib/chef/knife/sous_list.rb

@@ -0,0 +1,22 @@
+require 'chef/knife'
+require 'knife_sous/processor_command'
+require 'knife_sous/namespace_presenter'
+
+class Chef
+  class Knife
+    class SousList < Knife
+      include KnifeSous::ProcessorCommand
+
+      banner "knife sous list"
+
+      def run
+        present_nodes
+      end
+
+      def present_nodes
+        ui.output(KnifeSous::NamespacePresenter.new(root_namespace).present)
+      end
+    end
+  end
+end
+

data/lib/chef/knife/sous_prepare.rb

@@ -0,0 +1,31 @@
+require 'chef/knife'
+require 'knife_sous/processor_command'
+require 'knife_sous/configure_command'
+require 'knife_sous/node_command'
+
+class Chef
+  class Knife
+    class SousPrepare < Knife
+      include KnifeSous::ProcessorCommand
+      include KnifeSous::ConfigureCommand
+      include KnifeSous::NodeCommand
+
+      deps do
+        require 'chef/knife/solo_prepare'
+      end
+
+      banner "knife sous prepare [NAMESPACE] NODE"
+
+      def solo_prepare_node(node)
+        solo_cook_command = configure_command(Chef::Knife::SoloPrepare.new, node)
+        Chef::Knife::SoloPrepare.load_deps
+        solo_cook_command.name_args << node.hostname
+        solo_cook_command.name_args << node.node_config
+        solo_cook_command.run
+      end
+
+      alias_method :solo_command, :solo_prepare_node
+    end
+  end
+end
+

data/lib/knife_sous/configure_command.rb

@@ -0,0 +1,12 @@
+module KnifeSous
+  module ConfigureCommand
+
+    module_function
+
+    def configure_command(command, node)
+      command.config.merge!(node.config)
+      command
+    end
+  end
+end
+

data/lib/knife_sous/dsl_definitions.rb

@@ -0,0 +1,29 @@
+require 'knife_sous/namespace'
+require 'knife_sous/node'
+
+module KnifeSous
+  module DSL
+    def self.included(base)
+      base.class_eval do
+        base.extend Forwardable
+        def_delegators :children, :<<, :first, :last, :[], :map, :each, :to_a, :keep_if, :empty?
+      end
+    end
+
+    def children
+      @children ||= []
+    end
+
+    def namespace(name, &block)
+      namespace = Namespace.new(name)
+      namespace.evaluate_block(&block)
+      children << namespace
+    end
+
+    def node(name, args={})
+      node = Node.new(name, args)
+      children << node
+    end
+  end
+end
+

data/lib/knife_sous/dsl_wrapper.rb

@@ -0,0 +1,8 @@
+module KnifeSous
+  module DSLWrapper
+    def evaluate_block(&block)
+      yield self if block_given?
+    end
+  end
+end
+

data/lib/knife_sous/mixins/hash_mixins.rb

@@ -0,0 +1,14 @@
+module KnifeSous
+  module HashMixins
+
+    module_function
+
+    def normalize_hash(hash)
+      Hash[hash.map{ |k, v| [sanitize_key(k), v.to_s] }]
+    end
+
+    def sanitize_key(key)
+      key.to_s.downcase.gsub(/-/, '_').to_sym
+    end
+  end
+end

data/lib/knife_sous/namespace.rb

@@ -0,0 +1,19 @@
+require 'knife_sous/node'
+
+require 'knife_sous/dsl_definitions'
+require 'knife_sous/dsl_wrapper'
+
+
+module KnifeSous
+  class Namespace
+    include DSL
+    include DSLWrapper
+
+    attr_reader :name
+
+    def initialize(name)
+      @name = name.to_s
+    end
+  end
+end
+

data/lib/knife_sous/namespace_presenter.rb

@@ -0,0 +1,27 @@
+require 'knife_sous/node_presenter'
+
+module KnifeSous
+  class NamespacePresenter
+    attr_reader :namespace
+
+    def initialize(namespace)
+      @namespace = namespace
+    end
+
+    def present
+      return @namespace.name if @namespace.empty?
+
+      nodes = ""
+      @namespace.each do |child|
+        nodes << "#{@namespace.name} #{presenter(child).present}".lstrip
+      end
+      nodes
+    end
+
+    def presenter(item)
+      klass = item.is_a?(Node) ? NodePresenter : NamespacePresenter
+      klass.new(item)
+    end
+  end
+end
+

data/lib/knife_sous/node.rb

@@ -0,0 +1,47 @@
+require 'knife_sous/dsl_wrapper'
+require 'knife_sous/mixins/hash_mixins'
+
+module KnifeSous
+  class Node
+    include KnifeSous::HashMixins
+
+    attr_reader :name, :config
+
+    def initialize(name, config_args={})
+      @name = name.to_s
+      @config = {}
+      update_config(config_args)
+    end
+
+    def update_config(other_config={})
+      config = convert_aliases(normalize_hash(other_config))
+      @config.merge!(config)
+    end
+
+    def convert_aliases(config_hash = {})
+      config_aliases.each do |key, value|
+        if config_hash.has_key?(key)
+          config_hash[value] = config_hash[key]
+          config_hash.delete(key)
+        end
+      end
+      config_hash
+    end
+
+    def node_name
+      @config[:node_name] || @name
+    end
+
+    def method_missing(method_name, *args, &block)
+      @config[method_name.to_sym]
+    end
+
+    def config_aliases
+      {
+        ssh_config_file: :ssh_config,
+        ip: :hostname
+      }
+    end
+  end
+end
+

data/lib/knife_sous/node_command.rb

@@ -0,0 +1,36 @@
+module KnifeSous
+  module NodeCommand
+
+    def search_for_target
+      result = search(name_args)
+      if result.nil?
+        ui.error "Can't find node. Run `knife sous list` to see nodes"
+        exit 1
+      end
+      result
+    end
+
+    def run
+      check_args
+      search_result = search_for_target
+      process_result(search_result)
+    end
+
+    def process_result(result)
+      if result.is_a? KnifeSous::Namespace
+        result.each { |child| process_result(child) }
+      else
+        solo_command(result)
+      end
+    end
+
+    def check_args
+      unless name_args.size > 0
+        ui.fatal "You need to specificy a node or namespace"
+        show_usage
+        exit 1
+      end
+    end
+  end
+end
+

data/lib/knife_sous/node_presenter.rb

@@ -0,0 +1,13 @@
+module KnifeSous
+  class NodePresenter
+    attr_reader :node
+
+    def initialize(node)
+      @node = node
+    end
+
+    def present
+      "#{@node.name}\n"
+    end
+  end
+end

data/lib/knife_sous/processor_command.rb

@@ -0,0 +1,52 @@
+require 'knife_sous/namespace'
+
+module KnifeSous
+  module ProcessorCommand
+    def self.included(other)
+      other.class_eval do
+        option :node_manifest_file,
+          :long        => '--node-manifest-file NODE_MANIFEST_FILE',
+          :description => 'Alternate location for node manifest config file',
+          :default => File.join("nodes", "nodes.rb")
+      end
+    end
+
+    def validate_config!
+      unless File.exists?(manifest_file_path)
+        ui.fatal("Couldn't find #{manifest_file_path}")
+        exit 1
+      end
+      unless File.readable?(manifest_file_path)
+        ui.fatal("Can't read #{manifest_file_path}")
+        exit 1
+      end
+      true
+    end
+
+    def search(target_qualifiers = [])
+      namespace = root_namespace
+      target_qualifiers.each do |target_qualifier|
+        return nil if namespace.nil?
+        namespace = namespace.keep_if { |item| item.name == target_qualifier }.first
+      end
+      namespace
+    end
+
+    def root_namespace
+      @root_namespace ||= process_config
+    end
+
+    def manifest_file_path
+      config[:node_manifest_file] ||= default_config[:node_manifest_file]
+      @manifest_file_path ||= File.expand_path(File.join(Dir.pwd, config[:node_manifest_file]))
+    end
+
+    def process_config
+      validate_config!
+      root_namespace = Namespace.new('')
+      root_namespace.instance_eval(File.read(manifest_file_path), manifest_file_path)
+      root_namespace
+    end
+  end
+end
+

data/lib/knife_sous.rb

@@ -0,0 +1,3 @@
+
+require 'chef/knife/sous_list'
+