lib/reek/smells/smells.rb in reek-0.3.1 vs lib/reek/smells/smells.rb in reek-1.0.0

@@ -1,24 +1,85 @@
-$:.unshift File.dirname(__FILE__)
-
 require 'reek/smells/control_couple'
 require 'reek/smells/duplication'
 require 'reek/smells/feature_envy'
+require 'reek/smells/large_class'
 require 'reek/smells/long_method'
 require 'reek/smells/long_parameter_list'
 require 'reek/smells/long_yield_list'
 require 'reek/smells/nested_iterators'
 require 'reek/smells/uncommunicative_name'
 require 'reek/smells/utility_function'
+require 'yaml'
 
-module Reek
+class Hash
+  def value_merge!(other)
+    other.keys.each do |key|
+      self[key].adopt!(other[key])
+    end
+    self
+  end
   
-  SMELLS = {
-    :defn => [
-      Smells::UncommunicativeName,
-      Smells::LongMethod,
+  def adopt!(other)
+    other.keys.each do |key|
+      ov = other[key]
+      if Array === ov and has_key?(key)
+        self[key] += ov
+      else
+        self[key] = ov
+      end
+    end
+    self
+  end
+  
+  def adopt(other)
+    self.deep_copy.adopt!(other)
+  end
+  
+  def deep_copy
+    YAML::load(YAML::dump(self))
+  end
+end
+
+module Reek
+  class SmellConfig
+
+    SMELL_CLASSES = [
+      Smells::ControlCouple,
       Smells::Duplication,
+      Smells::FeatureEnvy,
+      Smells::LargeClass,
+      Smells::LongMethod,
+      Smells::LongParameterList,
+      Smells::LongYieldList,
+      Smells::NestedIterators,
+      Smells::UncommunicativeName,
       Smells::UtilityFunction,
-      Smells::FeatureEnvy
-      ]
-  }
+    ]
+
+    def initialize
+      defaults_file = File.join(File.dirname(__FILE__), '..', '..', '..', 'config', 'defaults.reek')
+      @config = YAML.load_file(defaults_file)
+    end
+
+    def smell_listeners()
+      result = Hash.new {|hash,key| hash[key] = [] }
+      SMELL_CLASSES.each { |smell| smell.listen(result, @config) }
+      return result
+    end
+
+    def load_local(file)
+      path = File.expand_path(file)
+      all_reekfiles(path).each do |rfile|
+        cf = YAML.load_file(rfile)
+        @config.value_merge!(cf)
+      end
+      self
+    end
+
+    def all_reekfiles(path)
+      return [] unless File.exist?(path)
+      parent = File.dirname(path)
+      return [] if path == parent
+      all_reekfiles(parent) + Dir["#{path}/*.reek"]
+    end
+  end
 end