madscience 0.0.1

data/cookbooks/windows/files/default/handlers/windows_reboot_handler.rb

@@ -0,0 +1,76 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Copyright:: Copyright (c) 2011 Opscode, Inc
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+class WindowsRebootHandler < Chef::Handler
+  include Chef::Mixin::ShellOut
+
+  def initialize(allow_pending_reboots = true, timeout = 60, reason = "Opscode Chef initiated reboot")
+    @allow_pending_reboots = allow_pending_reboots
+    @timeout = timeout
+    @reason = reason
+  end
+
+  def report
+    log_message, reboot = begin
+      if reboot_requested?
+        ["chef_handler[#{self.class}] requested reboot will occur in #{timeout} seconds", true]
+      elsif reboot_pending?
+        if @allow_pending_reboots
+          ["chef_handler[#{self.class}] reboot pending - automatic reboot will occur in #{timeout} seconds", true]
+        else
+          ["chef_handler[#{self.class}] reboot pending but handler not configured to act on pending reboots - please reboot node manually", false]
+        end
+      else
+        ["chef_handler[#{self.class}] no reboot requested or pending", false]
+      end
+    end
+
+    Chef::Log.warn(log_message)
+    shell_out!("shutdown /r /t #{timeout} /c \"#{reason}\"") if reboot
+  end
+
+  private
+  # reboot cause CHEF says so:
+  # reboot explicitly requested in our cookbook code
+  def reboot_requested?
+    node.run_state[:reboot_requested] == true
+  end
+
+  # reboot cause WIN says so:
+  # reboot pending because of some configuration action we performed
+  def reboot_pending?
+    # Any files listed here means reboot needed
+    (Registry.key_exists?('HKLM\SYSTEM\CurrentControlSet\Control\Session Manager\PendingFileRenameOperations') &&
+      Registry.get_value('HKLM\SYSTEM\CurrentControlSet\Control\Session Manager','PendingFileRenameOperations').any?) ||
+    # 1 for any value means reboot pending
+    # "9306cdfc-c4a1-4a22-9996-848cb67eddc3"=1
+    (Registry.key_exists?('HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\WindowsUpdate\Auto Update\RebootRequired') &&
+      Registry.get_values('HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\WindowsUpdate\Auto Update\RebootRequired').select{|v| v[2] == 1 }.any?) ||
+    # 1 or 2 for 'Flags' value means reboot pending
+    (Registry.key_exists?('HKLM\SOFTWARE\Microsoft\Updates\UpdateExeVolatile') &&
+      [1,2].include?(Registry::get_value('HKLM\SOFTWARE\Microsoft\Updates\UpdateExeVolatile','Flags')))
+  end
+
+  def timeout
+    node.run_state[:reboot_timeout] || node['windows']['reboot_timeout'] || @timeout
+  end
+
+  def reason
+    node.run_state[:reboot_reason] || @reason
+  end
+end

data/cookbooks/windows/libraries/feature_base.rb

@@ -0,0 +1,59 @@
+class Chef
+  class Provider
+    class WindowsFeature
+      module Base
+
+        def action_install
+          unless installed?
+            install_feature(@new_resource.feature_name)
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} installed feature")
+          else
+            Chef::Log.debug("#{@new_resource} is already installed - nothing to do")
+          end
+        end
+
+        def action_remove
+          if installed?
+            remove_feature(@new_resource.feature_name)
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} removed")
+          else
+            Chef::Log.debug("#{@new_resource} feature does not exist - nothing to do")
+          end
+        end
+
+        def action_delete
+          if available?
+            delete_feature(@new_resource.feature_name)
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} deleted")
+          else
+            Chef::Log.debug("#{@new_resource} feature is not installed - nothing to do")
+          end
+        end
+
+        def install_feature(name)
+          raise Chef::Exceptions::UnsupportedAction, "#{self.to_s} does not support :install"
+        end
+
+        def remove_feature(name)
+          raise Chef::Exceptions::UnsupportedAction, "#{self.to_s} does not support :remove"
+        end
+
+        def delete_feature(name)
+          raise Chef::Exceptions::UnsupportedAction, "#{self.to_s} does not support :delete"
+        end
+
+        def installed?
+          raise Chef::Exceptions::Override, "You must override installed? in #{self.to_s}"
+        end
+
+        def available?
+          raise Chef::Exceptions::Override, "You must override available? in #{self.to_s}"
+        end
+      end
+    end
+  end
+end
+      

