console_kit 0.1.5 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6519b1ab2c0164341a30cf828095d91fe8e3872f7456abf0041a4bc58401ec54
-  data.tar.gz: 33df556d5e20599248763f074a0f8379171bcacddb3996dede936980a1fbe6ae
+  metadata.gz: acd47ead1008fc3bcdc25b0d995b735f69d1e6f5a8d5f8442c67dc5ee86bb9a3
+  data.tar.gz: 55bd88158a92c6741eab85bda17d4b61828dd101e94cd5ab02097688f8994302
 SHA512:
-  metadata.gz: 5d5041a668f6c03adbd7cd41fd902d2a9ff420c0b3da700ae1b56e3b9d5773dfb009bce276ef970eb4e0b09db6aedf41a7404a1a9c8d8c1a8a8a98e5269235ce
-  data.tar.gz: d6db97c81c56974ff3652dca1fceded6d917556def258670b1710aa8603ef5f9a593b5e426b74ab87b3a380c5288eb2f6688346674a3add582de1e6d09502e45
+  metadata.gz: 9d1214289e7859342493f2930557554e05ad3ccb3200e041d3ade9b2f95a710b6f111eaea9bfc40f0192e98cabd1010f7be6c71469066d2fd702a74d70a33132
+  data.tar.gz: 4564b76383af8ab6d60c6ee79644fc78a1be8633520ffe1c74a26a69936ebbacb7db8b20dd84c28c2433f532198c46c1c140ff7d31cd323d613559624fabb713

data/CHANGELOG.md

@@ -6,6 +6,47 @@ 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.
+- **Isolated Tenant Selection:** Each thread maintains its own tenant selection for safety, while sharing the global configuration.
+- **Seamless Rails Reloading:** Full support for Rails `reload!`; your selected tenant and context are now automatically preserved after code reloads.
+- **Reliable Tenant Switching:** Switching or clearing tenants now correctly resets all database connections (SQL and MongoDB) to their default state.
+- **Flexible Tenant Selection:** Users can now select tenants by typing their names (case-insensitive) in addition to index numbers.
+- **Session Control:** Added support for `exit` or `quit` commands directly at the selection prompt to terminate the console session.
+- **Safe Mode:** Added a "Skip" option (0) to load the console without any tenant configuration.
+- **Improved Configuration Validation:** Enhanced startup checks to provide clearer feedback if the configuration or context class is incorrectly defined.
+- **Custom SQL Base Class:** New configuration option to specify a custom base class for SQL connections.
+
+### Changed
+- **Modernized CLI Interface:** Redesigned the tenant selection menu and prompts for a cleaner, more intuitive user experience.
+- **Enhanced Error Feedback:** Improved messaging for invalid selections and missing configurations.
+- **Optimized Performance:** Refactored internal discovery and configuration logic for better reliability in large applications.
+
+### Fixed
+- Fixed a bug where tenant context was lost after running `reload!` in the Rails console.
+- Fixed an issue where database connections could remain tied to a previous tenant after the context was cleared.
+- Resolved all stability and code quality warnings.
+- Fixed timestamp formatting in console output.
+
+---
+
 ## [0.1.5] - 2025-10-12
 ### Added
 - Minor Bug Fixes
