move site and collections console commands to locals (#853)

bridgetownrb · Mar 15, 2024 · c8ab4a1 · c8ab4a1
1 parent c7738e4
commit c8ab4a1
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/bridgetown-core/lib/bridgetown-core/commands/console.rb b/bridgetown-core/lib/bridgetown-core/commands/console.rb
@@ -2,17 +2,11 @@
 
 module Bridgetown
   module ConsoleMethods
-    def site
-      Bridgetown::Current.site
-    end
-
-    def collections
-      site.collections
-    end
-
     def reload!
       Bridgetown.logger.info "Reloading site..."
 
+      site = Bridgetown::Current.site
+
       I18n.reload! # make sure any locale files get read again
       Bridgetown::Hooks.trigger :site, :pre_reload, site
       Bridgetown::Hooks.clear_reloadable_hooks
@@ -98,7 +92,14 @@ def console # rubocop:disable Metrics
 
         IRB::ExtendCommandBundle.include ConsoleMethods
         IRB.setup(nil)
-        workspace = IRB::WorkSpace.new
+        workspace = IRB::WorkSpace.new(
+          begin
+            site = Bridgetown::Current.site
+            collections = site.collections
+
+            binding
+          end
+        )
         irb = IRB::Irb.new(workspace)
         IRB.conf[:IRB_RC]&.call(irb.context)
         IRB.conf[:MAIN_CONTEXT] = irb.context