data/cookbooks/windows/libraries/matchers.rb

@@ -0,0 +1,450 @@
+if defined?(ChefSpec)
+
+  #
+  # Assert that a +windows_package+ resource exists in the Chef run with the
+  # action +:install+. Given a Chef Recipe that installs "Node.js" as a
+  # +windows_package+:
+  #
+  #     windows_package 'Node.js' do
+  #       source 'http://nodejs.org/dist/v0.10.26/x64/node-v0.10.26-x64.msi'
+  #       action :install
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_package+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_package+ was installed
+  #   expect(chef_run).to install_windows_package('Node.js')
+  #
+  # @example Assert that a +windows_package+ was _not_ installed
+  #   expect(chef_run).to_not install_windows_package('7-zip')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def install_windows_package(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_package, :install, resource_name)
+  end
+
+  #
+  # Assert that a +windows_package+ resource exists in the Chef run with the
+  # action +:remove+. Given a Chef Recipe that removes "Node.js" as a
+  # +windows_package+:
+  #
+  #     windows_package 'Node.js' do
+  #       action :remove
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_package+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_package+ was installed
+  #   expect(chef_run).to remove_windows_package('Node.js')
+  #
+  # @example Assert that a +windows_package+ was _not_ removed
+  #   expect(chef_run).to_not remove_windows_package('7-zip')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def remove_windows_package(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_package, :remove, resource_name)
+  end
+
+
+
+  #
+  # Assert that a +windows_feature+ resource exists in the Chef run with the
+  # action +:install+. Given a Chef Recipe that installs "NetFX3" as a
+  # +windows_feature+:
+  #
+  #     windows_feature 'NetFX3' do
+  #       action :install
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_feature+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_feature+ was installed
+  #   expect(chef_run).to install_windows_feature('NetFX3')
+  #
+  # @example Assert that a +windows_feature+ was _not_ installed
+  #   expect(chef_run).to_not install_windows_feature('NetFX3')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def install_windows_feature(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_feature, :install, resource_name)
+  end
+
+  #
+  # Assert that a +windows_feature+ resource exists in the Chef run with the
+  # action +:remove+. Given a Chef Recipe that removes "NetFX3" as a
+  # +windows_feature+:
+  #
+  #     windows_feature 'NetFX3' do
+  #       action :remove
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_feature+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_feature+ was removed
+  #   expect(chef_run).to remove_windows_feature('NetFX3')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def remove_windows_feature(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_feature, :remove, resource_name)
+  end
+
+  #
+  # Assert that a +windows_feature+ resource exists in the Chef run with the
+  # action +:delete+. Given a Chef Recipe that deletes "NetFX3" as a
+  # +windows_feature+:
+  #
+  #     windows_feature 'NetFX3' do
+  #       action :delete
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_feature+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_feature+ was deleted
+  #   expect(chef_run).to delete_windows_feature('NetFX3')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def delete_windows_feature(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_feature, :delete, resource_name)
+  end
+
+
+
+  #
+  # Assert that a +windows_task+ resource exists in the Chef run with the
+  # action +:create+. Given a Chef Recipe that creates "mytask" as a
+  # +windows_task+:
+  #
+  #     windows_task 'mytask' do
+  #       command 'mybatch.bat'
+  #       action :create
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_task+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_task+ was created
+  #   expect(chef_run).to create_windows_task('mytask')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def create_windows_task(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_task, :create, resource_name)
+  end
+
+  #
+  # Assert that a +windows_task+ resource exists in the Chef run with the
+  # action +:delete+. Given a Chef Recipe that deletes "mytask" as a
+  # +windows_task+:
+  #
+  #     windows_task 'mytask' do
+  #       action :delete
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_task+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_task+ was deleted
+  #   expect(chef_run).to delete_windows_task('mytask')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def delete_windows_task(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_task, :delete, resource_name)
+  end
+
+  #
+  # Assert that a +windows_task+ resource exists in the Chef run with the
+  # action +:run+. Given a Chef Recipe that runs "mytask" as a
+  # +windows_task+:
+  #
+  #     windows_task 'mytask' do
+  #       action :run
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_task+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_task+ was run
+  #   expect(chef_run).to run_windows_task('mytask')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def run_windows_task(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_task, :run, resource_name)
+  end
+
+  #
+  # Assert that a +windows_task+ resource exists in the Chef run with the
+  # action +:change+. Given a Chef Recipe that changes "mytask" as a
+  # +windows_task+:
+  #
+  #     windows_task 'mytask' do
+  #       action :change
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_task+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_task+ was changed
+  #   expect(chef_run).to change_windows_task('mytask')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def change_windows_task(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_task, :change, resource_name)
+  end
+
+
+
+  #
+  # Assert that a +windows_path+ resource exists in the Chef run with the
+  # action +:add+. Given a Chef Recipe that adds "C:\7-Zip" to the Windows
+  # PATH env var
+  #
+  #     windows_path 'C:\7-Zip' do
+  #       action :add
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_path+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_path+ was added
+  #   expect(chef_run).to add_windows_path('C:\7-Zip')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def add_windows_path(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_path, :add, resource_name)
+  end
+
+  #
+  # Assert that a +windows_path+ resource exists in the Chef run with the
+  # action +:remove+. Given a Chef Recipe that removes "C:\7-Zip" from the
+  # Windows PATH env var
+  #
+  #     windows_path 'C:\7-Zip' do
+  #       action :remove
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_path+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_path+ was removed
+  #   expect(chef_run).to remove_windows_path('C:\7-Zip')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def remove_windows_path(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_path, :remove, resource_name)
+  end
+
+
+
+  #
+  # Assert that a +windows_batch+ resource exists in the Chef run with the
+  # action +:run+. Given a Chef Recipe that runs a batch script
+  #
+  #     windows_batch "unzip_and_move_ruby" do
+  #       code <<-EOH
+  #       7z.exe x #{Chef::Config[:file_cache_path]}/ruby-1.8.7-p352-i386-mingw32.7z
+  #          -oC:\\source -r -y
+  #       xcopy C:\\source\\ruby-1.8.7-p352-i386-mingw32 C:\\ruby /e /y
+  #       EOH
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_path+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_path+ was removed
+  #   expect(chef_run).to run_windows_batch('unzip_and_move_ruby')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def run_windows_batch(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_batch, :run, resource_name)
+  end
+
+
+
+  #
+  # Assert that a +windows_pagefile+ resource exists in the Chef run with the
+  # action +:set+. Given a Chef Recipe that sets a pagefile
+  #
+  #     windows_pagefile "pagefile" do
+  #       system_managed true
+  #       initial_size 1024
+  #       maximum_size 4096
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_pagefile+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_pagefile+ was set
+  #   expect(chef_run).to set_windows_pagefile('pagefile').with(
+  #     initial_size: 1024)
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def set_windows_pagefile(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_pagefile, :set, resource_name)
+  end
+
+
+
+  #
+  # Assert that a +windows_zipfile+ resource exists in the Chef run with the
+  # action +:unzip+. Given a Chef Recipe that extracts "SysinternalsSuite.zip"
+  # to c:/bin
+  #
+  #     windows_zipfile "c:/bin" do
+  #       source "http://download.sysinternals.com/Files/SysinternalsSuite.zip"
+  #       action :unzip
+  #       not_if {::File.exists?("c:/bin/PsExec.exe")}
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_zipfile+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_zipfile+ was unzipped
+  #   expect(chef_run).to unzip_windows_zipfile_to('c:/bin')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def unzip_windows_zipfile_to(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_zipfile, :unzip, resource_name)
+  end
+
+  #
+  # Assert that a +windows_zipfile+ resource exists in the Chef run with the
+  # action +:zip+. Given a Chef Recipe that zips "c:/src"
+  # to c:/code.zip
+  #
+  #     windows_zipfile "c:/code.zip" do
+  #       source "c:/src"
+  #       action :zip
+  #     end
+  #
+  # The Examples section demonstrates the different ways to test a
+  # +windows_zipfile+ resource with ChefSpec.
+  #
+  # @example Assert that a +windows_zipfile+ was zipped
+  #   expect(chef_run).to zip_windows_zipfile_to('c:/code.zip')
+  #
+  #
+  # @param [String, Regex] resource_name
+  #   the name of the resource to match
+  #
+  # @return [ChefSpec::Matchers::ResourceMatcher]
+  #
+  def zip_windows_zipfile_to(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_zipfile, :zip, resource_name)
+  end
+
+
+  # All the other less commonly used LWRPs
+  def create_windows_shortcut(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_shortcut, :create, resource_name)
+  end
+
+  def create_windows_auto_run(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_auto_run, :create, resource_name)
+  end
+
+  def remove_windows_auto_run(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_auto_run, :remove, resource_name)
+  end
+
+  def create_windows_printer(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_printer, :create, resource_name)
+  end
+
+  def delete_windows_printer(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_printer, :delete, resource_name)
+  end
+
+  def create_windows_printer_port(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_printer_port, :create, resource_name)
+  end
+
+  def delete_windows_printer_port(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_printer_port, :delete, resource_name)
+  end
+
+  def request_windows_reboot(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_reboot, :request, resource_name)
+  end
+
+  def cancel_windows_reboot(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_reboot, :cancel, resource_name)
+  end
+
+  def create_windows_shortcut(resource_name)
+    ChefSpec::Matchers::ResourceMatcher.new(:windows_shortcut, :create, resource_name)
+  end
+end

