brainzlab 0.1.2 → 0.1.3

data/lib/brainzlab/sentinel/client.rb

@@ -1,15 +1,15 @@
 # frozen_string_literal: true
 
-require "net/http"
-require "json"
-require "uri"
+require 'net/http'
+require 'json'
+require 'uri'
 
 module BrainzLab
   module Sentinel
     class Client
       def initialize(config)
         @config = config
-        @base_url = config.sentinel_url || "https://sentinel.brainzlab.ai"
+        @base_url = config.sentinel_url || 'https://sentinel.brainzlab.ai'
       end
 
       # List all registered hosts
@@ -17,14 +17,14 @@ module BrainzLab
         params = { page: page, per_page: per_page }
         params[:status] = status if status
 
-        response = request(:get, "/api/v1/hosts", params: params)
+        response = request(:get, '/api/v1/hosts', params: params)
 
         return [] unless response.is_a?(Net::HTTPSuccess)
 
         data = JSON.parse(response.body, symbolize_names: true)
         data[:hosts] || []
       rescue StandardError => e
-        log_error("list_hosts", e)
+        log_error('list_hosts', e)
         []
       end
 
@@ -36,7 +36,7 @@ module BrainzLab
 
         JSON.parse(response.body, symbolize_names: true)
       rescue StandardError => e
-        log_error("get_host", e)
+        log_error('get_host', e)
         nil
       end
 
@@ -44,9 +44,9 @@ module BrainzLab
       # @param host_id [String] Host ID
       # @param period [String] Time period (1h, 6h, 24h, 7d, 30d)
       # @param metrics [Array<String>] Specific metrics to fetch (cpu, memory, disk, network)
-      def get_metrics(host_id, period: "1h", metrics: nil)
+      def get_metrics(host_id, period: '1h', metrics: nil)
         params = { period: period }
-        params[:metrics] = metrics.join(",") if metrics
+        params[:metrics] = metrics.join(',') if metrics
 
         response = request(:get, "/api/v1/hosts/#{host_id}/metrics", params: params)
 
@@ -54,12 +54,12 @@ module BrainzLab
 
         JSON.parse(response.body, symbolize_names: true)
       rescue StandardError => e
-        log_error("get_metrics", e)
+        log_error('get_metrics', e)
         nil
       end
 
       # Get top processes for a host
-      def get_processes(host_id, sort_by: "cpu", limit: 20)
+      def get_processes(host_id, sort_by: 'cpu', limit: 20)
         params = { sort_by: sort_by, limit: limit }
 
         response = request(:get, "/api/v1/hosts/#{host_id}/processes", params: params)
@@ -69,7 +69,7 @@ module BrainzLab
         data = JSON.parse(response.body, symbolize_names: true)
         data[:processes] || []
       rescue StandardError => e
-        log_error("get_processes", e)
+        log_error('get_processes', e)
         []
       end
 
@@ -79,14 +79,14 @@ module BrainzLab
         params[:host_id] = host_id if host_id
         params[:status] = status if status
 
-        response = request(:get, "/api/v1/containers", params: params)
+        response = request(:get, '/api/v1/containers', params: params)
 
         return [] unless response.is_a?(Net::HTTPSuccess)
 
         data = JSON.parse(response.body, symbolize_names: true)
         data[:containers] || []
       rescue StandardError => e
-        log_error("list_containers", e)
+        log_error('list_containers', e)
         []
       end
 
@@ -98,12 +98,12 @@ module BrainzLab
 
         JSON.parse(response.body, symbolize_names: true)
       rescue StandardError => e
-        log_error("get_container", e)
+        log_error('get_container', e)
         nil
       end
 
       # Get container metrics
-      def get_container_metrics(container_id, period: "1h")
+      def get_container_metrics(container_id, period: '1h')
         params = { period: period }
 
         response = request(:get, "/api/v1/containers/#{container_id}/metrics", params: params)
@@ -112,7 +112,7 @@ module BrainzLab
 
         JSON.parse(response.body, symbolize_names: true)
       rescue StandardError => e
-        log_error("get_container_metrics", e)
+        log_error('get_container_metrics', e)
         nil
       end
 
@@ -123,14 +123,14 @@ module BrainzLab
         params[:status] = status if status
         params[:severity] = severity if severity
 
-        response = request(:get, "/api/v1/alerts", params: params)
+        response = request(:get, '/api/v1/alerts', params: params)
 
         return [] unless response.is_a?(Net::HTTPSuccess)
 
         data = JSON.parse(response.body, symbolize_names: true)
         data[:alerts] || []
       rescue StandardError => e
-        log_error("get_alerts", e)
+        log_error('get_alerts', e)
         []
       end
 
