com.proofpoint.discovery.cli 1.0

data/bin/dcurl

@@ -0,0 +1,43 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'httpclient'
+require 'json'
+require 'discovery/discovery'
+
+begin
+  Verbose = !ARGV.delete("--DEBUG").nil? 
+
+  out_args = ARGV.map do |arg|
+    md = /(\w+):\/\/([^\/]+)(\/.*)/.match(arg)
+    next arg if md.nil?
+  
+    protocol, host, path = *md[1..-1]
+  
+  
+    md = /^(?:(\w+)\.)*?(\w+)(?:\.(\w+))?$/.match(host)
+    next arg if md.nil?
+  
+    pool, type, environment = *md[1..-1]
+    
+    pool ||= "general"  
+  
+    puts "Looking up service '#{type}' in pool '#{pool}' in environment '#{environment}'" if Verbose
+    discovery = Discovery::Discovery.for_environment(environment)
+    base_url = discovery.lookup(type, pool).map { |service_descriptor| service_descriptor['properties'][protocol] }.compact.first
+  
+    if base_url.nil? then
+      puts "Service type #{type} in pool #{pool} not found"
+      exit 42
+    end
+  
+    "#{base_url}#{path}"
+  end
+  puts "Executing curl #{out_args.map{|v| "\"#{v}\"" }.join(" ")}" if Verbose
+  exec "curl", *out_args
+  
+rescue => e
+  puts e.message
+  exit 1
+end
+

data/bin/djconsole

@@ -0,0 +1,38 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'httpclient'
+require 'json'
+require 'discovery/discovery'
+
+begin
+  Verbose = !ARGV.delete("--DEBUG").nil? 
+
+  md = /^(?:(\w+)\.)*?(\w+)(?:\.(\w+))?$/.match(ARGV[0] || "")
+  raise "USAGE: djconsole POOL.TYPE.ENVIRONMENT" if ARGV.length != 1 || md.nil?
+
+  pool, type, environment = *md[1..-1]
+
+  pool ||= "general"
+
+  puts "Looking up service '#{type}' in pool '#{pool}' in environment '#{environment}'" if Verbose
+  discovery = Discovery::Discovery.for_environment(environment)
+
+  node_jmx = Hash[ discovery.lookup("jmx", pool).map{|sd| [ sd['nodeId'], sd['properties']['jmx'] ] } ]
+
+  service_descriptors = discovery.lookup(type, pool)
+  base_url = service_descriptors.map{ |sd| node_jmx[sd['nodeId']] }.compact.first
+
+  if base_url.nil? then
+    puts "Service type #{type} in pool #{pool} not found"
+    exit 42
+  end
+
+  puts "Executing jconsole \"#{base_url}\"" if Verbose
+  exec "jconsole", base_url
+  
+rescue => e
+  puts e.message
+  exit 1
+end
+

data/lib/discovery/discovery.rb

@@ -0,0 +1,50 @@
+module Discovery
+  class Discovery
+    def self.for_environment(environment)
+      environment ||= "local"
+
+      discoveryrc = File.expand_path("~/.discoveryrc")
+      unless File.exist?(discoveryrc) then
+        File.open(discoveryrc, "w"){ |f| f.puts("local = http://localhost:8080") }
+      end
+
+      value = File.readlines(discoveryrc).map { |l| l.strip.split(/\s*=\s*/, 2) }.select { |k, v| k == environment }.map { |k, v| v }.first
+      raise "No such environment '#{environment}'" if value.nil?
+      discovery_uris = value.split(/\s*,\s*/)
+      return Discovery.new(discovery_uris)
+    end
+
+    def initialize(discovery_uris, client = HTTPClient.new)
+      @client = client
+      @discovery_uris = discovery_uris
+    end
+
+    def lookup(type, pool)
+      service_descriptors = []
+      @discovery_uris.each do |discovery_uri|
+        begin
+          url = "#{discovery_uri}/v1/service/#{type}/#{pool}"
+          puts "Getting services at \"#{url}\"" if Verbose
+          response = @client.get(url)
+          if response.status == HTTP::Status::OK then
+            body = response.body
+            if body.is_a?(HTTP::Message::Body) then
+              body = body.content
+            end
+            service_descriptors = JSON.parse(body)["services"]
+            break
+          end
+        rescue => e
+          # ignored
+          p "ERROR:: #{e}" if Verbose
+        end
+      end
+
+      if service_descriptors.nil? then
+        return nil
+      end
+
+      return service_descriptors
+    end
+  end
+end

data/lib/discovery/version.rb

@@ -0,0 +1,3 @@
+module Discovery
+  VERSION = "1.0.0"
+end

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification 
+name: com.proofpoint.discovery.cli
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: "1.0"
+platform: ruby
+authors: 
+- Dain Sundstrom
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-06-13 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: httpclient
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.2.0
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: json_pure
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.5.1
+  type: :runtime
+  version_requirements: *id002
+description: Discovery command line interface
+email: 
+- dain@iq80.com
+executables: 
+- dcurl
+- djconsole
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- bin/dcurl
+- bin/djconsole
+- lib/discovery/discovery.rb
+- lib/discovery/version.rb
+has_rdoc: true
+homepage: https://github.com/dain/discovery-server
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+requirements: []
+
+rubyforge_project: discovery
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: discovery
+test_files: []
+