server-density 0.0.1

data/Gemfile

@@ -0,0 +1,10 @@
+source 'http://rubygems.org'
+
+gem 'rest-client', '1.6.1'
+gem 'yajl-ruby', '0.7.8', :require => 'yajl'
+
+group :test do
+  gem 'rake', '0.8.7'
+  gem 'fakeweb', '1.3.0'
+  gem 'minitest', '1.7.2', :require => 'minitest/unit'
+end

data/Gemfile.lock

@@ -0,0 +1,20 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    fakeweb (1.3.0)
+    mime-types (1.16)
+    minitest (1.7.2)
+    rake (0.8.7)
+    rest-client (1.6.1)
+      mime-types (>= 1.16)
+    yajl-ruby (0.7.8)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  fakeweb (= 1.3.0)
+  minitest (= 1.7.2)
+  rake (= 0.8.7)
+  rest-client (= 1.6.1)
+  yajl-ruby (= 0.7.8)

data/README

@@ -0,0 +1,13 @@
+Quick experiment. Use at your own risk.
+
+Setup:
+
+$ bundle install
+
+Test:
+
+$ bundle exec rake
+
+To create or find a server agent key for IP 1.1.1.1:
+
+server-density-key --username foo --password bar --account example.serverdensity.com --ip 1.1.1.1

data/Rakefile

@@ -0,0 +1,13 @@
+require 'rubygems'
+require 'bundler/setup'
+Bundler.require(:default, :test)
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), 'lib'))
+require 'server_density'
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+end
+
+task :default => :test

data/bin/server-density-key

@@ -0,0 +1,34 @@
+#!/usr/bin/env ruby
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib', 'server_density'))
+require 'optparse'
+
+options = {}
+
+OptionParser.new do |opts|
+  opts.banner = "Usage:\n server-density-key [options]"
+
+  opts.on("--ip IP", "IP Address") do |i|
+    options[:ip] = i
+  end
+
+  opts.on("--username USERNAME", "Server Density Username") do |u|
+    options[:username] = u
+  end
+
+  opts.on("--password PASSWORD", "Server Density Password") do |p|
+    options[:password] = p
+  end
+
+  opts.on("--account ACCOUNT", "Server Density Account i.e. example.serverdensity.com") do |a|
+    options[:account] = a
+  end
+end.parse!
+
+ip = options.delete(:ip)
+unless ip
+  puts "ERROR" # TODO polish
+  exit(1)
+end
+client = ServerDensity::Client.new(options)
+server = client.servers.find_or_create(ip)
+print server.agent_key

data/lib/server_density.rb

@@ -0,0 +1,102 @@
+require 'rubygems'
+require 'rest_client'
+require 'yajl'
+
+module ServerDensity
+  class Client
+    attr_reader :username, :password, :account
+    def initialize(options={})
+      valid = [:username, :password, :account].all? { |param| options[param] != nil && options[param] != '' }
+      unless valid
+        puts "ERROR" # TODO polish
+        exit(1)
+      end
+      @username = options[:username]
+      @password = options[:password]
+      @account  = options[:account]
+    end
+
+    def servers
+      @servers ||= Servers.new(self)
+    end
+  end
+
+  class Server
+    attr_reader :id, :agent_key, :ip
+    def initialize(data)
+      @id        = data['serverId']
+      @ip        = data['ip']
+      @agent_key = data['agentKey']
+    end
+  end
+
+  class CommandResult
+    def initialize(response)
+      @parsed_response = Yajl::Parser.parse(response)
+      @status = @parsed_response['status']
+    end
+
+    def success?
+      @status == 1
+    end
+
+    def data
+      @parsed_response['data'] if success?
+    end
+
+    def error_code
+      @parsed_response['error']['code'] unless success?
+    end
+
+    def error_message
+      @parsed_response['error']['message'] unless success?
+    end
+  end
+
+  class Servers
+    def initialize(client)
+      @client = client
+    end
+
+    def all
+      result = get("servers/list")
+      if result.success?
+        result.data['servers'].map { |data| Server.new(data) }
+      end
+    end
+
+    def find_or_create(ip)
+      server = find(ip)
+      server ||= create(ip)
+    end
+
+    def find(ip)
+      all.detect { |server| server.ip == ip }
+    end
+
+    def create(ip)
+      result = post("servers/add", {
+        :name => ip,
+        :ip   => ip
+      })
+      if result.success?
+        Server.new(result.data.merge('ip' => ip))
+      end
+    end
+
+    def get(command)
+      response = begin
+        response = RestClient.get("https://#{@client.username}:#{@client.password}@api.serverdensity.com/1.0/?account=#{@client.account}&c=#{command}")
+      rescue
+        puts "ERROR" # TODO polish
+        exit(1)
+      end
+      CommandResult.new(response)
+    end
+
+    def post(command, params)
+      response = RestClient.post("https://#{@client.username}:#{@client.password}@api.serverdensity.com/1.0/?account=#{@client.account}&c=#{command}", params)
+      CommandResult.new(response)
+    end
+  end
+end

data/server-density.gemspec

