rabbitmqadmin-cli 1.0.4 → 1.0.5

data/README

@@ -4,4 +4,6 @@ http://www.rabbitmq.com/management-cli.html
 
 Once rabbitmqadmin is on the path you can use this gem in rake, capistrano or regular ruby.
 
-Added parameters to connect to any host/port
+v1.0.3 Added parameters to connect to any host/port
+v1.0.4 Added class based test_connection(host, port) method and use it in initialize to raise if no connection
+v1.0.5 Added set_permissions method, non destructive, will create vhost and/or user if necessary and set permissions

data/lib/rabbitmqadmin-cli.rb

@@ -1,12 +1,29 @@
 require 'json'
 
 class RabbitMqAdminCli
+
+  def self.test_connect(host, port, scheme = "http")
+    `wget --spider -T 1 -t 1 #{scheme}://#{host}:#{port}/img/rabbitmqlogo.png >/dev/null 2>&1`
+    case $?.exitstatus
+    when 0, 6
+      [0, "All OK"]
+    when 4
+      [1, "Attempt to connect to #{host}:#{port} timed out"]
+    when 5
+      [2, "Using scheme #{scheme}, SSL verification failed"]
+    else
+      [3, "General problem, cannot determine if RabbitMq http server is running on #{host}:#{port}"]
+    end
+  end
+
   def initialize admin_user = 'guest', admin_password = 'guest', host = 'localhost', port = 55672
     @admin_user     = admin_user
     @admin_password = admin_password
     @host = host
     @port = port
     @verbose = false
+    code, msg = self.class.test_connect(@host, @port)
+    raise msg unless code == 0
   end
 
   attr_accessor :verbose, :host, :port
@@ -59,6 +76,10 @@ class RabbitMqAdminCli
     parse execute("#{stub} list vhosts")
   end
 
+  def list_permissions
+    parse execute("#{stub} list permissions")
+  end
+
   def purge_queues vhost
     list_queues(vhost).each { |queue| execute "#{stub vhost} purge queue name=#{queue[:name]}" }
   end
@@ -71,10 +92,36 @@ class RabbitMqAdminCli
     list_vhosts.any? { |h| h[:name] == vhost }
   end
 
+  def permissions_exist? vhost, user
+    permission = list_permissions.select {|perm| perm[:user] == user && perm[:vhost] == vhost}
+    return false if permission.empty?
+    permission.first.values_at(:configure,:read,:write).all? {|val| val == ".*"}
+  end
+
+  def set_permissions vhost, user, password, tags = ''
+    return if permissions_exist?(vhost, user)
+
+    create_vhost = !vhost_exists?(vhost)
+    create_user  = !user_exists?(user)
+
+    if create_vhost && create_user
+      initialize_vhost(vhost)
+      initialize_user(vhost, user, password, tags)
+    elsif !create_vhost && create_user
+      initialize_user(vhost, user, password, tags)
+    elsif create_vhost && !create_user
+      initialize_vhost(vhost)
+      add_user_to_vhost(vhost, user)
+    else
+      add_user_to_vhost(vhost, user)
+    end
+
+  end
+
   private
 
   def add_user_to_vhost vhost, user
-    execute "#{stub} -V #{vhost} declare permission vhost=#{vhost} user=#{user} configure=\"\.\*\" write=\"\.\*\" read=\"\.\*\""
+    execute %Q(#{stub} -V #{vhost} declare permission vhost=#{vhost} user=#{user} configure=".*" write=".*" read=".*")
   end
 
   def execute command
@@ -83,6 +130,8 @@ class RabbitMqAdminCli
   end
 
   def parse json
+    puts json if @verbose
+    return unless json.is_a?(String)
     JSON.parse json, :symbolize_names => true
   end
 end

data/lib/rabbitmqadmin-cli/version.rb

@@ -1,3 +1,3 @@
 module RabbitMqAdminCli
-  VERSION = "1.0.4"
+  VERSION = "1.0.5"
 end

data/rabbitmqadmin-cli.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name        = 'rabbitmqadmin-cli'
-  s.version     = '1.0.4'
+  s.version     = '1.0.5'
   s.platform    = RUBY_PLATFORM.to_s == 'java' ? 'java' : Gem::Platform::RUBY
   s.authors     = ['Lee Henson', 'Guy Boertje']
   s.email       = ['lee.m.henson@gmail.com', 'guyboertje@gmail.com']

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rabbitmqadmin-cli
 version: !ruby/object:Gem::Version
-  version: 1.0.4
+  version: 1.0.5
   prerelease: 
 platform: ruby
 authors:
@@ -14,7 +14,7 @@ date: 2011-10-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json
-  requirement: &77963690 !ruby/object:Gem::Requirement
+  requirement: &14786860 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -22,7 +22,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *77963690
+  version_requirements: *14786860
 description: Requires rabbitmqadmin to be in your path
 email:
 - lee.m.henson@gmail.com
@@ -57,7 +57,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.10
+rubygems_version: 1.8.6
 signing_key: 
 specification_version: 3
 summary: Ruby wrapper for calling out the rabbitmqadmin cli script