healthier 0.1.5 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9b9085660c0bf98cddf325db77e9d7f7a655c34e4af17e0febcc2d0468d42a10
-  data.tar.gz: b4cbbda7e758b1375c07d3ff98efa31ac85a52a9ede9c67dd2a06a95c75f9103
+  metadata.gz: 851390d021df1149267018e9ec714cee68168c2f1f9b02c0998790c33da63005
+  data.tar.gz: d6468040657b6ef60a75d68650c054966e359c9e16e076104e15022daf0ac41c
 SHA512:
-  metadata.gz: 522eea93d714a4afe55e18f20be294674a3406915b31c858567471117aa6e9089d8a8269517efb6bd2b8c29a70d895109e1047a93c7ec68a739d856b8656d6a2
-  data.tar.gz: 65e6082188ace39b2013404301fa137596b501d0f301fa0679697311038217642a08e29a7b5786fb406fdecf54016d90d49e6f652e90b89a9ae101b2db29448a
+  metadata.gz: c4d56fa78ad0f10d2d3cacf39277926636883acbba04c51f0bf85f5cc9888026f3ab4c50484c79a7d7d71e9a644c324ed8421e7a7c6ae5de407380fb114bf397
+  data.tar.gz: d53429a445fdc042ec7e73eb9a5f6fe80ec017314a85c287d42074e7fefca2fc67c0417e374effd23c858fa43685dd3916bded4522310ed8e9d58c7d5cca78f3

data/README.md

@@ -33,25 +33,37 @@ $ gem install healthier
 ```
 
 This will generate these two files:
+
 1. config/healthier.yml
+
 ```yaml
 ---
 healthier:
   depends_on:
-# Uncomment accordingly
-#    - name: 'postgresql'
-#    - name: 'mongodb'
-#    - name: 'redis'
-#    - name: 'rabbitmq'
+  # Uncomment accordingly
+  #    - name: 'postgresql'
+  #      config: {}
+  #    - name: 'mongodb'
+  #      config: {}
+  #    - name: 'redis'
+  #      config: {}
+  #    - name: 'rabbitmq'
+  #      config: {}
 ```
+NOTE: In future, config above will accept a hash each service's configuration
+
 2. config/initializers/healthier.rb
+
 ```ruby
 Healthier.setup do |config|
+  # If you don't specify auth_mechanism, then it defaults to http_basic.
+  # Currently, only http_basic and bearer_token are supported
+  config.auth_mechanism = 'bearer_token'
   config.depends_on = config.healthier.call
 end
 ```
 
-In your application routes.rb file:
+In your application routes.rb file(namespace according to your need):
 
 ```ruby
 mount Healthier::Engine => '/healthier'
