knife-node-attribute 0.2.0

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/Gemfile.lock

@@ -0,0 +1,58 @@
+PATH
+  remote: .
+  specs:
+    knife-attribute (0.1.0)
+      chef (>= 0.10.4)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    chef (11.6.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)
+    erubis (2.7.0)
+    highline (1.6.19)
+    ipaddress (0.8.0)
+    json (1.7.7)
+    mime-types (1.25)
+    mixlib-authentication (1.3.0)
+      mixlib-log
+    mixlib-cli (1.3.0)
+    mixlib-config (1.1.2)
+    mixlib-log (1.6.0)
+    mixlib-shellout (1.2.0)
+    net-ssh (2.6.8)
+    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.18.0)
+      ipaddress
+      mixlib-cli
+      mixlib-config
+      mixlib-log
+      mixlib-shellout
+      systemu
+      yajl-ruby
+    rest-client (1.6.7)
+      mime-types (>= 1.16)
+    systemu (2.5.2)
+    yajl-ruby (1.1.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  knife-attribute!

data/LICENSE

@@ -0,0 +1,19 @@
+Copyright (C) 2013 Daniel Schauenberg
+
+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,35 @@
+# knife-node-attribute
+
+## Overview
+knife plugin to interact with the various attributes on a nodes.
+
+## Usage
+Show all attributes on the node (or a specific one if given):
+
+    knife node attribute show foo.example.org
+    knife node attribute show foo.example.org normal [foo:bar]
+    knife node attribute show foo.example.org default [foo:bar]
+    knife node attribute show foo.example.org override [foo:bar]
+
+Delete a specific attribute on a node:
+
+    knife node attribute delete foo.example.org normal foo:bar
+    knife node attribute delete foo.example.org default foo:bar
+    knife node attribute delete foo.example.org override foo:bar
+
+
+## Installation
+Just get it from rubygems:
+
+    gem install knife-node-attribute
+
+
+## Contribute
+- fork the project
+- make your awesome changes
+- send a pull request
+
+## Thanks
+[@jonlives](https://github.com/jonlives) for his plugins which I took as
+examples and [@jgoulah](http://github.com/jgoulah) for the
+inspiration.

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require 'bundler/gem_tasks'

data/knife-node-attribute.gemspec

@@ -0,0 +1,17 @@
+$:.push File.expand_path('../lib', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.name          = 'knife-node-attribute'
+  gem.version       = '0.2.0'
+  gem.authors       = ["Daniel Schauenberg"]
+  gem.email         = 'd@unwiredcouch.com'
+  gem.homepage      = 'https://github.com/mrtazz/knife-node-attribute'
+  gem.summary       = "knife plugin to show or delete node attributes"
+  gem.description   = "knife plugin to show or delete node attributes"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.name          = "knife-node-attribute"
+  gem.require_paths = ["lib"]
+
+  gem.add_runtime_dependency 'chef', '>= 0.10.4'
+end

data/lib/chef/knife/knife-node-attribute.rb

@@ -0,0 +1,164 @@
+require 'chef/knife'
+
+
+ATTRIBUTE_CLASSES = [:normal, :default, :override]
+
+module KnifeNodeAttribute
+
+  module Helpers
+    def get_node(name)
+
+      puts "Looking for an fqdn of #{name} or name of #{name}"
+
+      searcher = Chef::Search::Query.new
+      result = searcher.search(:node, "fqdn:#{name} OR name:#{name}")
+
+      knife_search = Chef::Knife::Search.new
+      node = result.first.first
+      if node.nil?
+        ui.error "Could not find a node with the fqdn of #{name} or name of #{name}"
+        exit 1
+      end
+      return node
+    end
+
+
+    # helper methods to get attributes from a node
+    def get_all_attributes(node)
+      {"default" => get_default_attributes(node),
+       "normal" => get_normal_attributes(node),
+       "override" => get_override_attributes(node)}
+    end
+
+    def get_default_attributes(node)
+      node.default_attrs
+    end
+
+    def get_normal_attributes(node)
+      node.normal_attrs
+    end
+
+    def get_override_attributes(node)
+      node.override_attrs
+    end
+
+  end
+
+  class NodeAttributeShow < Chef::Knife
+    include KnifeNodeAttribute::Helpers
+
+    banner 'knife node attribute show HOSTNAME [NORMAL|DEFAULT|OVERRIDE] [colon:separated:key]'
+
+    def parse_args
+      unless node_name = name_args[0]
+        show_usage
+        ui.error "You need to specify a node"
+        exit 1
+      end
+      if name_args[1].nil?
+        attribute_class = :all
+      else
+        attribute_class = name_args[1].downcase.to_sym
+        # if we passed a valid attribute class, the next one could be an
+        # attribute name. If it's invalid, it might be the attribute name
+        if ATTRIBUTE_CLASSES.include? attribute_class
+          attribute_name = name_args[2] || ""
+        else
+          attribute_name = attribute_class
+          attribute_class = :all
+        end
+      end
+
+      {:nodename => node_name,
+       :class => attribute_class,
+       :name => attribute_name.split(":")}
+    end
+
+
+    def run
+      args = parse_args
+      method = "get_#{args[:class]}_attributes".to_sym
+
+      node = get_node(args[:nodename])
+      attribute_hash = self.send(method, node)
+      # this eval business is awful, I should probably come up with something
+      # smarter
+      getter =  args[:name].empty? ? "" : "['#{args[:name].join("']['")}']"
+      if args[:name].nil?
+        pp attribute_hash
+      else
+        begin
+          pp eval "attribute_hash#{getter}"
+        rescue NoMethodError
+          ui.error "Key not found"
+        end
+      end
+    end
+
+  end
+
+  class NodeAttributeDelete < Chef::Knife
+    include KnifeNodeAttribute::Helpers
+
+    banner 'knife node attribute delete HOSTNAME NORMAL|DEFAULT|OVERRIDE colon:separated:key'
+
+    def parse_args
+      unless node_name = name_args[0]
+        show_usage
+        ui.error "You need to specify a node"
+        exit 1
+      end
+      if name_args[1].nil?
+        show_usage
+        ui.error "No attribute level given. (e.g. override, normal, default)"
+        exit 1
+      else
+        attribute_class = name_args[1].downcase.to_sym
+        # if we passed a valid attribute class, the next one could be an
+        # attribute name. If it's invalid, it might be the attribute name
+        unless ATTRIBUTE_CLASSES.include? attribute_class
+          show_usage
+          ui.error "Wrong attribute level given. (e.g. override, normal, default)"
+          exit 1
+        end
+      end
+
+      unless attribute_name = name_args[2]
+        show_usage
+        ui.error "No attribute name given."
+        exit 1
+      end
+
+      {:nodename => node_name,
+       :class => attribute_class,
+       :name => attribute_name.split(":")}
+    end
+
+    def run
+      args = parse_args
+      method = "#{args[:class]}_attrs".to_sym
+
+      node = get_node(args[:nodename])
+      # this eval business is awful, I should probably come up with something
+      # smarter
+      if args[:name].length == 1
+        delete_method =  "node.#{method}.delete('#{args[:name].last}')"
+      else
+        parent_hash = args[:name][0, args[:name].length - 1]
+        delete_method =  "node.#{method}['#{parent_hash.join("']['")}'].delete('#{args[:name].last}')"
+      end
+      begin
+        eval delete_method
+        node.save
+        msg = parent_hash.nil? ? "" : " from parent structure #{parent_hash.join(':')}"
+        puts "Key #{args[:name].last}#{msg} deleted."
+      rescue StandardError => e
+        ui.error "An error occured while trying to delete the node key:"
+        puts e.to_s
+      end
+    end
+
+  end
+
+end
+

data/lib/knife-node-attribute.rb

@@ -0,0 +1,4 @@
+module KnifeNodeAttribute
+  VERSION = "0.2.0"
+end
+

metadata

@@ -0,0 +1,69 @@
+--- !ruby/object:Gem::Specification
+name: knife-node-attribute
+version: !ruby/object:Gem::Version
+  version: 0.2.0
+  prerelease: 
+platform: ruby
+authors:
+- Daniel Schauenberg
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-09-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: chef
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.10.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.10.4
+description: knife plugin to show or delete node attributes
+email: d@unwiredcouch.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README.md
+- Rakefile
+- knife-node-attribute.gemspec
+- lib/chef/knife/knife-node-attribute.rb
+- lib/knife-node-attribute.rb
+homepage: https://github.com/mrtazz/knife-node-attribute
+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.25
+signing_key: 
+specification_version: 3
+summary: knife plugin to show or delete node attributes
+test_files: []
+has_rdoc: