Adding puppetqd executable.

This uses the backported Application class, with a couple
of backported monkey-patches so the class works as expected
but in 0.24.

Signed-off-by: Luke Kanies <luke@madstop.com>

2 files changed, 160 insertions, 0 deletions

diff --git a/bin/puppetqd b/bin/puppetqd
new file mode 100644
index 000000000..14a15b519
--- /dev/null
+++ b/bin/puppetqd
@@ -0,0 +1,53 @@
+#!/usr/bin/env ruby
+# == Synopsis 
+#
+# Retrieve serialized records from a queue and process them in order.
+#
+# = Usage
+#
+#   puppetqd  [-d|--debug] [-v|--verbose]
+#
+# = Description
+#
+# This is a simple application that just processes entities in a queue as they
+# are recieved.
+#
+# = Options
+#
+# Note that any configuration parameter that's valid in the configuration file
+# is also a valid long argument.  For example, 'server' is a valid configuration
+# parameter, so you can specify '--server <servername>' as an argument.
+#
+# See the configuration file documentation at
+# http://reductivelabs.com/trac/puppet/wiki/ConfigurationReference for
+# the full list of acceptable parameters. A commented list of all
+# configuration options can also be generated by running puppetd with
+# '--genconfig'.
+#
+# debug::
+#   Enable full debugging.
+#
+# help::
+#   Print this help message
+#
+# verbose::
+#   Turn on verbose reporting.
+#
+# version::
+#   Print the puppet version number and exit.
+#
+# = Example
+#
+#   puppetqd
+#
+# = Author
+#
+# Luke Kanies
+#
+# = Copyright
+#
+# Copyright (c) 2009 Reductive Labs, LLC
+# Licensed under the GNU Public License
+
+require 'puppet/application/puppetqd'
+Puppet::Application[:puppetqd].run
diff --git a/lib/puppet/application/puppetqd.rb b/lib/puppet/application/puppetqd.rb
new file mode 100644
index 000000000..995668bec
--- /dev/null
+++ b/lib/puppet/application/puppetqd.rb
@@ -0,0 +1,107 @@
+require 'puppet'
+require 'puppet/application'
+require 'puppet/node/catalog'
+require 'puppet/indirector/catalog/queue'
+
+
+# BACKPORT - this method should be removed when merged into master.
+class Puppet::Util::Settings
+    # Generate the list of valid arguments, in a format that OptionParser can
+    # understand, and add them to the passed option list.
+    def optparse_addargs(options)
+        # Add all of the config parameters as valid options.
+        self.each { |name, element|
+            options << element.optparse_args
+        }
+
+        return options
+    end
+end
+
+# BACKPORT - this method should be removed when merged into master.
+class Puppet::Util::Settings::CElement
+    # get the arguments in OptionParser format
+    def optparse_args
+        if short
+            ["--#{name}", "-#{short}", desc, :REQUIRED]
+        else
+            ["--#{name}", desc, :REQUIRED]
+        end
+    end
+end
+
+Puppet::Application.new(:puppetqd) do
+
+    should_parse_config
+
+    preinit do
+        # Do an initial trap, so that cancels don't get a stack trace.
+        trap(:INT) do
+            $stderr.puts "Cancelling startup"
+            exit(0)
+        end
+
+        {
+            :verbose => false,
+            :debug => false
+        }.each do |opt,val|
+            options[opt] = val
+        end
+
+        @args = {}
+    end
+
+    option("--debug","-d")
+    option("--verbose","-v")
+
+    command(:main) do
+        Puppet::Node::Catalog::Queue.subscribe do |catalog|
+            # Once you have a Puppet::Node::Catalog instance, calling save() on it should suffice
+            # to put it through to the database via its active_record indirector (which is determined
+            # by the terminus_class = :active_record setting above)
+            catalog.save
+        end
+    end
+
+    # This is the main application entry point.
+    # BACKPORT - this method should be removed when merged into master.
+    # This method had to be added because Puppet.settings.parse takes no
+    # arguments in master but requires an argument in 0.24.x.
+    def run
+        run_preinit
+        parse_options
+        Puppet.settings.parse(Puppet[:config]) if should_parse_config?
+        run_setup
+        run_command
+    end
+
+    # Handle the logging settings.
+    def setup_logs
+        if options[:debug] or options[:verbose]
+            Puppet::Util::Log.newdestination(:console)
+            if options[:debug]
+                Puppet::Util::Log.level = :debug
+            else
+                Puppet::Util::Log.level = :info
+            end
+        end
+
+        unless options[:setdest]
+            Puppet::Util::Log.newdestination(:syslog)
+        end
+    end
+
+    setup do
+        unless Puppet.features.stomp?
+            raise ArgumentError, "Could not load 'stomp', which must be present for queueing to work"
+        end
+
+        setup_logs
+
+        if Puppet.settings.print_configs?
+            exit(Puppet.settings.print_configs ? 0 : 1)
+        end
+
+        Puppet::Node::Catalog.terminus_class = :active_record
+    end
+end