pry-byetypo 1.0.2 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7faf45157b7a6b110d22c13122b713a51f25b9e9c31d928f4d5511a2e9ffbcb3
-  data.tar.gz: 4f462974bcc85139f06652b8d7ca28374fea6fbff9096388054197917f4bb104
+  metadata.gz: dd76efc826578597fe9cecb0ec6357a275d8cb8c37aaa1e238681f0235287027
+  data.tar.gz: 11eeb72895ab4150815e61f932a7aa8d13dc2ae6d6b3564c81d3ff9624c8ec89
 SHA512:
-  metadata.gz: 2156f5efe4854c4ac4298a314eb48006d629f289a59e6d58d4078c0373f8e4db776d93b3247fd61fba01536e9109b26ad3881ec46f1bca6f09c4c8c5194ea1c7
-  data.tar.gz: 44b4e2b0c73133bead16a8ed5f7660de909862fe3b2b7b156bcc0f4eb9ca4d7054e8eac6360766851dbe0cfa4ee74d25c5b065169509979ff36a3ebe14cefa21
+  metadata.gz: 9f55811aaaeeeb06521f50108d8fab819b00c461b2fd635519ad3559baf409ee256e241d5315a39ec2e238aaae6e330f2d84bcd0cfbc1b3b9f04e9b26d8b9d39
+  data.tar.gz: 954ecd0a20d6fb34df1592cf36f8c2dd5d893d21f0060a7ccba7a4769ead0800917d6ee08aa8c294b8937e8eaa450c77fe7ac3704f4da3e2ea4466257e306552

data/Gemfile.lock

@@ -1,10 +1,9 @@
 PATH
   remote: .
   specs:
-    pry-byetypo (1.0.2)
+    pry-byetypo (1.2.0)
       colorize (~> 1.1.0)
       pry (>= 0.13, < 0.15)
-      rails (~> 7.0)
 
 GEM
   remote: https://rubygems.org/
@@ -146,7 +145,7 @@ GEM
     psych (5.1.2)
       stringio
     racc (1.7.3)
-    rack (3.0.8)
+    rack (3.0.9)
     rack-session (2.0.0)
       rack (>= 3.0.0)
     rack-test (2.1.0)
@@ -253,6 +252,7 @@ PLATFORMS
 DEPENDENCIES
   byebug
   pry-byetypo!
+  rails (~> 7.0)
   rake (~> 13.0)
   rspec (~> 3.0)
   standard (~> 1.3)

data/README.md

@@ -1,29 +1,32 @@
-# PRY-BYETYPO
+# PRY-BYETYPO 👋
 
 Autocorrects typos in your Pry console.
 
-This small Pry plugin captures exceptions that could be due to typos and deduces the correct command based on your database information.
+This small Pry plugin captures exceptions that may arise from typos and deduces the correct command based on your database information and session history.
 
 #### Before
 
 ```ruby
-[1] pry(main)> Usert.last
-NameError: uninitialized constant Usert
-from (pry):3:in `__pry__'
+[1] pry(main)> result = 1
+=> 1
+[2] pry(main)> resilt
+NameError: undefined local variable or method `resilt' for main:Object
+from (pry):2:in `__pry__'
 ```
 
 #### After
 
 ```ruby
