console_kit 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8e392416a4b0f32756b184aefa7431d71ce45a733479e477f9c54f9e14affcd6
-  data.tar.gz: 3b15e7006400dc93794f84e7e14828f573639c67452c0ccb4922ece8fcab9c89
+  metadata.gz: acd47ead1008fc3bcdc25b0d995b735f69d1e6f5a8d5f8442c67dc5ee86bb9a3
+  data.tar.gz: 55bd88158a92c6741eab85bda17d4b61828dd101e94cd5ab02097688f8994302
 SHA512:
-  metadata.gz: c6f6b2d49d2c2e2d776af957764804dfbebb397a7fbb52449013a77bed92d1ff817672154d546a304bbc92c9c385f14e5402bee250046e8b001564169f52881d
-  data.tar.gz: e9c24b447c674eb693a05b9552484e293a483b547e04620e78145f9c865010c27b45f487ab16b453bce8bd9a7bb8c4c180c9464abef3cda20f7e5678fdc932bd
+  metadata.gz: 9d1214289e7859342493f2930557554e05ad3ccb3200e041d3ade9b2f95a710b6f111eaea9bfc40f0192e98cabd1010f7be6c71469066d2fd702a74d70a33132
+  data.tar.gz: 4564b76383af8ab6d60c6ee79644fc78a1be8633520ffe1c74a26a69936ebbacb7db8b20dd84c28c2433f532198c46c1c140ff7d31cd323d613559624fabb713

data/CHANGELOG.md

