add deep_freeze refinement to support

pakyow-core/lib/pakyow/core/app.rb

@@ -2,6 +2,7 @@
 require "pakyow/support/definable"
 require "pakyow/support/hookable"
 require "pakyow/support/recursive_require"
+require "pakyow/support/deep_freeze"
 
 require "pakyow/core/helpers"
 require "pakyow/core/router"
@@ -318,6 +319,10 @@ def reset
       end
     end
 
+    using Support::DeepFreeze
+
+    unfreezable :builder
+
     # @api private
     def initialize(environment, builder: nil, &block)
       @environment = environment

pakyow-core/lib/pakyow/core/router.rb

@@ -748,30 +748,6 @@ def merge(router)
         merge_templates(router.templates)
       end
 
-      # @api private
-      def freeze
-        # TODO: let's instead have a deep freeze method that freezes the object and its ivars, recursively
-        hooks.each do |_, hooks_arr|
-          hooks_arr.each(&:freeze)
-          hooks_arr.freeze
-        end
-
-        children.each(&:freeze)
-
-        routes.each do |_, routes_arr|
-          routes_arr.each(&:freeze)
-          routes_arr.freeze
-        end
-
-        matcher.freeze
-        hooks.freeze
-        children.freeze
-        routes.freeze
-        templates.freeze
-
-        super
-      end
-
       # @api private
       def exception_for_class(klass, exceptions: {})
         self.exceptions.merge(exceptions)[klass]

pakyow-core/lib/pakyow/core/routing/route.rb

@@ -59,19 +59,6 @@ def populated_path(path_to_self, **params)
         }.join("/"))
       end
 
-      def freeze
-        hooks.each do |_, hooks_arr|
-          hooks_arr.each(&:freeze)
-          hooks_arr.freeze
-        end
-
-        path.freeze
-        pipeline.freeze
-        hooks.freeze
-
-        super
-      end
-
       protected
 
       def create_matcher_from_path(path)

pakyow-support/lib/pakyow/support/deep_freeze.rb

@@ -0,0 +1,52 @@
+module Pakyow
+  module Support
+    module DeepFreeze
+      refine Object.singleton_class do
+        def unfreezable_variables
+          @unfreezable_variables ||= []
+        end
+
+        private
+
+        def unfreezable(*ivars)
+          ivars = ivars.map { |ivar| "@#{ivar}".to_sym }
+          unfreezable_variables.concat(ivars)
+          unfreezable_variables.uniq!
+        end
+      end
+
+      refine Object do
+        def deep_freeze
+          return self if frozen?
+
+          self.freeze
+
+          freezable_variables.each do |name|
+            instance_variable_get(name).deep_freeze
+          end
+
+          self
+        end
+
+        private
+
+        def freezable_variables
+          instance_variables - self.class.unfreezable_variables
+        end
+      end
+
+      refine Array do
+        def deep_freeze
+          return self if frozen?
+
+          self.freeze
+
+          each(&:deep_freeze)
+
+          self
+        end
+      end
+    end
+  end
+end
+

pakyow-support/lib/pakyow/support/definable.rb

@@ -1,4 +1,5 @@
 require "pakyow/support/deep_dup"
+require "pakyow/support/deep_freeze"
 
 module Pakyow
   module Support
@@ -28,6 +29,7 @@ module Support
     #
     module Definable
       using DeepDup
+      using DeepFreeze
 
       def self.included(base)
         base.extend ClassAPI
@@ -58,7 +60,7 @@ def initialize(&block)
         end
 
         # instance state is now immutable
-        freeze
+        deep_freeze
       end
 
       # Returns register instances for state.
@@ -68,13 +70,6 @@ def state_for(type)
         @state[type].instances
       end
 
-      # @api private
-      def freeze
-        @state.each { |_, state| state.freeze }
-        @state.freeze
-        super
-      end
-
       module ClassAPI
         attr_reader :state, :inherited_state
 
@@ -201,12 +196,6 @@ def register(instance, priority: :default)
         reprioritize!
       end
 
-      def freeze
-        instances.each(&:freeze)
-        instances.freeze
-        super
-      end
-
       def reset
         object.reset
         @instances = []

pakyow-support/spec/deep_freeze_spec.rb

@@ -0,0 +1,99 @@
+require "pakyow/support/deep_freeze"
+
+using Pakyow::Support::DeepFreeze
+
+class MyObject
+  unfreezable :fire
+
+  attr_reader :fire, :water
+
+  def initialize
+    @fire = 'fire'
+    @water = 'water'
+  end
+end
+
+RSpec.describe Pakyow::Support::DeepFreeze do
+  let :obj_with_unfreezable { 
+    MyObject.new
+  }
+
+  describe "deep_freeze" do
+    it "returns the object" do
+      obj = Object.new
+
+      expect(obj.deep_freeze.object_id).to be(obj.object_id)
+    end
+
+    it "deep freezes an Object without ivars" do
+      obj = Object.new
+      obj.deep_freeze
+
+      expect(obj).to be_frozen
+    end
+
+    it "deep freezes an Object with ivars" do
+      ivar_value = Object.new
+      obj = Object.new
+      obj.instance_variable_set(:@ivar_name, ivar_value)
+
+      obj.deep_freeze
+
+      expect(obj).to be_frozen
+      expect(ivar_value).to be_frozen
+    end
+
+    it "deep freezes ivars" do
+      objects = Array.new(5) { Object.new }
+      objects.reduce do |parent, child|
+        parent.instance_variable_set(:@child, child)
+      end
+
+      objects.first.deep_freeze
+
+      objects.each do |obj|
+        expect(obj).to be_frozen
+      end
+    end
+
+    it "deep freezes ivars without loop" do
+      objects = Array.new(5, Object.new)
+      objects.reduce do |parent, child|
+        parent.instance_variable_set(:@child, child)
+      end
+
+      objects.first.deep_freeze
+
+      objects.each do |obj|
+        expect(obj).to be_frozen
+      end
+    end
+
+    it "deep freezes array items" do
+      objects = Array.new(5) { Object.new }
+
+      objects.deep_freeze
+
+      objects.each do |obj|
+        expect(obj).to be_frozen
+      end
+
+      expect(objects).to be_frozen
+    end
+
+    it "doesn't freeze unfreezeable" do
+      obj_with_unfreezable.deep_freeze
+
+      expect(obj_with_unfreezable).to be_frozen
+      expect(obj_with_unfreezable.water).to be_frozen
+      expect(obj_with_unfreezable.fire).not_to be_frozen
+    end
+
+    it "returns an array of unfreezeable ivars" do
+      vars = MyObject.unfreezable_variables
+
+      expect(vars).to be_instance_of(Array)
+    end
+  end
+end
+