lib/grumlin/with_extension.rb in grumlin-0.23.0 vs lib/grumlin/with_extension.rb in grumlin-1.0.0.rc1

@@ -1,28 +1,26 @@
 # frozen_string_literal: true
 
-module Grumlin
-  module WithExtension
-    def with(name, value)
-      prev = self
-      strategy = if is_a?(with_step_class)
-                   prev = previous_step
-                   TraversalStrategies::OptionsStrategy.new(args.first.value.merge(name => value))
-                 else
-                   TraversalStrategies::OptionsStrategy.new({ name => value })
-                 end
-      with_step_class.new(:withStrategies, args: [strategy], previous_step: prev)
-    end
+module Grumlin::WithExtension
+  def with(name, value)
+    prev = self
+    strategy = if is_a?(with_step_class)
+                 prev = previous_step
+                 Grumlin::TraversalStrategies::OptionsStrategy.new(args.first.value.merge(name => value))
+               else
+                 Grumlin::TraversalStrategies::OptionsStrategy.new({ name => value })
+               end
+    with_step_class.new(:withStrategies, args: [strategy], previous_step: prev)
+  end
 
-    private
+  private
 
-    def with_step_class
-      @with_step_class ||= Class.new(shortcuts.step_class) do
-        include WithExtension
+  def with_step_class
+    @with_step_class ||= Class.new(shortcuts.step_class) do
+      include Grumlin::WithExtension
 
-        def with_step_class
-          self.class
-        end
+      def with_step_class
+        self.class
       end
     end
   end
 end