johnsbrn-has_many_polymorphs 2.13

data/lib/has_many_polymorphs/configuration.rb

@@ -0,0 +1,19 @@
+
+=begin rdoc
+Access the <tt>has_many_polymorphs_options</tt> hash in your Rails::Initializer.run#after_initialize block if you need to modify the behavior of Rails::Initializer::HasManyPolymorphsAutoload.
+=end
+
+module Rails #:nodoc:
+  class Configuration
+
+    def has_many_polymorphs_options
+      ::HasManyPolymorphs.options
+    end
+    
+    def has_many_polymorphs_options=(hash)
+      ::HasManyPolymorphs.options = HashWithIndifferentAccess.new(hash)
+    end
+    
+  end  
+end
+

data/lib/has_many_polymorphs/debugging_tools.rb

@@ -0,0 +1,103 @@
+
+=begin rdoc
+
+Debugging tools for Has_many_polymorphs. 
+
+Enable the different tools by setting the environment variable HMP_DEBUG. Settings with special meaning are <tt>"ruby-debug"</tt>, <tt>"trace"</tt>, and <tt>"dependencies"</tt>.
+
+== Code generation
+
+Enabled by default when HMP_DEBUG is set.
+
+Ouputs a folder <tt>generated_models/</tt> in RAILS_ROOT containing valid Ruby files explaining all the ActiveRecord relationships set up by the plugin, as well as listing the line in the plugin that called each particular association method.
+
+== Ruby-debug
+
+Enable by setting HMP_DEBUG to <tt>"ruby-debug"</tt>.
+
+Starts <tt>ruby-debug</tt> for the life of the process.
+
+== Trace
+
+Enable by setting HMP_DEBUG to <tt>"ruby-debug"</tt>.
+
+Outputs an indented trace of relevant method calls as they occur.
+
+== Dependencies
+
+Enable by setting HMP_DEBUG to <tt>"dependencies"</tt>.
+
+Turns on Rails' default dependency logging.
+
+=end
+
+_logger_warn "debug mode enabled"
+
+class << ActiveRecord::Base
+  COLLECTION_METHODS = [:belongs_to, :has_many, :has_and_belongs_to_many, :has_one, 
+      :has_many_polymorphs, :acts_as_double_polymorphic_join].each do |method_name|
+    alias_method "original_#{method_name}".to_sym, method_name
+    undef_method method_name
+  end      
+
+  unless defined? GENERATED_CODE_DIR
+    GENERATED_CODE_DIR = "#{RAILS_ROOT}/generated_models"
+  
+    begin
+      system "rm -rf #{GENERATED_CODE_DIR}" 
+      Dir.mkdir GENERATED_CODE_DIR
+    rescue Errno::EACCES
+      _logger_warn "no permissions for generated code dir: #{GENERATED_CODE_DIR}"
+    end
+
+    if File.exist? GENERATED_CODE_DIR
+      alias :original_method_missing :method_missing
+      def method_missing(method_name, *args, &block)
+        if COLLECTION_METHODS.include? method_name.to_sym
+          Dir.chdir GENERATED_CODE_DIR do
+            filename = "#{demodulate(self.name.underscore)}.rb"
+            contents = File.open(filename).read rescue "\nclass #{self.name}\n\nend\n"
+            line = caller[1][/\:(\d+)\:/, 1]
+            contents[-5..-5] = "\n  #{method_name} #{args[0..-2].inspect[1..-2]},\n     #{args[-1].inspect[1..-2].gsub(" :", "\n     :").gsub("=>", " => ")}\n#{ block ? "     #{block.inspect.sub(/\@.*\//, '@')}\n" : ""}     # called from line #{line}\n\n"
+            File.open(filename, "w") do |file| 
+              file.puts contents             
+            end
+          end
+          # doesn't actually display block contents
+          self.send("original_#{method_name}", *args, &block)
+        else
+          self.send(:original_method_missing, method_name, *args, &block)
+        end
+      end
+    end      
+    
+  end
+end
+
+case ENV['HMP_DEBUG']
+
+  when "ruby-debug"
+    require 'rubygems'
+    require 'ruby-debug'
+    Debugger.start
+    _logger_warn "ruby-debug enabled."
+
+  when "trace"
+    _logger_warn "method tracing enabled"  
+    $debug_trace_indent = 0
+    set_trace_func (proc do |event, file, line, id, binding, classname|
+      if id.to_s =~ /instantiate/ #/IRB|Wirble|RubyLex|RubyToken|Logger|ConnectionAdapters|SQLite3|MonitorMixin|Benchmark|Inflector|Inflections/ 
+        if event == 'call'
+          puts (" " * $debug_trace_indent) + "#{event}ed #{classname}\##{id} from #{file.split('/').last}::#{line}"
+          $debug_trace_indent += 1
+        elsif event == 'return'
+          $debug_trace_indent -= 1 unless $debug_trace_indent == 0
+          puts (" " * $debug_trace_indent) + "#{event}ed #{classname}\##{id}"
+        end
+      end
+    end)
+    
+  when "dependencies"
+    _logger_warn "dependency activity being logged"
+    (::Dependencies.log_activity = true) rescue nil
+end

