zapix3 0.1.0

data/lib/zapix/proxies/hosts.rb

@@ -0,0 +1,77 @@
+require_relative 'base'
+class Hosts < Base
+
+  def get_id(name)
+    if(exists?(name))
+      client.host_get({'filter' => {'host' => name}}).first['hostid']
+    else
+      raise NonExistingHost, "Host #{name} does not exist !"
+    end
+  end
+
+  def create(options={})
+    client.host_create(options) unless exists?(options["host"])
+  end
+
+  def create_or_update(options={})
+    raise EmptyHostname, 'Host name cannot be empty !' if options['host'].nil?
+    if exists?(options['host'])
+      id = get_id(options['host'])
+      options.merge!('hostid' => id)
+      client.host_update(options)
+    else
+      create(options)
+    end
+  end
+
+  def unlink_and_clear_templates(options={})
+    template_ids = options['template_ids'].map { |id| {'templateid' => id}}
+    client.host_update({'hostid' => options['host_id'], 'templates_clear' => template_ids})
+  end
+
+  def update_templates(options={})
+    template_ids = options['template_ids'].map { |id| {'templateid' => id}}
+    client.host_update({'hostid' => options['host_id'], 'templates' => template_ids})
+  end
+
+  def update_macros(options={})
+    client.host_update('hostid' => options['host_id'], 'macros' => options['macros'])
+  end
+
+  def exists?(name)
+    result = client.host_get({'filter' => {'host' => name}})
+    if (result == nil || result.empty?)
+      false
+    else
+      true
+    end
+  end
+
+  def get_all
+    host_names_and_ids = client.host_get({'output' => ['name']})
+
+    # the fucking api ALWAYS returns the ids and that's
+    # why we need to extract the names separately
+
+    extract_host_names(host_names_and_ids)
+  end
+
+  def delete(name)
+    if exists?(name)
+      client.host_delete([get_id(name)])
+    else
+      raise NonExistingHost, "Host #{name} cannot be deleted because it does not exist !"
+    end
+  end
+
+  class NonExistingHost < StandardError; end
+  class EmptyHostname < StandardError; end
+
+  private
+
+  def extract_host_names(hosts_and_ids)
+    hosts_and_ids.map do |current_host|
+      current_host['name']
+    end
+  end
+end

data/lib/zapix/proxies/scenarios.rb

@@ -0,0 +1,34 @@
+require_relative 'base'
+
+class Scenarios < Base
+  def create(options)
+    client.httptest_create(options) unless exists?(options)
+  end
+
+  def get_id(options)
+    client.httptest_get({
+      'filter' => {'name' => options['name'], 'hostid' => options['hostid']}}).first['httptestid']
+  end
+
+  def delete(options)
+    client.httptest_delete(options)
+  end
+
+  def exists?(options)
+    result = client.httptest_get({
+      'countOutput' => true,
+      'filter' => {'name' => options['name'],
+      'hostid' => options['hostid']}})
+
+    result.to_i >= 1 ? true : false
+  end
+
+  def get_all
+    scenarios = client.httptest_get({'output' => 'extend'})
+    names = extract_names(scenarios)
+  end
+
+  def extract_names(scenarios)
+    scenarios.map {|scenario| scenario['name']} 
+  end
+end

data/lib/zapix/proxies/templates.rb

@@ -0,0 +1,30 @@
+require_relative 'base'
+class Templates < Base
+
+  def exists?(name)
+    result = client.template_get({'filter' => {'name' => name}})
+    if (result == nil || result.empty?)
+      false
+    else
+      true
+    end
+  end
+
+  def create(options)
+    client.template_create(options) unless exists?(options['host'])
+  end
+
+  def get_id(name)
+    if(exists?(name))
+      client.template_get({'filter' => {'name' => name}}).first['templateid']
+    else
+      raise NonExistingTemplate, "Template #{name} does not exist !"
+    end
+  end
+
+  def get_templates_for_host(id)
+    client.template_get({'hostids' => [id]}).map { |result_set| result_set['templateid'] }
+  end
+
+  class NonExistingTemplate < StandardError; end
+end

data/lib/zapix/proxies/triggers.rb

