Switched puppetmasterd to use the new-style server plumbing.

The code is much cleaner, and it seems to be mostly
functional, but we have to pick a strategy for signing
the host's certificate on first startup.  Also, I haven't
actually done end-to-end testing yet, which needs the certs
working first.

1 files changed, 20 insertions, 102 deletions

diff --git a/bin/puppetmasterd b/bin/puppetmasterd
index b4733e604..11b0d7fd0 100755
--- a/bin/puppetmasterd
+++ b/bin/puppetmasterd
@@ -8,8 +8,7 @@
 # = Usage
 #
 #   puppetmasterd [-D|--daemonize|--no-daemonize] [-d|--debug] [-h|--help]
-#       [-l|--logdest <file>|console|syslog] [--nobucket] [--nonodes]
-#       [-v|--verbose] [-V|--version]
+#       [-l|--logdest <file>|console|syslog] [-v|--verbose] [-V|--version]
 #
 # = Description
 #
@@ -22,7 +21,7 @@
 # parameter, so you can specify '--ssldir <directory>' as an argument.
 #
 # See the configuration file documentation at
-# http://reductivelabs.com/projects/puppet/reference/configref.html for
+# 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 puppetmasterdd with
 # '--genconfig'.
@@ -44,16 +43,6 @@
 #   Defaults to sending messages to syslog, or the console
 #   if debugging or verbosity is enabled.
 #
-# nobucket::
-#   Do not function as a file bucket.
-#
-# nonodes::
-#   Do not use individual node designations; each node will receive the result
-#   of evaluating the entire configuration.
-#
-# noreports::
-#   Do not start the reports server.
-#
 # verbose::
 #   Enable verbosity.
 #
@@ -81,16 +70,12 @@ end
 
 require 'getoptlong'
 require 'puppet'
-require 'puppet/network/handler'
-require 'puppet/sslcertificates'
+require 'puppet/network/server'
 
 options = [
 	[ "--debug",	"-d",			GetoptLong::NO_ARGUMENT ],
 	[ "--help",		"-h",			GetoptLong::NO_ARGUMENT ],
 	[ "--logdest",	"-l",			GetoptLong::REQUIRED_ARGUMENT ],
-	[ "--nobucket",        			GetoptLong::NO_ARGUMENT ],
-	[ "--noreports",       			GetoptLong::NO_ARGUMENT ],
-	[ "--nonodes",         			GetoptLong::NO_ARGUMENT ],
 	[ "--verbose",	"-v",			GetoptLong::NO_ARGUMENT ],
     [ "--version",  "-V",           GetoptLong::NO_ARGUMENT ]
 ]
@@ -100,15 +85,7 @@ Puppet.settings.addargs(options)
 
 result = GetoptLong.new(*options)
 
-master = {}
-ca = {}
-report = {}
-bucket = {}
-
 options = {
-    :havereport => true,
-    :havebucket => true,
-    :havemaster => true,
     :setdest => false,
     :verbose => false,
     :debug => false
@@ -128,14 +105,6 @@ begin
                     puts "No help available unless you have RDoc::usage installed"
                     exit
                 end
-            when "--noreports"
-                options[:havereport] = false
-            when "--nomaster"
-                options[:havemaster] = false
-            when "--nobucket"
-                options[:havebucket] = false
-            when "--nonodes"
-                master[:UseNodes] = false
             when "--logdest"
                 begin
                     Puppet::Util::Log.newdestination(arg)
@@ -193,85 +162,39 @@ Puppet::Node.cache_class = :yaml
 
 require 'etc'
 
-handlers = {
-    :Status => {},
-    :FileServer => {}
-}
-
-if options[:havemaster]
-    handlers[:Master] = master
-end
-
-if options[:havereport]
-    handlers[:Report] = report
-end
-
-if Puppet[:ca]
-    handlers[:CA] = ca
-end
-
-if options[:havebucket]
-    handlers[:FileBucket] = bucket
-end
-
 if Puppet[:parseonly]
     begin
-        Puppet::Network::Handler.master.new(master)
+        Puppet::Parser::Interpreter.new.parser(Puppet[:environment])
     rescue => detail
-        if Puppet[:trace]
-            puts detail.backtrace
-        end
-        $stderr.puts detail
-        exit(32)
+        Puppet.err detail
+        exit 1
     end
-    # we would have already exited if the file weren't syntactically correct
     exit(0)
 end
 
-webserver = server = nil
-begin
-    case Puppet[:servertype]
-    when "webrick"
-        # use the default, um, everything
-        require 'puppet/network/http_server/webrick'
-        webserver = server = Puppet::Network::HTTPServer::WEBrick.new(:Handlers => handlers)
-    when "mongrel":
-        require 'puppet/network/http_server/mongrel'
-        server = Puppet::Network::HTTPServer::Mongrel.new(handlers)
-        addr = Puppet[:bindaddress]
-        if addr == ""
-            addr =  "127.0.0.1"
-        end
-        webserver = Mongrel::HttpServer.new(addr, Puppet[:masterport])
-        webserver.register("/", server)
-    else
-        Puppet.err "Invalid server type %s" % Puppet[:servertype]
-        exit(45)
-    end
-rescue => detail
-    if Puppet[:trace]
-        puts detail.backtrace
-    end
-    $stderr.puts detail
-    exit(1)
-end
+require 'puppet/file_serving/content'
+require 'puppet/file_serving/metadata'
+require 'puppet/checksum'
+
+xmlrpc_handlers = [:Status, :FileServer, :Master, :Report, :CA, :Filebucket]
+rest_handlers = [:file_content, :file_metadata, :certificate, :facts, :catalog, :report, :checksum]
+
+server = Puppet::Network::Server.new(:handlers => rest_handlers, :xmlrpc_handlers => xmlrpc_handlers)
 
 if Process.uid == 0
     begin
         Puppet::Util.chuser
     rescue => detail
-        if Puppet[:debug]
-            puts detail.backtrace
-        end
+        puts detail.backtrace if Puppet[:trace]
         $stderr.puts "Could not change user to %s: %s" % [Puppet[:user], detail]
         exit(39)
     end
 end
 
-# Mongrel doesn't shut down like webrick; we really need to write plugins for it.
-if Puppet[:servertype] == "webrick"
-    Puppet.newservice(server)
-end
+# Tell Puppet to manage this service for us, which has it starting and stopping
+# as appropriate.
+Puppet.newservice(server)
+
 Puppet.settraps
 
 if Puppet[:daemonize]
@@ -279,10 +202,5 @@ if Puppet[:daemonize]
 end
 
 Puppet.notice "Starting Puppet server version %s" % [Puppet.version]
-case Puppet[:servertype]
-when "webrick"
-    Puppet.start
-when "mongrel":
-    webserver.run.join
-end
 
+Puppet.start