data/lib/has_many_polymorphs/rake_task_redefine_task.rb

@@ -0,0 +1,35 @@
+
+# Redefine instead of chain a Rake task
+# http://www.bigbold.com/snippets/posts/show/2032
+
+module Rake
+  module TaskManager
+    def redefine_task(task_class, args, &block)
+      task_name, deps = resolve_args(args)
+      task_name = task_class.scope_name(@scope, task_name)
+      deps = [deps] unless deps.respond_to?(:to_ary)
+      deps = deps.collect {|d| d.to_s }
+      task = @tasks[task_name.to_s] = task_class.new(task_name, self)
+      task.application = self
+      task.add_comment(@last_comment)
+      @last_comment = nil
+      task.enhance(deps, &block)
+      task
+    end
+  end
+  class Task
+    class << self
+      def redefine_task(args, &block)
+        Rake.application.redefine_task(self, args, &block)
+      end
+    end
+  end
+end
+
+class Object
+  def silently
+    stderr, stdout, $stderr, $stdout = $stderr, $stdout, StringIO.new, StringIO.new
+    yield
+    $stderr, $stdout = stderr, stdout
+  end
+end

data/lib/has_many_polymorphs/reflection.rb

@@ -0,0 +1,58 @@
+module ActiveRecord #:nodoc:
+  module Reflection #:nodoc:
+    
+    module ClassMethods #:nodoc:
+      
+      # Update the default reflection switch so that <tt>:has_many_polymorphs</tt> types get instantiated. 
+      # It's not a composed method so we have to override the whole thing.
+      def create_reflection(macro, name, options, active_record)
+        case macro
+          when :has_many, :belongs_to, :has_one, :has_and_belongs_to_many
+            klass = options[:through] ? ThroughReflection : AssociationReflection
+            reflection = klass.new(macro, name, options, active_record)
+          when :composed_of
+            reflection = AggregateReflection.new(macro, name, options, active_record)
+          # added by has_many_polymorphs #
+          when :has_many_polymorphs
+            reflection = PolymorphicReflection.new(macro, name, options, active_record)
+        end
+        write_inheritable_hash :reflections, name => reflection
+        reflection
+      end
+      
+    end
+
+    class PolymorphicError < ActiveRecordError #:nodoc:
+    end
+    
+=begin rdoc
+
+The reflection built by the <tt>has_many_polymorphs</tt> method. 
+
+Inherits from ActiveRecord::Reflection::AssociationReflection.
+
+=end
+
+    class PolymorphicReflection < ThroughReflection
+      # Stub out the validity check. Has_many_polymorphs checks validity on macro creation, not on reflection.
+      def check_validity! 
+        # nothing
+      end                 
+
+      # Return the source reflection.
+      def source_reflection
+        # normally is the has_many to the through model, but we return ourselves, 
+        # since there isn't a real source class for a polymorphic target
+        self
+      end      
+      
+      # Set the classname of the target. Uses the join class name.
+      def class_name
+        # normally is the classname of the association target
+        @class_name ||= options[:join_class_name]
+      end
+                     
+    end
+ 
+  end
+end

data/lib/has_many_polymorphs/support_methods.rb

@@ -0,0 +1,84 @@
+
+class String
+
+  # Changes an underscored string into a class reference.
+  def _as_class
+    # classify expects self to be plural
+    self.classify.constantize
+  end
+
+  # For compatibility with the Symbol extensions.
+  alias :_singularize :singularize
+  alias :_pluralize :pluralize
+  alias :_classify :classify
+end
+
+class Symbol
+  
+  # Changes an underscored symbol into a class reference.
+  def _as_class; self.to_s._as_class; end
+  
+  # Changes a plural symbol into a singular symbol.
+  def _singularize; self.to_s.singularize.to_sym; end
+
+  # Changes a singular symbol into a plural symbol.
+  def _pluralize; self.to_s.pluralize.to_sym; end
+
+  # Changes a symbol into a class name string.
+  def _classify; self.to_s.classify; end
+end
+
+class Array
+
+  # Flattens the first level of self.
+  def _flatten_once
+    self.inject([]){|r, el| r + Array(el)}
+  end
+
+  # Rails 1.2.3 compatibility method. Copied from http://dev.rubyonrails.org/browser/trunk/activesupport/lib/active_support/core_ext/array/extract_options.rb?rev=7217 
+  def _extract_options!
+    last.is_a?(::Hash) ? pop : {}
+  end
+end
+
+class Hash
+
+  # An implementation of select that returns a Hash.
+  def _select
+    Hash[*self.select do |key, value|
+      yield key, value
+    end._flatten_once]
+  end
+end
+
+class Object
+
+  # Returns the metaclass of self.
+  def _metaclass; (class << self; self; end); end
+
+  # Logger shortcut.
+  def _logger_debug s
+    s = "** has_many_polymorphs: #{s}"
+    RAILS_DEFAULT_LOGGER.debug(s) if RAILS_DEFAULT_LOGGER
+  end  
+
+  # Logger shortcut.  
+  def _logger_warn s
+    s = "** has_many_polymorphs: #{s}"
+    if RAILS_DEFAULT_LOGGER
+      RAILS_DEFAULT_LOGGER.warn(s) 
+    else
+      $stderr.puts(s)
+    end    
+  end
+  
+end
+
+class ActiveRecord::Base
+
+  # Return the base class name as a string.
+  def _base_class_name
+    self.class.base_class.name.to_s
+  end
+  
+end

