rzabbix 0.1 → 0.2

data/README.md

@@ -11,8 +11,8 @@ To install RZabbix, run the following command:
 To use RZabbix:
 
     gem 'rzabbix'
-    zbx = RZabbix::Base.new("zabbix.example.com", "user", "password")
-    host = zbx.find_host("my-dns-example.com")
+    RZabbix::Base.set_credentials("zabbix.example.com", "user", "password")
+    host = RZabbix::Host.find("my-dns-example.com")
 
 ## External Links
 

data/lib/rzabbix.rb

@@ -2,4 +2,42 @@ module RZabbix
   
 end
 
-require './lib/base'
+require 'rzabbix/connection'
+require 'rzabbix/base'
+require 'rzabbix/host'
+require 'rzabbix/host_group'
+require 'rzabbix/template'
+require 'rzabbix/version'
+
+class Hash
+  
+  def rzbx_symbolize_keys
+    inject({}) do |options, (key, value)|
+      options[(key.to_sym rescue key) || key] = value
+      options
+    end
+  end
+
+
+  def rzbx_recursively_symbolize_keys
+    result = self.rzbx_symbolize_keys
+    result.keys.each do |key|
+      if result[key].is_a?(Hash) || result[key].is_a?(Array)
+        result[key] = result[key].rzbx_recursively_symbolize_keys
+      end
+    end
+    result
+  end
+end
+
+class Array
+  def rzbx_recursively_symbolize_keys
+    self.map do |item|
+      if item.is_a?(Hash) || item.is_a?(Array)
+        item.rzbx_recursively_symbolize_keys
+      else
+        item
+      end
+    end
+  end
+end

data/lib/rzabbix/base.rb

@@ -1,9 +1,38 @@
+require 'json'
+require 'uri'
+require 'net/http'
+require 'net/https'
+
 module RZabbix
   
   class Base
     
+    def self.perform_request(*args)
+      Connection.perform_request(*args)
+    end
     
+    ###########################################################################################################
+    # Instance methods
+    ###########################################################################################################
     
+    attr_accessor :attributes
+    
+    def initialize(attributes)
+      super()
+      self.attributes = self.class.default_attributes.merge(attributes)
+    end
+    
+    def [](key)
+      self.attributes[key.to_sym]
+    end
+    
+    def to_json(*args)
+      self.attributes.to_json(*args)
+    end
+    
+    def to_hash
+      self.attributes
+    end
   end
   
 end

data/lib/rzabbix/connection.rb

@@ -0,0 +1,108 @@
+module RZabbix
+  
+  class Connection
+    
+    API_OUTPUT_SHORTEN = "shorten"
+    API_OUTPUT_REFER   = "refer"
+    API_OUTPUT_EXTEND  = "extend"
+    
+    class ResponseCodeError < StandardError; end
+    
+    class << self
+      attr_accessor :credentials
+      attr_accessor :use_ssl
+      attr_accessor :auth
+    end
+    
+    self.use_ssl = false
+    
+    def self.set_credentials(api_url, api_user, api_password)
+      self.credentials = {:api_url=>api_url, :api_user=>api_user, :api_password=>api_password}
+    end
+    
+    def self.next_request_id
+      @request_id = @request_id ? @request_id+1 : 1
+    end
+    
+    def self.auth
+      @auth ||= begin
+        auth_message = {
+          'auth' => nil,
+          'method' => 'user.authenticate',
+          'params' => {
+            'user' => Connection.credentials[:api_user],
+            'password' => Connection.credentials[:api_password],
+            '0' => '0'
+          }
+        }
+        do_request(auth_message)
+      end
+    end
+    
+    
+    def self.perform_request(controller, action, params)
+      message = message_for(controller, action, params)
+      do_request(message)
+    end
+    
+    def self.message_for(controller, action, params = {})
+      { 
+        'method' => "#{controller}.#{action}",
+        'params' => params.to_hash.merge(:output=>API_OUTPUT_EXTEND),
+        'auth' => Connection.auth
+      }
+    end
+    
+    def self.do_request(message)
+      id = next_request_id
+  
+      message['id'] = id
+      message['jsonrpc'] = '2.0'
+  
+      message_json = JSON.generate(message)
+      
+      #puts "#{JSON.parse(message_json).inspect}\n\n\n"
+      
+      uri = URI.parse(Connection.credentials[:api_url])
+      http = Net::HTTP.new(uri.host, uri.port)
+      if (use_ssl)
+        http.use_ssl = true
+        http.verify_mode = OpenSSL::SSL::VERIFY_NONE  
+      end
+      
+      request = Net::HTTP::Post.new(uri.request_uri)
+      request.add_field('Content-Type', 'application/json-rpc')
+      request.body=(message_json)
+      responce = http.request(request)
+      
+      if ( responce.code != "200" ) then
+        raise ResponseCodeError.new("Responce code from [" + credentials[:api_url] + "] is " + responce.code)
+      end
+  
+      responce_body_hash = JSON.parse(responce.body)
+  
+      #if not ( responce_body_hash['id'] == id ) then
+      # raise Zabbix::InvalidAnswerId.new("Wrong ID in zabbix answer")
+      #end
+  
+  
+      # Check errors in zabbix answer. If error exist - raise exception Zabbix::Error
+      if ( error = responce_body_hash['error'] ) then
+        error_message = error['message']
+        error_data = error['data']
+        error_code = error['code']
+  
+        e_message = "Code: [" + error_code.to_s + "]. Message: [" + error_message +\
+              "]. Data: [" + error_data + "]."
+  
+        raise StandardError.new(e_message)
+      end
+      
+      result = responce_body_hash['result'] 
+      result.respond_to?(:rzbx_recursively_symbolize_keys) ? result.rzbx_recursively_symbolize_keys : result
+    end
+  
+    
+  end
+  
+end

