restfully-addons 1.0.0

data/LICENSE

@@ -0,0 +1 @@
+TODO

data/README.md

@@ -0,0 +1,24 @@
+# restfully-addons
+
+Some additional helpers for RESTful APIs, to deal with SSH connections or
+other protocols.
+
+Long-term goal is to add support for these protocols into Restfully.
+
+## Installation
+
+    $ gem install restfully-addons
+
+## Usage
+
+Just require the library you need in your scripts. For instance, requiring
+addons for the BonFIRE API would be done as follows:
+
+    require 'restfully'
+    require 'restfully/addons/bonfire'
+    
+    # your code
+
+## Authors
+
+* Cyril Rohr <cyril.rohr@inria.fr>

data/Rakefile

@@ -0,0 +1 @@
+# TODO: test

data/lib/restfully/addons/bonfire/ssh.rb

@@ -0,0 +1,62 @@
+require 'net/ssh/gateway'
+require 'net/scp'
+require 'net/sftp'
+
+module Restfully
+  class Resource
+    def ssh(user = nil, opts = {}, &block)
+      raise NotImplementedError unless uri.to_s =~ /\/computes\/\w+$/
+      @ssh ||= SSH.new(session, self)
+      @ssh.run(@ssh.ip, user, opts, &block) if block
+      @ssh
+    end
+  end
+  
+  class SSH
+    attr_reader :session, :resource
+
+    def initialize(session, resource)
+      @session = session
+      @resource = resource
+    end
+    
+    def run(fqdn, user, options = {}, &block)
+      user ||= 'root'
+      session.logger.info "Trying to SSH into #{user}@#{fqdn}..."
+      options[:keys] ||= session.config[:keys]
+      gateway = session.config[:gateway]
+      if gateway
+        gateway_handler = Net::SSH::Gateway.new(gateway, session.config[:username], options)
+        gateway_handler.ssh(fqdn, user, options, &block)
+        gateway_handler.shutdown!
+      else
+        Net::SSH.start(fqdn, user, options, &block)
+      end
+    end
+    
+    def ip
+      @ip ||= accessible?
+    end
+    
+    def accessible?
+      return false if resource['nic'].empty?
+      accessible_nic = resource['nic'].find{|nic|
+        ip = nic['ip']
+        if ip.nil? || ip =~ /^192.168/
+          true
+        else
+          begin
+            Timeout.timeout(10) do
+              run(ip, 'root') {|s| s.exec!("hostname") }
+            end
+            true
+          rescue Exception => e
+            session.logger.info  "Can't SSH yet to #{ip}. Reason: #{e.class.name}, #{e.message}."
+            false
+          end
+        end
+      }
+      accessible_nic && accessible_nic['ip']
+    end
+  end
+end

data/lib/restfully/addons/bonfire/zabbix.rb