data/test/fixtures/bow_wows.yml

@@ -0,0 +1,10 @@
+rover:
+  id: 1
+  name: Rover
+  created_at: "2007-01-01 12:00:00"
+  updated_at: "2007-01-04 10:00:00"
+spot:
+  id: 2
+  name: Spot
+  created_at: "2007-01-02 12:00:00"
+  updated_at: "2007-01-03 10:00:00"

data/test/fixtures/cats.yml

@@ -0,0 +1,18 @@
+chloe:
+  id: 1
+  cat_type: Kitten
+  name: Chloe
+  created_at: "2007-04-01 12:00:00"
+  updated_at: "2007-04-04 10:00:00"
+alice:
+  id: 2
+  cat_type: Kitten
+  name: Alice
+  created_at: "2007-04-02 12:00:00"
+  updated_at: "2007-04-03 10:00:00"
+toby:
+  id: 3
+  cat_type: Tabby
+  name: Toby
+  created_at: "2007-04-02 12:00:00"
+  updated_at: "2007-04-03 10:00:00"

data/test/fixtures/fish.yml

@@ -0,0 +1,12 @@
+swimmy:
+  id: 1
+  name: Swimmy
+  speed: 10
+  created_at: "2007-02-01 12:00:00"
+  updated_at: "2007-02-04 10:00:00"
+jaws:
+  id: 2
+  name: Jaws
+  speed: 20
+  created_at: "2007-02-02 12:00:00"
+  updated_at: "2007-02-03 10:00:00"

data/test/fixtures/frogs.yml

@@ -0,0 +1,5 @@
+froggy:
+  id: 1
+  name: Froggy
+  created_at: "2007-05-01 12:00:00"
+  updated_at: "2007-05-04 10:00:00"

data/test/fixtures/people.yml

@@ -0,0 +1,7 @@
+bob:
+  id: 1
+  name: Bob 
+  age: 45
+  created_at: "2007-04-01 12:00:00"
+  updated_at: "2007-04-04 10:00:00"
+  

data/test/fixtures/petfoods.yml

@@ -0,0 +1,11 @@
+kibbles:
+  the_petfood_primary_key: 1
+  name: Kibbles
+  created_at: "2007-06-01 12:00:00"
+  updated_at: "2007-06-04 10:00:00"
+bits:
+  the_petfood_primary_key: 2
+  name: Bits
+  created_at: "2007-06-02 12:00:00"
+  updated_at: "2007-06-03 10:00:00"
+  

data/test/fixtures/whales.yml

@@ -0,0 +1,5 @@
+shamu:
+  id: 1
+  name: Shamu
+  created_at: "2007-03-01 12:00:00"
+  updated_at: "2007-03-04 10:00:00"

data/test/fixtures/wild_boars.yml

@@ -0,0 +1,10 @@
+puma:
+  id: 1
+  name: Puma
+  created_at: "2007-07-01 12:00:00"
+  updated_at: "2007-07-04 10:00:00"
+jacrazy:
+  id: 2
+  name: Jacrazy
+  created_at: "2007-07-02 12:00:00"
+  updated_at: "2007-07-03 10:00:00"

data/test/generator/tagging_generator_test.rb

@@ -0,0 +1,42 @@
+require 'fileutils'
+require File.dirname(__FILE__) + '/../test_helper'
+
+class TaggingGeneratorTest < Test::Unit::TestCase
+  
+  def setup
+    Dir.chdir RAILS_ROOT do
+      truncate
+
+      # Revert environment lib requires
+      FileUtils.cp "config/environment.rb.canonical", "config/environment.rb"
+      
+      # Delete generator output
+      ["app/models/tag.rb", "app/models/tagging.rb", 
+        "test/unit/tag_test.rb", "test/unit/tagging_test.rb", 
+        "test/fixtures/tags.yml", "test/fixtures/taggings.yml",
+        "lib/tagging_extensions.rb",
+        "db/migrate/010_create_tags_and_taggings.rb"].each do |file|
+          File.delete file if File.exist? file
+      end
+      
+      # Rebuild database
+      Echoe.silence do
+        system("ruby #{HERE}/setup.rb")
+      end
+    end
+  end
+  
+  alias :teardown :setup
+
+  def test_generator
+    Dir.chdir RAILS_ROOT do
+      Echoe.silence do
+        assert system("script/generate tagging Stick Stone -q -f")
+        assert system("rake db:migrate")
+        assert system("rake db:fixtures:load")
+        assert system("rake test:units")      
+      end
+    end
+  end
+  
+end