data/lib/rzabbix/host.rb

@@ -0,0 +1,63 @@
+module RZabbix
+  
+  class Host < Base
+    
+    def self.find(host_id)
+      hosts = perform_request(:host, :get, :hostids=>[host_id], :output=>"extend")
+      self.new(hosts.first.rzbx_recursively_symbolize_keys) unless hosts.empty?
+    end
+  
+    def self.find_host_id_by_hostname(hostname)
+      hosts = perform_request(:host, :get, :filter => {:host=>hostname})
+      hosts.first[:hostid]
+    end
+    
+    def self.find_by_hostname(hostname)
+      hosts = perform_request(:host, :get, :filter => {:host=>hostname}, :output=>"extend")
+      self.new(hosts.first) unless hosts.empty?
+    end
+    
+    def self.create(attributes)
+      host = self.new(attributes)
+      result = perform_request(:host, :create, host)
+      result && result[:hostids] ? self.find(result[:hostids].first) : nil
+    end
+    
+    def self.default_attributes
+      {
+        #:hostid=>nil,             #int	    Host ID	
+        :host=>nil,               #string	  Host name.
+        :port=>10050,             #int	    Port number.	
+        :status=>0,               #int	    Host Status.	
+        :useip=>0,                #int	    Use IP.	
+        :dns=>'',                 #string	  DNS.	
+        :ip=>'0.0.0.0',           #string	  IP.	
+        :proxy_hostid=> 0,        #int	    Proxy Host ID.	
+        :useipmi=> 0 ,            #int	    Use IPMI.	
+        :ipmi_ip=>'',             #string	  IPMAI IP.	
+        :ipmi_port=>623,          #int	    IPMI port.	
+        :ipmi_authtype=>0,        #int	    IPMI authentication type.	
+        :ipmi_privilege=>0,       #int	    IPMI privilege.	
+        :ipmi_username=>'',      #string	  IPMI username.	
+        :ipmi_password=>''       #string	  IPMI password.
+      }
+    end
+    
+    def to_hash(*args)
+      host_hash = self.attributes.inject({}) do |hash, (attr, v)|
+        case attr.to_sym
+          when :groups
+            hash[:groups] = v.map {|g| { :groupid=>g.attributes[:groupid]}}
+          when :templates
+            hash[:templates] = v.map {|t| {:templateid=>t.attributes[:templateid]}}
+          else
+            hash[attr] = v
+        end
+        hash
+      end
+      host_hash
+    end
+    
+  end
+  
+end

data/lib/rzabbix/host_group.rb

