diff options
| -rw-r--r-- | lib/puppet/network/server.rb | 44 |
1 files changed, 22 insertions, 22 deletions
diff --git a/lib/puppet/network/server.rb b/lib/puppet/network/server.rb index 50e3bd686..9e6b82469 100644 --- a/lib/puppet/network/server.rb +++ b/lib/puppet/network/server.rb @@ -8,44 +8,44 @@ class Puppet::Network::Server raise(ArgumentError, "Must specify :address or configure Puppet :bindaddress.") @port = args[:port] || Puppet[:masterport] || raise(ArgumentError, "Must specify :port or configure Puppet :masterport") - @protocols = [] - @listening = false - @routes = {} - self.register(args[:handlers]) if args[:handlers] + @protocols = [] + @listening = false + @routes = {} + self.register(args[:handlers]) if args[:handlers] end def register(*indirections) - raise ArgumentError, "Indirection names are required." if indirections.empty? - indirections.flatten.each { |i| @routes[i.to_sym] = true } + raise ArgumentError, "Indirection names are required." if indirections.empty? + indirections.flatten.each { |i| @routes[i.to_sym] = true } end def unregister(*indirections) - raise "Cannot unregister indirections while server is listening." if listening? - indirections = @routes.keys if indirections.empty? - - indirections.flatten.each do |i| - raise(ArgumentError, "Indirection [%s] is unknown." % i) unless @routes[i.to_sym] - end + raise "Cannot unregister indirections while server is listening." if listening? + indirections = @routes.keys if indirections.empty? + + indirections.flatten.each do |i| + raise(ArgumentError, "Indirection [%s] is unknown." % i) unless @routes[i.to_sym] + end - indirections.flatten.each do |i| - @routes.delete(i.to_sym) - end + indirections.flatten.each do |i| + @routes.delete(i.to_sym) + end end def listening? - @listening + @listening end def listen - raise "Cannot listen -- already listening." if listening? - http_server.listen(@routes.dup) - @listening = true + raise "Cannot listen -- already listening." if listening? + http_server.listen(@routes.dup) + @listening = true end def unlisten - raise "Cannot unlisten -- not currently listening." unless listening? - http_server.unlisten - @listening = false + raise "Cannot unlisten -- not currently listening." unless listening? + http_server.unlisten + @listening = false end def http_server_class |