@@ -0,0 +1,130 @@
+require 'json'
+
+class Array
+  def sum
+    inject(:+)
+  end
+
+  def avg
+    sum.to_f / size
+  end
+
+  def median
+    sorted = sort
+    (sorted[size/2] + sorted[(size+1)/2]) / 2
+  end
+end
+
+module Restfully
+  class Resource
+    def zabbix
+      raise NotImplementedError unless uri.to_s =~ /\/experiments\/[0-9]+$/
+      @zabbix ||= Zabbix.new(session, self)
+    end
+  end
+  
+  class Zabbix
+    attr_reader :session, :experiment
+    def initialize(session, experiment, opts = {})
+      @session = session
+      @username = opts[:username] || "Admin"
+      @password = opts[:password] || "zabbix"
+      @experiment = experiment
+      @token, @request_id = nil, 0
+      @uri = @experiment.uri.to_s+"/zabbix"
+      @max_attempts = 5
+    end
+
+    def request(method, params = {})
+      begin
+        authenticate if @token.nil? && method != "user.authenticate"
+        @request_id += 1
+        q = { "jsonrpc" => "2.0", "auth" => @token, "id" => @request_id,
+              "method" => method, "params" => params }
+        resource = @session.post(@uri,
+          JSON.dump(q),
+          :head => {
+            'Content-Type' => 'application/json',
+            'Accept' => 'application/json'
+          }
+        )
+
+        # That fucking zabbix API returns "text/plain" as content-type...
+        h = JSON.parse(resource.response.body)
+
+        if h['error']
+          if h['error']['data'] == "Not authorized"
+            @token = nil
+            request(method, params)
+          else
+            raise StandardError, "Received error: #{h.inspect}" if h['error']
+          end
+        else
+          h['result']
+        end
+      rescue Restfully::HTTP::Error => e
+        # retry ad vitam eternam
+        sleep 5
+        retry
+      end
+    end
+
+    def authenticate
+      @token = request("user.authenticate", {"user" => @username, "password" => @password})
+    end
+    
+    def metric(name, options = {})
+      hosts = [options.delete(:hosts) || []].flatten.map{|h|
+        [h['name'], h['id']].join("-")
+      }
+      items = request("item.get", {
+        :filter => {
+          "host" => hosts[0],
+          "key_" => name.to_s
+        },
+        "output" => "extend"
+      }).map{|i| i['itemid']}
+
+      history = case options[:type]
+      when :numeric
+        0
+      else
+        1 # STRING
+      end
+
+      # Most recent last
+      now = Time.now.to_i
+      results = request("history.get", {
+        "itemids" => items[0..1],
+        # FIX once we can correctly specify metric type
+        "history" => history, 
+        "output" => "extend",
+        "time_from" => now-3600,
+        "time_till" => now
+      })
+
+      Metric.new(name, results, options)
+    end
+
+    class Metric
+      def initialize(name, results, opts = {})
+        @name = name
+        @results = results
+        @opts = opts
+      end
+
+      def values
+        @results.map{|r| 
+          case @opts[:type]
+          when :numeric
+            r['value'].to_f
+          else
+            r['value']
+          end
+        }.reverse
+      end
+    end
+    
+  end
+  
+end

data/lib/restfully/addons/bonfire.rb

@@ -0,0 +1,3 @@
+require 'restfully'
+require 'restfully/addons/bonfire/ssh'
+require 'restfully/addons/bonfire/zabbix'

data/lib/restfully/addons/version.rb

@@ -0,0 +1,5 @@
+module Restfully
+  module Addons
+    VERSION = "1.0.0"
+  end
+end

data/lib/restfully/addons.rb

@@ -0,0 +1 @@
+require 'restfully/addons/bonfire'

metadata

@@ -0,0 +1,135 @@
+--- !ruby/object:Gem::Specification
+name: restfully-addons
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Cyril Rohr
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-07-29 00:00:00.000000000 +02:00
+default_executable: 
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: restfully
+  requirement: &2156314120 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2156314120
+- !ruby/object:Gem::Dependency
+  name: net-ssh-gateway
+  requirement: &2156313660 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2156313660
+- !ruby/object:Gem::Dependency
+  name: net-scp
+  requirement: &2156313240 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2156313240
+- !ruby/object:Gem::Dependency
+  name: net-sftp
+  requirement: &2156312820 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2156312820
+- !ruby/object:Gem::Dependency
+  name: net-ssh-multi
+  requirement: &2156312400 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2156312400
+- !ruby/object:Gem::Dependency
+  name: libxml-ruby
+  requirement: &2156311980 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2156311980
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: &2156563360 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.8'
+  type: :development
+  prerelease: false
+  version_requirements: *2156563360
+description: Addons for Restfully
+email:
+- cyril.rohr@inria.fr
+executables: []
+extensions: []
+extra_rdoc_files:
+- LICENSE
+- README.md
+files:
+- lib/restfully/addons/bonfire/ssh.rb
+- lib/restfully/addons/bonfire/zabbix.rb
+- lib/restfully/addons/bonfire.rb
+- lib/restfully/addons/version.rb
+- lib/restfully/addons.rb
+- Rakefile
+- LICENSE
+- README.md
+has_rdoc: true
+homepage: http://github.com/crohr/restfully-addons
+licenses: []
+post_install_message: 
+rdoc_options:
+- --charset=UTF-8
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '1.8'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '1.3'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: Addons for Restfully
+test_files: []