repsheet_visualizer 0.0.3 → 0.0.4

data/bin/repsheet_visualizer

@@ -2,11 +2,19 @@
 
 begin
   require 'repsheet_visualizer/application/app.rb'
+  if ARGV.count == 2
+    RepsheetVisualizer.set :redis_host, ARGV[0]
+    RepsheetVisualizer.set :redis_port, ARGV[1]
+  end
   RepsheetVisualizer.run!
 rescue LoadError => e
   require 'rubygems'
   path = File.expand_path '../../lib', __FILE__
   $:.unshift(path) if File.directory?(path) && !$:.include?(path)
   require 'repsheet_visualizer/application/app.rb'
+  if ARGV.count == 2
+    RepsheetVisualizer.set :redis_host, ARGV[0]
+    RepsheetVisualizer.set :redis_port, ARGV[1]
+  end
   RepsheetVisualizer.run!
 end

data/lib/repsheet_visualizer/application/app.rb

@@ -13,8 +13,15 @@ class RepsheetVisualizer < Sinatra::Base
     end
   end
 
+  def redis_connection
+    host = defined?(settings.redis_host) ? settings.redis_host : "localhost"
+    port = defined?(settings.redis_port) ? settings.redis_port : 6379
+
+    Redis.new(:host => host, :port => port)
+  end
+
   get '/' do
-    redis = Redis.new(:host => "localhost", :port => 6379)
+    redis = redis_connection
     data = redis.keys("*:requests").map {|d| d.split(":").first}.reject {|ip| ip.empty?}
     @actors = {}
     data.each do |actor|
@@ -27,14 +34,14 @@ class RepsheetVisualizer < Sinatra::Base
   end
 
   get '/activity/:ip' do
-    redis = Redis.new(:host => "localhost", :port => 6379)
+    redis = redis_connection
     @ip = params[:ip]
     @activity = redis.lrange("#{@ip}:requests", 0, -1)
     erb :activity
   end
 
   post '/action' do
-    redis = Redis.new(:host => "localhost", :port => 6379)
+    redis = redis_connection
     if params["action"] == "allow"
       redis.set("#{params[:ip]}:repsheet:blacklist", "false")
     else
@@ -44,7 +51,7 @@ class RepsheetVisualizer < Sinatra::Base
   end
 
   get '/breakdown' do
-    redis = Redis.new(:host => "localhost", :port => 6379)
+    redis = redis_connection
     @data = {}
     offenders = redis.keys("*:repsheet").map {|o| o.split(":").first}
     offenders.each do |offender|

data/lib/repsheet_visualizer/version.rb

@@ -1,3 +1,3 @@
 module RepsheetVisualizer
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: repsheet_visualizer
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-20 00:00:00.000000000 Z
+date: 2013-05-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redis