@@ -0,0 +1,40 @@
+require_relative 'base'
+class Triggers < Base
+
+  def create(options)
+    client.trigger_create(options)
+  end
+
+  def delete(*trigger_ids)
+    client.trigger_delete(trigger_ids)
+  end
+
+  def get_id(options)
+    result = client.trigger_get({'output' => 'extend',
+      'expandExpression' => true})
+    id = extract_id(result, options['expression'])
+    unless id.nil?
+      id
+    else
+      raise NonExistingTrigger, "Trigger with expression #{options['expression']} not found."
+    end
+  end
+
+  class NonExistingTrigger < StandardError; end
+
+  private
+
+  def extract_id(triggers, expression)
+    result = nil
+    triggers.each do |trigger|
+      if trigger['expression'] == expression
+        result = trigger['triggerid']
+        break
+      end
+    end
+    result
+  end
+
+end
+
+

data/lib/zapix/proxies/usergroups.rb

@@ -0,0 +1,35 @@
+require_relative 'base'
+
+class Usergroups < Base
+
+  def create(options)
+    client.usergroup_create(options) unless exists?(options)
+  end
+
+  def exists?(options)
+    #client.usergroup_exists(options)
+    result = client.usergroup_get({'filter' => {'name' => options['name']}})
+    if result.empty? || result == nil
+      false
+    else
+      true
+    end
+  end
+
+  def get_id(options)
+    if(exists?(options))
+      result = client.usergroup_get({
+        'filter' => {'name' => options['name']}})
+      result.first['usrgrpid']
+    else
+      raise NonExistingUsergroup, "Usergroup #{options['name']} does not exist !"
+    end
+  end
+
+  def delete(*group_ids)
+    client.usergroup_delete(group_ids)
+  end
+
+  class NonExistingUsergroup < StandardError; end
+
+end

data/lib/zapix/proxies/users.rb

@@ -0,0 +1,28 @@
+require_relative 'base'
+
+class Users < Base
+  def create(options)
+    client.user_create(options) unless exists?(options)
+  end
+
+  def exists?(options)
+    result = client.user_get({'filter' => {'alias' => options['alias']}})
+    result.empty? ? false : true
+  end
+
+  def get_id(options)
+    if(exists?(options))
+      client.user_get({'filter' => {'alias' => options['alias']}}).first['userid']
+    else
+      raise NonExistingUser, "User #{options['alias']} does not exist !"
+    end
+  end
+
+  def delete(user_ids)
+    client.user_delete(user_ids)
+  end
+
+  class NonExistingUser < StandardError; end
+end
+
+

data/lib/zapix/version.rb

@@ -0,0 +1,3 @@
+module Zapix
+  VERSION = "0.1.0"
+end

data/lib/zapix/zabbix_classes/host.rb

@@ -0,0 +1,47 @@
+class Host
+
+  attr_accessor :group_ids, :template_ids, :interfaces, :macros
+
+  def initialize()
+    @group_ids = Array.new
+    @template_ids = Array.new
+    @interfaces = Array.new
+    @macros = Array.new
+    @properties = Hash.new
+  end
+
+  def add_name(name)
+    @properties.merge!('host' => name)
+  end
+
+  def add_group_ids(*ids)
+    ids.each do |id|
+      group_ids << {'groupid' => id}
+    end
+    @properties.merge!('groups' => group_ids)
+  end
+
+  def add_interfaces(*ifaces)
+    interfaces.concat(ifaces)
+    @properties.merge!('interfaces' => interfaces)
+  end
+
+  def add_template_ids(*ids)
+    ids.each do |id|
+      template_ids << {'templateid' => id}
+    end
+    @properties.merge!('templates' => template_ids)
+  end
+
+  def add_macros(*host_macros)
+    macros.concat(host_macros)
+    @properties.merge!('macros' => host_macros)
+  end
+
+  def to_hash
+    @properties
+  end
+
+end
+
+

data/lib/zapix/zabbix_classes/interface.rb

