lib/tfw/state.rb in tfw-0.1.9 vs lib/tfw/state.rb in tfw-0.1.10

@@ -1,19 +1,21 @@
 # frozen_string_literal: true
 
-# Singleton class to keep the stack
-class State
-  include Singleton
+module TFW
+  # Singleton class to keep the stack
+  class State
+    include Singleton
 
-  def initialize
-    @stack = TFDSL::Stack.new
-  end
+    def initialize
+      @stack = TFDSL::Stack.new
+    end
 
-  def stack(&block)
-    @stack.instance_eval(&block) if block_given?
-    @stack
-  end
+    def stack(&block)
+      @stack.instance_eval(&block) if block_given?
+      @stack
+    end
 
-  def reset
-    @stack = TFDSL::Stack.new
+    def reset
+      @stack = TFDSL::Stack.new
+    end
   end
 end