zabbix-rails 0.1.9 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aa818a8ba839cf3da0d79d2bd76a5bce34c9b153408320c57e87aa8c62c28700
-  data.tar.gz: bf33db9b88d475e5ebf71caa669efe7be39a5976a0f1a844978a157181a89a58
+  metadata.gz: e91a0b65f3d8c11a21c801e896756b8b33c2a3ac0cbcdb5aaadc51ff77347a3f
+  data.tar.gz: 0cb4b629c690749a26f01ff2d6c3dad7ecb1aa78b17882e38b119333de361803
 SHA512:
-  metadata.gz: 20114878297c5b7113f204881462c466b45e8512e500f5e3e6ede5a56ddc0869872bfdf7446142948133467c3a53e021f0a2b5a4d0d5aaf06a3e2574940448e4
-  data.tar.gz: 1d3a1613710a430c2485e2354776df1e6cf8121ce203241a980c51ecd24c7363c3cb92d78d635ecfe83531927c4a02ea5f90e1eefb7147d674cc2d71b1d5b16d
+  metadata.gz: dab95a7df10f9833babcbe70aea7077df0c3b57bdc4ac3479a643d4c7450e16ebb80e216390fcf35c3c8a0b7cbdc05706634e469c43b0871b714fda6b7483283
+  data.tar.gz: 357db5c2018f7edf22d3066712fd47159eefc8a272474cddf4c5a6ef0fa4ea7cc596db6ce775f5c1b3b585c2aae5306eadd9c319e1da4746bf457805e5f4bb29

data/Rakefile

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "bundler/setup"
 
 require "bundler/gem_tasks"

data/lib/zabbix/chart.rb

@@ -0,0 +1,77 @@
+# frozen_string_literal: true
+
+require "active_support/concern"
+
+module Zabbix
+  module Chart
+    extend ActiveSupport::Concern
+    include Zabbix::Connector
+
+    # 请求生成 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)
+      base_url = Zabbix.url
+      host     = URI(base_url).host
+      conn     = Faraday::Connection.new(base_url)
+      # 请求接口
+      conn.get "#{url}" do |r|
+        r.params.merge!(data)
+        r.headers["Host"]   = host
+        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))
+      "/assets/chart/#{File.basename(filename)}"
+    end
+
+    # 保存图片
+    def save_file(filename, content)
+      directory_base = "app/assets/images/chart/"
+      # 生成绝对路径地址
+      FileUtils.mkdir_p("#{Rails.root}/#{directory_base}") unless Dir.exist?("#{Rails.root}/#{directory_base}")
+      filename.prepend(directory_base)
+
+      # 存储数据
+      File.open(filename, "wb") do |chart|
+        chart.write(content)
+      end
+    end
+
+    # 生成随机名称
+    def generate_name(itemid)
+      "#{SecureRandom.hex(8)}-#{itemid}.png"
+    end
+  end
+end

data/lib/zabbix/connector.rb

@@ -1,23 +1,32 @@
 # frozen_string_literal: true
 
 require "active_support/concern"
+require "faraday"
 
 module Zabbix
-  extend ::ActiveSupport::Concern
-
   module Connector
+    extend ::ActiveSupport::Concern
+
     # 开发测试环境需要单独开启缓存功能：bin/rails dev:cache
     # 将鉴权后的 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: 30.minutes) do
         Rails.logger.warn("Zabbix 正在请求 API 鉴权")
-        ZabbixManager.connect(url: Config.url, user: Config.user, password: Config.password, debug: Config.debug)
+        ZabbixManager.connect(url: Zabbix.url, user: Zabbix.user, password: Zabbix.password, debug: Zabbix.debug)
       rescue => e
         Rails.logger.warn("登录 ZABBIX 异常，请检查网络或登录凭证。原始报错信息: #{e}")
       end
     end
 
+    # 获取 zabbix 登录凭证缓存
+    def zabbix_token
+      Rails.cache.fetch("zabbix_token", expires_in: 30.minutes) 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)
@@ -42,5 +51,29 @@ module Zabbix
     rescue => e
       Rails.logger.warn("调用 delete_trigger_monitor 接口异常，原始报错信息: #{e}")
     end
+
+    # 请求 zabbix 后端认证，返回 Cookie
+    def authenticate_with_cookie
+      # 初始化会话
+      base_url = Zabbix.url
+      host     = URI(base_url).host
+      conn     = Faraday::Connection.new(base_url)
+
+      # 登录权限凭证
+      data = {
+        name:      Zabbix.user,
+        password:  Zabbix.password,
+        autologin: 1,
+        enter:     "Sign in",
+      }
+
+      # 请求认证
+      ret = conn.post "/index.php" do |r|
+        r.headers["Host"] = host
+        r.body            = data
+      end
+
+      ret.headers["set-cookie"].presence || nil
+    end
   end
 end

data/lib/zabbix/dns_monitor.rb

@@ -3,9 +3,9 @@
 require "active_support/concern"
 
 module Zabbix
-  extend ::ActiveSupport::Concern
-
   module DnsMonitor
+    extend ::ActiveSupport::Concern
+
     include Zabbix::Connector
 
     # 生成 DNS 触发器表达式

data/lib/zabbix/host_monitor.rb

@@ -3,9 +3,9 @@
 require "active_support/concern"
 
 module Zabbix
-  extend ::ActiveSupport::Concern
-
   module HostMonitor
+    extend ::ActiveSupport::Concern
+
     include Zabbix::Connector
 
     # 创建或更新 HostMonitor主机监控对象

data/lib/zabbix/item_trigger.rb

@@ -3,9 +3,9 @@
 require "active_support/concern"
 
 module Zabbix
-  extend ::ActiveSupport::Concern
-
   module ItemTrigger
+    extend ::ActiveSupport::Concern
+
     include Zabbix::Connector
 
     # 创建或新增触发器，返回触发器ID | 触发器的 DATA_STRUCTURE

data/lib/zabbix/version.rb

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

data/lib/zabbix-rails.rb

@@ -1,11 +1,12 @@
 # frozen_string_literal: true
 
+# 注意加载顺序
 require "zabbix/version"
-require "zabbix/config"
 require "zabbix/connector"
 require "zabbix/host_monitor"
 require "zabbix/item_trigger"
 require "zabbix/dns_monitor"
+require "zabbix/chart"
 
 # 加载外部依赖
 require "active_support/concern"
@@ -17,8 +18,8 @@ module Zabbix
   class << self
     attr_accessor :url, :user, :password, :debug
 
-    def configure(&block)
-      Config.configure(&block)
+    def config
+      yield self
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: zabbix-rails
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.2.3
 platform: ruby
 authors:
 - WENWU.YAN
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-03-26 00:00:00.000000000 Z
+date: 2022-04-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -35,7 +35,7 @@ files:
 - Rakefile
 - lib/tasks/zabbix_tasks.rake
 - lib/zabbix-rails.rb
-- lib/zabbix/config.rb
+- lib/zabbix/chart.rb
 - lib/zabbix/connector.rb
 - lib/zabbix/dns_monitor.rb
 - lib/zabbix/engine.rb

data/lib/zabbix/config.rb

@@ -1,13 +0,0 @@
-# frozen_string_literal: true
-
-module Zabbix
-  class Config
-    class << self
-      attr_accessor :url, :user, :password, :debug
-
-      def configure
-        yield self
-      end
-    end
-  end
-end