@@ -0,0 +1,29 @@
+Gem::Specification.new do |s|
+  s.specification_version = 2 if s.respond_to? :specification_version=
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.name               = "server-density"
+  s.version            = "0.0.1"
+  s.summary            = "API wrapper for Server Density"
+  s.authors            = ["Jon Crosby"]
+  s.email              = "jon@path.com"
+  s.executables        = ["server-density-key"]
+  s.default_executable = 'server-density-key'
+  s.files              = %w[
+    Gemfile
+    Gemfile.lock
+    README
+    Rakefile
+    server-density.gemspec
+    bin/server-density-key
+    lib/server_density.rb
+    test/helper.rb
+    test/test_server_density.rb
+  ]
+  s.test_files         = s.files.select {|path| path =~ /^test\/test_.*\.rb/}
+  s.rubygems_version   = "1.3.7"
+  s.add_dependency 'rest-client', '1.6.1'
+  s.add_dependency 'yajl-ruby', '0.7.8'
+  s.add_development_dependency 'rake', '0.8.7'
+  s.add_development_dependency 'minitest', '1.7.2'
+  s.add_development_dependency 'fakeweb', '1.3.0'
+end

data/test/helper.rb

@@ -0,0 +1,7 @@
+dir = File.dirname(File.expand_path(__FILE__))
+$LOAD_PATH.unshift(dir)
+$LOAD_PATH.unshift(File.join(dir, '..', 'lib'))
+require 'server_density'
+require 'fakeweb'
+require 'minitest/unit'
+MiniTest::Unit.autorun

data/test/test_server_density.rb

@@ -0,0 +1,48 @@
+require File.join(File.expand_path(File.dirname(__FILE__)), 'helper')
+
+class TestServerDensity < MiniTest::Unit::TestCase
+  def setup
+    @client = ServerDensity::Client.new(:username => 'foo', :password => 'bar', :account => 'example.serverdensity.com')
+    fake_list_response = '{"status":1,"data":{"servers":[{"serverId":1,"ip":"0.0.0.0","name":"example-server","group":"utility","location":"basement","provider":"EC2","agentKey":"decafbad","os":"linux"},{"serverId":2,"ip":"0.0.0.1","name":"example-server-2","group":"utility","location":"basement","provider":"EC2","agentKey":"deadbeef","os":"linux"}]}}'
+    FakeWeb.register_uri(:get, 'https://foo:bar@api.serverdensity.com/1.0/?account=example.serverdensity.com&c=servers/list', :body => fake_list_response)
+    fake_create_response = '{"status":1,"data":{"serverId":42,"agentKey":"decafdecafbeef"}}'
+    FakeWeb.register_uri(:post, 'https://foo:bar@api.serverdensity.com/1.0/?account=example.serverdensity.com&c=servers/add', :body => fake_create_response)
+  end
+
+  def test_list_servers
+    servers = @client.servers.all
+    assert_equal(2, servers.size)
+    assert_equal('decafbad', servers.first.agent_key)
+    assert_equal('0.0.0.1', servers.last.ip)
+  end
+
+  def test_find
+    server = @client.servers.find('0.0.0.0')
+    assert(server)
+    assert_equal('0.0.0.0', server.ip)
+    assert_equal('decafbad', server.agent_key)
+  end
+
+  def test_find_or_create_with_match
+    server = @client.servers.find_or_create('0.0.0.0')
+    assert(server)
+    assert_equal('0.0.0.0', server.ip)
+    assert_equal('decafbad', server.agent_key)
+  end
+
+  def test_find_or_create_without_match
+    server = @client.servers.find_or_create('1.1.1.1')
+    assert(server)
+    assert_equal('1.1.1.1', server.ip)
+    assert_equal(42, server.id)
+    assert_equal('decafdecafbeef', server.agent_key)
+  end
+
+  def test_create
+    server = @client.servers.create('1.1.1.1')
+    assert(server)
+    assert_equal('1.1.1.1', server.ip)
+    assert_equal(42, server.id)
+    assert_equal('decafdecafbeef', server.agent_key)
+  end
+end

metadata

@@ -0,0 +1,154 @@
+--- !ruby/object:Gem::Specification 
+name: server-density
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Jon Crosby
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-11-13 00:00:00 -08:00
+default_executable: server-density-key
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rest-client
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 13
+        segments: 
+        - 1
+        - 6
+        - 1
+        version: 1.6.1
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: yajl-ruby
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 19
+        segments: 
+        - 0
+        - 7
+        - 8
+        version: 0.7.8
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 49
+        segments: 
+        - 0
+        - 8
+        - 7
+        version: 0.8.7
+  type: :development
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: minitest
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 15
+        segments: 
+        - 1
+        - 7
+        - 2
+        version: 1.7.2
+  type: :development
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: fakeweb
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 27
+        segments: 
+        - 1
+        - 3
+        - 0
+        version: 1.3.0
+  type: :development
+  version_requirements: *id005
+description: 
+email: jon@path.com
+executables: 
+- server-density-key
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- Gemfile
+- Gemfile.lock
+- README
+- Rakefile
+- server-density.gemspec
+- bin/server-density-key
+- lib/server_density.rb
+- test/helper.rb
+- test/test_server_density.rb
+has_rdoc: true
+homepage: 
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 2
+summary: API wrapper for Server Density
+test_files: 
+- test/test_server_density.rb