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

@@ -3,12 +3,12 @@
   module Registry
   	extend self
 
   	def farms(config)
   		Dir.glob(format(config.classpath, '*')).sort.map do |f|
-        sub = Regexp.new(".*#{config.classpath.sub(/%s/,'([^/]*)')}")
-        f.sub(sub, "\\1")
+        extract_path = Regexp.new(".*#{config.classpath.sub(/%s/,'([^/]*)')}")
+        f.sub(extract_path, "\\1")
       end
   	end
 
     def nodes(config)
       Dir.glob(File.join(config.encpath, '*.yaml')).sort.map do |f|
@@ -23,22 +23,35 @@
     end
 
     def reload_nodes()
       @_nodes_parameters = {}
       @_nodes_modules = {}
+      @_farms_modules = {}
     end
 
-    def nodes_parameters(config, env = 'production')
+    def nodes_parameters(config, env = 'production', reload = true)
+      reload_nodes if reload
       @_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 nodes_modules(config, env = 'production')
+    def farms_modules(config, env = 'production')
+      @_farms_modules ||= {}
+      regex = Regexp.new('\s*include\s*([-_a-z0-9]*)')
+      extract_path = Regexp.new(".*#{config.classpath.sub(/%s/,'([^/]*)')}")
+      @_farms_modules[env] ||= Dir.glob(format(config.classpath, '*')).sort.reduce({}) do |a, f|
+        name = f.sub(extract_path, "\\1")
+        a[name] = find_item(f, regex)
+        a
+      end
+    end
+
+    def nodes_modules(config, env = 'production', filter = nil)
       @_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
@@ -48,20 +61,63 @@
       end
     end
 
     def farms_counted(config, env = 'production', reload = false)
       reload_nodes if reload
+      extract_path = Regexp.new(".*#{config.classpath.sub(/%s/,'([^/]*)')}")
       Dir.glob(format(config.classpath, '*')).sort.reduce({}) do |a, f|
-        sub = Regexp.new(".*#{config.classpath.sub(/%s/,'([^/]*)')}")
-        name = f.sub(sub, "\\1")
+        name = f.sub(extract_path, "\\1")
         a[name] = nodes_parameters(config, env).select { |k, v| v['farm'] == name }.length
         a
       end
     end
 
+    def farms_nodes(config, env = 'production', reload = false, filter = nil)
+      reload_nodes if reload
+      extract_path = Regexp.new(".*#{config.classpath.sub(/%s/,'([^/]*)')}")
+      Dir.glob(format(config.classpath, '*')).sort.reduce({}) do |a, f|
+        name = f.sub(extract_path, "\\1")
+        a[name] = nodes_parameters(config, env).select { |k, v| v['farm'] == name }.keys
+        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|
+        mod = File.basename(mod)
+        acc[mod] = farms_modules(config, env).select { |k, v| v.include? mod }.length
+        acc
+      end
+    end
+
+    def modules_farms(config, env = 'production', reload = false)
+      reload_nodes if reload
+      Dir.glob(File.join(config.modulepath, '*')).sort.reduce({}) do |acc, mod|
+        mod = File.basename(mod)
+        acc[mod] = farms_modules(config, env).select { |k, v| v.include? mod }.keys
+        acc
+      end
+    end
+
+    def modules_nodes(config, env = 'production', reload = false, filter = nil)
+      reload_nodes if reload
+      Dir.glob(File.join(config.modulepath, '*')).sort.reduce({}) do |acc, mod|
+        if filter and Regexp.new(filter[0]).match(mod)
+          mod = File.basename(mod)
+          farms = farms_modules(config, env).select { |k, v| v.include? mod }
+          # acc[mod] = farms.map { |f| farms_nodes(config, env, reload, f).values }
+          acc[mod] = farms
+        end
+        acc
+      end
+    end
+
+    def find_item(file, regexp)
+      File.readlines(file).reduce([]) do |acc, line|
+        if regexp.match line
+          acc.push $1
+        end
         acc
       end
     end
 
 	end