@@ -58,10 +99,8 @@ This project adheres to [Semantic Versioning](https://semver.org/).
   - Tenant-specific database configuration.
   - Colorized console output for improved UX.
 
----
-
-## [Unreleased]
-
+[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
 [0.1.3]: https://github.com/Soumyadeep-ai/console_kit/releases/tag/v0.1.3

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,33 +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 tenants are configured).
-You can also manually interact with it:
+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.
 
-### Get Current Tenant
-```ruby
-ConsoleKit.current_tenant
-# => :tenant_one
-```
+### 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.
+
+### Console Helpers
+
+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 Disable 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,9 @@ Once a new version is released, the previous version is branched and locked, and
 
 | Version | Supported          |
 | ------- | ------------------ |
-| 0.1.5   | :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:                |
 | 0.1.2   | :x:                |

data/lib/console_kit/configuration.rb

@@ -3,16 +3,43 @@
 module ConsoleKit
   # Stores ConsoleKit configurations such as tenant map and context behavior
   class Configuration
-    attr_reader :pretty_output, :tenants, :context_class
+    attr_accessor :pretty_output, :tenants, :sql_base_class
+    attr_writer :context_class
 
-    def initialize(tenants: nil, context_class: nil)
+    def initialize
       @pretty_output = true
-      @tenants = tenants
-      @context_class = context_class
+      @tenants = nil
+      @context_class = nil
+      @sql_base_class = 'ApplicationRecord'
     end
 
-    %i[pretty_output tenants context_class].each do |attr|
-      define_method("#{attr}=") { |value| instance_variable_set("@#{attr}", value) }
+    def context_class
+      case @context_class
+      when String, Symbol then resolve_context_class
+      else @context_class
+      end
+    end
+
+    def validate
+      validate!
+      true
+    rescue Error
+      false
+    end
+
+    def validate!
+      raise Error, 'ConsoleKit: `tenants` is not configured.' if @tenants.blank?
+      raise Error, 'ConsoleKit: `tenants` must be a Hash.' unless @tenants.is_a?(Hash)
+      raise Error, 'ConsoleKit: `context_class` is not configured.' if @context_class.blank?
+    end
+
+    private
+
+    def resolve_context_class
+      @context_class.to_s.constantize
+    rescue NameError
+      raise Error, "ConsoleKit: context_class '#{@context_class}' could not be found. " \
+                   'Ensure the class is defined before configuration is accessed.'
     end
   end
 end

data/lib/console_kit/connections/base_connection_handler.rb

@@ -1,18 +1,26 @@
 # frozen_string_literal: true
 
+require 'active_support/core_ext/class/subclasses'
+
 module ConsoleKit
   module Connections
     # Parent class for connection handlers
     class BaseConnectionHandler
+      class << self
+        def registry = descendants
+      end
+
       attr_reader :context
 
       def initialize(context) = @context = context
+      def connect = raise NotImplementedError, "#{self.class} must implement #connect"
+      def available? = raise NotImplementedError, "#{self.class} must implement #available?"
 
-      def connect
-        raise NotImplementedError, "#{self.class} must implement #connect"
-      end
+      private
 
-      def available? = false
+      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,12 +14,14 @@ module ConsoleKit
           handler_classes.filter_map do |klass|
             handler = klass.new(context)
             handler if handler.available?
+          rescue NotImplementedError
+            nil
           end
         end
 
         private
 
-        def handler_classes = BaseConnectionHandler.descendants
+        def handler_classes = BaseConnectionHandler.registry
       end
     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,26 +1,26 @@
 # 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
-        return if tenant_mongo_db.blank?
-
-        Output.print_info("Switching to MongoDB client: #{tenant_mongo_db}")
-        Mongoid.override_client(tenant_mongo_db)
+        db = context_attribute(:tenant_mongo_db).presence
+        Output.print_info(switch_message(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)
+
+      private
+
+      def switch_message(db)
+        db ? "Switching to MongoDB client: #{db}" : 'Resetting MongoDB client to default'
+      end
     end
   end
 end

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,24 +1,33 @@
 # frozen_string_literal: true
 
-require 'forwardable'
 require_relative 'base_connection_handler'
 
 module ConsoleKit
   module Connections
     # Handles SQL connections
     class SqlConnectionHandler < BaseConnectionHandler
-      extend Forwardable
+      def connect
+        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_delegator :@context, :tenant_shard
+      def available? = sql_base_class_name.to_s.safe_constantize.present?
 
-      def connect
-        return if tenant_shard.blank?
+      private
+
+      def base_class
+        klass = sql_base_class_name.to_s.safe_constantize
+        return klass if klass
+
+        raise Error, "ConsoleKit: sql_base_class '#{sql_base_class_name}' could not be found."
+      end
 
-        Output.print_info("Establishing SQL connection to shard: #{tenant_shard}")
-        ApplicationRecord.establish_connection(tenant_shard.to_sym)
+      def connection_message(shard)
+        shard ? "Establishing SQL connection to shard: #{shard}" : 'Resetting SQL connection to default'
       end
 
-      def available? = defined?(ApplicationRecord)
+      def sql_base_class_name = ConsoleKit.configuration.sql_base_class
     end
   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

@@ -15,24 +15,55 @@ module ConsoleKit
     }.freeze
 
     class << self
+      def silent = Thread.current[:console_kit_silent]
+
+      def silent=(val)
+        Thread.current[:console_kit_silent] = val
+      end
+
+      def silence
+        old_silent = silent
+        self.silent = true
+        yield
+      ensure
+        self.silent = old_silent
+      end
+
       TYPES.each_key do |type|
-        define_method("print_#{type}") do |text, timestamp: false|
-          formatted = (type == :header ? "\n=== #{text} ===" : text)
-          print_with(type, formatted, timestamp)
+        define_method("print_#{type}") do |text, timestamp: false, newline: (type != :prompt)|
+          return if silent
+
+          formatted = (type == :header ? "\n--- #{text} ---" : text)
+          print_with(type, formatted, timestamp, { newline: newline })
         end
       end
 
+      def print_list(items, header: nil)
+        return if silent
+
+        print_header(header) if header
+        items.each { |item| puts "  #{item}" }
+      end
+
+      def print_raw(text)
+        return if silent
+
+        puts text
+      end
+
       # Backtrace prints always with timestamp, no param
       def print_backtrace(exception)
-        exception&.backtrace&.each { |line| print_with(:trace, "    #{line}", true) }
+        return if silent
+
+        exception&.backtrace&.each { |line| print_with(:trace, "    #{line}", true, { newline: true }) }
       end
 
       private
 
-      def print_with(type, text, timestamp)
+      def print_with(type, text, timestamp, opts = {})
         meta = TYPES.fetch(type)
         message = build_message(text, meta[:symbol], timestamp)
-        output(message, meta[:color])
+        emit(message, meta[:color], opts.fetch(:newline, true))
       end
 
       def build_message(text, symbol, timestamp)
@@ -43,10 +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)
-        return puts message unless ConsoleKit.configuration.pretty_output && color
-
-        puts "\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,6 +3,16 @@
 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
 end

