madscience 0.0.1

data/cookbooks/windows/providers/package.rb

@@ -0,0 +1,252 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Cookbook Name:: windows
+# Provider:: package
+#
+# 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.
+#
+
+if RUBY_PLATFORM =~ /mswin|mingw32|windows/
+  require 'win32/registry'
+end
+
+require 'chef/mixin/shell_out'
+require 'chef/mixin/language'
+
+include Chef::Mixin::ShellOut
+include Windows::Helper
+
+# the logic in all action methods mirror that of
+# the Chef::Provider::Package which will make
+# refactoring into core chef easy
+
+action :install do
+  # If we specified a version, and it's not the current version, move to the specified version
+  if @new_resource.version != nil && @new_resource.version != @current_resource.version
+    install_version = @new_resource.version
+  # If it's not installed at all, install it
+  elsif @current_resource.version == nil
+    install_version = candidate_version
+  end
+
+  if install_version
+    Chef::Log.info("Installing #{@new_resource} version #{install_version}")
+    status = install_package(@new_resource.package_name, install_version)
+    if status
+      @new_resource.updated_by_last_action(true)
+    end
+  end
+end
+
+action :upgrade do
+  if @current_resource.version != candidate_version
+    orig_version = @current_resource.version || "uninstalled"
+    Chef::Log.info("Upgrading #{@new_resource} version from #{orig_version} to #{candidate_version}")
+    status = upgrade_package(@new_resource.package_name, candidate_version)
+    if status
+      @new_resource.updated_by_last_action(true)
+    end
+  end
+end
+
+action :remove do
+  if removing_package?
+    Chef::Log.info("Removing #{@new_resource}")
+    remove_package(@current_resource.package_name, @new_resource.version)
+    @new_resource.updated_by_last_action(true)
+  else
+  end
+end
+
+def removing_package?
+  if @current_resource.version.nil?
+    false # nothing to remove
+  elsif @new_resource.version.nil?
+    true # remove any version of a package
+  elsif @new_resource.version == @current_resource.version
+    true # remove the version we have
+  else
+    false # we don't have the version we want to remove
+  end
+end
+
+def expand_options(options)
+  options ? " #{options}" : ""
+end
+
+# these methods are the required overrides of
+# a provider that extends from Chef::Provider::Package
+# so refactoring into core Chef should be easy
+
+def load_current_resource
+  @current_resource = Chef::Resource::WindowsPackage.new(@new_resource.name)
+  @current_resource.package_name(@new_resource.package_name)
+  @current_resource.version(nil)
+
+  unless current_installed_version.nil?
+    @current_resource.version(current_installed_version)
+  end
+
+  @current_resource
+end
+
+def current_installed_version
+  @current_installed_version ||= begin
+    if installed_packages.include?(@new_resource.package_name)
+      installed_packages[@new_resource.package_name][:version]
+    end
+  end
+end
+
+def candidate_version
+  @candidate_version ||= begin
+    @new_resource.version || 'latest'
+  end
+end
+
+def install_package(name,version)
+  Chef::Log.debug("Processing #{@new_resource} as a #{installer_type} installer.")
+  install_args = [cached_file(@new_resource.source, @new_resource.checksum), expand_options(unattended_installation_flags), expand_options(@new_resource.options)]
+  Chef::Log.info("Starting installation...this could take awhile.")
+  Chef::Log.debug "Install command: #{ sprintf(install_command_template, *install_args) }"
+  shell_out!(sprintf(install_command_template, *install_args), {:timeout => @new_resource.timeout, :returns => @new_resource.success_codes})
+end
+
+def remove_package(name, version)
+  uninstall_string = installed_packages[@new_resource.package_name][:uninstall_string]
+  Chef::Log.info("Registry provided uninstall string for #{@new_resource} is '#{uninstall_string}'")
+  uninstall_command = begin
+    if uninstall_string =~ /msiexec/i
+      "#{uninstall_string} /qn"
+    else
+      uninstall_string.gsub!('"','')
+      "start \"\" /wait /d\"#{::File.dirname(uninstall_string)}\" #{::File.basename(uninstall_string)}#{expand_options(@new_resource.options)} /S"
+    end
+  end
+  Chef::Log.info("Removing #{@new_resource} with uninstall command '#{uninstall_command}'")
+  shell_out!(uninstall_command, {:returns => @new_resource.success_codes})
+end
+
+private
+
+def install_command_template
+  case installer_type
+  when :msi
+    "msiexec%2$s \"%1$s\"%3$s"
+  else
+    "start \"\" /wait \"%1$s\"%2$s%3$s"
+  end
+end
+
+def uninstall_command_template
+  case installer_type
+  when :msi
+    "msiexec %2$s %1$s"
+  else
+    "start \"\" /wait /d%1$s %2$s %3$s"
+  end
+end
+
+# http://unattended.sourceforge.net/installers.php
+def unattended_installation_flags
+  case installer_type
+  when :msi
+    # this is no-ui
+    "/qn /i"
+  when :installshield
+    "/s /sms"
+  when :nsis
+    "/S /NCRC"
+  when :inno
+    #"/sp- /silent /norestart"
+    "/verysilent /norestart"
+  when :wise
+    "/s"
+  else
+  end
+end
+
+def installed_packages
+  @installed_packages || begin
+    installed_packages = {}
+    # Computer\HKEY_LOCAL_MACHINE\Software\Microsoft\Windows\CurrentVersion\Uninstall
+    installed_packages.merge!(extract_installed_packages_from_key(::Win32::Registry::HKEY_LOCAL_MACHINE)) #rescue nil
+    # 64-bit registry view
+    # Computer\HKEY_LOCAL_MACHINE\Software\Wow6464Node\Microsoft\Windows\CurrentVersion\Uninstall
+    installed_packages.merge!(extract_installed_packages_from_key(::Win32::Registry::HKEY_LOCAL_MACHINE, (::Win32::Registry::Constants::KEY_READ | 0x0100))) #rescue nil
+    # 32-bit registry view
+    # Computer\HKEY_LOCAL_MACHINE\Software\Wow6432Node\Microsoft\Windows\CurrentVersion\Uninstall
+    installed_packages.merge!(extract_installed_packages_from_key(::Win32::Registry::HKEY_LOCAL_MACHINE, (::Win32::Registry::Constants::KEY_READ | 0x0200))) #rescue nil
+    # Computer\HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Uninstall
+    installed_packages.merge!(extract_installed_packages_from_key(::Win32::Registry::HKEY_CURRENT_USER)) #rescue nil
+    installed_packages
+  end
+end
+
+def extract_installed_packages_from_key(hkey = ::Win32::Registry::HKEY_LOCAL_MACHINE, desired = ::Win32::Registry::Constants::KEY_READ)
+  uninstall_subkey = 'Software\Microsoft\Windows\CurrentVersion\Uninstall'
+  packages = {}
+  begin
+    ::Win32::Registry.open(hkey, uninstall_subkey, desired) do |reg|
+      reg.each_key do |key, wtime|
+        begin
+          k = reg.open(key, desired)
+          display_name = k["DisplayName"] rescue nil
+          version = k["DisplayVersion"] rescue "NO VERSION"
+          uninstall_string = k["UninstallString"] rescue nil
+          if display_name
+            packages[display_name] = {:name => display_name,
+                                      :version => version,
+                                      :uninstall_string => uninstall_string}
+          end
+        rescue ::Win32::Registry::Error
+        end
+      end
+    end
+  rescue ::Win32::Registry::Error
+  end
+  packages
+end
+
+def installer_type
+  @installer_type || begin
+    if @new_resource.installer_type
+      @new_resource.installer_type
+    else
+      basename = ::File.basename(cached_file(@new_resource.source, @new_resource.checksum))
+      if basename.split(".").last.downcase == "msi" # Microsoft MSI
+        :msi
+      else
+        # search the binary file for installer type
+        contents = ::Kernel.open(::File.expand_path(cached_file(@new_resource.source)), "rb") {|io| io.read } # TODO limit data read in
+        case contents
+        when /inno/i # Inno Setup
+          :inno
+        when /wise/i # Wise InstallMaster
+          :wise
+        when /nsis/i # Nullsoft Scriptable Install System
+          :nsis
+        else
+          # if file is named 'setup.exe' assume installshield
+          if basename == "setup.exe"
+            :installshield
+          else
+            raise Chef::Exceptions::AttributeNotFound, "installer_type could not be determined, please set manually"
+          end
+        end
+      end
+    end
+  end
+end

