diff options
Diffstat (limited to 'distro/windows/service_manager.rb')
| -rw-r--r-- | distro/windows/service_manager.rb | 164 |
1 files changed, 164 insertions, 0 deletions
diff --git a/distro/windows/service_manager.rb b/distro/windows/service_manager.rb new file mode 100644 index 0000000000..8d71448d6e --- /dev/null +++ b/distro/windows/service_manager.rb @@ -0,0 +1,164 @@ +# +# 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 + class Windows + class ServiceManager + include Config + include Mixlib::CLI + + option :action, + :short => "-a ACTION", + :long => "--action ACTION", + :default => "start", + :description => "Action to carry out on the resource; one of 'install', 'uninstall', 'start', 'stop', 'pause', or 'resume'" + + 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 :splay, + :short => "-s SECONDS", + :long => "--splay SECONDS", + :description => "The splay time for running at intervals, in seconds", + :proc => lambda { |s| s.to_i } + + option :interval, + :short => "-i SECONDS", + :long => "--interval SECONDS", + :description => "Set the number of seconds to wait between chef-client runs", + :proc => lambda { |s| s.to_i } + + 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' + ruby = File.join(RbConfig::CONFIG['bindir'], 'ruby') + path = File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'lib', 'chef', 'application', 'windows_service.rb')) + + opts = "" + opts << " -c #{config[:config_file]}" if config[:config_file] + opts << " -L #{config[:log_location]}" if config[:log_location] + opts << " -i #{config[:interval]}" if config[:interval] + opts << " -s #{config[:splay]}" if config[:splay] + + # Quote the full paths to deal with possible spaces in the path name. + # Also ensure all forward slashes are backslashes + cmd = "\"#{ruby}\" \"#{path}\" #{opts}".gsub(File::SEPARATOR, File::ALT_SEPARATOR) + + Win32::Service.new( + :service_name => config[:name], + :display_name => config[:display_name], + :description => config[:description], + :binary_path_name => cmd) + puts "Service '#{config[:name]}' has successfully been 'installed'." + when 'start' + # TODO: allow override of startup parameters here? + take_action('start', RUNNING) if Win32::Service.exists?(config[:name]) + when 'stop' + take_action('stop', STOPPED) if Win32::Service.exists?(config[:name]) + when 'uninstall', 'delete' + take_action('stop', STOPPED) + if Win32::Service.exists?(config[:name]) + Win32::Service.delete(config[:name]) + puts "Service #{config[:name]} deleted" + end + when 'pause' + take_action('pause', PAUSED) + when 'resume' + take_action('resume', RUNNING) + end + end + + private + + # Just some state constants + STOPPED = "stopped" + RUNNING = "running" + PAUSED = "paused" + + def take_action(action=nil, desired_state=nil) + if Win32::Service.exists?(config[:name]) + if current_state != desired_state + Win32::Service.send(action, config[:name]) + wait_for_state(desired_state) + puts "Service '#{config[:name]}' is now '#{current_state}'." + else + puts "Service '#{config[:name]}' is already '#{desired_state}'." + end + else + puts "Cannot '#{action}' service '#{config[:name]}', service does not exist." + end + end + + def current_state + Win32::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 + end +end + +Chef::Windows::ServiceManager.new.run |