-[1] pry(main)> Usert.last
-I, [2024-01-13T20:00:16.280710 #694]  ERROR -- : NameError: uninitialized constant Usert
-I, [2024-01-13T20:00:16.281237 #694]  INFO -- : Running: User.last
-2024-01-13 20:00:16.345175 D [694:9200 log_subscriber.rb:130] ActiveRecord::Base --   User Load (1.0ms)  SELECT "users".* FROM "users" WHERE "users"."deleted_at" IS NULL ORDER BY "users"."id" DESC LIMIT $1  [["LIMIT", 1]]
-=> #<User id: 1, email: "yo@email.com">
+[1] pry(main)> result = 1
+=> 1
+[2] pry(main)> resilt
+E, [2024-01-31T17:11:16.344161 #3739] ERROR -- : undefined local variable or method `resilt' for main:Object
+I, [2024-01-31T17:11:16.344503 #3739]  INFO -- : Running: result
+=> 1
 ```
 
 > [!NOTE]
-> So far, this plugin is not framework agnostic, and it requires Rails 7 or later.
+> Currently, this plugin is not truly ORM-agnostic; to fully benefit from it, ActiveRecord is required.
 
 ## Installation
 
@@ -46,24 +49,59 @@ gem install pry-byetypo
 2. Start your daily work.
 3. Let `pry-byetypo` remove frictions. 🚀
 
-## Under the hood
-
-### 1. Byetypo dictionary
+## Byetypo dictionary
 
-When you open a new Pry console, the gem will generate a `byetypo_dictionary.pstore` file containing three pieces of information:
+When you open a new Pry console, the gem will generate a `byetypo_dictionary.pstore` file containing four pieces of information:
 
 - A list of the ActiveRecord models in your application (e.g. `User`, `Account`).
 - A list of the ActiveRecord associations in your application (e.g. `user`, `users`, `account`, `accounts`).
+- Unique identifiers for each active Pry instance, populated with the variable history of the current session. (e.g. `result = 1` will store `result`)
 - A timestamp representing the last time the `byetypo_dictionary` was updated. (by default updated every week).
 
 This file is generated at the root of your application by default. If you want to update its location, you can configure the path by adding a `BYETYPO_STORE_PATH` entry in your `.env` file.
 
-### 2. Captured exceptions
+## Captured exceptions
+
+### NameError - undefined local variable or method
+
+This error occurs when you mispelled a variable in your REPL. The gem will catch that exception and will try find the closest matches. If so, it will run the command with the (potential) corrected variable.
+
+##### Before
+
+```ruby
+[1] pry(main)> result = 1
+=> 1
+[2] pry(main)> resilt
+NameError: undefined local variable or method `resilt' for main:Object
+from (pry):2:in `__pry__'
+```
+
+##### After
+
+```ruby
+[1] pry(main)> result = 1
+=> 1
+[2] pry(main)> resilt
+E, [2024-01-31T17:11:16.344161 #3739] ERROR -- : undefined local variable or method `resilt' for main:Object
+I, [2024-01-31T17:11:16.344503 #3739]  INFO -- : Running: result
+=> 1
+```
 
-#### NameError
+### NameError - uninitialized constant
 
 This error occurs when you mispelled a model in your REPL. The gem will catch that exception and will try find the closest matches. If so, it will run the command with the (potential) corrected model.
 
+##### Before
+
+```ruby
+[2] pry(main)> Usert.last
+NameError: uninitialized constant Usert
+from (pry):2:in `__pry__'
+[3] pry(main)>
+```
+
+##### After
+
 ```ruby
 [1] pry(main)> Usert.last
 I, [2024-01-13T20:00:16.280710 #694]  ERROR -- : uninitialized constant Usert
@@ -71,19 +109,19 @@ I, [2024-01-13T20:00:16.281237 #694]  INFO -- : Running: User.last
 => #<User id: 1, email: "yo@email.com">
 ```
 
-#### ActiveRecord::ConfigurationError
+### ActiveRecord::ConfigurationError
 
 Raised when association is being configured improperly or user tries to use offset and limit together with `ActiveRecord::Base.has_many` or `ActiveRecord::Base.has_and_belongs_to_many` associations.
+This plugin will look into the `byetypo_dictionary` file to find the closest match and run the correct query.
 
-eg:
+##### Before
 
 ```ruby
 [6] pry(main)> User.joins(:group).where(groups: { name: "Landlord" }).last
 ActiveRecord::ConfigurationError: Can't join 'User' to association named 'group'; perhaps you misspelled it?
 ```
 
-This plugin will look into the `byetypo_dictionary` file to find the closest match and run the correct query.
-
+##### After
 
 ```ruby
 [1] pry(main)> User.joins(:group).where(groups: { name: "Landlord" })
@@ -93,9 +131,12 @@ I, [2024-01-13T22:45:16.297972 #1079]  INFO -- : Running: User.joins(:groups).wh
 => []
 ```
 
-#### ActiveRecord::StatementInvalid
+### ActiveRecord::StatementInvalid
 
 The query attempts to reference columns or conditions related to a table, but the table is not properly included in the FROM clause.
+This plugin will look into the `byetypo_dictionary` file to find the closest match and run the correct query.
+
+##### Before
 
 ```ruby
 [1] pry(main)> User.joins(:groups).where(grous: { name: "Landlord" }).last
@@ -103,7 +144,7 @@ ActiveRecord::StatementInvalid: PG::UndefinedTable: ERROR:  missing FROM-clause
 LINE 1: ..."group_id" WHERE "users"."deleted_at" IS NULL AND "grous"."n...
 ```
 
-This plugin will look into the `byetypo_dictionary` file to find the closest match and run the correct query.
+##### After
 
 ```ruby
 1] pry(main)> User.joins(:groups).where(grous: { name: "Landlord" }).last
@@ -116,11 +157,11 @@ I, [2024-01-14T23:50:49.273177 #1248]  INFO -- : Running: User.joins(:groups).wh
 
 Pry-byetypo is linked to your development database. During initialization, it will attempt to establish a connection to retrieve the tables available in your project. It will fetch the information for the development environment from the `database.yml` file.
 
-### Unreadable database URL (URI::InvalidURIError)
+#### Unreadable database URL (URI::InvalidURIError)
 
 If the database connection string is not readable, the gem will be unable to establish a connection. If you encounter such an issue, make sure to add a `DATABASE_URL` variable to your `.env` file with the easily readable URL of your database.
 
-### Unreadable connection pool (ActiveRecord::ConnectionTimeoutError)
+#### Unreadable connection pool (ActiveRecord::ConnectionTimeoutError)
 
 If the number of connections in your pool is not readable, you may encounter an `ActiveRecord::ConnectionTimeoutError`. If you experience this issue, make sure to add a `DATABASE_POOL` variable to your `.env` file.
 

data/lib/pry-byetypo/constants/errors.rb

@@ -0,0 +1,7 @@
+module Constants
+  module Errors
+    UNINITIALIZED_CONSTANT = "uninitialized constant".freeze
+    UNDEFINED_VARIABLE = "undefined local variable".freeze
+    UNDEFINED_TABLE = "UndefinedTable".freeze
+  end
+end

data/lib/pry-byetypo/exceptions/active_record/base.rb

@@ -7,10 +7,6 @@ module Exceptions
     class Base < ExceptionsBase
       private
 
-      def corrected_word
-        @corrected_word ||= spell_checker(associations_dictionary).correct(unknown_from_exception).first
-      end
-
       def corrected_cmd
         @corrected_cmd ||= last_cmd.gsub(/\b#{unknown_from_exception}\b/, corrected_word)
       end
@@ -19,8 +15,8 @@ module Exceptions
         exception.to_s.match(exception_regexp)[1]
       end
 
-      def associations_dictionary
-        @associations_dictionary ||= store.transaction { |s| s["associations"] }
+      def dictionary
+        store.transaction { |s| s["associations"] }
       end
     end
   end

data/lib/pry-byetypo/exceptions/active_record/uninitialized_constant.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Exceptions
+  module ActiveRecord
+    class UninitializedConstant < Base
+      private
+
+      def unknown_from_exception
+        exception.to_s.split.last
+      end
+
+      def corrected_cmd
+        @corrected_cmd ||= last_cmd.gsub(unknown_from_exception, corrected_word)
+      end
+
+      def dictionary
+        @ar_models_dictionary ||= store.transaction { |s| s["active_record_models"] }
+      end
+    end
+  end
+end

data/lib/pry-byetypo/exceptions/exceptions_base.rb

@@ -1,17 +1,22 @@
 # frozen_string_literal: true
 
+require "logger"
+require "colorize"
 require_relative "../base"
 require_relative "../setup/store"
-require "colorize"
 
 class ExceptionsBase < Base
   include Setup::Store
 
   def call
-    logger.error(exception.to_s.colorize(color: :light_red, mode: :bold))
-    logger.info("Running: #{corrected_cmd}".colorize(color: :green, mode: :bold))
-
-    pry.eval(corrected_cmd)
+    if corrected_word
+      logger.error(exception.to_s.colorize(color: :light_red, mode: :bold))
+      logger.info("Running: #{corrected_cmd}".colorize(color: :green, mode: :bold))
+
+      pry.eval(corrected_cmd)
+    else
+      Pry::ExceptionHandler.handle_exception(output, exception, pry)
+    end
   end
 
   private
@@ -24,7 +29,11 @@ class ExceptionsBase < Base
     @pry = pry
   end
 
-  def spell_checker(dictionary)
+  def corrected_word
+    @corrected_word ||= spell_checker.correct(unknown_from_exception).first
+  end
+
+  def spell_checker
     DidYouMean::SpellChecker.new(dictionary: dictionary)
   end
 

data/lib/pry-byetypo/exceptions/name_error/handler.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require_relative "../exceptions_base"
+require_relative "../../constants/errors"
+require_relative "../active_record/uninitialized_constant"
+require_relative "undefined_variable"
+
+module Exceptions
+  module NameError
+    class Handler < ExceptionsBase
+      attr_reader :exception, :output, :pry
+
+      def initialize(output, exception, pry)
+        @output = output
+        @exception = exception
+        @pry = pry
+      end
+
+      # FIXME: https://github.com/rubocop/rubocop-performance/issues/438
+      # rubocop:disable Performance/ConstantRegexp
+      def call
+        case exception.message
+        in /#{Constants::Errors::UNINITIALIZED_CONSTANT}/
+          ActiveRecord::UninitializedConstant.call(output, exception, pry)
+        in /#{Constants::Errors::UNDEFINED_VARIABLE}/
+          UndefinedVariable.call(output, exception, pry)
+        else
+          Pry::ExceptionHandler.handle_exception(output, exception, pry)
+        end
+      end
+      # rubocop:enable Performance/ConstantRegexp
+    end
+  end
+end

data/lib/pry-byetypo/exceptions/name_error/undefined_variable.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require_relative "../exceptions_base"
+
+module Exceptions
+  module NameError
+    class UndefinedVariable < ExceptionsBase
+      private
+
+      def corrected_cmd
+        @corrected_cmd ||= last_cmd.gsub(/\b#{unknown_from_exception}\b/, corrected_word)
+      end
+
+      def unknown_from_exception
+        exception.to_s.match(exception_regexp)[1]
+      end
+
+      def dictionary
+        store.transaction { |s| s[pry_instance_uid] }
+      end
+
+      def exception_regexp
+        /`(\w+)'/
+      end
+
+      # Use the current binding identifier as pry instance uid.
+      def pry_instance_uid
+        pry.current_binding.to_s
+      end
+    end
+  end
+end

data/lib/pry-byetypo/exceptions_handler.rb

@@ -3,14 +3,12 @@
 require_relative "base"
 require_relative "exceptions/active_record/statement_invalid"
 require_relative "exceptions/active_record/configuration_error"
-require_relative "exceptions/name_error"
+require_relative "exceptions/name_error/handler"
+require_relative "constants/errors"
 
 class ExceptionsHandler < Base
   attr_reader :exception, :output, :pry
 
-  UNINITIALIZED_CONSTANT = "uninitialized constant"
-  UNDEFINED_TABLE = "UndefinedTable"
-
   def initialize(output, exception, pry)
     @output = output
     @exception = exception
@@ -19,20 +17,15 @@ class ExceptionsHandler < Base
 
   def call
     case exception
-    in NameError => error
-      # eg: Usert.last
+    in NameError
       # NameError is a Superclass for all undefined statement.
-      # In this context We only need to one including an `uninitialized constant` error.
-      return pry_exception_handler unless error.message.include?(UNINITIALIZED_CONSTANT)
-      Exceptions::NameError.call(output, exception, pry)
-    in ActiveRecord::StatementInvalid => error
-      # eg: User.joins(:groups).where(grous: { name: "Landlord" }).last
+      Exceptions::NameError::Handler.call(output, exception, pry)
+    in ActiveRecord::StatementInvalid
       # ActiveRecord::StatementInvalid is a Superclass for all database execution errors.
       # We only need to one including an `UndefinedTable` error.
-      return pry_exception_handler unless error.message.include?(UNDEFINED_TABLE)
+      return pry_exception_handler unless exception.message.include?(Constants::Errors::UNDEFINED_TABLE)
       Exceptions::ActiveRecord::StatementInvalid.call(output, exception, pry)
     in ActiveRecord::ConfigurationError
-      # eg:  User.joins(:group).where(groups: { name: "Landlord" }).last
       Exceptions::ActiveRecord::ConfigurationError.call(output, exception, pry)
     else
       pry_exception_handler

data/lib/pry-byetypo/session/clear_history.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require_relative "../base"
+require_relative "../setup/store"
+
+module Session
+  class ClearHistory < Base
+    include Setup::Store
+
+    attr_reader :pry
+
+    def initialize(pry)
+      @pry = pry
+    end
+
+    def call
+      pry_instance_to_remove = pry.push_initial_binding.join
+      store.transaction { store.delete(pry_instance_to_remove) }
+    end
+  end
+end

data/lib/pry-byetypo/session/populate_history.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require_relative "../base"
+require_relative "../setup/store"
+
+module Session
+  class PopulateHistory < Base
+    include Setup::Store
+
+    attr_reader :binding
+
+    def initialize(binding)
+      @binding = binding
+    end
+
+    def call
+      return unless is_assignement_variables?
+
+      store.transaction do
+        store.abort unless variables_to_store
+
+        store[pry_instance_uid].push(*variables_to_store)
+      end
+    end
+
+    private
+
+    # Unique instance identifier (e.g., a Pry opened tab)
+    def pry_instance_uid
+      binding.binding_stack.join
+    end
+
+    def variables_to_store
+      @variables_to_store ||= last_cmd.split("=").first.strip.split(",")
+    end
+
+    def last_cmd
+      binding.eval_string.strip
+    end
+
+    # Returns true if the last command seems to be an assignment of variables, false otherwise.
+    #
+    # Examples
+    #
+    #   is_assignment_variables?("user_last, user_first = User.last, User.first")
+    #   # => true
+    #
+    #   is_assignment_variables?("user_last = User.last")
+    #   # => true
+    #
+    #   is_assignment_variables?("user_last")
+    #   # => false
+    def is_assignement_variables?
+      last_cmd.include?("=")
+    end
+  end
+end

data/lib/pry-byetypo/setup/application_dictionary.rb

@@ -1,73 +1,35 @@
 # frozen_string_literal: true
 
-require_relative "checks/database_url"
-require_relative "checks/database_pool"
+require_relative "../base"
 require_relative "store"
-
+require_relative "dictionary/active_record"
 require "pstore"
 
 module Setup
-  class ApplicationDictionary
+  class ApplicationDictionary < Base
     include Store
 
-    def initialize
-      establish_db_connection
+    def initialize(binding)
+      @binding = binding
+    end
+
+    def call
+      Setup::Dictionary::ActiveRecord.initialize! if defined?(ActiveRecord)
       populate_store
     end
 
     private
 
-    SEVEN_DAYS = 604800
+    attr_reader :binding
 
     def populate_store
-      store.transaction do
-        store.abort unless staled_store?
-
-        store["active_record_models"] = populate_active_record_models_dictionary
-        store["associations"] = populate_associations
-        store["synced_at"] = Time.now
-      end
-    end
-
-    def establish_db_connection
-      configuration_checks
-      ActiveRecord::Base.establish_connection(development_database_config)
-    end
-
-    def populate_active_record_models_dictionary
-      Zeitwerk::Loader.eager_load_all
-      ActiveRecord::Base.descendants.map { |model| model.name }
-    end
-
-    def populate_associations
-      table_names = ActiveRecord::Base.connection.tables
-      singularize_table_names = table_names.map { |a| a.chop }
-      table_names.zip(singularize_table_names).flatten
-    end
-
-    def configuration_checks
-      Setup::Checks::DatabaseUrl.check(development_database_config, logger)
-      Setup::Checks::DatabasePool.check(development_database_config, logger)
-    end
-
-    def database_config
-      YAML.safe_load(File.read("./config/database.yml"), aliases: true)
-    end
-
-    def development_database_config
-      @development_database_config ||= database_config["development"]
-    end
-
-    # By default we update the store every week.
-    # TODO: Make it configurable
-    def staled_store?
-      return true if store["synced_at"].nil?
-
-      (store["synced_at"] + SEVEN_DAYS) <= Time.now
+      # Create a table with unique instance identifier information to store variables history.
+      store.transaction { store[pry_instance_uid] = [] }
     end
 
-    def logger
-      @logger = Logger.new($stdout)
+    # Use the binding identifier as pry instance uid.
+    def pry_instance_uid
+      binding.to_s
     end
   end
 end

data/lib/pry-byetypo/setup/dictionary/active_record.rb

@@ -0,0 +1,81 @@
+# frozen_string_literal: true
+
+require_relative "../checks/database_url"
+require_relative "../checks/database_pool"
+require_relative "../store"
+
+require "pstore"
+
+module Setup
+  module Dictionary
+    class ActiveRecord
+      class << self
+        include Store
+
+        def initialize!
+          establish_db_connection
+          populate_store
+        end
+
+        private
+
+        def populate_store
+          store.transaction do
+            store.commit unless staled_store?
+
+            store["active_record_models"] = populate_active_record_models_dictionary
+            store["associations"] = populate_associations
+            store["synced_at"] = Time.now
+          end
+        end
+
+        def establish_db_connection
+          configuration_checks
+          ::ActiveRecord::Base.establish_connection(development_database_config)
+        end
+
+        def populate_active_record_models_dictionary
+          Zeitwerk::Loader.eager_load_all
+          ::ActiveRecord::Base.descendants.map { |model| format_active_record_model(model) }.flatten
+        end
+
+        def populate_associations
+          table_names = ::ActiveRecord::Base.connection.tables
+          singularize_table_names = table_names.map { |a| a.chop }
+          table_names.zip(singularize_table_names).flatten
+        end
+
+        # This method takes an ActiveRecord model as an argument and formats its name and module information.
+        # If the model is within a module namespace, it returns an array containing the model's name and an array of its modules.
+        # If the model is not within a module, it returns just the model's name as a string.
+        def format_active_record_model(model)
+          modules = model.name.split("::")
+          return model.name, modules if modules.count > 1
+
+          model.name
+        end
+
+        def development_database_config
+          @development_database_config ||= database_config["development"]
+        end
+
+        def configuration_checks
+          Setup::Checks::DatabaseUrl.check(development_database_config, logger)
+          Setup::Checks::DatabasePool.check(development_database_config, logger)
+        end
+
+        def database_config
+          YAML.safe_load(File.read(database_config_path), aliases: true)
+        end
+
+        def logger
+          @logger = Logger.new($stdout)
+        end
+
+        def database_config_path
+          ENV["DB_CONFIG_PATH"] || "./config/database.yml"
+        end
+      end
+    end
+  end
+end

data/lib/pry-byetypo/setup/store.rb

@@ -1,6 +1,7 @@
 module Setup
   module Store
     DEFAULT_STORE_PATH = "byetypo_dictionary.pstore"
+    SEVEN_DAYS = 604800
 
     def store
       @store ||= PStore.new(store_path)
@@ -9,5 +10,13 @@ module Setup
     def store_path
       ENV["BYETYPO_STORE_PATH"] || DEFAULT_STORE_PATH
     end
+
+    # By default we update the store every week.
+    # TODO: Make it configurable
+    def staled_store?
+      return true if store["synced_at"].nil?
+
+      (store["synced_at"] + SEVEN_DAYS) <= Time.now
+    end
   end
 end

data/lib/pry-byetypo/version.rb

@@ -2,6 +2,6 @@
 
 class Pry
   module Byetypo
-    VERSION = "1.0.2"
+    VERSION = "1.2.0"
   end
 end

data/lib/pry-byetypo.rb

@@ -1,15 +1,24 @@
 # frozen_string_literal: true
 
 require "pry"
-require "zeitwerk"
 
-require_relative "pry-byetypo/version"
 require_relative "pry-byetypo/setup/application_dictionary"
+require_relative "pry-byetypo/session/clear_history"
 require_relative "pry-byetypo/exceptions_handler"
+require_relative "pry-byetypo/session/populate_history"
+require_relative "pry-byetypo/version"
 
 module Pry::Byetypo
-  Pry.config.hooks.add_hook(:before_session, :eager_loading) do |output, exception, pry|
-    Setup::ApplicationDictionary.new
+  Pry.config.hooks.add_hook(:before_session, :create_dictionary) do |_output, binding, _pry|
+    Setup::ApplicationDictionary.call(binding)
+  end
+
+  Pry.config.hooks.add_hook(:after_read, :populate_session_history) do |_output, binding, _pry|
+    Session::PopulateHistory.call(binding)
+  end
+
+  Pry.config.hooks.add_hook(:after_session, :clear_session_history) do |_output, _binding, pry|
+    Session::ClearHistory.call(pry)
   end
 
   # TODO: Adds max_attempts

data/pry-byetypo.gemspec

@@ -5,14 +5,14 @@ require_relative "lib/pry-byetypo/version"
 Gem::Specification.new do |gem|
   gem.name = "pry-byetypo"
   gem.version = Pry::Byetypo::VERSION
-  gem.authors = ["morissetcl"]
+  gem.authors = ["Clément Morisset"]
   gem.email = ["morissetcl87@gmail.com"]
 
   gem.summary = "Autocorrects typos in your Pry console"
-  gem.description = "This small Pry plugin captures exceptions that could be due to typos and deduces the correct command based on your database information."
+  gem.description = "This Pry plugin captures exceptions that could be due to typos and deduces the correct command based on your database information."
   gem.homepage = "https://github.com/morissetcl/pry-byetypo"
   gem.license = "MIT"
-  gem.required_ruby_version = ">= 2.6.0"
+  gem.required_ruby_version = ">= 2.7.0"
 
   gem.metadata["homepage_uri"] = gem.homepage
   gem.metadata["source_code_uri"] = "https://github.com/morissetcl/pry-byetypo"
@@ -28,7 +28,7 @@ Gem::Specification.new do |gem|
   gem.bindir = "exe"
   gem.executables = gem.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
   gem.require_paths = ["lib"]
+  gem.add_development_dependency "rails", "~> 7.0"
   gem.add_runtime_dependency "colorize", "~> 1.1.0"
   gem.add_runtime_dependency "pry", ">= 0.13", "< 0.15"
-  gem.add_runtime_dependency "rails", "~> 7.0"
 end

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: pry-byetypo
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.2.0
 platform: ruby
 authors:
-- morissetcl
+- Clément Morisset
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-01-29 00:00:00.000000000 Z
+date: 2024-02-21 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.0'
 - !ruby/object:Gem::Dependency
   name: colorize
   requirement: !ruby/object:Gem::Requirement
@@ -44,22 +58,8 @@ dependencies:
     - - "<"
       - !ruby/object:Gem::Version
         version: '0.15'
-- !ruby/object:Gem::Dependency
-  name: rails
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '7.0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '7.0'
-description: This small Pry plugin captures exceptions that could be due to typos
-  and deduces the correct command based on your database information.
+description: This Pry plugin captures exceptions that could be due to typos and deduces
+  the correct command based on your database information.
 email:
 - morissetcl87@gmail.com
 executables: []
@@ -77,16 +77,22 @@ files:
 - Rakefile
 - lib/pry-byetypo.rb
 - lib/pry-byetypo/base.rb
+- lib/pry-byetypo/constants/errors.rb
 - lib/pry-byetypo/exceptions/active_record/base.rb
 - lib/pry-byetypo/exceptions/active_record/configuration_error.rb
 - lib/pry-byetypo/exceptions/active_record/statement_invalid.rb
+- lib/pry-byetypo/exceptions/active_record/uninitialized_constant.rb
 - lib/pry-byetypo/exceptions/exceptions_base.rb
-- lib/pry-byetypo/exceptions/name_error.rb
+- lib/pry-byetypo/exceptions/name_error/handler.rb
+- lib/pry-byetypo/exceptions/name_error/undefined_variable.rb
 - lib/pry-byetypo/exceptions_handler.rb
+- lib/pry-byetypo/session/clear_history.rb
+- lib/pry-byetypo/session/populate_history.rb
 - lib/pry-byetypo/setup/application_dictionary.rb
 - lib/pry-byetypo/setup/checks/base.rb
 - lib/pry-byetypo/setup/checks/database_pool.rb
 - lib/pry-byetypo/setup/checks/database_url.rb
+- lib/pry-byetypo/setup/dictionary/active_record.rb
 - lib/pry-byetypo/setup/store.rb
 - lib/pry-byetypo/version.rb
 - pry-byetypo.gemspec
@@ -106,7 +112,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 2.6.0
+      version: 2.7.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="

data/lib/pry-byetypo/exceptions/name_error.rb

@@ -1,25 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "exceptions_base"
-
-module Exceptions
-  class NameError < ExceptionsBase
-    private
-
-    def unknown_from_exception
-      exception.to_s.split.last
-    end
-
-    def corrected_word
-      @corrected_word ||= spell_checker(ar_models_dictionary).correct(unknown_from_exception).first
-    end
-
-    def corrected_cmd
-      @corrected_cmd ||= last_cmd.gsub(unknown_from_exception, corrected_word)
-    end
-
-    def ar_models_dictionary
-      @ar_models_dictionary ||= store.transaction { |s| s["active_record_models"] }
-    end
-  end
-end