@@ -138,7 +138,7 @@ module BrainzLab
       def report_metrics(host_id:, metrics:, timestamp: nil)
         response = request(
           :post,
-          "/internal/agent/report",
+          '/internal/agent/report',
           body: {
             host_id: host_id,
             metrics: metrics,
@@ -149,21 +149,21 @@ module BrainzLab
 
         response.is_a?(Net::HTTPSuccess) || response.is_a?(Net::HTTPAccepted)
       rescue StandardError => e
-        log_error("report_metrics", e)
+        log_error('report_metrics', e)
         false
       end
 
       def provision(project_id:, app_name:)
         response = request(
           :post,
-          "/api/v1/projects/provision",
+          '/api/v1/projects/provision',
           body: { project_id: project_id, app_name: app_name },
           use_service_key: true
         )
 
         response.is_a?(Net::HTTPSuccess) || response.is_a?(Net::HTTPCreated)
       rescue StandardError => e
-        log_error("provision", e)
+        log_error('provision', e)
         false
       end
 
@@ -172,36 +172,34 @@ module BrainzLab
       def request(method, path, headers: {}, body: nil, params: nil, use_service_key: false, use_agent_key: false)
         uri = URI.parse("#{@base_url}#{path}")
 
-        if params
-          uri.query = URI.encode_www_form(params)
-        end
+        uri.query = URI.encode_www_form(params) if params
 
         http = Net::HTTP.new(uri.host, uri.port)
-        http.use_ssl = uri.scheme == "https"
+        http.use_ssl = uri.scheme == 'https'
         http.open_timeout = 10
         http.read_timeout = 30
 
         request = case method
-        when :get
-          Net::HTTP::Get.new(uri)
-        when :post
-          Net::HTTP::Post.new(uri)
-        when :put
-          Net::HTTP::Put.new(uri)
-        when :delete
-          Net::HTTP::Delete.new(uri)
-        end
-
-        request["Content-Type"] = "application/json"
-        request["Accept"] = "application/json"
+                  when :get
+                    Net::HTTP::Get.new(uri)
+                  when :post
+                    Net::HTTP::Post.new(uri)
+                  when :put
+                    Net::HTTP::Put.new(uri)
+                  when :delete
+                    Net::HTTP::Delete.new(uri)
+                  end
+
+        request['Content-Type'] = 'application/json'
+        request['Accept'] = 'application/json'
 
         if use_service_key
-          request["X-Service-Key"] = @config.sentinel_master_key || @config.secret_key
+          request['X-Service-Key'] = @config.sentinel_master_key || @config.secret_key
         elsif use_agent_key
-          request["X-Agent-Key"] = @config.sentinel_agent_key || @config.sentinel_api_key || @config.secret_key
+          request['X-Agent-Key'] = @config.sentinel_agent_key || @config.sentinel_api_key || @config.secret_key
         else
           auth_key = @config.sentinel_api_key || @config.secret_key
-          request["Authorization"] = "Bearer #{auth_key}" if auth_key
+          request['Authorization'] = "Bearer #{auth_key}" if auth_key
         end
 
         headers.each { |k, v| request[k] = v }

data/lib/brainzlab/sentinel/provisioner.rb

@@ -37,7 +37,7 @@ module BrainzLab
       end
 
       def detect_project_id
-        ENV["BRAINZLAB_PROJECT_ID"]
+        ENV.fetch('BRAINZLAB_PROJECT_ID', nil)
       end
     end
   end

data/lib/brainzlab/sentinel.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative "sentinel/client"
-require_relative "sentinel/provisioner"
+require_relative 'sentinel/client'
+require_relative 'sentinel/provisioner'
 
 module BrainzLab
   module Sentinel
@@ -45,7 +45,7 @@ module BrainzLab
       # @example
       #   metrics = BrainzLab::Sentinel.metrics(host_id, period: "24h", metrics: ["cpu", "memory"])
       #
-      def metrics(host_id, period: "1h", metrics: nil)
+      def metrics(host_id, period: '1h', metrics: nil)
         return nil unless enabled?
 
         ensure_provisioned!
@@ -63,7 +63,7 @@ module BrainzLab
       # @example
       #   procs = BrainzLab::Sentinel.processes(host_id, sort_by: "memory", limit: 10)
       #
-      def processes(host_id, sort_by: "cpu", limit: 20)
+      def processes(host_id, sort_by: 'cpu', limit: 20)
         return [] unless enabled?
 
         ensure_provisioned!
@@ -105,7 +105,7 @@ module BrainzLab
       # @param container_id [String] Container ID
       # @param period [String] Time period
       # @return [Hash, nil] Container metrics
-      def container_metrics(container_id, period: "1h")
+      def container_metrics(container_id, period: '1h')
         return nil unless enabled?
 
         ensure_provisioned!

data/lib/brainzlab/signal/client.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require "net/http"
-require "json"
-require "uri"
+require 'net/http'
+require 'json'
+require 'uri'
 
 module BrainzLab
   module Signal
@@ -12,15 +12,15 @@ module BrainzLab
       end
 
       def send_alert(alert)
-        post("/api/v1/alerts", alert)
+        post('/api/v1/alerts', alert)
       end
 
       def send_notification(notification)
-        post("/api/v1/notifications", notification)
+        post('/api/v1/notifications', notification)
       end
 
       def trigger_rule(payload)
-        post("/api/v1/rules/trigger", payload)
+        post('/api/v1/rules/trigger', payload)
       end
 
       private
@@ -28,24 +28,22 @@ module BrainzLab
       def post(path, body)
         uri = URI.parse("#{base_url}#{path}")
         http = Net::HTTP.new(uri.host, uri.port)
-        http.use_ssl = uri.scheme == "https"
+        http.use_ssl = uri.scheme == 'https'
         http.open_timeout = 5
         http.read_timeout = 10
 
         request = Net::HTTP::Post.new(uri.path)
-        request["Content-Type"] = "application/json"
-        request["Authorization"] = "Bearer #{api_key}"
-        request["User-Agent"] = "brainzlab-sdk/#{BrainzLab::VERSION}"
+        request['Content-Type'] = 'application/json'
+        request['Authorization'] = "Bearer #{api_key}"
+        request['User-Agent'] = "brainzlab-sdk/#{BrainzLab::VERSION}"
         request.body = body.to_json
 
         response = http.request(request)
 
-        unless response.is_a?(Net::HTTPSuccess)
-          BrainzLab.debug_log("[Signal] Request failed: #{response.code} - #{response.body}")
-        end
+        BrainzLab.debug_log("[Signal] Request failed: #{response.code} - #{response.body}") unless response.is_a?(Net::HTTPSuccess)
 
         response
-      rescue => e
+      rescue StandardError => e
         BrainzLab.debug_log("[Signal] Request error: #{e.message}")
         nil
       end

data/lib/brainzlab/signal/provisioner.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require "net/http"
-require "json"
-require "uri"
+require 'net/http'
+require 'json'
+require 'uri'
 
 module BrainzLab
   module Signal
@@ -16,7 +16,7 @@ module BrainzLab
         return unless @config.signal_url && !@config.signal_url.to_s.empty?
         return unless @config.secret_key && !@config.secret_key.to_s.empty?
 
-        BrainzLab.debug_log("[Signal] Auto-provisioning project...")
+        BrainzLab.debug_log('[Signal] Auto-provisioning project...')
         provision_project
       end
 
@@ -25,16 +25,16 @@ module BrainzLab
       def provision_project
         uri = URI.parse("#{@config.signal_url}/api/v1/projects/provision")
         http = Net::HTTP.new(uri.host, uri.port)
-        http.use_ssl = uri.scheme == "https"
+        http.use_ssl = uri.scheme == 'https'
         http.open_timeout = 10
         http.read_timeout = 30
 
         request = Net::HTTP::Post.new(uri.path)
-        request["Content-Type"] = "application/json"
-        request["Authorization"] = "Bearer #{@config.secret_key}"
-        request["User-Agent"] = "brainzlab-sdk/#{BrainzLab::VERSION}"
+        request['Content-Type'] = 'application/json'
+        request['Authorization'] = "Bearer #{@config.secret_key}"
+        request['User-Agent'] = "brainzlab-sdk/#{BrainzLab::VERSION}"
         request.body = {
-          name: @config.service || "default",
+          name: @config.service || 'default',
           environment: @config.environment
         }.to_json
 
@@ -42,12 +42,12 @@ module BrainzLab
 
         if response.is_a?(Net::HTTPSuccess)
           data = JSON.parse(response.body)
-          @config.signal_api_key = data["api_key"]
-          BrainzLab.debug_log("[Signal] Project provisioned successfully")
+          @config.signal_api_key = data['api_key']
+          BrainzLab.debug_log('[Signal] Project provisioned successfully')
         else
           BrainzLab.debug_log("[Signal] Provisioning failed: #{response.code} - #{response.body}")
         end
-      rescue => e
+      rescue StandardError => e
         BrainzLab.debug_log("[Signal] Provisioning error: #{e.message}")
       end
     end

data/lib/brainzlab/signal.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative "signal/client"
-require_relative "signal/provisioner"
+require_relative 'signal/client'
+require_relative 'signal/provisioner'
 
 module BrainzLab
   module Signal
@@ -21,7 +21,7 @@ module BrainzLab
         return unless BrainzLab.configuration.signal_valid?
 
         payload = {
-          type: "alert",
+          type: 'alert',
           name: name,
           message: message,
           severity: severity.to_s,
@@ -50,7 +50,7 @@ module BrainzLab
 
         channels = Array(channel)
         payload = {
-          type: "notification",
+          type: 'notification',
           channels: channels,
           message: message,
           title: title,
@@ -73,7 +73,7 @@ module BrainzLab
         return unless BrainzLab.configuration.signal_valid?
 
         payload = {
-          type: "trigger",
+          type: 'trigger',
           rule: rule_name,
           context: context,
           timestamp: Time.now.utc.iso8601(3),
@@ -87,8 +87,8 @@ module BrainzLab
       # Send a test alert to verify configuration
       def test!
         alert(
-          "test_alert",
-          "This is a test alert from BrainzLab Signal SDK",
+          'test_alert',
+          'This is a test alert from BrainzLab Signal SDK',
           severity: :info,
           data: { test: true, sdk_version: BrainzLab::VERSION }
         )