data/lib/console_kit/setup.rb

@@ -9,27 +9,44 @@ module ConsoleKit
   # Does the initial setup
   module Setup
     class << self
-      attr_reader :current_tenant
+      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)
+        Thread.current[:console_kit_current_tenant] = val
+      end
 
       def setup = run_setup
-      def tenant_setup_successful? = !@current_tenant.to_s.empty?
+      def tenant_setup_successful? = !current_tenant.to_s.empty?
+
+      def reapply
+        return unless tenant_setup_successful?
+
+        Output.silence { TenantConfigurator.configure_tenant(current_tenant) }
+      end
 
       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?
+
+        clear_current_tenant
+        return skip_tenant_message if %i[exit none].include?(key)
 
-        @current_tenant = nil
-        setup
+        configure(key)
       end
 
       private
 
       def run_setup
         return if tenant_setup_successful?
-        return Output.print_error('No tenants configured.') if no_tenants?
 
+        ConsoleKit.configuration.validate!
         select_and_configure
       rescue StandardError => e
         handle_error(e)
@@ -37,17 +54,57 @@ module ConsoleKit
 
       def select_and_configure
         key = select_tenant_key
-        return Output.print_error('No tenant selected. Loading without tenant configuration.') unless key
+        return handle_selection_result(key) if %i[exit abort none].include?(key) || key.blank?
 
         configure(key)
       end
 
+      def handle_selection_result(key)
+        exit_on_key if %i[exit abort].include?(key)
+
+        case key
+        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
+        Output.print_info('Exiting console...')
+        Kernel.exit
+      end
+
       def configure(key)
         TenantConfigurator.configure_tenant(key)
         return unless TenantConfigurator.configuration_success
 
-        @current_tenant = key
+        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
@@ -59,7 +116,17 @@ module ConsoleKit
       def single_tenant? = tenants.size == 1
       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 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,7 +7,18 @@ module ConsoleKit
   # For tenant configuration
   module TenantConfigurator
     class << self
-      attr_reader :configuration_success
+      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)
+        Thread.current[:console_kit_configuration_success] = val
+      end
 
       def configure_tenant(key)
         constants = ConsoleKit.configuration.tenants[key]&.[](:constants)
