knife-flip 0.1.0 → 0.1.2

data/README.md

@@ -1,6 +1,6 @@
 # knife-flip
 
-A knife plugin to move a node from one environment to another
+A knife plugin to move a node, or all nodes in a role, to a specific environment
 
 ## What it does
 
@@ -10,3 +10,9 @@ knife node flip mynode.foo.com myenv
 
 will move the node mynode.foo.com into the environment myenv
 
+
+````
+knife role flip MyRole myenv
+````
+
+will move all nodes containing the role MyRole in their runlists into the environment myenv

data/knife-flip.gemspec

@@ -13,14 +13,14 @@ Gem::Specification.new do |s|
   ## If your rubyforge_project name is different, then edit it and comment out
   ## the sub! line in the Rakefile
   s.name              = 'knife-flip'
-  s.version           = '0.1.0'
-  s.date              = '2012-01-28'
+  s.version           = '0.1.2'
+  s.date              = '2012-02-01'
   s.rubyforge_project = 'knife-flip'
 
   ## Make sure your summary is short. The description may be as long
   ## as you like.
-  s.summary     = "A knife plugin to move a node from one environment to another"
-  s.description = "A knife plugin to move a node from one environment to another"
+  s.summary     = "A knife plugin to move a node, or all nodes in a role, to a specific environment"
+  s.description = "A knife plugin to move a node, or all nodes in a role, to a specific environment"
 
   ## List the primary authors. If there are a bunch of authors, it's probably
   ## better to set the email to an email list or something. If you don't have
@@ -50,7 +50,8 @@ Gem::Specification.new do |s|
     README.md
     Rakefile
     knife-flip.gemspec
-    lib/chef/knife/flip.rb
+    lib/chef/knife/nodeflip.rb
+    lib/chef/knife/roleflip.rb
     lib/knife-flip.rb
   ]
   # = MANIFEST =

data/lib/chef/knife/roleflip.rb

@@ -0,0 +1,100 @@
+#
+# Author:: Jon Cowie (<jonlives@gmail.com>)
+# Copyright:: Copyright (c) 2011 Jon Cowie
+# License:: GPL
+
+
+require 'chef/knife'
+
+module KnifeFlip
+  class RoleFlip < Chef::Knife
+
+    deps do
+      require 'chef/search/query'
+      require 'chef/knife/search'
+    end
+
+    banner "knife role flip ROLENAME ENVIRONMENT"
+
+    def run
+      unless @role = name_args[0]
+        ui.error "You need to specify a role to search against"
+        exit 1
+      end
+
+      unless @new_env = name_args[1]
+        ui.error "You need to specify an environment to move nodes to"
+        exit 1
+      end
+      
+      puts "Checking for a role called #{@role} to flip nodes from..."
+
+      searcher = Chef::Search::Query.new
+      result = searcher.search(:role, "name:#{@role}")
+      
+      knife_search = Chef::Knife::Search.new
+      env = result.first.first
+      if env.nil?
+        puts "Could not find an role named #{@role}. Can't update nodes which run a non-existant role!"
+        exit 1
+      else
+        puts "Found!\n"
+      end
+      
+        
+      puts "Checking for an environment called #{@new_env} to update to..."
+
+      searcher = Chef::Search::Query.new
+      result = searcher.search(:environment, "name:#{@new_env}")
+      
+      knife_search = Chef::Knife::Search.new
+      env = result.first.first
+      if env.nil?
+        puts "Could not find an environment named #{@new_env}. Please create it before trying to put nodes in it!"
+        exit 1
+      else
+        puts "Found!\n"
+      end
+
+      #puts "Setting environment to #{@environment}"
+      #node.chef_environment(@environment)
+      #node.save
+      
+      q_nodes = Chef::Search::Query.new
+      node_query = "role:#{@role}"
+      query_nodes = URI.escape(node_query,
+                         Regexp.new("[^#{URI::PATTERN::UNRESERVED}]"))
+
+      result_items = []
+      result_count = 0
+
+      #ui.msg("\nFinding all nodes in environment #{@old_env} and moving them to environment #{@new_env}...\n")
+      
+      begin
+        q_nodes.search('node', query_nodes) do |node_item|
+          
+          node_item.chef_environment(@new_env)
+          node_item.save
+          formatted_item_node = format_for_display(node_item)
+          if formatted_item_node.respond_to?(:has_key?) && !formatted_item_node.has_key?('id')
+            formatted_item_node['id'] = node_item.has_key?('id') ? node_item['id'] : node_item.name
+          end
+          ui.msg("Moving #{formatted_item_node.name} to environment #{@new_env}...")
+          result_items << formatted_item_node
+          result_count += 1
+        end
+      rescue Net::HTTPServerException => e
+        msg = Chef::JSONCompat.from_json(e.response.body)["error"].first
+        ui.error("knife role flip failed: #{msg}")
+        exit 1
+      end
+
+      if ui.interchange?
+        output({:results => result_count, :rows => result_items})
+      else
+        ui.msg "#{result_count} Nodes updated"
+      end
+
+    end
+  end
+end

data/lib/knife-flip.rb

@@ -1,3 +1,3 @@
 module KnifeFlip
-  VERSION = "0.1.0"
+  VERSION = "0.1.2"
 end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 1
-  - 0
-  version: 0.1.0
+  - 2
+  version: 0.1.2
 platform: ruby
 authors: 
 - Jon Cowie
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-01-28 00:00:00 +00:00
+date: 2012-02-01 00:00:00 +00:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -32,7 +32,7 @@ dependencies:
         version: 0.10.4
   type: :runtime
   version_requirements: *id001
-description: A knife plugin to move a node from one environment to another
+description: A knife plugin to move a node, or all nodes in a role, to a specific environment
 email: jonlives@gmail.com
 executables: []
 
@@ -44,7 +44,8 @@ files:
 - README.md
 - Rakefile
 - knife-flip.gemspec
-- lib/chef/knife/flip.rb
+- lib/chef/knife/nodeflip.rb
+- lib/chef/knife/roleflip.rb
 - lib/knife-flip.rb
 has_rdoc: true
 homepage: https://github.com/jonlives/knife-flip
@@ -77,6 +78,6 @@ rubyforge_project: knife-flip
 rubygems_version: 1.3.7
 signing_key: 
 specification_version: 2
-summary: A knife plugin to move a node from one environment to another
+summary: A knife plugin to move a node, or all nodes in a role, to a specific environment
 test_files: []