@@ -0,0 +1,56 @@
+module RZabbix
+  
+  class HostGroup < Base
+    
+    def self.find_by_name(name)
+      groups = perform_request(:hostgroup, :get, :filter => {:name=>name}, :output=>"extend")
+      self.new(groups.first.rzbx_recursively_symbolize_keys) unless groups.empty?
+    end
+    
+    def self.default_attributes
+      {                      
+        :nodeids => nil,
+        :groupids => nil,
+        :hostids => nil,
+        :templateids => nil,
+        :graphids => nil,
+        :triggerids => nil,
+        :maintenanceids => nil,
+        :monitored_hosts => nil,
+        :templated_hosts => nil,
+        :real_hosts => nil,
+        :not_proxy_hosts => nil,
+        :with_items => nil,
+        :with_monitored_items => nil,
+        :with_historical_items => nil,
+        :with_triggers => nil,
+        :with_monitored_triggers => nil,
+        :with_httptests => nil,
+        :with_monitored_httptests => nil,
+        :with_graphs => nil,
+        :editable => nil,
+        :nopermissions => nil
+        
+#        :filter => nil,
+#        :pattern => '',
+#        
+#        :output => Connection::API_OUTPUT_REFER,
+#        :extendoutput => nil,
+#        :select_hosts => nil,
+#        :select_templates => nil,
+#        :count => nil,
+#        :preservekeys => nil,
+#        :sortfield => '',
+#        :sortorder => '',
+#        :limit => nil,
+#        :limitSelects => nil
+      }
+    end
+    
+    def to_json(*args)
+      self.attributes.to_json(*args)
+    end
+    
+  end
+  
+end

data/lib/rzabbix/template.rb

@@ -0,0 +1,55 @@
+module RZabbix
+  
+  class Template < Base
+    
+    def self.find_by_name(name)
+      templates_hash = perform_request(:template, :get, :filter => {:host=>name}, :output=>"extend")
+      if templates_hash.kind_of?(Hash) && !templates_hash.keys.empty?
+        template_hash = templates_hash[templates_hash.keys.first]
+        self.new(template_hash.rzbx_recursively_symbolize_keys)
+      end
+    end
+    
+    def self.default_attributes
+      {                           
+        :nodeids => nil,
+        :groupids => nil,
+        :templateids => nil,
+        :parentTemplateids => nil,
+        :hostids => nil,
+        :graphids => nil,
+        :itemids => nil,
+        :triggerids => nil,
+        :with_items => nil,
+        :with_triggers => nil,
+        :with_graphs => nil,
+        :editable => nil,
+        :nopermissions => nil
+        
+#        :filter => nil,
+#        :pattern => '',
+#        
+#        :output => API_OUTPUT_REFER,
+#        :extendoutput => nil,
+#        :select_groups => nil,
+#        :select_hosts => nil,
+#        :select_templates => nil,
+#        :selectParentTemplates => nil,
+#        :select_items => nil,
+#        :select_triggers => nil,
+#        :select_graphs => nil,
+#        :select_applications => nil,
+#        :select_macros => nil,
+#        :countOutput => nil,
+#        :groupCount => nil,
+#        :preservekeys => nil,
+#        :sortfield => '',
+#        :sortorder => '',
+#        :limit => nil,
+#        :limitSelects => nil
+      }
+    end
+    
+  end
+  
+end

data/lib/rzabbix/version.rb

@@ -1,3 +1,3 @@
 module RZabbix
-  VERSION = "0.1" 
+  VERSION = "0.2" 
 end

metadata

@@ -1,12 +1,12 @@
 --- !ruby/object:Gem::Specification 
 name: rzabbix
 version: !ruby/object:Gem::Version 
-  hash: 9
+  hash: 15
   prerelease: false
   segments: 
   - 0
-  - 1
-  version: "0.1"
+  - 2
+  version: "0.2"
 platform: ruby
 authors: 
 - Neer Friedman
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-12-07 00:00:00 +02:00
+date: 2010-12-09 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -41,8 +41,12 @@ extensions: []
 extra_rdoc_files: []
 
 files: 
+- lib/rzabbix/host_group.rb
 - lib/rzabbix/base.rb
 - lib/rzabbix/version.rb
+- lib/rzabbix/host.rb
+- lib/rzabbix/template.rb
+- lib/rzabbix/connection.rb
 - lib/rzabbix.rb
 - README.md
 - ROADMAP.md