torquebox-console 0.2.1 → 0.2.2

data/bin/tbconsole

@@ -54,8 +54,10 @@ EOT
   end
 
   desc "connect", "Runs the CLI console"
+  method_option :hostname, :type => :string, :default => "localhost"
+  method_option :port, :type => :numeric, :default => 8675
   def connect
-    TorqueBox::Console::Client.connect
+    TorqueBox::Console::Client.connect({ :host => options[:hostname], :port => options[:port] })
   end
 
   protected

data/lib/torquebox/console/client.rb

@@ -18,21 +18,22 @@ require 'readline'
 module TorqueBox
   module Console
     class Client
-      HEADERS = { "accept-version" => "1.1", "host" => "localhost" }
-      HOSTS   = [{:host => "localhost", :port => 8675}]
-      PARAMS  = { :connect_headers => HEADERS, :hosts => HOSTS, :max_reconnect_attempts => -1 }
+      DEFAULT_HEADERS = { "accept-version" => "1.1" }
+      DEFAULT_HOST = { :host => "localhost", :port => 8675 }
+      DEFAULT_PARAMS  = { :max_reconnect_attempts => -1 }
 
       attr_accessor :client, :closed
 
-      def initialize
+      def initialize (host = DEFAULT_HOST)
+        build_globals(host)
         @closed = false
-        @client = Stomp::Client.new( PARAMS )
+        @client = Stomp::Client.new(@params)
       rescue Stomp::Error::MaxReconnectAttempts
         puts "Cannot connect to TorqueBox. Are you sure the server is running?"
       end
 
-      def self.connect
-        Client.new.run
+      def self.connect (host = DEFAULT_HOST)
+        Client.new(host).run
       end
 
       def run
@@ -57,7 +58,7 @@ module TorqueBox
           while !received_prompt && !closed
             sleep 0.05
           end
-          while !closed && (input = Readline.readline( prompt, true ))
+          while !closed && (input = Readline.readline(prompt, true))
             received_prompt = false
             client.publish("/stomplet/console", input) unless closed
             while !received_prompt && !closed
@@ -69,6 +70,14 @@ module TorqueBox
           $stderr.close
         end
       end
+      
+      protected
+      
+      def build_globals (host)
+        @hosts = [host]
+        @headers = DEFAULT_HEADERS.merge({ :host => host[:host] })
+        @params = DEFAULT_PARAMS.merge({ :hosts => @hosts, :connect_headers => @headers })
+      end
     end
   end
 end

data/lib/torquebox/console/version.rb

@@ -1,5 +1,5 @@
 module TorqueBox
   module Console
-    VERSION = "0.2.1"
+    VERSION = "0.2.2"
   end
 end

metadata

@@ -2,7 +2,7 @@
 name: torquebox-console
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors: 
   - Lance Ball
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-12-04 00:00:00 Z
+date: 2013-01-02 00:00:00 Z
 dependencies: 
   - !ruby/object:Gem::Dependency 
     name: pry