initial version of a mixlib-cli based wrapper for managing a chef-client windows service

1 files changed, 138 insertions, 0 deletions

diff --git a/chef/distro/windows/service.rb b/chef/distro/windows/service.rb
new file mode 100644
index 0000000000..f1b1401b6f
--- /dev/null
+++ b/chef/distro/windows/service.rb
@@ -0,0 +1,138 @@
+#
+# 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.
+#
+
+require 'win32/service'
+require 'rbconfig'
+require 'mixlib/cli'
+
+class Chef::Service::Windows
+  include Win32
+  include Config
+  include Mixlib::CLI
+
+  option :action
+    :short => "-a ACTION",
+    :long  => "--action ACTION",
+    :default => "start",
+    :description => ""
+
+  option :name,
+    :short => "-n NAME",
+    :long  => "--name NAME",
+    :default => "chef-client",
+    :description => "The service name to use."
+
+  option :display_name,
+    :long  => "--display_name NAME",
+    :default => "chef-client",
+    :description => "The display name to use for the service."
+
+  option :description,
+    :short => "-d DESCRIPTION",
+    :long  => "--description DESCRIPTION",
+    :default => "chef-client",
+    :description => "The description for the service."
+
+  option :config_file,
+    :short => "-c CONFIG",
+    :long  => "--config CONFIG",
+    :default => "#{ENV['SYSTEMDRIVE']}/chef/client.rb",
+    :description => "The configuration file to use"
+
+  option :log_location,
+    :short        => "-L LOGLOCATION",
+    :long         => "--logfile LOGLOCATION",
+    :description  => "Set the log file location",
+    :default => "#{ENV['SYSTEMDRIVE']}/chef/client.log",
+
+  option :help,
+    :short        => "-h",
+    :long         => "--help",
+    :description  => "Show this message",
+    :on           => :tail,
+    :boolean      => true,
+    :show_options => true,
+    :exit         => 0
+
+  def run
+    parse_options
+
+    case config[:action]
+    when 'install'
+
+      # Quote the full path to deal with possible spaces in the path name.
+      ruby = File.join(RbConfig::CONFIG['bindir'], 'ruby')
+      path = ' "' + File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'lib', 'chef', 'daemon', 'windows.rb')) + '"'
+      # ensure all forward slashes are backslashes
+      options = " -c #{config[:config_file]} -l #{config[:log_location]}"
+      cmd = (ruby + path + options).gsub(File::SEPARATOR, File::ALT_SEPARATOR)
+
+      Service.new(
+         :service_name     => config[:name],
+         :display_name     => config[:display_name],
+         :description      => config[:description],
+         :binary_path_name => cmd
+      )
+
+    when 'start'
+      take_action('start', RUNNING)
+    when 'stop'
+      take_action('stop', STOPPED)
+    when 'uninstall', 'delete'
+
+      if current_state != STOPPED
+         Service.stop(config[:name])
+      end
+      wait_for_state(STOPPED)
+      Service.delete(config[:name])
+      puts 'Service #{config[:name]} deleted'
+
+    when 'pause'
+      take_action('pause', PAUSED)
+    when 'resume'
+      take_action('resume', RUNNING)
+    end
+  end
+
+  private
+
+  def take_action(action=nil, desired_state=nil)
+    if current_state != desired_state
+       Service.send(action, config[:name])
+       wait_for_state(action)
+       puts "Service #{config[:name]} #{action}ed"
+    else
+       puts "Already #{desired_state}"
+    end
+  end
+
+  def current_state
+    Service.status(config[:name]).current_state
+  end
+
+  # Helper method that waits for a status to change its state since state
+  # changes aren't usually instantaneous.
+  def wait_for_state(desired_state)
+    while current_state != desired_state
+      puts "One moment...#{current_state}"
+      sleep 1
+    end
+  end
+end
+
+Chef::Service::Windows.run