1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207
|
#!/usr/bin/env ruby
#
# = Synopsis
#
# Run a +puppet+ script as a cfengine module.
#
# = Usage
#
# puppet_module [-h|--help] [-V|--version] [-d|--debug] [-v|--verbose]
# [-l|--logdest <file>]<file>
#
# = Description
#
# This is the standalone puppet execution script; use it to execute
# individual scripts that you write. If you need to execute site-wide
# scripts, use +puppetd+ and +puppetmasterd+.
#
# = Options
#
# Note that any configuration parameter that's valid in the configuration file
# is also a valid long argument. For example, 'ssldir' is a valid configuration
# parameter, so you can specify '--ssldir <directory>' as an argument.
#
# See the configuration file for the full list of acceptable parameters.
#
# debug::
# Enable full debugging.
#
# help::
# Print this help message
#
# logdest::
# Where to send messages. Choose between syslog, the console, and a log file.
# Defaults to sending messages to the console.
#
# verbose::
# Print extra information.
#
# = Author
#
# Luke Kanies
#
# = Copyright
#
# Copyright (c) 2005 Reductive Labs, LLC
# Licensed under the GNU Public License
require 'puppet'
require 'puppet/server'
require 'puppet/client'
require 'getoptlong'
$haveusage = true
begin
require 'rdoc/usage'
rescue LoadError
$haveusage = false
end
options = [
[ "--debug", "-d", GetoptLong::NO_ARGUMENT ],
[ "--help", "-h", GetoptLong::NO_ARGUMENT ],
[ "--logdest", "-l", GetoptLong::REQUIRED_ARGUMENT ],
[ "--verbose", "-v", GetoptLong::NO_ARGUMENT ],
[ "--use-nodes", GetoptLong::NO_ARGUMENT ],
[ "--version", "-V", GetoptLong::NO_ARGUMENT ]
]
# Add all of the config parameters as valid options.
Puppet.config.addargs(options)
result = GetoptLong.new(*options)
debug = false
verbose = false
noop = false
logfile = false
parseonly = false
master = {
:Local => true
}
setdest = false
begin
result.each { |opt,arg|
case opt
when "--version"
puts "%s" % Puppet.version
exit
when "--help"
if $haveusage
RDoc::usage && exit
else
puts "No help available unless you have RDoc::usage installed"
exit
end
when "--use-nodes"
master[:UseNodes] = true
when "--verbose"
Puppet::Log.level = :info
Puppet::Log.newdestination(:console)
verbose = true
when "--debug"
Puppet::Log.level = :debug
Puppet::Log.newdestination(:console)
debug = true
when "--logdest"
begin
Puppet::Log.newdestination arg
setdest=true
rescue => detail
$stderr.puts detail.to_s
end
else
Puppet.config.handlearg(opt, arg)
end
}
rescue GetoptLong::InvalidOption => detail
$stderr.puts "Try '#{$0} --help'"
#if $haveusage
# RDoc::usage(1,'usage')
#end
exit(1)
end
# Now parse the config
if Puppet[:config] and File.exists? Puppet[:config]
Puppet.config.parse(Puppet[:config])
end
client = nil
server = nil
[:INT, :TERM].each do |signal|
trap(signal) do
Puppet.notice "Caught #{signal}; shutting down"
[client, server].each { |obj|
if obj
obj.shutdown
end
}
end
end
Puppet.genconfig
Puppet.genmanifest
unless ARGV.length > 0
$stderr.puts "You must pass a script to parse"
exit(14)
end
unless setdest
Puppet::Log.newdestination(:syslog)
end
master[:Manifest] = ARGV.shift
unless ENV.include?("CFALLCLASSES")
$stderr.puts "Cfengine classes must be passed to the module"
exit(15)
end
master[:UseNodes] = false
classes = ENV["CFALLCLASSES"].split(":")
if classes.empty?
$stderr.puts "Could not find any cfengine classes"
exit(16)
end
master[:Classes] = classes
begin
server = Puppet::Server::Master.new(master)
rescue => detail
$stderr.puts detail
exit(1)
end
begin
client = Puppet::Client::MasterClient.new(
:Master => server,
:Cache => false
)
rescue => detail
$stderr.puts detail
exit(1)
end
if parseonly
exit(0)
end
begin
client.getconfig
client.apply
rescue => detail
Puppet.err detail
exit(1)
end
|