zabbix 0.1.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,21 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Matthew Knopp
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,50 @@
+= zabbix-rb
+            _     _     _                _     
+   ______ _| |__ | |__ (_)_  __     _ __| |__  
+  |_  / _` | '_ \| '_ \| \ \/ /____| '__| '_ \ 
+   / / (_| | |_) | |_) | |>  <_____| |  | |_) |
+  /___\__,_|_.__/|_.__/|_/_/\_\    |_|  |_.__/ 
+
+I threw this together this afternoon. It's absulotely untested. I only just 
+corrected a few syntax errors. 
+
+== usage 
+
+=== easy
+
+  Zabbix::Sender::Easy.run(:"com.yammer.cron.brithday_email_job") do |zbx|
+    Users.each_with_index do |user, idx|
+      HappyBirthday.send if user.birthday?
+  
+      ## let zabbix know we're alive every N users, whatever
+      zbx.send_heartbeat if idx % N == 0
+    end
+  end
+
+=== tedious
+
+  zbx = Zabbix::Sender.new
+  zbx.send_start(:"com.yammer.cron.birthday_email_job")
+  
+  Users.each_with_index do |user, idx|
+    HappyBirthday.send if user.birthday? 
+  
+    ## let zabbix know we're alive every N users, whatever 
+    zbx.send_heartbeat if idx % N == 0
+  end
+  zbx.send_stop(:"com.yammer.cron.birthday_email_job")
+
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Matthew Knopp. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,52 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name        = "zabbix"
+    gem.summary     = %Q{send data to zabbix from ruby}
+    gem.description = %Q{send data to zabbix from ruby}
+    gem.email       = "mknopp@yammer-inc.com"
+    gem.homepage    = "http://github.com/mhat/zabbix"
+    gem.authors     = ["Matthew Knopp"]
+    gem.add_development_dependency "thoughtbot-shoulda", ">= 0"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs    << 'lib' << 'test'
+  test.pattern  = 'test/**/test_*.rb'
+  test.verbose  = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs    << 'test'
+    test.pattern  = 'test/**/test_*.rb'
+    test.verbose  = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version       = File.exist?('VERSION') ? File.read('VERSION') : ""
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title    = "zabbix #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/lib/zabbix.rb

@@ -0,0 +1,31 @@
+module Zabbix
+end
+
+require "zabbix/agent/configuration"
+require "zabbix/sender"
+require "zabbix/sender/easy"
+
+## we could use zabbix-sender, but that appears to be unecessary; we can instead 
+## simply talk to the zabbit-server. we get there via the parent. 
+## 
+## thinking about what we want to be able to do there are a few types of messages
+## we want to send to zabbix:
+##   1. that a process has starting 
+##   2. that a process is running 
+##   3. that a process has finished 
+## 
+## there will be more in the future, but this is the minimum; it would be nice 
+## to expose two interfaces: 
+
+## Zabbix::Sender.run(:task_name, opts={}) do |zbx|
+##   ... 
+##   zbx.heartbeat
+## end
+
+## zbx = Zabbit::Zender::Easy.new(opts={})
+## zbx.send_start
+## zbx.send_heartbeat
+## zbx.send_end
+
+## That's pretty much it. 
+

data/lib/zabbix/agent/configuration.rb

@@ -0,0 +1,89 @@
+module Zabbix::Agent
+  class Configuration
+    def initialize(config={})
+      @config = config
+    end
+
+    def server
+      @config['Server']
+    end
+
+    def server_port
+      @config['ServerPort']
+    end
+
+    def listen_port
+      @config['ListenPort']
+    end
+
+    def listen_ip
+      @config['ListenIP']
+    end
+
+    def source_ip
+      @config['SourceIP']
+    end
+
+    def start_agents
+      @config['StartAgents']
+    end
+
+    def refresh_active_checks
+      @config['RefreshActiveChecks']
+    end
+
+    def disable_active
+      @config['DisableActive']
+    end
+
+    def enable_remote_commands
+      @config['EnableRemoteCommands']
+    end
+
+    def debug_level
+      @config['DebugLevel']
+    end
+
+    def pid_file
+      @config['PidFile']
+    end
+
+    def log_file
+      @config['LogFile']
+    end
+
+    def timeout
+      @config['Timeout']
+    end
+
+    def arbitrary(key)
+      @config[key]
+    end 
+
+    def self.read(zabbix_conf_file=nil)
+      zabbix_conf_file ||= "/etc/zabbix/zabbix-agentd.conf"
+      zabbix_conf        = {} 
+
+      File.open(zabbix_conf_file).each do |line|
+        ## skip comments
+        next if line =~ /^(\s+)?#/ 
+
+        ## strip tail comments
+        line.gsub!(/#.*/, '')
+
+        ## zabbix splits on equals 
+        key, value = line.split("=", 2)
+        key.chomp!
+        value.chomp!
+
+        ## zabbix keys look like strings
+        next unless key  =~ /[A-Za-z0-9]+/
+
+        ## cool
+        zabbix_conf[key] = value
+      end
+
+      Configuration.new(zabbix_conf)
+    end
+  end
+end

data/lib/zabbix/sender.rb

@@ -0,0 +1,82 @@
+require 'socket'
+require 'yajl'
+
+class Zabbix::Sender
+
+  DEFAULT_SERVER_PORT = 10051
+
+  attr_reader :configured 
+
+  def initialize(opts={})
+    if opts[:config_file]
+      config = Zabbix::Agent::Configuration.read(opts[:config_file])
+      @host  = config.server
+      @port  = config.server_port || DEFAULT_SERVER_PORT
+    end
+
+    if opts[:host]
+      @host = opts[:host]
+      @port = opts[:port] || DEFAULT_SERVER_PORT
+    end
+
+  end
+
+
+  def configured?
+    @configured ||= !(@host.nil? and @port.nil?)
+  end
+
+
+  def connect
+    TCPSocket.new(@host, @port)
+  end
+
+
+  def send_start(key, opts={})
+    send_data("#{key}.start", 1, opts)
+  end
+
+
+  def send_stop(key, opts={}) 
+    send_data("#{key}.stop", 1, opts)
+  end
+
+
+  def send_heartbeat(key, msg="", opts={})
+    send_data("#{key}.heartbeat", msg, opts)
+  end
+
+  def send_data(key, value, opts={})
+    return false unless configured? 
+    host   = opts[:host] || Socket.gethostname
+    ts     = opts[:ts]   || Time.now.to_i 
+    socket = connect 
+
+    # construct json
+    json = Yajl::Encoder.encode({ 
+      :request => "agent data",
+      :clock   => ts,
+      :data    => [{
+        :host       => host,
+        :key        => key,
+        :value      => value,
+        :clock      => ts
+      }]
+    })
+
+    # send the data
+    socket.write "ZBXD\x01"
+    socket.write [json.size].pack('q')
+    socket.write json
+    socket.flush
+
+    # read the response message if desired
+    #header = socket.read(5)
+    #len    = socket.read(8)
+    #len    = len.unpack('q').shift
+    #msg    = socket.read(len)
+    socket.close 
+
+    return true
+  end  
+end

data/lib/zabbix/sender/easy.rb

@@ -0,0 +1,8 @@
+class Zabbix::Sender::Easy
+  def self.run (task, opts={}, &block)
+    zbx = Zabbix::Sender.new(opts)
+    zbx.send_start 
+    block.call(zbx)
+    zbx.send_stop 
+  end
+end

data/test/helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'zabbix'
+
+class Test::Unit::TestCase
+end

data/test/test_zabbix.rb

@@ -0,0 +1,32 @@
+require 'helper'
+
+class TestZabbix < Test::Unit::TestCase
+
+  should "not be configured" do 
+    zbx = Zabbix::Sender.new 
+    assert_equal false, zbx.configured?
+  end
+
+  should "be configured by args" do
+    zbx = Zabbix::Sender.new :host => 'localhost'
+    assert_equal true, zbx.configured?
+  end
+ 
+  should "be configured by zabbix_agentd.conf" do
+    config_file = "#{File.dirname(__FILE__)}/zabbix_agentd.conf"
+    zbx = Zabbix::Sender.new :config_file => config_file
+    assert_equal true, zbx.configured?
+  end 
+
+  should "send some data" do
+    config_file = "#{File.dirname(__FILE__)}/zabbix_agentd.conf"
+    zbx  = Zabbix::Sender.new :config_file => config_file
+    key  = :"foo.bar.baz"
+    host = "testhost.example.com"
+    assert_equal true, zbx.send_start(key, :host => host)
+    assert_equal true, zbx.send_heartbeat(key, "8==D~~~ ADAM", :host => host)
+    assert_equal true, zbx.send_stop(key, :host => host)
+  end
+
+  
+end

data/test/zabbix_agentd.conf

@@ -0,0 +1,2 @@
+Server=localhost
+ServerPort=10051

metadata

@@ -0,0 +1,94 @@
+--- !ruby/object:Gem::Specification 
+name: zabbix
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Matthew Knopp
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-11-30 00:00:00 -08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thoughtbot-shoulda
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id001
+description: send data to zabbix from ruby
+email: mknopp@yammer-inc.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- lib/zabbix.rb
+- lib/zabbix/agent/configuration.rb
+- lib/zabbix/sender.rb
+- lib/zabbix/sender/easy.rb
+- test/helper.rb
+- test/test_zabbix.rb
+- test/zabbix_agentd.conf
+has_rdoc: true
+homepage: http://github.com/mhat/zabbix
+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 
+      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: 3
+summary: send data to zabbix from ruby
+test_files: 
+- test/helper.rb
+- test/test_zabbix.rb