zabbix_graph 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0381d0fbc2dd4ef15d01c46370ae50ec9d8405f9
-  data.tar.gz: 28fc072e6e504120877680cd6e2fb27106c710f1
+  metadata.gz: c03d15e5d41ea5ddcd6169c0a4306711d539688e
+  data.tar.gz: d44f97c90120f97b536e4063dda69c3586457e3e
 SHA512:
-  metadata.gz: 506332bb3df617d3d618f0b0f482e785e92e1b39ae97dd97ef6990ace9bc3fca440fd9b2de96eef064d456ff0e0e40fac9f0fe6c6d64f75bdb5fb1e6c5cd12b1
-  data.tar.gz: b4132ae8cdc5d07872fcc8e66f4ff5f4ed3916c9ac48fd5ab91e974ad0bcf2e9755736b6cbde1a21c5f58e8d8e7f177880e79cefac9f6ab57a168c85be55ca14
+  metadata.gz: 8556bd51e3b65b5e67122896ac22bc3b8efcaa56d2f3fea8ee5afe3eb53d8f9b4c7f4c9364a2fe3546633d3a68c6fa24280c98b77460bd404603ca96048d70e8
+  data.tar.gz: 8b85e0e2cbc2797728e2302dcd49e090a2b1f7c1e5907b13e031a2d2705118dbc3bd8c7494ef39ee6dc0564b79e2a6ddbf167a08a4794423e14bb14454dc6835

data/README.md

@@ -1,26 +1,19 @@
 # ZabbixGraph
 
-TODO: Write a gem description
+Select hosts and items with peco, and open adhoc graph.
 
 ## Installation
 
-Add this line to your application's Gemfile:
-
-```ruby
-gem 'zabbix_graph'
-```
-
-And then execute:
-
-    $ bundle
-
-Or install it yourself as:
-
     $ gem install zabbix_graph
 
 ## Usage
 
-TODO: Write usage instructions here
+```
+$ export ZABBIX_URL=https://your-zabbix.example.com
+$ export ZABBIX_USER=...
+$ export ZABBIX_PASSWORD=...
+$ zabbix_graph
+```
 
 ## Contributing
 

data/lib/zabbix_graph/version.rb

@@ -1,3 +1,3 @@
 module ZabbixGraph
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/zabbix_graph.rb

@@ -1,18 +1,34 @@
 require "zabbix_graph/version"
 require "zabbixapi"
 require "peco_selector"
+require "optparse"
 
 module ZabbixGraph
   class CLI
     def self.start(argv)
-      Opener.new.select_and_open
+      Opener.new(parse_argv(argv)).select_and_open
+    end
+
+    def self.parse_argv(argv)
+      options = {}
+
+      parser = OptionParser.new
+      parser.on('--host-graph') { options[:host_graph] = true }
+      parser.on('--item-graph') { options[:item_graph] = true }
+      parser.parse!(argv)
+
+      options
     end
   end
 
   class Opener
+    def initialize(options)
+      @options = options
+      @all_hosts = zbx.hosts.get({})
+    end
+
     def select_and_open
-      hosts = zbx.hosts.get({})
-      hosts = PecoSelector.select_from(hosts.sort_by do |h|
+      hosts = PecoSelector.select_from(@all_hosts.sort_by do |h|
         h['host']
       end.map do |h|
         [h['host'], h]
@@ -37,17 +53,85 @@ module ZabbixGraph
         end
       end
 
-      query = [['action', 'batchgraph'], ['graphtype', '0']]
-      selected_items.each do |i|
-        query << ['itemids[]', i['itemid']]
+      if @options[:host_graph]
+        open_host_graph(selected_items)
+      elsif @options[:item_graph]
+        open_item_graph(selected_items)
+      else
+        open_history(selected_items)
       end
+    end
 
-      url = URI.join(zabbix_url, "/history.php?#{URI.encode_www_form(query)}")
+    private
+
+    def open_history(items)
+      url = URI.join(zabbix_url, "/history.php?#{query_string_from_items(items)}")
 
       system 'open', url.to_s
     end
 
-    private
+    def open_host_graph(items)
+      host_items = items.group_by do |i|
+        i['hostid']
+      end
+
+      grouped_items = host_items.map do |hostid, items|
+        host = @all_hosts.find {|h| h['hostid'] == hostid }
+        [host['name'], items]
+      end.sort_by do |hostname, _|
+        hostname
+      end
+        
+      open_grouped_items(grouped_items)
+    end
+
+    def open_item_graph(items)
+      key_items = items.group_by do |i|
+        [i['name'], i['key_']]
+      end
+
+      grouped_items = key_items.sort_by do |name_key, _|
+        name_key.join
+      end.map do |name_key, items|
+        ["#{name_key[0]} (#{name_key[1]})", items]
+      end
+
+      open_grouped_items(grouped_items)
+    end
+
+    def open_grouped_items(grouped_items)
+      html = ""
+      grouped_items.each do |name, items|
+        src = URI.join(zabbix_url, "/chart.php?#{query_string_from_items(items)}").to_s
+        html << "<div>"
+        html << "<h2>" << name << "</h2>"
+        html << %{<img src="#{src}">}
+        html << "</div>"
+      end
+
+      path = File.join(temp_html_dir, "#{Time.now.to_f.to_s}.html")
+      open(path, 'w') do |f|
+        f.write html
+      end
+
+      system 'open', path
+    end
+
+    def temp_html_dir
+      dir = "/tmp/zabbix_graph"
+      FileUtils.mkdir_p(dir)
+
+      dir
+    end
+
+    def query_string_from_items(items)
+      query = [['action', 'batchgraph'], ['graphtype', '0']]
+      items.each do |i|
+        query << ['itemids[]', i['itemid']]
+      end
+
+      URI.encode_www_form(query)
+    end
 
     def zbx
       @zbx ||= ZabbixApi.connect(

data/zabbix_graph.gemspec

@@ -18,7 +18,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_dependency "zabbixapi", "~> 2.4.2"
-  spec.add_dependency "peco_selector"
+  spec.add_dependency "peco_selector", "~> 0.0.3"
 
   spec.add_development_dependency "bundler", "~> 1.7"
   spec.add_development_dependency "rake", "~> 10.0"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: zabbix_graph
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Ryota Arai
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-03-10 00:00:00.000000000 Z
+date: 2015-03-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: zabbixapi
@@ -28,16 +28,16 @@ dependencies:
   name: peco_selector
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.0.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.0.3
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -103,7 +103,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 2.4.5
 signing_key: 
 specification_version: 4
 summary: Open Zabbix adhoc graph