Added .listening to REST server, handle listen states and transitions.

1 files changed, 15 insertions, 0 deletions

diff --git a/lib/puppet/network/rest_server.rb b/lib/puppet/network/rest_server.rb
index 0ffa8ff16..7c745f321 100644
--- a/lib/puppet/network/rest_server.rb
+++ b/lib/puppet/network/rest_server.rb
@@ -1,5 +1,6 @@
 class Puppet::Network::RESTServer # :nodoc:
   @@routes = {}
+  @@listening = false
   
   def self.register(*indirections)
     raise ArgumentError, "indirection names are required" if indirections.empty?
@@ -17,4 +18,18 @@ class Puppet::Network::RESTServer # :nodoc:
   def self.reset
     self.unregister(@@routes.keys) unless @@routes.keys.empty?
   end
+  
+  def self.listening?
+    @@listening
+  end
+  
+  def self.listen
+    raise "Cannot listen -- already listening" if @@listening
+    @@listening = true
+  end
+  
+  def self.unlisten
+    raise "Cannot unlisten -- not currently listening" unless @@listening
+    @@listening = false
+  end
 end