@@ -19,22 +30,34 @@ module ConsoleKit
       end
 
       def clear
-        @configuration_success = false
-        %i[tenant_shard tenant_mongo_db partner_identifier].each do |attr|
-          ConsoleKit.configuration.context_class.public_send("#{attr}=", nil)
-        end
+        ctx = ConsoleKit.configuration.context_class
+        return unless ctx
+
+        reset_tenant(ctx)
         Output.print_info('Tenant context has been cleared.')
       end
 
       private
 
+      def reset_tenant(ctx)
+        self.configuration_success = false
+        reset_context_attributes(ctx)
+        setup_connections(ctx)
+      end
+
+      def reset_context_attributes(ctx)
+        available_context_attributes(ctx).each do |attr|
+          ctx.public_send("#{attr}=", nil)
+        end
+      end
+
       def validate_constants!(constants)
         missing = %i[shard partner_code] - constants.keys
-        raise "Tenant constants missing keys: #{missing.join(', ')}" unless missing.empty?
+        raise Error, "Tenant constants missing keys: #{missing.join(', ')}" unless missing.empty?
       end
 
       def missing_config_error(key)
-        @configuration_success = false
+        self.configuration_success = false
         Output.print_error("No configuration found for tenant: #{key}")
       end
 
@@ -44,26 +67,51 @@ module ConsoleKit
         configure_success(key)
       end
 
+      def handler_available?(handler_class)
+        handler_class.new(nil).available?
+      rescue NotImplementedError, StandardError
+        false
+      end
+
+      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
-        ctx.tenant_shard = constant[:shard]
-        ctx.tenant_mongo_db = constant[:mongo_db]
-        ctx.partner_identifier = constant[:partner_code]
-
+        assign_context_attributes(ctx, constant)
         setup_connections(ctx)
       end
 
+      def assign_context_attributes(ctx, constant)
+        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)
         ConsoleKit::Connections::ConnectionManager.available_handlers(context).each(&:connect)
       end
 
       def configure_success(key)
         Output.print_success("Tenant set to: #{key}")
-        @configuration_success = true
+        self.configuration_success = true
       end
 
       def handle_error(error, key)
-        @configuration_success = false
+        self.configuration_success = false
         Output.print_error("Failed to configure tenant '#{key}': #{error.message}")
         Output.print_backtrace(error)
       end

data/lib/console_kit/tenant_selector.rb

@@ -10,35 +10,56 @@ module ConsoleKit
 
     class << self
       def select
-        attempt_selection(RETRY_LIMIT)
+        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
         selection = parse_user_selection
-        selection ? resolve_selection(selection) : attempt_selection(retries_left - 1)
+        return :abort if selection == :abort
+        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_info('  0. Load without tenant (no tenant configuration)')
+        Output.print_list(menu_items)
+      end
 
+      def menu_items
+        items = ['0. Skip (load without tenant configuration)']
         ConsoleKit.tenants.keys.each_with_index do |key, index|
-          Output.print_info("  #{index + 1}. #{key} (partner: #{tenant_partner(key)})")
+          items << "#{index + 1}. #{key} (partner: #{tenant_partner(key)})"
         end
+        items
       end
 
       def tenant_partner(key) = ConsoleKit.tenants.dig(key, :constants, :partner_code) || 'N/A'
 
       def parse_user_selection
         input = read_input_with_default
-        return handle_invalid_input('Invalid input. Please enter a number.') unless valid_integer?(input)
+        return :abort if input == :abort
+        return :exit if %w[exit quit].include?(input.downcase)
+        return find_tenant_by_name(input) unless valid_integer?(input)
+
+        validate_index_range(input.to_i)
+      end
 
-        index = input.to_i
+      def find_tenant_by_name(input)
+        match = ConsoleKit.tenants.keys.find { |k| k.to_s.casecmp(input).zero? }
+        return match if match
+
+        handle_invalid_input("Invalid selection: '#{input}'. Please enter a number or tenant name.")
+      end
+
+      def validate_index_range(index)
         unless valid_selection_index?(index)
           return handle_invalid_input("Selection must be between 0 and #{max_index}.")
         end
