lib/hieracles/registry.rb in hieracles-0.3.5 vs lib/hieracles/registry.rb in hieracles-0.3.6

@@ -20,24 +20,49 @@
       Dir.glob(File.join(config.modulepath, '*')).sort.map do |f|
         File.basename(f)
       end
     end
 
-    def nodes_data(config, env = 'production', reload = false)
-      @_nodes_data = {} if reload || !@_nodes_data
-      @_nodes_data[env] ||= Dir.glob(File.join(config.encpath, '*.yaml')).sort.reduce({}) do |a, f|
+    def reload_nodes()
+      @_nodes_parameters = {}
+      @_nodes_modules = {}
+    end
+
+    def nodes_parameters(config, env = 'production')
+      @_nodes_parameters ||= {}
+      @_nodes_parameters[env] ||= Dir.glob(File.join(config.encpath, '*.yaml')).sort.reduce({}) do |a, f|
         fqdn = File.basename(f, '.yaml')
         a[fqdn] = YAML.load_file(f)['parameters']
         a
       end
     end
 
-    def farms_counted(config, env = 'production')
+    def nodes_modules(config, env = 'production')
+      @_nodes_modules ||= {}
+      @_nodes_modules[env] ||= Dir.glob(File.join(config.encpath, '*.yaml')).sort.reduce({}) do |a, f|
+        YAML.load_file(f)['classes'].each do |cl|
+
+        end
+        fqdn = File.basename(f, '.yaml')
+        a[fqdn] = YAML.load_file(f)['parameters']
+        a
+      end
+    end
+
+    def farms_counted(config, env = 'production', reload = false)
+      reload_nodes if reload
       Dir.glob(format(config.classpath, '*')).sort.reduce({}) do |a, f|
         sub = Regexp.new(".*#{config.classpath.sub(/%s/,'([^/]*)')}")
         name = f.sub(sub, "\\1")
-        a[name] = nodes_data(config, env).select { |k, v| v['farm'] == name }.length
+        a[name] = nodes_parameters(config, env).select { |k, v| v['farm'] == name }.length
         a
+      end
+    end
+
+    def modules_counted(config, env = 'production', reload = false)
+      reload_nodes if reload
+      Dir.glob(File.join(config.modulepath, '*')).sort.reduce({}) do |acc, mod|
+        acc
       end
     end
 
 	end
 end