@@ -64,6 +76,23 @@ USERNAME: 'demouser'
 PASSWORD: 'demouser@2023'
 ```
 
+ENVs for Bearer token authentication:
+
+```bash
+BEARER_TOKEN: 'e66cfadb3ce37a714fdf3df237dcfded9ed9a0f9ccb09e6a744d731e9214bae6fac7bf6d27cab3969702b7aa54c81fe1cff1842ac03b3d5f1a918e1f5061dcb80b80363976ae5f3b916c793fdfa5283ce5f387286152d62e2dcaa5c8c0f857ceb2cda045'
+```
+
+However, if you want to leave the API open, just set:
+
+```ruby
+  Healthier.auth_mechanism = :none
+```
+
+Example Curl:
+```curl
+ curl -H "Bearer b5f3396a876e4f2ec828d11514c94d95d2f2330d251496a78cbe0e9bc330d775" -H "Content-Type: application/json" 'https://migros.selise.dev/api/v2/cnc_rsp/healthier/ping'
+```
+
 ## Contributing
 
 Contribution directions go here.

data/app/controllers/healthier/application_controller.rb

@@ -3,5 +3,6 @@
 module Healthier
   # ApplicationController
   class ApplicationController < ActionController::Base
+    include Errors::ApiErrors
   end
 end

data/app/lib/errors/api_errors.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module Errors
+  module ApiErrors
+    def self.included(base)
+      base.class_eval do
+        rescue_from StandardError, with: :bad_request
+      end
+    end
+
+    def bad_request(error)
+      render_error(:bad_request, error: { message: error.message })
+    end
+
+    def render_success(status, options = {})
+      options[:success] = true
+      render json: options, status: status
+    end
+
+    def render_error(status, options = {})
+      options[:error] = { message: I18n.t("api_errors.#{status}") } if options[:error].blank?
+      options[:success] = false
+      Rollbar.error(options[:error])
+      render json: options, status: status
+    end
+
+    def render_silent_error(status, options = {})
+      options[:error] = { message: I18n.t("api_errors.#{status}") } if options[:error].blank?
+      options[:success] = false
+      render json: options, status: status
+    end
+  end
+end

data/lib/checkers/mongodb_checker.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module Checkers
+  # Mongodb health checker
+  class MongodbChecker
+    class << self
+      def check
+        raise 'Please make sure you have Mongodb installed' unless defined?(::Mongo::Client)
+
+        begin
+          @client = Mongo::Client.new(
+            ENV.fetch('MONGO_DB_URI', nil) || ['localhost:27017'],
+            max_retries: 2,
+            retry_interval: 1
+          )
+          @client.command(ping: 1)
+          true
+        rescue Mongo::Error::SocketError, Mongo::Error::NoServerAvailable
+          false
+        ensure
+          @client&.close
+        end
+      end
+    end
+  end
+end

data/lib/checkers/postgresql_checker.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module Checkers
+  # Postgres health checker
+  class PostgresqlChecker
+    class << self
+      def check
+        raise 'Please make sure you have Postgres installed' unless defined?(::ActiveRecord::Base)
+
+        begin
+          # @conn = ::PG.connect(dbname: ENV['DATABASE_NAME'])
+          # @conn.exec('SELECT 1')
+          @conn = ActiveRecord::Base.connection
+          true
+        rescue ::PG::ConnectionBad
+          false
+        ensure
+          @conn&.close
+        end
+      end
+    end
+  end
+end

data/lib/checkers/rabbitmq_checker.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Checkers
+  # Rabbitmq health checker
+  class RabbitmqChecker
+    class << self
+      def check
+        raise 'Please make sure you have RabbitMQ installed' unless defined?(::Bunny)
+
+        begin
+          @conn = ::Bunny.new
+          @conn.start
+          channel = @conn.create_channel
+          queue = channel.queue('health_check')
+          queue.publish('ping')
+          true
+        rescue ::Bunny::Exception, ::Bunny::TCPConnectionFailed, ::Bunny::PossibleAuthenticationFailureError
+          false
+        ensure
+          @conn&.close
+        end
+      end
+    end
+  end
+end

data/lib/checkers/redis_checker.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Checkers
+  # Redis health checker
+  class RedisChecker
+    class << self
+      def check
+        raise 'Please make sure you have Redis installed' unless defined?(::Redis)
+
+        begin
+          @redis = Redis.new
+          @redis.ping == 'PONG'
+        rescue Redis::CannotConnectError
+          false
+        ensure
+          @redis&.close
+        end
+      end
+    end
+  end
+end

data/lib/generators/healthier/config_generator.rb

@@ -10,6 +10,9 @@ module Healthier
         # frozen_string_literal: true
 
         Healthier.setup do |config|
+          # Currently, there are :basic_auth(default), :bearer_token for API authentication
+          # But if you want to leave the API open, then just set auth_mechanism to :none to skip authentication
+          # config.auth_mechanism = :bearer_token
           config.depends_on = config.healthier.call
         end
       RUBY
@@ -22,9 +25,13 @@ module Healthier
           # Uncomment accordingly
           depends_on:
         #    - name: 'postgresql'
+        #      config: {}
         #    - name: 'mongodb'
+        #      config: {}
         #    - name: 'redis'
+        #      config: {}
         #    - name: 'rabbitmq'
+        #      config: {}
       YAML
     end
   end

data/lib/healthier/api_authenticator.rb

@@ -3,25 +3,31 @@
 module Healthier
   # Engine
   class ApiAuthenticator
-    attr_accessor :controller
+    attr_accessor :controller, :request
 
     def initialize(controller)
       @controller = controller
+      @request = controller.request
     end
 
     def authenticate
+      return true if Healthier.auth_mechanism == :none
+
       send("authenticate_with_#{Healthier.auth_mechanism}")
     end
 
     def authenticate_with_bearer_token
+      tok = request.authorization.to_s
+      raise StandardError, 'Access Denied' unless tok.starts_with?('Bearer')
+
       controller.authenticate_with_http_token do |token, _options|
-        ENV['BEARER_TOKEN'] == token
+        raise StandardError, 'Invalid token' unless ENV['BEARER_TOKEN'] == token
       end
     end
 
     def authenticate_with_basic_auth
       controller.authenticate_or_request_with_http_basic do |username, password|
-        username == ENV['USERNAME'] && password == ENV['PASSWORD']
+        raise StandardError, 'Invalid credentials' unless username == ENV['USERNAME'] && password == ENV['PASSWORD']
       end
     end
   end

data/lib/healthier/engine.rb

@@ -1,35 +1,15 @@
 # frozen_string_literal: true
 
+require_relative '../checkers/mongodb_checker'
+require_relative '../checkers/postgresql_checker'
+require_relative '../checkers/rabbitmq_checker'
+require_relative '../checkers/redis_checker'
+
 module Healthier
   # Engine
   class Engine < ::Rails::Engine
     isolate_namespace Healthier
 
-    PING_CONF = HashWithIndifferentAccess.new(
-      {
-        postgresql: {
-          pinger: -> { ActiveRecord::Base.connection },
-          message: 'PostgreSQL service is not running. Please check your connection'
-        },
-        mongodb: {
-          pinger: -> { Mongoid.default_client.command(ping: 1) },
-          message: 'MongoDB service is not running. Please check your connection'
-        },
-        redis: {
-          pinger: -> { Redis.new },
-          message: 'Redis service is not running. Please check your connection'
-        },
-        rabbitmq: {
-          pinger: -> { Bunny.new },
-          message: 'RabbitMQ service is not running. Please check your connection'
-        }
-      }
-    )
-
-    # TODO: Refactor this whole code
-    SUCCESS = 200
-    FAILURE = 400
-
     def ping!
       conf['depends_on'].each_with_object({}) do |dependant, obj|
         obj[dependant['name']] = ping_it(dependant['name'])
@@ -41,9 +21,15 @@ module Healthier
     end
 
     def ping_it(service)
-      return { status: SUCCESS, message: 'pong' } if PING_CONF[service]['pinger'].call
+      checked = checker(service).check
+
+      return { status: Healthier::SUCCESS, message: 'pong' } if checked == true
+
+      { status: Healthier::FAILURE, message: "#{service.camelize} service is not running. Please check your connection" }
+    end
 
-      { status: FAILURE, message: PING_CONF[service]['message'] }
+    def checker(service)
+      "::Checkers::#{service.camelize}Checker".constantize
     end
   end
 end

data/lib/healthier/version.rb

@@ -2,5 +2,5 @@
 
 module Healthier
   # MAJOR.MINOR.PATCH
-  VERSION = '0.1.5'
+  VERSION = '0.1.7'
 end

data/lib/healthier.rb

@@ -10,6 +10,9 @@ module Healthier
     yield self
   end
 
+  SUCCESS = 200
+  FAILURE = 503
+
   mattr_accessor :healthier, default: -> { YAML.load_file(Rails.root.join('config/healthier.yml')) }
   mattr_accessor :auth_mechanism, default: 'basic_auth'
   mattr_accessor :depends_on, default: {}

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: healthier
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.7
 platform: ruby
 authors:
 - Nima Yonten
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-05-03 00:00:00.000000000 Z
+date: 2023-05-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -36,7 +36,12 @@ files:
 - Rakefile
 - app/controllers/healthier/application_controller.rb
 - app/controllers/healthier/checks_controller.rb
+- app/lib/errors/api_errors.rb
 - config/routes.rb
+- lib/checkers/mongodb_checker.rb
+- lib/checkers/postgresql_checker.rb
+- lib/checkers/rabbitmq_checker.rb
+- lib/checkers/redis_checker.rb
 - lib/generators/healthier/config_generator.rb
 - lib/healthier.rb
 - lib/healthier/api_authenticator.rb
@@ -51,6 +56,7 @@ metadata:
   homepage_uri: https://github.com/selisebt/healthier
   source_code_uri: https://github.com/selisebt/healthier
   changelog_uri: https://github.com/selisebt/healthier
+  rubygems_mfa_required: 'true'
 post_install_message:
 rdoc_options: []
 require_paths: