console_kit 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8c68e98f9ea76d56039a31aee01f4642c23bc89e9b41868f988efff84f891647
-  data.tar.gz: 87be1db75fcb8d80851409a1903fe136d09e6275e5db9797b74f18df28603c63
+  metadata.gz: 634e983d59fba9a10d608945d11f0be809c4b3008612c804489a0ee18fd8bcf0
+  data.tar.gz: 646bf39efa3b81058524433d88852906313c20e0b584b1e205386bae45b8e638
 SHA512:
-  metadata.gz: f339cda7362c7a913918b992395ebd84ab4db2601a8c4d778302ae2579a1adfffab19d8eb63ca669b1d669ed9c58b589bfbe18c0d4ce758b83667b5935ebf304
-  data.tar.gz: c171cf918f2ea0ebe30c292f7b09847f3974c2fc39c55e21e36845c0399e04d7bc1b6558a8022bf860a57ae25403148ce62ef58748a7aad527ef55a870e6c5dd
+  metadata.gz: 70f8f1282f3e6a63588187a86de442b663f17efd40773f7ad6c2916a21917549917d5fa9fba688774e18608f173e8126cee711cb468faf94efdda1ef7653151d
+  data.tar.gz: 436bfcb527f71341dd2f6fe1b865a884b4e4208a4dec75820e2bc46068de85db706f2e037e2b1aa94f07091600e204c35ef89cff612dd2bb103ae6d1baa414f7

data/.reek.yml

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

data/.rubocop.yml

@@ -1,5 +1,10 @@
 AllCops:
   TargetRubyVersion: 3.1
+  NewCops: enable
+
+plugins:
+  - rubocop-rspec
+  - rubocop-rake
 
 Metrics/BlockLength:
   Exclude:

data/CHANGELOG.md

@@ -6,6 +6,24 @@ This project adheres to [Semantic Versioning](https://semver.org/).
 
 ---
 
+## [0.1.4] - 2025-09-30
+### Added
+- Minor Fixes and Improvements
+
+---
+
+## [0.1.3] - 2025-08-12
+### Added
+- `ConsoleKit.current_tenant` method to retrieve the current tenant at runtime.
+- `ConsoleKit.reset_current_tenant` to reset tenant selection.
+- `pretty_output` configuration added with ability to manually toggle CLI verbosity.
+
+### Changed
+- Refactored internal logic for improved maintainability and future extensibility.
+- Enhanced test coverage for better reliability and edge case handling.
+
+---
+
 ## [0.1.2] - 2025-07-23
 ### Added
 - Changelog added.
@@ -38,6 +56,8 @@ This project adheres to [Semantic Versioning](https://semver.org/).
 
 ## [Unreleased]
 
+[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
 [0.1.2]: https://github.com/Soumyadeep-ai/console_kit/releases/tag/v0.1.2
 [0.1.1]: https://github.com/Soumyadeep-ai/console_kit/releases/tag/v0.1.1
 [0.1.0]: https://github.com/Soumyadeep-ai/console_kit/releases/tag/v0.1.0

data/README.md

@@ -1,5 +1,11 @@
 # ConsoleKit
 
+![Gem Version](https://img.shields.io/gem/v/console_kit.svg)
+![Gem Downloads](https://img.shields.io/gem/dt/console_kit.svg)
+![Build Status](https://github.com/Soumyadeep-ai/console_kit/actions/workflows/release.yml/badge.svg)
+![License](https://img.shields.io/github/license/Soumyadeep-ai/console_kit)
+![Ruby](https://img.shields.io/badge/ruby-%3E=3.1.0-red)
+
 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.
@@ -52,9 +58,39 @@ ConsoleKit.configure do |config|
   }
 
   config.context_class = CurrentContext
+
+  # Optional: Toggle pretty CLI output
+  config.pretty_output = true
 end
 ```
 
+## 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:
+
+### Get Current Tenant
+```ruby
+ConsoleKit.current_tenant
+# => :tenant_one
+```
+
+### Reset Current Tenant
+```ruby
+ConsoleKit.reset_current_tenant
+# => nil
+```
+
+### Manually Disable Pretty Output
+```ruby
+ConsoleKit.enable_pretty_output
+```
+
+### Manually Disable Pretty Output
+```ruby
+ConsoleKit.disable_pretty_output
+```
+
 ## 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.
@@ -63,7 +99,7 @@ To install this gem onto your local machine, run `bundle exec rake install`. To
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/Soumyadeep-ai/console_kit. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/Soumyadeep-ai/console_kit/blob/main/CODE_OF_CONDUCT.md).
+Bug reports and pull requests are welcome on GitHub at [Console Kit](https://github.com/Soumyadeep-ai/console_kit). This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/Soumyadeep-ai/console_kit/blob/main/CODE_OF_CONDUCT.md).
 
 ## License
 

data/SECURITY.md

@@ -0,0 +1,28 @@
+# Security Policy
+
+## Supported Versions
+
+We do **not maintain multiple supported versions**. Only the latest released version is eligible for security updates.
+
+Once a new version is released, the previous version is branched and locked, and will no longer receive updates — including security patches.
+
+| Version | Supported          |
+| ------- | ------------------ |
+| 0.1.4   | :white_check_mark: |
+| 0.1.3   | :x:                |
+| 0.1.2   | :x:                |
+| 0.1.1   | :x:                |
+| 0.1.0   | :x:                |
+
+## Reporting a Vulnerability
+
+If you discover a security vulnerability, please use the GitHub [Security Advisories](https://github.com/Soumyadeep-ai/console_kit/security/advisories/new) feature to report it privately.
+
+We follow this process:
+- All reported vulnerabilities are reviewed promptly.
+- A new version with security fixes will be released.
+- Once validated, we will publish an advisory and issue a patched release if necessary.
+
+Please avoid disclosing vulnerabilities publicly until we’ve had a chance to review and respond.
+
+For more details, visit [GitHub Docs: Reporting a Vulnerability](https://docs.github.com/en/code-security/security-advisories/guidance-on-reporting-and-writing/privately-reporting-a-security-vulnerability).

data/lib/console_kit/configuration.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module ConsoleKit
+  # Stores ConsoleKit configurations such as tenant map and context behavior
+  class Configuration
+    attr_reader :pretty_output, :tenants, :context_class
+
+    def initialize(tenants: nil, context_class: nil)
+      @pretty_output = true
+      @tenants = tenants
+      @context_class = context_class
+    end
+
+    %i[pretty_output tenants context_class].each do |attr|
+      define_method("#{attr}=") { |value| instance_variable_set("@#{attr}", value) }
+    end
+  end
+end

data/lib/console_kit/connections/base_connection_handler.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module ConsoleKit
+  module Connections
+    # Parent class for connection handlers
+    class BaseConnectionHandler
+      attr_reader :context
+
+      def initialize(context) = @context = context
+
+      def connect
+        raise NotImplementedError, "#{self.class} must implement #connect"
+      end
+
+      def available? = false
+    end
+  end
+end

data/lib/console_kit/connections/connection_manager.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require_relative 'sql_connection_handler'
+require_relative 'mongo_connection_handler'
+
+module ConsoleKit
+  module Connections
+    # Manages available connection handlers
+    class ConnectionManager
+      class << self
+        def available_handlers(context)
+          handler_classes.filter_map do |klass|
+            handler = klass.new(context)
+            handler if handler.available?
+          end
+        end
+
+        private
+
+        def handler_classes = BaseConnectionHandler.descendants
+      end
+    end
+  end
+end

data/lib/console_kit/connections/mongo_connection_handler.rb

@@ -0,0 +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)
+      rescue NoMethodError
+        Output.print_warning('Mongoid.override_client is not defined.')
+      end
+
+      def available? = defined?(Mongoid)
+    end
+  end
+end

data/lib/console_kit/connections/sql_connection_handler.rb

@@ -0,0 +1,24 @@
+# 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
+        return if tenant_shard.blank?
+
+        Output.print_info("Establishing SQL connection to shard: #{tenant_shard}")
+        ApplicationRecord.establish_connection(tenant_shard.to_sym)
+      end
+
+      def available? = defined?(ApplicationRecord)
+    end
+  end
+end

data/lib/console_kit/output.rb

@@ -3,40 +3,50 @@
 module ConsoleKit
   # Handles Console outputs
   module Output
-    class << self
-      def print_error(text)
-        print_message("[✗] #{text}", '1;31') # Red
-      end
+    PREFIX = '[ConsoleKit]'
+    TYPES = {
+      error: { symbol: '[✗]', color: '1;31' },
+      success: { symbol: '[✓]', color: '1;32' },
+      warning: { symbol: '[!]', color: '1;33' },
+      prompt: { symbol: nil,   color: '1;36' },
+      header: { symbol: nil,   color: '1;34' },
+      trace: { symbol: nil, color: '0;90' },
+      info: { symbol: nil, color: nil }
+    }.freeze
 
-      def print_success(text)
-        print_message("[✓] #{text}", '1;32') # Green
+    class << self
+      TYPES.each_key do |type|
+        define_method("print_#{type}") do |text, timestamp: false|
+          formatted = (type == :header ? "\n=== #{text} ===" : text)
+          print_with(type, formatted, timestamp)
+        end
       end
 
+      # Backtrace prints always with timestamp, no param
       def print_backtrace(exception)
-        exception.backtrace.each { |line| print_message("    #{line}", '0;90') } # Dim gray
+        exception&.backtrace&.each { |line| print_with(:trace, "    #{line}", true) }
       end
 
-      def print_header(text)
-        print_message("\n=== #{text} ===", '1;34') # Bold Blue
-      end
+      private
 
-      def print_info(text)
-        print_message(text)
+      def print_with(type, text, timestamp)
+        meta = TYPES.fetch(type)
+        message = build_message(text, meta[:symbol], timestamp)
+        output(message, meta[:color])
       end
 
-      def print_prompt(text)
-        print_message(text, '1;36') # Cyan
+      def build_message(text, symbol, timestamp)
+        "#{PREFIX} #{timestamp_prefix(timestamp)}#{symbol_prefix(symbol)}#{text}"
       end
 
-      def print_warning(text)
-        print_message("[!] #{text}", '1;33') # Yellow
-      end
+      def prefix_for(value) = value ? yield(value) : ''
+      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} " }
 
-      private
+      def output(message, color)
+        return puts message unless ConsoleKit.configuration.pretty_output && color
 
-      def print_message(text, color = nil)
-        msg = "[ConsoleKit] #{text}"
-        puts color ? "\e[#{color}m#{msg}\e[0m" : msg
+        puts "\e[#{color}m#{message}\e[0m"
       end
     end
   end

data/lib/console_kit/railtie.rb

@@ -1,10 +1,8 @@
 # frozen_string_literal: true
 
 module ConsoleKit
-  # Railtie
+  # Railtie for integrating ConsoleKit with Rails console.
   class Railtie < Rails::Railtie
-    console do
-      ConsoleKit.setup
-    end
+    console { ConsoleKit::Setup.setup }
   end
 end

data/lib/console_kit/setup.rb

@@ -6,46 +6,64 @@ require_relative 'output'
 
 # Core Logic for initial Setup
 module ConsoleKit
-  class << self
-    attr_accessor :tenants, :context_class
+  # Does the initial setup
+  module Setup
+    class << self
+      attr_reader :current_tenant
 
-    def setup
-      return Output.print_error('No tenants configured.') if tenants.nil? || tenants.empty?
+      def setup = run_setup
+      def tenant_setup_successful? = !@current_tenant.to_s.empty?
 
-      tenant_key = resolve_tenant_key
-      return Output.print_error('No tenant selected. Loading without tenant configuration.') unless tenant_key
+      def reset_current_tenant
+        return warn_no_tenants unless tenants?
 
-      initialize_tenant(tenant_key)
-    rescue StandardError => e
-      handle_setup_error(e)
-    end
+        warn_reset if @current_tenant
+        TenantConfigurator.clear if @current_tenant
 
-    def configure
-      yield self
-    end
+        @current_tenant = nil
+        setup
+      end
 
-    private
+      private
 
-    def resolve_tenant_key
-      single_tenant? || non_interactive? ? tenants.keys.first : TenantSelector.select(tenants, tenants.keys)
-    end
+      def run_setup
+        return Output.print_error('No tenants configured.') if no_tenants?
 
-    def single_tenant?
-      tenants.size == 1
-    end
+        select_and_configure
+      rescue StandardError => e
+        handle_error(e)
+      end
 
-    def non_interactive?
-      !$stdin.tty?
-    end
+      def select_and_configure
+        key = select_tenant_key
+        return Output.print_error('No tenant selected. Loading without tenant configuration.') unless key
 
-    def initialize_tenant(tenant_key)
-      TenantConfigurator.configure_tenant(tenant_key, tenants, context_class)
-      Output.print_success("Tenant initialized: #{tenant_key}")
-    end
+        configure(key)
+      end
+
+      def configure(key)
+        TenantConfigurator.configure_tenant(key)
+        return unless TenantConfigurator.configuration_success
+
+        @current_tenant = key
+        Output.print_success("Tenant initialized: #{key}")
+      end
+
+      def tenants = ConsoleKit.configuration.tenants
+      def context_class = ConsoleKit.configuration.context_class
+      def tenants? = tenants&.any?
+      def no_tenants? = !tenants?
+      def select_tenant_key = auto_select? ? tenants.keys.first : TenantSelector.select
+      def auto_select? = single_tenant? || non_interactive?
+      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 handle_setup_error(error)
-      Output.print_error("Error setting up tenant: #{error.message}")
-      Output.print_backtrace(error)
+      def handle_error(error)
+        Output.print_error("Error setting up tenant: #{error.message}")
+        Output.print_backtrace(error)
+      end
     end
   end
 end

data/lib/console_kit/tenant_configurator.rb

@@ -1,36 +1,71 @@
 # frozen_string_literal: true
 
 require_relative 'output'
+require_relative 'connections/connection_manager'
 
 module ConsoleKit
   # For tenant configuration
   module TenantConfigurator
     class << self
-      def configure_tenant(key, tenants, context_class)
-        config = tenants[key]
-        return Output.print_error("No configuration found for tenant: #{key}") unless config
+      attr_reader :configuration_success
 
-        constants = config[:constants]
-        apply_context(context_class, constants)
-        setup_database_connections(context_class)
+      def configure_tenant(key)
+        constants = ConsoleKit.configuration.tenants[key]&.[](:constants)
+        return missing_config_error(key) unless constants
 
-        Output.print_success("Tenant set to: #{key}")
+        perform_configuration(key, constants)
       rescue StandardError => e
-        Output.print_error("Failed to configure tenant '#{key}': #{e.message}")
-        Output.print_backtrace(e)
+        handle_error(e, key)
+      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
+        Output.print_info('Tenant context has been cleared.')
       end
 
       private
 
-      def apply_context(context_class, constants)
-        context_class.tenant_shard = constants[:shard]
-        context_class.tenant_mongo_db = constants[:mongo_db]
-        context_class.partner_identifier = constants[:partner_code]
+      def validate_constants!(constants)
+        missing = %i[shard partner_code] - constants.keys
+        raise "Tenant constants missing keys: #{missing.join(', ')}" unless missing.empty?
+      end
+
+      def missing_config_error(key)
+        @configuration_success = false
+        Output.print_error("No configuration found for tenant: #{key}")
+      end
+
+      def perform_configuration(key, constants)
+        validate_constants!(constants)
+        apply_context(constants)
+        configure_success(key)
+      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]
+
+        setup_connections(ctx)
+      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
       end
 
-      def setup_database_connections(context_class)
-        ApplicationRecord.establish_connection(context_class.tenant_shard.to_sym)
-        Mongoid.override_client(context_class.tenant_mongo_db.to_s)
+      def handle_error(error, key)
+        @configuration_success = false
+        Output.print_error("Failed to configure tenant '#{key}': #{error.message}")
+        Output.print_backtrace(error)
       end
     end
   end

data/lib/console_kit/tenant_selector.rb

@@ -5,57 +5,64 @@ require_relative 'output'
 module ConsoleKit
   # For tenant selection
   module TenantSelector
-    class << self
-      def select(tenants, keys)
-        print_tenant_selection_menu(tenants, keys)
-
-        max_attempts = 3
-        max_attempts.times do
-          index = prompt_user_for_selection(keys.size)
-          return nil if index.zero?
-          return keys[index - 1] if index.positive?
-        end
+    RETRY_LIMIT = 3
+    DEFAULT_SELECTION = '1'
 
-        nil
+    class << self
+      def select
+        attempt_selection(RETRY_LIMIT)
       end
 
       private
 
-      def print_tenant_selection_menu(tenants, keys)
+      def attempt_selection(retries_left)
+        return nil if retries_left.zero?
+
+        print_tenant_selection_menu
+        selection = parse_user_selection
+        selection ? resolve_selection(selection) : attempt_selection(retries_left - 1)
+      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)')
 
-        keys.each_with_index do |key, index|
-          partner = tenants.dig(key, :constants, :partner_code) || 'N/A'
-          Output.print_info("  #{index + 1}. #{key} (partner: #{partner})")
+        ConsoleKit.tenants.keys.each_with_index do |key, index|
+          Output.print_info("  #{index + 1}. #{key} (partner: #{tenant_partner(key)})")
         end
       end
 
-      def prompt_user_for_selection(max_index)
-        Output.print_prompt("\nEnter the number of the tenant you want (or press Enter for default '1'): ")
-        input = $stdin.gets&.chomp&.strip
-        input = '1' if input.to_s.empty?
+      def tenant_partner(key) = ConsoleKit.tenants.dig(key, :constants, :partner_code) || 'N/A'
 
-        return invalid_input_response unless valid_integer?(input)
+      def parse_user_selection
+        input = read_input_with_default
+        return handle_invalid_input('Invalid input. Please enter a number.') unless valid_integer?(input)
 
-        parsed_index = input.to_i
-        return invalid_range_response(max_index) unless parsed_index.between?(0, max_index)
+        index = input.to_i
+        unless valid_selection_index?(index)
+          return handle_invalid_input("Selection must be between 0 and #{max_index}.")
+        end
 
-        parsed_index
+        index
       end
 
-      def valid_integer?(input)
-        input.match?(/\A\d+\z/)
+      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
       end
 
-      def invalid_input_response
-        Output.print_warning('Invalid input. Please enter a number.')
-        -1
-      end
+      def handle_invalid_input(message) = Output.print_warning(message).then { nil }
+      def valid_integer?(input) = input.match?(/\A\d+\z/)
+      def max_index = ConsoleKit.tenants.size
+      def valid_selection_index?(index) = index.between?(0, max_index)
+
+      def resolve_selection(index)
+        return nil if index.zero?
 
-      def invalid_range_response(max_index)
-        Output.print_warning("Selection must be between 0 and #{max_index}.")
-        -1
+        ConsoleKit.tenants.keys[index - 1]
       end
     end
   end

data/lib/console_kit/version.rb

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

data/lib/console_kit.rb

@@ -1,10 +1,30 @@
 # frozen_string_literal: true
 
 require_relative 'console_kit/version'
+require_relative 'console_kit/configuration'
 require_relative 'console_kit/setup'
 require_relative 'console_kit/railtie' if defined?(Rails::Railtie)
 
+# Main module for console kit
 module ConsoleKit
+  # Base error class for ConsoleKit-related exceptions.
   class Error < StandardError; end
-  # Your code goes here...
+
+  class << self
+    def configure = yield(configuration)
+
+    def configuration = Thread.current[:console_kit_configuration] ||= Configuration.new
+    def reset_configuration! = Thread.current[:console_kit_configuration] = nil
+
+    %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) }
+    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
 end

data/lib/generators/console_kit/install_generator.rb

@@ -9,20 +9,26 @@ module ConsoleKit
     class InstallGenerator < Rails::Generators::Base
       source_root File.expand_path('templates', __dir__)
 
+      class_option :force, type: :boolean, default: false, desc: 'Overwrite existing files'
+
       def copy_initializer
+        force = options[:force]
         initializer_path = Rails.root.join('config', 'initializers', 'console_kit.rb')
 
-        if File.exist?(initializer_path)
-          say_status('skipped', "Initializer already exists: #{initializer_path}", :yellow)
+        if File.exist?(initializer_path) && !force
+          say_status :skipped, "Initializer already exists: #{initializer_path}", :yellow
         else
-          template 'console_kit.rb', 'config/initializers/console_kit.rb'
-          say_status('created', "Initializer generated at #{initializer_path}", :green)
+          template 'console_kit.rb', 'config/initializers/console_kit.rb', force: force
+          say_status :created, "Initializer generated at #{initializer_path}", :green
         end
       end
 
       def remind_about_customization
         say "\n✅ Setup complete!", :green
-        say '👉 Please update `config/initializers/console_kit.rb` to set your `tenants` and `context_class`.', :green
+        say '📄 Modify `config/initializers/console_kit.rb`:', :green
+        %w[tenants context_class].each do |field|
+          say "  - Set `#{field}` (required)", :green
+        end
       end
     end
   end

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

@@ -5,15 +5,20 @@
 
 Rails.application.config.after_initialize do
   ConsoleKit.configure do |config|
-    # TODO: Set your tenants source in the following format
+    # TODO: Set your tenants source, example:
     # {
-    #   key:
-    #   {
-    #     constants:
-    #     {
-    #       shard: # Active Record Shard
-    #       mongo_db: # Mongo Shard (If Mongo Is being used)
-    #       partner_code: # If partners are needed
+    #   tenant_a: {
+    #     constants: {
+    #       shard: :shard_1,
+    #       mongo_db: 'mongo_db_1',
+    #       partner_code: 'partner_a'
+    #     }
+    #   },
+    #   tenant_b: {
+    #     constants: {
+    #       shard: :shard_2,
+    #       mongo_db: 'mongo_db_2',
+    #       partner_code: 'partner_b'
     #     }
     #   }
     # }
@@ -21,5 +26,18 @@ Rails.application.config.after_initialize do
 
     # TODO: Set your context class (e.g., CurrentContext)
     config.context_class = nil
+
+    # Toggle pretty output on/off (default: true)
+    config.pretty_output = true
+
+    if config.tenants.nil?
+      warn '[ConsoleKit] Warning: `tenants` is not configured. ' \
+           'Please set it in config/initializers/console_kit.rb'
+    end
+
+    if config.context_class.nil?
+      warn '[ConsoleKit] Warning: `context_class` is not configured. ' \
+           'Please set it in config/initializers/console_kit.rb'
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: console_kit
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Soumyadeep Pal
@@ -44,6 +44,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".reek.yml"
 - ".rspec"
 - ".rubocop.yml"
 - CHANGELOG.md
@@ -51,7 +52,13 @@ files:
 - 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/mongo_connection_handler.rb
+- lib/console_kit/connections/sql_connection_handler.rb
 - lib/console_kit/output.rb
 - lib/console_kit/railtie.rb
 - lib/console_kit/setup.rb
@@ -69,6 +76,7 @@ metadata:
   homepage_uri: https://github.com/Soumyadeep-ai/console_kit
   source_code_uri: https://github.com/Soumyadeep-ai/console_kit
   changelog_uri: https://github.com/Soumyadeep-ai/console_kit/blob/main/CHANGELOG.md
+  rubygems_mfa_required: 'true'
 rdoc_options: []
 require_paths:
 - lib