data/cookbooks/windows/providers/pagefile.rb

@@ -0,0 +1,153 @@
+#
+# Author:: Kevin Moser (<kevin.moser@nordstrom.com>)
+# Cookbook Name:: windows
+# Provider:: pagefile
+#
+# Copyright:: 2012, Nordstrom, 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.
+#
+
+include Chef::Mixin::ShellOut
+include Windows::Helper
+
+action :set do
+  pagefile = @new_resource.name
+  initial_size = @new_resource.initial_size
+  maximum_size = @new_resource.maximum_size
+  system_managed = @new_resource.system_managed
+  automatic_managed = @new_resource.automatic_managed
+  updated = false
+
+  if automatic_managed
+    unless automatic_managed?
+      set_automatic_managed
+      updated = true
+    end
+  else
+    if automatic_managed?
+      unset_automatic_managed
+      updated = true
+    end
+
+    # Check that the resource is not just trying to unset automatic managed, if it is do nothing more
+    if (initial_size && maximum_size) || system_managed
+      unless exists?(pagefile)
+        create(pagefile)
+      end
+
+      if system_managed
+        unless max_and_min_set?(pagefile, 0, 0)
+          set_system_managed(pagefile)
+          updated = true
+        end
+      else
+        unless max_and_min_set?(pagefile, initial_size, maximum_size)
+          set_custom_size(pagefile, initial_size, maximum_size)
+          updated = true
+        end
+      end
+    end
+  end
+
+  @new_resource.updated_by_last_action(updated)
+end
+
+action :delete do
+  pagefile = @new_resource.name
+  updated = false
+
+  if exists?(pagefile)
+    delete(pagefile)
+    updated = true
+  end
+
+  @new_resource.updated_by_last_action(updated)
+end
+
+
+private
+def exists?(pagefile)
+  @exists ||= begin
+    cmd = shell_out("#{wmic} pagefileset where SettingID=\"#{get_setting_id(pagefile)}\" list /format:list", {:returns => [0]})
+    cmd.stderr.empty? && (cmd.stdout =~ /SettingID=#{get_setting_id(pagefile)}/i)
+  end
+end
+
+def max_and_min_set?(pagefile, min, max)
+  @max_and_min_set ||= begin
+    cmd = shell_out("#{wmic} pagefileset where SettingID=\"#{get_setting_id(pagefile)}\" list /format:list", {:returns => [0]})
+    cmd.stderr.empty? && (cmd.stdout =~ /InitialSize=#{min}/i) && (cmd.stdout =~ /MaximumSize=#{max}/i)
+  end
+end
+
+def create(pagefile)
+  Chef::Log.debug("Creating pagefile #{pagefile}")
+  cmd = shell_out("#{wmic} pagefileset create name=\"#{win_friendly_path(pagefile)}\"")
+  check_for_errors(cmd.stderr)
+end
+
+def delete(pagefile)
+  Chef::Log.debug("Removing pagefile #{pagefile}")
+  cmd = shell_out("#{wmic} pagefileset where SettingID=\"#{get_setting_id(pagefile)}\" delete")
+  check_for_errors(cmd.stderr)
+end
+
+def automatic_managed?
+  @automatic_managed ||= begin
+    cmd = shell_out("#{wmic} computersystem where name=\"%computername%\" get AutomaticManagedPagefile /format:list")
+    cmd.stderr.empty? && (cmd.stdout =~ /AutomaticManagedPagefile=TRUE/i)
+  end
+end
+
+def set_automatic_managed
+  Chef::Log.debug("Setting pagefile to Automatic Managed")
+  cmd = shell_out("#{wmic} computersystem where name=\"%computername%\" set AutomaticManagedPagefile=True")
+  check_for_errors(cmd.stderr)
+end
+
+def unset_automatic_managed
+  Chef::Log.debug("Setting pagefile to User Managed")
+  cmd = shell_out("#{wmic} computersystem where name=\"%computername%\" set AutomaticManagedPagefile=False")
+  check_for_errors(cmd.stderr)
+end
+
+def set_custom_size(pagefile, min, max)
+  Chef::Log.debug("Setting #{pagefile} to InitialSize=#{min} & MaximumSize=#{max}")
+  cmd = shell_out("#{wmic} pagefileset where SettingID=\"#{get_setting_id(pagefile)}\" set InitialSize=#{min},MaximumSize=#{max}", {:returns => [0]})
+  check_for_errors(cmd.stderr)
+end
+
+def set_system_managed(pagefile)
+  Chef::Log.debug("Setting #{pagefile} to System Managed")
+  cmd = shell_out("#{wmic} pagefileset where SettingID=\"#{get_setting_id(pagefile)}\" set InitialSize=0,MaximumSize=0", {:returns => [0]})
+  check_for_errors(cmd.stderr)
+end
+
+def get_setting_id(pagefile)
+  pagefile = win_friendly_path(pagefile)
+  pagefile = pagefile.split("\\")
+  "#{pagefile[1]} @ #{pagefile[0]}"
+end
+
+def check_for_errors(stderr)
+  unless stderr.empty?
+    Chef::Log.fatal(stderr)
+  end
+end
+
+def wmic
+  @wmic ||= begin
+    locate_sysnative_cmd("wmic.exe")
+  end
+end

data/cookbooks/windows/providers/path.rb

@@ -0,0 +1,35 @@
+#
+# Author:: Paul Morotn (<pmorton@biaprotect.com>)
+# Cookbook Name:: windows
+# Provider:: path
+#
+# Copyright:: 2011, Business Intelligence Associates, 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.
+#
+
+action :add do
+  env "PATH" do
+    action :modify
+    delim ::File::PATH_SEPARATOR
+    value new_resource.path
+  end
+end
+
+action :remove do
+  env "PATH" do
+    action :delete
+    delim ::File::PATH_SEPARATOR
+    value new_resource.path
+  end
+end

data/cookbooks/windows/providers/printer.rb

@@ -0,0 +1,100 @@
+#
+# Author:: Doug Ireton (<doug.ireton@nordstrom.com>)
+# Cookbook Name:: windows
+# Provider:: printer
+#
+# Copyright:: 2012, Nordstrom, 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.
+#
+
+# Support whyrun
+def whyrun_supported?
+  true
+end
+
+action :create do
+  if @current_resource.exists
+    Chef::Log.info "#{ @new_resource } already exists - nothing to do."
+  else
+    converge_by("Create #{ @new_resource }") do
+      create_printer
+    end
+  end
+end
+
+action :delete do
+  if @current_resource.exists
+    converge_by("Delete #{ @new_resource }") do
+      delete_printer
+    end
+  else
+    Chef::Log.info "#{ @current_resource } doesn't exist - can't delete."
+  end
+end
+
+def load_current_resource
+  @current_resource = Chef::Resource::WindowsPrinter.new(@new_resource.name)
+  @current_resource.name(@new_resource.name)
+
+  if printer_exists?(@current_resource.name)
+    # TODO: Set @current_resource printer properties from registry
+    @current_resource.exists = true
+  end
+end
+
+
+private
+
+PRINTERS_REG_KEY = 'HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows NT\CurrentVersion\Print\Printers\\'.freeze unless defined?(PRINTERS_REG_KEY)
+
+def printer_exists?(name)
+  printer_reg_key = PRINTERS_REG_KEY + name
+  Chef::Log.debug "Checking to see if this reg key exists: '#{ printer_reg_key }'"
+  Registry.key_exists?(printer_reg_key)
+end
+
+def create_printer
+
+  # Create the printer port first
+  windows_printer_port new_resource.ipv4_address do
+  end
+
+  port_name = "IP_#{ new_resource.ipv4_address }"
+
+  powershell_script "Creating printer: #{ new_resource.name }" do
+    code <<-EOH
+
+      Set-WmiInstance -class Win32_Printer `
+        -EnableAllPrivileges `
+        -Argument @{ DeviceID   = "#{ new_resource.device_id }";
+                     Comment    = "#{ new_resource.comment }";
+                     Default    = "$#{ new_resource.default }";
+                     DriverName = "#{ new_resource.driver_name }";
+                     Location   = "#{ new_resource.location }";
+                     PortName   = "#{ port_name }";
+                     Shared     = "$#{ new_resource.shared }";
+                     ShareName  = "#{ new_resource.share_name }";
+                  }
+    EOH
+  end
+end
+
+def delete_printer
+  powershell_script "Deleting printer: #{ new_resource.name }" do
+    code <<-EOH
+      $printer = Get-WMIObject -class Win32_Printer -EnableAllPrivileges -Filter "name = '#{ new_resource.name }'"
+      $printer.Delete()
+    EOH
+  end
+end

data/cookbooks/windows/providers/printer_port.rb

@@ -0,0 +1,102 @@
+#
+# Author:: Doug Ireton (<doug.ireton@nordstrom.com>)
+# Cookbook Name:: windows
+# Provider:: printer_port
+#
+# Copyright:: 2012, Nordstrom, 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.
+#
+
+# Support whyrun
+def whyrun_supported?
+  true
+end
+
+action :create do
+  if @current_resource.exists
+    Chef::Log.info "#{ @new_resource } already exists - nothing to do."
+  else
+    converge_by("Create #{ @new_resource }") do
+      create_printer_port
+    end
+  end
+end
+
+action :delete do
+  if @current_resource.exists
+    converge_by("Delete #{ @new_resource }") do
+      delete_printer_port
+    end
+  else
+    Chef::Log.info "#{ @current_resource } doesn't exist - can't delete."
+  end
+end
+
+def load_current_resource
+  @current_resource = Chef::Resource::WindowsPrinterPort.new(@new_resource.name)
+  @current_resource.name(@new_resource.name)
+  @current_resource.ipv4_address(@new_resource.ipv4_address)
+  @current_resource.port_name(@new_resource.port_name || "IP_#{ @new_resource.ipv4_address }")
+
+  if port_exists?(@current_resource.port_name)
+    # TODO: Set @current_resource port properties from registry
+    @current_resource.exists = true
+  end
+end
+
+
+private
+
+PORTS_REG_KEY = 'HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Control\Print\Monitors\Standard TCP/IP Port\Ports\\'.freeze unless defined?(PORTS_REG_KEY)
+
+def port_exists?(name)
+  port_reg_key = PORTS_REG_KEY + name
+
+  Chef::Log.debug "Checking to see if this reg key exists: '#{ port_reg_key }'"
+  Registry.key_exists?(port_reg_key)
+end
+
+
+def create_printer_port
+
+  port_name = new_resource.port_name || "IP_#{ new_resource.ipv4_address }"
+
+  # create the printer port using PowerShell
+  powershell_script "Creating printer port #{ new_resource.port_name }" do
+    code <<-EOH
+
+      Set-WmiInstance -class Win32_TCPIPPrinterPort `
+        -EnableAllPrivileges `
+        -Argument @{ HostAddress = "#{ new_resource.ipv4_address }";
+                     Name        = "#{ port_name }";
+                     Description = "#{ new_resource.port_description }";
+                     PortNumber  = "#{ new_resource.port_number }";
+                     Protocol    = "#{ new_resource.port_protocol }";
+                     SNMPEnabled = "$#{ new_resource.snmp_enabled }";
+                  }
+    EOH
+  end
+end
+
+def delete_printer_port
+
+  port_name = new_resource.port_name || "IP_#{ new_resource.ipv4_address }"
+
+  powershell_script "Deleting printer port: #{ new_resource.port_name }" do
+    code <<-EOH
+      $port = Get-WMIObject -class Win32_TCPIPPrinterPort -EnableAllPrivileges -Filter "name = '#{ port_name }'"
+      $port.Delete()
+    EOH
+  end
+end

data/cookbooks/windows/providers/reboot.rb

@@ -0,0 +1,31 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Cookbook Name:: windows
+# Provider:: reboot
+#
+# 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.
+#
+
+action :request do
+  node.run_state[:reboot_requested] = true
+  node.run_state[:reboot_timeout] = @new_resource.timeout
+  node.run_state[:reboot_reason] = @new_resource.reason
+end
+
+action :cancel do
+  node.run_state.delete(:reboot_requested)
+  node.run_state.delete(:reboot_timeout)
+  node.run_state.delete(:reboot_reason)
+end