knife-bulkchangeenvironment 0.1.0

data/README.md

@@ -0,0 +1,16 @@
+# knife-bulkchangeenv
+
+A plugin for Chef::Knife which lets you move all nodes in one environment into another.
+
+## Preface
+
+Searches for all nodes in a particular environment and moves them to another
+
+## What it does
+
+knife node bulk_change_environment "_default" development
+will move all nodes in the default environment (in quotes because of the underscore) into the development environment.
+
+## Notes
+Please be careful when using this plugin, especially if you're already using environments with version constraints configured. You don't want all your nodes moved to the wrong one by mistake!
+

data/Rakefile

@@ -0,0 +1,122 @@
+require 'rubygems'
+require 'rake'
+require 'date'
+
+#############################################################################
+#
+# Helper functions
+#
+#############################################################################
+
+def name
+  @name ||= Dir['*.gemspec'].first.split('.').first
+end
+
+def version
+  line = File.read("lib/#{name}.rb")[/^\s*VERSION\s*=\s*.*/]
+  line.match(/.*VERSION\s*=\s*['"](.*)['"]/)[1]
+end
+
+def date
+  Date.today.to_s
+end
+
+def rubyforge_project
+  name
+end
+
+def gemspec_file
+  "#{name}.gemspec"
+end
+
+def gem_file
+  "#{name}-#{version}.gem"
+end
+
+def replace_header(head, header_name)
+  head.sub!(/(\.#{header_name}\s*= ').*'/) { "#{$1}#{send(header_name)}'"}
+end
+
+#############################################################################
+#
+# Standard tasks
+#
+#############################################################################
+
+task :default => :validate
+
+desc "Open an irb session preloaded with this library"
+task :console do
+  sh "irb -rubygems -r ./lib/#{name}.rb"
+end
+
+#############################################################################
+#
+# Custom tasks (add your own tasks here)
+#
+#############################################################################
+
+
+
+#############################################################################
+#
+# Packaging tasks
+#
+#############################################################################
+
+desc "Create tag v#{version} and build and push #{gem_file} to Rubygems"
+task :release => :build do
+  unless `git branch` =~ /^\* master$/
+    puts "You must be on the master branch to release!"
+    exit!
+  end
+  sh "git commit --allow-empty -a -m 'Release #{version}'"
+  sh "git tag v#{version}"
+  sh "git push origin master"
+  sh "git push origin v#{version}"
+  sh "gem push pkg/#{name}-#{version}.gem"
+end
+
+desc "Build #{gem_file} into the pkg directory"
+task :build => :gemspec do
+  sh "mkdir -p pkg"
+  sh "gem build #{gemspec_file}"
+  sh "mv #{gem_file} pkg"
+end
+
+desc "Generate #{gemspec_file}"
+task :gemspec => :validate do
+  # read spec file and split out manifest section
+  spec = File.read(gemspec_file)
+  head, manifest, tail = spec.split("  # = MANIFEST =\n")
+
+  # replace name version and date
+  replace_header(head, :name)
+  replace_header(head, :version)
+  replace_header(head, :date)
+  #comment this out if your rubyforge_project has a different name
+  replace_header(head, :rubyforge_project)
+
+  # determine file list from git ls-files
+  files = `git ls-files`.
+    split("\n").
+    sort.
+    reject { |file| file =~ /^\./ }.
+    reject { |file| file =~ /^(rdoc|pkg)/ }.
+    map { |file| "    #{file}" }.
+    join("\n")
+
+  # piece file back together and write
+  manifest = "  s.files = %w[\n#{files}\n  ]\n"
+  spec = [head, manifest, tail].join("  # = MANIFEST =\n")
+  File.open(gemspec_file, 'w') { |io| io.write(spec) }
+  puts "Updated #{gemspec_file}"
+end
+
+desc "Validate #{gemspec_file}"
+task :validate do
+  unless Dir['VERSION*'].empty?
+    puts "A `VERSION` file at root level violates Gem best practices."
+    exit!
+  end
+end

data/knife-bulkchangeenv.gemspec

@@ -0,0 +1,61 @@
+## This is the rakegem gemspec template. Make sure you read and understand
+## all of the comments. Some sections require modification, and others can
+## be deleted if you don't need them. Once you understand the contents of
+## this file, feel free to delete any comments that begin with two hash marks.
+## You can find comprehensive Gem::Specification documentation, at
+## http://docs.rubygems.org/read/chapter/20
+Gem::Specification.new do |s|
+  s.specification_version = 2 if s.respond_to? :specification_version=
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.rubygems_version = '1.3.5'
+
+  ## Leave these as is they will be modified for you by the rake gemspec task.
+  ## If your rubyforge_project name is different, then edit it and comment out
+  ## the sub! line in the Rakefile
+  s.name              = 'knife-bulkchangeenvironment'
+  s.version           = '0.1.0'
+  s.date              = '2012-01-28'
+  s.rubyforge_project = 'knife-bulkchangeenvironment'
+
+  ## Make sure your summary is short. The description may be as long
+  ## as you like.
+  s.summary     = "A knife plugin to move all nodes from one environment to another"
+  s.description = "A knife plugin to move all nodes from one environment to another"
+
+  ## 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
+  ## a custom homepage, consider using your GitHub URL or the like.
+  s.authors  = ["Jon Cowie"]
+  s.email    = 'jonlives@gmail.com'
+  s.homepage = 'https://github.com/jonlives/knife-bulkchangeenvironment'
+
+  ## This gets added to the $LOAD_PATH so that 'lib/NAME.rb' can be required as
+  ## require 'NAME.rb' or'/lib/NAME/file.rb' can be as require 'NAME/file.rb'
+  s.require_paths = %w[lib]
+
+  ## Specify any RDoc options here. You'll want to add your README and
+  ## LICENSE files to the extra_rdoc_files list.
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.extra_rdoc_files = %w[README.md]
+
+  ## List your runtime dependencies here. Runtime dependencies are those
+  ## that are needed for an end user to actually USE your code.
+  s.add_dependency('chef', [">= 0.10.4"])
+
+  ## Leave this section as-is. It will be automatically generated from the
+  ## contents of your Git repository via the gemspec task. DO NOT REMOVE
+  ## THE MANIFEST COMMENTS, they are used as delimiters by the task.
+  # = MANIFEST =
+  s.files = %w[
+    README.md
+    Rakefile
+    knife-bulkchangeenv.gemspec
+    lib/chef/knife/bulkchangeenv.rb
+    lib/knife-bulkchangeenv.rb
+  ]
+  # = MANIFEST =
+
+  ## Test files will be grabbed from the file list. Make sure the path glob
+  ## matches what you actually use.
+  s.test_files = s.files.select { |path| path =~ /^test\/test_.*\.rb/ }
+end

data/lib/chef/knife/bulkchangeenv.rb

@@ -0,0 +1,100 @@
+#
+# Author:: Jon Cowie (<jonlives@gmail.com>)
+# Copyright:: Copyright (c) 2011 Jon Cowie
+# License:: GPL
+
+
+require 'chef/knife'
+
+module KnifeBulkChangeEnv
+  class NodeBulkChangeEnvironment < Chef::Knife
+
+    deps do
+      require 'chef/search/query'
+      require 'chef/knife/search'
+    end
+
+    banner "knife node bulk_change_environment OLDENVIRONMENT NEWENVIRONMENT"
+
+    def run
+      unless @old_env = name_args[0]
+        ui.error "You need to specify a Environment to update from"
+        exit 1
+      end
+
+      unless @new_env = name_args[1]
+        ui.error "You need to specify an environment to update to"
+        exit 1
+      end
+      
+      puts "Checking for an environment called #{@old_env} to update from..."
+
+      searcher = Chef::Search::Query.new
+      result = searcher.search(:environment, "name:#{@old_env}")
+      
+      knife_search = Chef::Knife::Search.new
+      env = result.first.first
+      if env.nil?
+        puts "Could not find an environment named #{@old_env}. Can't update nodes in a non-existant environment!"
+        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 = "chef_environment:#{@old_env}"
+      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("Updated #{formatted_item_node.name}...")
+          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 preflight 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-bulkchangeenv.rb

@@ -0,0 +1,3 @@
+module KnifeBulkChangeEnv
+  VERSION = "0.1.0"
+end

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification 
+name: knife-bulkchangeenvironment
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Jon Cowie
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-01-28 00:00:00 +00:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: chef
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 10
+        - 4
+        version: 0.10.4
+  type: :runtime
+  version_requirements: *id001
+description: A knife plugin to move all nodes from one environment to another
+email: jonlives@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.md
+files: 
+- README.md
+- Rakefile
+- knife-bulkchangeenv.gemspec
+- lib/chef/knife/bulkchangeenv.rb
+- lib/knife-bulkchangeenv.rb
+has_rdoc: true
+homepage: https://github.com/jonlives/knife-bulkchangeenvironment
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: knife-bulkchangeenvironment
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 2
+summary: A knife plugin to move all nodes from one environment to another
+test_files: []
+