@@ -47,11 +68,16 @@ module ConsoleKit
       end
 
       def read_input_with_default
-        prompt_message = "\nEnter the number of the tenant you want " \
-                         "(or press Enter for default '#{DEFAULT_SELECTION}'): "
-        Output.print_prompt(prompt_message)
-        input = $stdin.gets&.chomp&.strip
-        input.to_s.empty? ? DEFAULT_SELECTION : input
+        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)
+        input = raw_input.chomp.strip
+        input.empty? ? DEFAULT_SELECTION : input
       end
 
       def handle_invalid_input(message) = Output.print_warning(message).then { nil }
@@ -60,7 +86,7 @@ module ConsoleKit
       def valid_selection_index?(index) = index.between?(0, max_index)
 
       def resolve_selection(index)
-        return nil if index.zero?
+        return :none if index.zero?
 
         ConsoleKit.tenants.keys[index - 1]
       end

data/lib/console_kit/version.rb

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

data/lib/console_kit.rb

@@ -1,29 +1,51 @@
 # frozen_string_literal: true
 
+require 'active_support/core_ext/object/blank'
+require 'active_support/core_ext/object/inclusion'
+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 console kit
+# Main module for ConsoleKit
 module ConsoleKit
   # Base error class for ConsoleKit-related exceptions.
   class Error < StandardError; end
 
   class << self
     def configure = yield(configuration)
+    def configuration = @configuration ||= Configuration.new
+
+    def reset_configuration!
+      @configuration = nil
+      Setup.current_tenant = nil
+      TenantConfigurator.configuration_success = false if defined?(TenantConfigurator)
+    end
+
+    def pretty_output = configuration.pretty_output
+
+    def pretty_output=(val)
+      configuration.pretty_output = val
+    end
 
-    def configuration = Thread.current[:console_kit_configuration] ||= Configuration.new
-    def reset_configuration! = Thread.current[:console_kit_configuration] = nil
+    def tenants = configuration.tenants
 
-    %i[pretty_output tenants context_class].each do |name|
-      define_method(name) { configuration.public_send(name) }
-      define_method("#{name}=") { |val| configuration.public_send("#{name}=", val) }
+    def tenants=(val)
+      configuration.tenants = val
+    end
+
+    def context_class = configuration.context_class
+
+    def context_class=(val)
+      configuration.context_class = val
     end
 
     def current_tenant = Setup.current_tenant
     def reset_current_tenant = Setup.reset_current_tenant
-
     def enable_pretty_output = configuration.pretty_output = true
     def disable_pretty_output = configuration.pretty_output = false
   end

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

@@ -11,20 +11,27 @@ 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'
     #     }
     #   }
     # }
     config.tenants = nil
 
-    # TODO: Set your context class (e.g., CurrentContext)
+    # TODO: Set your context class (e.g., 'CurrentContext')
+    # Recommendation: Use a String to ensure the class is correctly re-resolved after `reload!`
     config.context_class = nil
 
     # Toggle pretty output on/off (default: true)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: console_kit
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  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
@@ -44,22 +30,22 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".reek.yml"
-- ".rspec"
-- ".rubocop.yml"
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
 - LICENSE.txt
 - README.md
-- Rakefile
 - SECURITY.md
 - lib/console_kit.rb
 - 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

data/.reek.yml

@@ -1,4 +0,0 @@
-detectors:
-  UncommunicativeVariableName:
-    exclude:
-      - e

data/.rspec

@@ -1,3 +0,0 @@
---format documentation
---color
---require spec_helper

data/.rubocop.yml

@@ -1,11 +0,0 @@
-AllCops:
-  TargetRubyVersion: 3.1
-  NewCops: enable
-
-plugins:
-  - rubocop-rspec
-  - rubocop-rake
-
-Metrics/BlockLength:
-  Exclude:
-    - 'spec/**/*_spec.rb'

data/Rakefile

@@ -1,12 +0,0 @@
-# frozen_string_literal: true
-
-require 'bundler/gem_tasks'
-require 'rspec/core/rake_task'
-
-RSpec::Core::RakeTask.new(:spec)
-
-require 'rubocop/rake_task'
-
-RuboCop::RakeTask.new
-
-task default: %i[spec rubocop]