data/cookbooks/windows/libraries/powershell_helper.rb

@@ -0,0 +1,59 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Cookbook Name:: windows
+# Library:: helper
+#
+# Copyright:: 2011, Opscode, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/mixin/shell_out'
+
+module Powershell
+  module Helper
+    include Chef::Mixin::ShellOut
+
+    def powershell_installed?
+      !powershell_version.nil?
+    end
+
+    def interpreter
+      # force 64-bit powershell from 32-bit ruby process
+      if ::File.exist?("#{ENV['WINDIR']}\\sysnative\\WindowsPowershell\\v1.0\\powershell.exe")
+        "#{ENV['WINDIR']}\\sysnative\\WindowsPowershell\\v1.0\\powershell.exe"
+      elsif ::File.exist?("#{ENV['WINDIR']}\\system32\\WindowsPowershell\\v1.0\\powershell.exe")
+        "#{ENV['WINDIR']}\\system32\\WindowsPowershell\\v1.0\\powershell.exe"
+      else
+        "powershell.exe"
+      end
+    end
+
+    def powershell_version
+      begin
+        cmd = shell_out("#{interpreter} -InputFormat none -Command \"& echo $PSVersionTable.psversion.major\"")
+        if cmd.stdout.empty? # PowerShell 1.0 doesn't have a $PSVersionTable
+          1
+        else
+          if cmd.stdout =~ /^(\d+)/
+            $1.to_i
+          else
+            nil
+          end
+        end
+      rescue Errno::ENOENT
+        nil
+      end
+    end
+  end
+end