lib/volt/server/rack/component_code.rb in volt-0.8.14 vs lib/volt/server/rack/component_code.rb in volt-0.8.15

@@ -2,24 +2,26 @@
 require 'volt/server/component_templates'
 require 'volt/server/rack/asset_files'
 
 # Takes in the name and all component paths and has a .code
 # method that returns all of the ruby setup code for the component.
-class ComponentCode
-  def initialize(component_name, component_paths, client=true)
-    @component_name = component_name
-    @component_paths = component_paths
-    @client = client
-  end
+module Volt
+  class ComponentCode
+    def initialize(component_name, component_paths, client=true)
+      @component_name  = component_name
+      @component_paths = component_paths
+      @client          = client
+    end
 
-  def code
-    code = ''
+    def code
+      code = ''
 
-    asset_files = AssetFiles.new(@component_name, @component_paths)
-    asset_files.component_paths.each do |component_path, component_name|
-      code << ComponentTemplates.new(component_path, component_name, @client).code
-      code << "\n\n"
-    end
+      asset_files = AssetFiles.new(@component_name, @component_paths)
+      asset_files.component_paths.each do |component_path, component_name|
+        code << ComponentTemplates.new(component_path, component_name, @client).code
+        code << "\n\n"
+      end
 
-    return code
+      return code
+    end
   end
-end
\ No newline at end of file
+end