@@ -6,6 +6,22 @@ This project adheres to [Semantic Versioning](https://semver.org/).
 
 ---
 
+## [1.1.0] - 2026-03-14
+### Added
+- **Redis Connection Handler:** Automatic Redis DB selection per tenant via `Redis.current.select`, with graceful fallback for Redis v5+ where `Redis.current` is deprecated.
+- **Elasticsearch Connection Handler:** Sets a per-tenant Elasticsearch index name prefix via thread-local storage and `Elasticsearch::Model.index_name_prefix=` (when available).
+- **Console Helpers:** New `switch_tenant`, `tenant_info`, and `tenants` methods available in the Rails console for quick tenant management.
+- **Custom Console Prompt:** IRB and Pry prompts now display the active tenant name (e.g., `[acme] main:001>`).
+- **Tenant Banner:** On successful tenant initialization, a banner now shows the tenant name, environment safety warnings (production in red, staging in yellow), and a summary of active connections.
+- **Environment Safety Warnings:** Production and staging environments are flagged with color-coded warnings at tenant setup time.
+- New tenant configuration keys: `redis_db`, `elasticsearch_prefix`, and `environment`.
+
+### Changed
+- `TenantConfigurator` now manages `tenant_redis_db` and `tenant_elasticsearch_prefix` context attributes alongside existing ones.
+- Generator template updated with examples for the new configuration keys.
+
+---
+
 ## [1.0.0] - 2026-03-01
 ### Added
 - **Global Configuration Persistence:** ConsoleKit settings now persist across the entire session and across multiple threads.
@@ -83,6 +99,7 @@ This project adheres to [Semantic Versioning](https://semver.org/).
   - Tenant-specific database configuration.
   - Colorized console output for improved UX.
 
+[1.1.0]: https://github.com/Soumyadeep-ai/console_kit/releases/tag/v1.1.0
 [1.0.0]: https://github.com/Soumyadeep-ai/console_kit/releases/tag/v1.0.0
 [0.1.5]: https://github.com/Soumyadeep-ai/console_kit/releases/tag/v0.1.5
 [0.1.4]: https://github.com/Soumyadeep-ai/console_kit/releases/tag/v0.1.4

data/README.md

@@ -8,7 +8,7 @@
 
 A simple and flexible multi-tenant console setup toolkit for Rails applications.
 
-ConsoleKit helps you manage tenant-specific database connections and context configuration via an easy CLI interface and Rails integration.
+ConsoleKit helps you manage tenant-specific database connections (SQL, MongoDB, Redis, Elasticsearch) and context configuration via an easy CLI interface and Rails integration.
 
 ## Installation
 
@@ -50,10 +50,24 @@ Then, edit config/initializers/console_kit.rb to define your tenants and context
 ConsoleKit.configure do |config|
   config.tenants = {
     tenant_one: {
-      constants: { shard: :tenant_one_db, mongo_db: :tenant_one_mongo, partner_code: 'partnerA' }
+      constants: {
+        shard: :tenant_one_db,
+        mongo_db: :tenant_one_mongo,
+        partner_code: 'partnerA',
+        redis_db: 1,
+        elasticsearch_prefix: 'tenant_one',
+        environment: 'production'
+      }
     },
     tenant_two: {
-      constants: { shard: :tenant_two_db, mongo_db: :tenant_two_mongo, partner_code: 'partnerB' }
+      constants: {
+        shard: :tenant_two_db,
+        mongo_db: :tenant_two_mongo,
+        partner_code: 'partnerB',
+        redis_db: 2,
+        elasticsearch_prefix: 'tenant_two',
+        environment: 'staging'
+      }
     }
   }
 
@@ -64,39 +78,72 @@ ConsoleKit.configure do |config|
 end
 ```
 
+## Supported Connections
+
+ConsoleKit automatically detects and manages connections for:
+
+| Connection      | Gem Required    | Config Key               | Behavior                                        |
+|-----------------|-----------------|--------------------------|------------------------------------------------|
+| SQL (ActiveRecord) | `activerecord` | `shard`               | Calls `establish_connection` on your base class |
+| MongoDB         | `mongoid`       | `mongo_db`              | Calls `Mongoid.override_database`              |
+| Redis           | `redis`         | `redis_db`              | Calls `Redis.current.select(db)`               |
+| Elasticsearch   | `elasticsearch` | `elasticsearch_prefix`  | Sets `Elasticsearch::Model.index_name_prefix=` |
+
+Handlers are only activated when their corresponding gem is loaded.
+
 ## Console Usage
 
-When launching the Rails console, ConsoleKit will prompt you to select a tenant (if multiple tenants are configured).
+When launching the Rails console, ConsoleKit will prompt you to select a tenant (if multiple tenants are configured). On selection, a tenant banner is displayed showing the tenant name, environment safety warnings, and active connections.
 
 ### Selection Options:
 - **Number or Name:** Select a tenant by its index or name (case-insensitive).
 - **0 (Skip):** Load the console without any tenant configuration.
 - **exit / quit:** Immediately terminate the console session.
 
-You can also manually interact with it:
+### Console Helpers
 
-### Get Current Tenant
-```ruby
-ConsoleKit.current_tenant
-# => :tenant_one
-```
+The following helper methods are available in your Rails console:
 
-### Reset Current Tenant
 ```ruby
-ConsoleKit.reset_current_tenant
-# => nil
+# Switch to a different tenant
+switch_tenant
+
+# Print details about the current tenant
+tenant_info
+
+# List all available tenants
+tenants
 ```
 
-### Manually Enable Pretty Output
-```ruby
-ConsoleKit.enable_pretty_output
+### Custom Prompt
+
+ConsoleKit automatically sets your IRB/Pry prompt to show the active tenant:
+
+```
+[tenant_one] main:001>
 ```
 
-### Manually Disable Pretty Output
+### Other Methods
+
 ```ruby
+# Get current tenant
+ConsoleKit.current_tenant
+# => :tenant_one
+
+# Reset and re-select tenant
+ConsoleKit.reset_current_tenant
+
+# Toggle pretty output
+ConsoleKit.enable_pretty_output
 ConsoleKit.disable_pretty_output
 ```
 
+### Environment Warnings
+
+When a tenant has an `environment` key in its constants:
+- **production**: A red warning is displayed at setup time.
+- **staging**: A yellow warning is displayed at setup time.
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/SECURITY.md

@@ -8,7 +8,8 @@ Once a new version is released, the previous version is branched and locked, and
 
 | Version | Supported          |
 | ------- | ------------------ |
-| 1.0.0   | :white_check_mark: |
+| 1.1.0   | :white_check_mark: |
+| 1.0.0   | :x:                |
 | 0.1.5   | :x:                |
 | 0.1.4   | :x:                |
 | 0.1.3   | :x:                |

data/lib/console_kit/connections/base_connection_handler.rb

@@ -14,7 +14,13 @@ module ConsoleKit
 
       def initialize(context) = @context = context
       def connect = raise NotImplementedError, "#{self.class} must implement #connect"
-      def available? = false
+      def available? = raise NotImplementedError, "#{self.class} must implement #available?"
+
+      private
+
+      def context_attribute(name)
+        @context.respond_to?(name, true) ? @context.send(name) : nil
+      end
     end
   end
 end

data/lib/console_kit/connections/connection_manager.rb

@@ -2,6 +2,8 @@
 
 require_relative 'sql_connection_handler'
 require_relative 'mongo_connection_handler'
+require_relative 'redis_connection_handler'
+require_relative 'elasticsearch_connection_handler'
 
 module ConsoleKit
   module Connections
@@ -12,6 +14,8 @@ module ConsoleKit
           handler_classes.filter_map do |klass|
             handler = klass.new(context)
             handler if handler.available?
+          rescue NotImplementedError
+            nil
           end
         end
 

data/lib/console_kit/connections/elasticsearch_connection_handler.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require_relative 'base_connection_handler'
+
+module ConsoleKit
+  module Connections
+    # Handles Elasticsearch connections
+    class ElasticsearchConnectionHandler < BaseConnectionHandler
+      def connect
+        prefix = context_attribute(:tenant_elasticsearch_prefix).presence
+        Output.print_info(switch_message(prefix))
+        Thread.current[:console_kit_elasticsearch_prefix] = prefix
+        apply_model_index_prefix(prefix)
+      end
+
+      def available? = defined?(Elasticsearch)
+
+      private
+
+      def apply_model_index_prefix(prefix)
+        return unless defined?(Elasticsearch::Model) && Elasticsearch::Model.respond_to?(:index_name_prefix=)
+
+        Elasticsearch::Model.index_name_prefix = prefix
+      end
+
+      def switch_message(prefix)
+        prefix ? "Setting Elasticsearch index prefix: #{prefix}" : 'Resetting Elasticsearch index prefix to default'
+      end
+    end
+  end
+end

data/lib/console_kit/connections/mongo_connection_handler.rb

@@ -1,22 +1,17 @@
 # frozen_string_literal: true
 
-require 'forwardable'
 require_relative 'base_connection_handler'
 
 module ConsoleKit
   module Connections
     # Handles MongoDB connections
     class MongoConnectionHandler < BaseConnectionHandler
-      extend Forwardable
-
-      def_delegator :@context, :tenant_mongo_db
-
       def connect
-        db = tenant_mongo_db.presence
+        db = context_attribute(:tenant_mongo_db).presence
         Output.print_info(switch_message(db))
-        Mongoid.override_client(db)
+        Mongoid.override_database(db)
       rescue NoMethodError
-        Output.print_warning('Mongoid.override_client is not defined.')
+        Output.print_warning('Mongoid.override_database is not available in this version of Mongoid.')
       end
 
       def available? = defined?(Mongoid)

data/lib/console_kit/connections/redis_connection_handler.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require_relative 'base_connection_handler'
+
+module ConsoleKit
+  module Connections
+    # Handles Redis connections
+    class RedisConnectionHandler < BaseConnectionHandler
+      DEFAULT_REDIS_DB = 0
+
+      def connect
+        db = context_attribute(:tenant_redis_db)
+        Output.print_info(switch_message(db))
+        select_redis_db(db.nil? ? DEFAULT_REDIS_DB : db)
+      end
+
+      def available? = defined?(Redis)
+
+      private
+
+      def select_redis_db(db)
+        if Redis.respond_to?(:current) && Redis.current
+          Redis.current.select(db)
+        elsif defined?(RedisClient) && db != DEFAULT_REDIS_DB
+          Output.print_warning("Redis DB #{db} configured but auto-select not supported with RedisClient. " \
+                               'Ensure your Redis configuration sets the correct DB.')
+        end
+      rescue NoMethodError
+        Output.print_warning('Redis.current is not available (deprecated in Redis v5+).')
+      end
+
+      def switch_message(db)
+        db ? "Switching to Redis DB: #{db}" : 'Resetting Redis connection to default'
+      end
+    end
+  end
+end

data/lib/console_kit/connections/sql_connection_handler.rb

@@ -1,22 +1,15 @@
 # frozen_string_literal: true
 
-require 'forwardable'
 require_relative 'base_connection_handler'
 
 module ConsoleKit
   module Connections
     # Handles SQL connections
     class SqlConnectionHandler < BaseConnectionHandler
-      extend Forwardable
-
-      def_delegator :@context, :tenant_shard
-
       def connect
-        shard = tenant_shard.presence&.to_sym
-        msg = shard ? "Establishing SQL connection to shard: #{shard}" : 'Resetting SQL connection to default'
-
-        Output.print_info("#{msg} via #{base_class}")
-        base_class.establish_connection(shard)
+        shard = context_attribute(:tenant_shard).presence&.to_sym
+        Output.print_info("#{connection_message(shard)} via #{base_class}")
+        shard ? base_class.establish_connection(shard) : base_class.establish_connection
       end
 
       def available? = sql_base_class_name.to_s.safe_constantize.present?
@@ -30,6 +23,10 @@ module ConsoleKit
         raise Error, "ConsoleKit: sql_base_class '#{sql_base_class_name}' could not be found."
       end
 
+      def connection_message(shard)
+        shard ? "Establishing SQL connection to shard: #{shard}" : 'Resetting SQL connection to default'
+      end
+
       def sql_base_class_name = ConsoleKit.configuration.sql_base_class
     end
   end

data/lib/console_kit/console_helpers.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module ConsoleKit
+  # Helper methods available in the Rails console
+  module ConsoleHelpers
+    def switch_tenant = ConsoleKit.reset_current_tenant
+
+    def tenant_info
+      tenant = ConsoleKit::Setup.current_tenant
+      unless tenant
+        ConsoleKit::Output.print_warning('No tenant is currently configured.')
+        return
+      end
+
+      constants = ConsoleKit.configuration.tenants[tenant]&.[](:constants) || {}
+      print_tenant_details(tenant, constants)
+    end
+
+    def tenants
+      names = ConsoleKit.configuration.tenants&.keys || []
+      ConsoleKit::Output.print_list(names, header: 'Available Tenants')
+      names
+    end
+
+    private
+
+    def print_tenant_details(tenant, constants)
+      ConsoleKit::Output.print_header("Tenant: #{tenant}")
+      {
+        'Partner' => :partner_code, 'Shard' => :shard, 'Mongo DB' => :mongo_db,
+        'Redis DB' => :redis_db, 'ES Prefix' => :elasticsearch_prefix, 'Environment' => :environment
+      }.each do |label, key|
+        ConsoleKit::Output.print_info("  #{label.ljust(13)}#{constants[key]}") unless constants[key].nil?
+      end
+      nil
+    end
+  end
+end

data/lib/console_kit/output.rb

@@ -34,7 +34,7 @@ module ConsoleKit
           return if silent
 
           formatted = (type == :header ? "\n--- #{text} ---" : text)
-          print_with(type, formatted, timestamp, newline: newline)
+          print_with(type, formatted, timestamp, { newline: newline })
         end
       end
 
@@ -55,15 +55,15 @@ module ConsoleKit
       def print_backtrace(exception)
         return if silent
 
-        exception&.backtrace&.each { |line| print_with(:trace, "    #{line}", true, newline: true) }
+        exception&.backtrace&.each { |line| print_with(:trace, "    #{line}", true, { newline: true }) }
       end
 
       private
 
-      def print_with(type, text, timestamp, newline: true)
+      def print_with(type, text, timestamp, opts = {})
         meta = TYPES.fetch(type)
         message = build_message(text, meta[:symbol], timestamp)
-        output(message, meta[:color], newline: newline)
+        emit(message, meta[:color], opts.fetch(:newline, true))
       end
 
       def build_message(text, symbol, timestamp)
@@ -74,11 +74,10 @@ module ConsoleKit
       def timestamp_prefix(timestamp) = prefix_for(timestamp) { Time.current.strftime('[%Y-%m-%d %H:%M:%S] ') }
       def symbol_prefix(symbol) = prefix_for(symbol) { |sym| "#{sym} " }
 
-      def output(message, color, newline: true)
-        method = newline ? :puts : :print
-        return send(method, message) unless ConsoleKit.configuration.pretty_output && color
-
-        send(method, "\e[#{color}m#{message}\e[0m")
+      def emit(message, color, newline)
+        writer = newline ? :puts : :print
+        formatted = ConsoleKit.configuration.pretty_output && color ? "\e[#{color}m#{message}\e[0m" : message
+        send(writer, formatted)
       end
     end
   end

data/lib/console_kit/prompt.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+module ConsoleKit
+  # Sets the console prompt to show the current tenant
+  module Prompt
+    class << self
+      def apply
+        apply_irb_prompt if defined?(IRB)
+        apply_pry_prompt if defined?(Pry)
+      end
+
+      private
+
+      def tenant_label
+        tenant = ConsoleKit::Setup.current_tenant
+        tenant ? "[#{tenant}]" : '[no-tenant]'
+      end
+
+      def apply_irb_prompt
+        conf = IRB.conf
+        conf[:PROMPT] ||= {}
+        conf[:PROMPT][:CONSOLE_KIT] = {
+          PROMPT_I: "#{tenant_label} %N(%m):%03n> ",
+          PROMPT_S: "#{tenant_label} %N(%m):%03n%l ",
+          PROMPT_C: "#{tenant_label} %N(%m):%03n* ",
+          RETURN: "=> %s\n"
+        }
+        conf[:PROMPT_MODE] = :CONSOLE_KIT
+      end
+
+      def apply_pry_prompt
+        Pry.config.prompt = Pry::Prompt.new(
+          'console_kit',
+          'ConsoleKit tenant prompt',
+          [
+            proc { |obj, nest_level, _pry_instance| "#{Prompt.send(:tenant_label)} (#{obj}):#{nest_level}> " },
+            proc { |obj, nest_level, _pry_instance| "#{Prompt.send(:tenant_label)} (#{obj}):#{nest_level}* " }
+          ]
+        )
+      end
+    end
+  end
+end

data/lib/console_kit/railtie.rb

@@ -3,7 +3,15 @@
 module ConsoleKit
   # Railtie for integrating ConsoleKit with Rails console.
   class Railtie < Rails::Railtie
-    console { ConsoleKit::Setup.setup }
+    console do
+      ConsoleKit::Setup.setup
+      ConsoleKit::Prompt.apply
+      if defined?(Pry)
+        TOPLEVEL_BINDING.receiver.extend(ConsoleKit::ConsoleHelpers)
+      elsif defined?(IRB::ExtendCommandBundle)
+        IRB::ExtendCommandBundle.include(ConsoleKit::ConsoleHelpers)
+      end
+    end
 
     config.to_prepare { ConsoleKit::Setup.reapply if defined?(Rails::Console) }
   end

data/lib/console_kit/setup.rb

@@ -9,6 +9,11 @@ module ConsoleKit
   # Does the initial setup
   module Setup
     class << self
+      ENVIRONMENT_WARNINGS = {
+        'production' => -> { Output.print_error('WARNING: You are connected to a PRODUCTION environment!') },
+        'staging' => -> { Output.print_warning('You are connected to a staging environment.') }
+      }.freeze
+
       def current_tenant = Thread.current[:console_kit_current_tenant]
 
       def current_tenant=(val)
@@ -27,11 +32,13 @@ module ConsoleKit
       def reset_current_tenant
         return warn_no_tenants unless tenants?
 
-        warn_reset if current_tenant
-        TenantConfigurator.clear if current_tenant
+        key = select_tenant_key
+        return cancel_switch if key == :abort || key.blank?
 
-        self.current_tenant = nil
-        setup
+        clear_current_tenant
+        return skip_tenant_message if %i[exit none].include?(key)
+
+        configure(key)
       end
 
       private
@@ -40,7 +47,6 @@ module ConsoleKit
         return if tenant_setup_successful?
 
         ConsoleKit.configuration.validate!
-
         select_and_configure
       rescue StandardError => e
         handle_error(e)
@@ -54,19 +60,17 @@ module ConsoleKit
       end
 
       def handle_selection_result(key)
-        exit_on_key(key) if key == :exit
+        exit_on_key if %i[exit abort].include?(key)
 
         case key
-        when :abort, :none
+        when :none
           Output.print_info('No tenant selected. Loading without tenant configuration.')
         when nil, ''
           Output.print_error('Tenant selection failed. Loading without tenant configuration.')
         end
       end
 
-      def exit_on_key(key)
-        return unless key == :exit
-
+      def exit_on_key
         Output.print_info('Exiting console...')
         Kernel.exit
       end
@@ -76,7 +80,31 @@ module ConsoleKit
         return unless TenantConfigurator.configuration_success
 
         self.current_tenant = key
+        Prompt.apply
+        print_tenant_banner(key)
+      end
+
+      def print_tenant_banner(key)
+        constants = ConsoleKit.configuration.tenants[key]&.[](:constants) || {}
+        env = constants[:environment]&.to_s&.downcase
         Output.print_success("Tenant initialized: #{key}")
+        print_environment_warning(env) if env
+        print_active_connections
+      end
+
+      def print_environment_warning(env) = ENVIRONMENT_WARNINGS[env]&.call
+
+      def print_active_connections
+        names = active_connection_names
+        Output.print_info("Active connections: #{names.join(', ')}") unless names.empty?
+      end
+
+      def active_connection_names
+        ctx = context_class
+        return [] unless ctx
+
+        handlers = ConsoleKit::Connections::ConnectionManager.available_handlers(ctx)
+        handlers.map { |h| h.class.name.demodulize.delete_suffix('ConnectionHandler') }
       end
 
       def tenants = ConsoleKit.configuration.tenants
@@ -89,6 +117,16 @@ module ConsoleKit
       def non_interactive? = !$stdin.tty?
       def warn_no_tenants = Output.print_warning('Cannot reset tenant: No tenants configured.')
       def warn_reset = Output.print_warning("Resetting tenant: #{current_tenant}")
+      def cancel_switch = Output.print_warning('Tenant switch cancelled.')
+      def skip_tenant_message = Output.print_info('No tenant selected. Loading without tenant configuration.')
+
+      def clear_current_tenant
+        if current_tenant
+          warn_reset
+          TenantConfigurator.clear
+        end
+        self.current_tenant = nil
+      end
 
       def handle_error(error)
         Output.print_error("Error setting up tenant: #{error.message}")

data/lib/console_kit/tenant_configurator.rb

@@ -7,6 +7,13 @@ module ConsoleKit
   # For tenant configuration
   module TenantConfigurator
     class << self
+      HANDLER_ATTRIBUTES = {
+        Connections::SqlConnectionHandler => :tenant_shard,
+        Connections::MongoConnectionHandler => :tenant_mongo_db,
+        Connections::RedisConnectionHandler => :tenant_redis_db,
+        Connections::ElasticsearchConnectionHandler => :tenant_elasticsearch_prefix
+      }.freeze
+
       def configuration_success = Thread.current[:console_kit_configuration_success]
 
       def configuration_success=(val)
@@ -39,7 +46,7 @@ module ConsoleKit
       end
 
       def reset_context_attributes(ctx)
-        %i[tenant_shard tenant_mongo_db partner_identifier].each do |attr|
+        available_context_attributes(ctx).each do |attr|
           ctx.public_send("#{attr}=", nil)
         end
       end
@@ -60,31 +67,38 @@ module ConsoleKit
         configure_success(key)
       end
 
-      def validate_context_interface!(ctx)
-        missing = required_interface_methods.reject { |s| ctx.respond_to?(s) }
-        return if missing.empty?
-
-        raise Error, "Context class #{ctx} does not implement the required interface. " \
-                     "Missing methods: #{missing.join(', ')}"
+      def handler_available?(handler_class)
+        handler_class.new(nil).available?
+      rescue NotImplementedError, StandardError
+        false
       end
 
-      def required_interface_methods
-        attributes = %i[tenant_shard tenant_mongo_db partner_identifier]
-        attributes + attributes.map { |a| "#{a}=" }
+      def available_context_attributes(ctx)
+        attributes = %i[partner_identifier]
+        HANDLER_ATTRIBUTES.each do |handler, attr|
+          attributes << attr if handler_available?(handler) && ctx.respond_to?("#{attr}=")
+        end
+        attributes.select { |attr| ctx.respond_to?("#{attr}=") }
       end
 
       def apply_context(constant)
         ctx = ConsoleKit.configuration.context_class
-        validate_context_interface!(ctx)
-
         assign_context_attributes(ctx, constant)
         setup_connections(ctx)
       end
 
       def assign_context_attributes(ctx, constant)
-        ctx.tenant_shard = constant[:shard]
-        ctx.tenant_mongo_db = constant[:mongo_db]
-        ctx.partner_identifier = constant[:partner_code]
+        attribute_to_constant = {
+          partner_identifier: :partner_code,
+          tenant_shard: :shard,
+          tenant_mongo_db: :mongo_db,
+          tenant_redis_db: :redis_db,
+          tenant_elasticsearch_prefix: :elasticsearch_prefix
+        }
+
+        available_context_attributes(ctx).each do |attr|
+          ctx.public_send("#{attr}=", constant[attribute_to_constant[attr]])
+        end
       end
 
       def setup_connections(context)

data/lib/console_kit/tenant_selector.rb

@@ -9,36 +9,36 @@ module ConsoleKit
     DEFAULT_SELECTION = '1'
 
     class << self
-      def select = attempt_selection(RETRY_LIMIT)
+      def select
+        RETRY_LIMIT.times do
+          result = attempt_selection
+          return result unless result == :retry
+        end
+        nil
+      end
 
       private
 
-      def attempt_selection(retries_left)
-        return nil if retries_left.zero?
-
+      def attempt_selection
         print_tenant_selection_menu
-        process_selection(retries_left)
-      end
-
-      def process_selection(retries_left)
         selection = parse_user_selection
         return :abort if selection == :abort
-        return attempt_selection(retries_left - 1) unless selection
+        return :retry unless selection
 
         selection.is_a?(Integer) ? resolve_selection(selection) : selection
       end
 
       def print_tenant_selection_menu
         Output.print_header('Multiple tenants detected. Please choose one:')
+        Output.print_list(menu_items)
+      end
 
-        items = []
-        items << '0. Skip (load without tenant configuration)'
-
+      def menu_items
+        items = ['0. Skip (load without tenant configuration)']
         ConsoleKit.tenants.keys.each_with_index do |key, index|
           items << "#{index + 1}. #{key} (partner: #{tenant_partner(key)})"
         end
-
-        Output.print_list(items)
+        items
       end
 
       def tenant_partner(key) = ConsoleKit.tenants.dig(key, :constants, :partner_code) || 'N/A'
@@ -69,9 +69,10 @@ module ConsoleKit
 
       def read_input_with_default
         Output.print_prompt("Selection (number or name) [#{DEFAULT_SELECTION}]: ")
-
         raw_input = $stdin.gets
         raw_input ? normalize_input(raw_input) : :abort
+      rescue Interrupt
+        :abort
       end
 
       def normalize_input(raw_input)

data/lib/console_kit/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ConsoleKit
-  VERSION = '1.0.0'
+  VERSION = '1.1.0'
 end

data/lib/console_kit.rb

@@ -7,6 +7,8 @@ require 'active_support/core_ext/string/inflections'
 require_relative 'console_kit/version'
 require_relative 'console_kit/configuration'
 require_relative 'console_kit/setup'
+require_relative 'console_kit/console_helpers'
+require_relative 'console_kit/prompt'
 require_relative 'console_kit/railtie' if defined?(Rails::Railtie)
 
 # Main module for ConsoleKit
@@ -21,7 +23,7 @@ module ConsoleKit
     def reset_configuration!
       @configuration = nil
       Setup.current_tenant = nil
-      TenantConfigurator.configuration_success = false
+      TenantConfigurator.configuration_success = false if defined?(TenantConfigurator)
     end
 
     def pretty_output = configuration.pretty_output

data/lib/generators/console_kit/templates/console_kit.rb

@@ -11,14 +11,20 @@ Rails.application.config.after_initialize do
     #     constants: {
     #       shard: :shard_1,
     #       mongo_db: 'mongo_db_1',
-    #       partner_code: 'partner_a'
+    #       partner_code: 'partner_a',
+    #       redis_db: 1,
+    #       elasticsearch_prefix: 'tenant_a',
+    #       environment: 'production'
     #     }
     #   },
     #   tenant_b: {
     #     constants: {
     #       shard: :shard_2,
     #       mongo_db: 'mongo_db_2',
-    #       partner_code: 'partner_b'
+    #       partner_code: 'partner_b',
+    #       redis_db: 2,
+    #       elasticsearch_prefix: 'tenant_b',
+    #       environment: 'staging'
     #     }
     #   }
     # }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: console_kit
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Soumyadeep Pal
@@ -9,20 +9,6 @@ bindir: exe
 cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
-- !ruby/object:Gem::Dependency
-  name: mongoid
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
 - !ruby/object:Gem::Dependency
   name: rails
   requirement: !ruby/object:Gem::Requirement
@@ -53,9 +39,13 @@ files:
 - lib/console_kit/configuration.rb
 - lib/console_kit/connections/base_connection_handler.rb
 - lib/console_kit/connections/connection_manager.rb
+- lib/console_kit/connections/elasticsearch_connection_handler.rb
 - lib/console_kit/connections/mongo_connection_handler.rb
+- lib/console_kit/connections/redis_connection_handler.rb
 - lib/console_kit/connections/sql_connection_handler.rb
+- lib/console_kit/console_helpers.rb
 - lib/console_kit/output.rb
+- lib/console_kit/prompt.rb
 - lib/console_kit/railtie.rb
 - lib/console_kit/setup.rb
 - lib/console_kit/tenant_configurator.rb