zabbix-rails 0.1.9 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aa818a8ba839cf3da0d79d2bd76a5bce34c9b153408320c57e87aa8c62c28700
-  data.tar.gz: bf33db9b88d475e5ebf71caa669efe7be39a5976a0f1a844978a157181a89a58
+  metadata.gz: d72e0a2099b3dd87de6d9a38ec3514bc2e8f6fa7720804a202457a278af39987
+  data.tar.gz: d92f35f0c3497663edc9f50f5a6fbf4815bde298fb50fd2eb2921ffa367077ed
 SHA512:
-  metadata.gz: 20114878297c5b7113f204881462c466b45e8512e500f5e3e6ede5a56ddc0869872bfdf7446142948133467c3a53e021f0a2b5a4d0d5aaf06a3e2574940448e4
-  data.tar.gz: 1d3a1613710a430c2485e2354776df1e6cf8121ce203241a980c51ecd24c7363c3cb92d78d635ecfe83531927c4a02ea5f90e1eefb7147d674cc2d71b1d5b16d
+  metadata.gz: 239955a4196412900aad1f05e4b455274355960d3643ef1fc6572f16b16b158f7ec7b856aa069046c857774c284eaf7ef9d2b6a8f60e9727b2a4bbef1dd674c9
+  data.tar.gz: 6d3b8303de360e70c3c590faa7ed39b3b22fab9a198fd552dbe150ebf5a4a4eec090110c8b199db4c16409d54ffdbd7d143aebb4d1e372ae671435e4621b856e

data/lib/zabbix/chart.rb

@@ -0,0 +1,69 @@
+require "active_support/concern"
+
+module Zabbix
+  extend ActiveSupport::Concern
+
+  module Chart
+    # 请求生成 graph_chart 对象
+    def graph_chart(graphid, height = 400, width = 900, start_at = "now-6h", end_at = "now")
+      # 构造数据结构
+      data = {
+        from:    start_at,
+        to:      end_at,
+        graphid: graphid,
+        height:  height,
+        width:   width
+      }
+      # 请求后端返回图片对象
+      _item_chart("/chart2.php", data)
+    end
+
+    # 请求生成 item_chart 对象
+    def item_chart(itemid, height = 400, width = 900, start_at = "now-6h", end_at = "now")
+      # 构造数据结构
+      data = {
+        from:    start_at,
+        to:      end_at,
+        itemids: [itemid.to_i],
+        height:  height,
+        width:   width
+      }
+      # 请求后端返回图片对象
+      _item_chart(data)
+    end
+
+    # 请求后端返回图形
+    def _item_chart(url = "/chart.php", data)
+      # 请求接口
+      @conn.get "#{url}" do |r|
+        r.params.merge!(data)
+        r.headers["Host"]   = Config.url.gsub!(%r{http(s)?://}, "")
+        r.headers["Accept"] = "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9"
+        r.headers["Cookie"] = zabbix_token
+      end.body
+    end
+
+    # 保存图片并返回文件名
+    def save_chart_graph(itemid, height = 400, width = 900, start_at = "now-6h", end_at = "now")
+      filename = generate_name(itemid)
+      save_file(filename, item_chart(itemid, height, width, start_at, end_at))
+      "chart/#{File.basename(filename)}"
+    end
+
+    # 保存图片
+    def save_file(filename, content)
+      # 生成绝对路径地址
+      filename.prepend("app/assets/images/chart/")
+
+      # 存储数据
+      File.open(filename, "wb") do |chart|
+        chart.write(content)
+      end
+    end
+
+    def generate_name(itemid)
+      require "uuidtools"
+      "#{UUIDTools::UUID.timestamp_create}-#{itemid}.png"
+    end
+  end
+end

data/lib/zabbix/connector.rb

@@ -10,7 +10,7 @@ module Zabbix
     # 将鉴权后的 zabbix-rails 对象缓存到 Rails，减少不必要的认证动作加速执行
     # https://guides.rubyonrails.org/caching_with_rails.html
     def zabbix_connector
-      Rails.cache.fetch("zabbix_connector", expires_in: 4.hours) do
+      Rails.cache.fetch("zabbix_connector", expires_in: 2.hours) do
         Rails.logger.warn("Zabbix 正在请求 API 鉴权")
         ZabbixManager.connect(url: Config.url, user: Config.user, password: Config.password, debug: Config.debug)
       rescue => e
@@ -18,6 +18,14 @@ module Zabbix
       end
     end
 
+    # 获取 zabbix 登录凭证缓存
+    def zabbix_token
+      Rails.cache.fetch("zabbix_token", expires_in: 2.hours) do
+        Rails.logger.warn("正在请求 zabbix/index.php 登录认证")
+        authenticate_with_cookie
+      end
+    end
+
     # 删除主机监控
     def delete_host_monitor(name)
       id = zabbix_connector.hosts.get_host_id(name)
@@ -43,4 +51,27 @@ module Zabbix
       Rails.logger.warn("调用 delete_trigger_monitor 接口异常，原始报错信息: #{e}")
     end
   end
+
+  # 请求 zabbix 后端认证，返回 Cookie
+  def authenticate_with_cookie
+    # 初始化会话
+    base_url = Config.url
+    @conn    = Faraday::Connection.new(base_url)
+
+    # 登录权限凭证
+    data = {
+      name:      Config.user,
+      password:  Config.password,
+      autologin: 1,
+      enter:     "Sign in",
+    }
+
+    # 请求认证
+    ret = @conn.post "/index.php" do |r|
+      r.headers["Host"] = base_url.gsub!(%r{http(s)?://}, "")
+      r.body            = data
+    end
+
+    ret.headers["set-cookie"].presence || nil
+  end
 end

data/lib/zabbix/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Zabbix
-  VERSION = "0.1.9"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: zabbix-rails
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.2.0
 platform: ruby
 authors:
 - WENWU.YAN
@@ -35,6 +35,7 @@ files:
 - Rakefile
 - lib/tasks/zabbix_tasks.rake
 - lib/zabbix-rails.rb
+- lib/zabbix/chart.rb
 - lib/zabbix/config.rb
 - lib/zabbix/connector.rb
 - lib/zabbix/dns_monitor.rb