@@ -0,0 +1,42 @@
+require 'active_record'
+
+class Interface
+  include ActiveModel::Validations
+
+  attr_reader :type, :main, :useip, :ip, :dns, :port
+  validates_presence_of :type, :main, :useip, :ip, :dns, :port
+
+  # for more info see
+  #  https://www.zabbix.com/documentation/2.0/manual/appendix/api/hostinterface/definitions#host_interface
+  # we assume ip and dns shall always be set
+
+  validates_inclusion_of :type, :in => 1..4
+  validates_inclusion_of :main, :in => 0..1
+  validates_inclusion_of :useip, :in => 0..1
+
+  def initialize(attributes)
+    @type = attributes['type'] ||= 1
+    @main = attributes['main'] ||= 1
+    @useip = attributes['useip'] ||= 1
+    @ip = attributes['ip'] = attributes['ip']
+    @dns = attributes['dns'] = attributes['dns']
+    @port = attributes['port'] = attributes['port'] ||= 10050
+    @result = {
+      'type' => type,
+      'main' => main,
+      'useip' => useip,
+      'ip' => ip,
+      'dns' => dns,
+      'port' => port
+    }
+  end
+
+  def read_attribute_for_validation(key)
+    send(key)
+  end
+
+  def to_hash
+    @result
+  end
+
+end

data/lib/zapix/zabbix_rpc_client.rb

@@ -0,0 +1,58 @@
+require 'net/http'
+require 'uri'
+require 'json'
+ 
+class ZabbixRPCClient
+
+  attr_reader :uri, :debug
+
+  def initialize(options)
+    @uri = URI.parse(options[:service_url])
+    @username = options[:username]
+    @password = options[:password]
+    @debug = options[:debug]
+    @auth_token = authenticate
+  end
+ 
+  def method_missing(name, *args)
+    method_name = map_name(name)
+
+    post_body = {
+      'method' => method_name,
+      'params' => args.first,
+      'id' => id,
+      'jsonrpc' => '2.0',
+      'auth' => @auth_token 
+    }.to_json
+
+    resp = JSON.parse( http_post_request(post_body) )
+    raise JSONRPCError, resp['error'] if resp['error']
+    puts "[DEBUG] Get answer: #{resp}" if debug
+    resp["result"]
+  end
+ 
+  def http_post_request(post_body)
+    http    = Net::HTTP.new(uri.host, uri.port)
+    http.use_ssl = true
+    request = Net::HTTP::Post.new(uri.request_uri)
+    request.content_type = 'application/json'
+    request.body = post_body
+    puts "[DEBUG] Send request: #{request.body}" if debug
+    http.request(request).body
+  end
+
+  def authenticate
+    p user_login({'user' => @username, 'password' => @password})
+    user_login({'user' => @username, 'password' => @password})
+  end
+
+  def id
+    rand(100000)
+  end
+
+  def map_name(name)
+   name.to_s.sub('_', '.')
+  end
+
+  class JSONRPCError < RuntimeError; end
+end

data/lib/zapix.rb

@@ -0,0 +1,57 @@
+require 'zapix/version'
+require_relative 'zapix/zabbix_rpc_client'
+
+class ZabbixAPI
+  attr_reader :client
+
+  def self.connect(options = {})
+    new(options)
+  end
+
+  def initialize(options = {})
+    @client = ZabbixRPCClient.new(options)
+    Dir["#{File.dirname(__FILE__)}/zapix/zabbix_classes/*.rb"].each { |f| load(f) }
+    Dir["#{File.dirname(__FILE__)}/zapix/proxies/*.rb"].each { |f| load(f) }
+  end
+
+  def hostgroups
+    @hostgroups ||= HostGroups.new(client)
+  end
+
+  def hosts
+    @hosts ||= Hosts.new(client)
+  end
+
+  def templates
+    @templates ||= Templates.new(client)
+  end
+
+  def applications
+    @applications ||= Applications.new(client)
+  end
+
+  def scenarios
+    @scenarios ||= Scenarios.new(client)
+  end
+
+  def triggers
+    @triggers ||= Triggers.new(client)
+  end
+
+  def hostinterfaces
+    @hostinterfaces ||= Hostinterfaces.new(client)
+  end
+
+  def actions
+    @actions ||= Actions.new(client)
+  end
+
+  def usergroups
+    @usergroups ||= Usergroups.new(client)
+  end
+
+  def users
+    @users ||= Users.new(client)
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,19 @@
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper"` to ensure that it is only
+# loaded once.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+require 'zapix'
+RSpec.configure do |config|
+
+  config.treat_symbols_as_metadata_keys_with_true_values = true
+  config.run_all_when_everything_filtered = true
+  config.filter_run :focus
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = 'random'
+end