diff --git a/lib/puppet/rails/lib/README b/lib/puppet/rails/lib/README new file mode 100644 index 000000000..8d2f90822 --- /dev/null +++ b/lib/puppet/rails/lib/README @@ -0,0 +1,4 @@ +Acts As Taggable +================= + +Allows for tags to be added to multiple classes. \ No newline at end of file diff --git a/lib/puppet/rails/lib/acts_as_taggable.rb b/lib/puppet/rails/lib/acts_as_taggable.rb new file mode 100644 index 000000000..e9bd03696 --- /dev/null +++ b/lib/puppet/rails/lib/acts_as_taggable.rb @@ -0,0 +1,58 @@ +module ActiveRecord + module Acts #:nodoc: + module Taggable #:nodoc: + def self.included(base) + base.extend(ClassMethods) + end + + module ClassMethods + def acts_as_taggable(options = {}) + write_inheritable_attribute(:acts_as_taggable_options, { + :taggable_type => ActiveRecord::Base.send(:class_name_of_active_record_descendant, self).to_s, + :from => options[:from] + }) + + class_inheritable_reader :acts_as_taggable_options + + has_many :taggings, :as => :taggable, :dependent => true + has_many :tags, :through => :taggings + + include ActiveRecord::Acts::Taggable::InstanceMethods + extend ActiveRecord::Acts::Taggable::SingletonMethods + end + end + + module SingletonMethods + def find_tagged_with(list) + find_by_sql([ + "SELECT #{table_name}.* FROM #{table_name}, tags, taggings " + + "WHERE #{table_name}.#{primary_key} = taggings.taggable_id " + + "AND taggings.taggable_type = ? " + + "AND taggings.tag_id = tags.id AND tags.name IN (?)", + acts_as_taggable_options[:taggable_type], list + ]) + end + end + + module InstanceMethods + def tag_with(list) + Tag.transaction do + taggings.destroy_all + + Tag.parse(list).each do |name| + if acts_as_taggable_options[:from] + send(acts_as_taggable_options[:from]).tags.find_or_create_by_name(name).on(self) + else + Tag.find_or_create_by_name(name).on(self) + end + end + end + end + + def tag_list + tags.collect { |tag| tag.name.include?(" ") ? "'#{tag.name}'" : tag.name }.join(" ") + end + end + end + end +end \ No newline at end of file diff --git a/lib/puppet/rails/lib/init.rb b/lib/puppet/rails/lib/init.rb new file mode 100644 index 000000000..0a07e5c8e --- /dev/null +++ b/lib/puppet/rails/lib/init.rb @@ -0,0 +1,5 @@ +require 'puppet/rails/lib/acts_as_taggable' +ActiveRecord::Base.send(:include, ActiveRecord::Acts::Taggable) + +require 'puppet/rails/lib/tagging' +require 'puppet/rails/lib/tag' diff --git a/lib/puppet/rails/lib/tag.rb b/lib/puppet/rails/lib/tag.rb new file mode 100644 index 000000000..ca31171b2 --- /dev/null +++ b/lib/puppet/rails/lib/tag.rb @@ -0,0 +1,40 @@ +class Tag < ActiveRecord::Base + has_many :taggings + + def self.parse(list) + tag_names = [] + + # first, pull out the quoted tags + list.gsub!(/\"(.*?)\"\s*/ ) { tag_names << $1; "" } + + # then, replace all commas with a space + list.gsub!(/,/, " ") + + # then, get whatever's left + tag_names.concat list.split(/\s/) + + # strip whitespace from the names + tag_names = tag_names.map { |t| t.strip } + + # delete any blank tag names + tag_names = tag_names.delete_if { |t| t.empty? } + + return tag_names + end + + def tagged + @tagged ||= taggings.collect { |tagging| tagging.taggable } + end + + def on(taggable) + taggings.create :taggable => taggable + end + + def ==(comparison_object) + super || name == comparison_object.to_s + end + + def to_s + name + end +end \ No newline at end of file diff --git a/lib/puppet/rails/lib/tagging.rb b/lib/puppet/rails/lib/tagging.rb new file mode 100644 index 000000000..e06e88a14 --- /dev/null +++ b/lib/puppet/rails/lib/tagging.rb @@ -0,0 +1,12 @@ +class Tagging < ActiveRecord::Base + belongs_to :tag + belongs_to :taggable, :polymorphic => true + + def self.tagged_class(taggable) + ActiveRecord::Base.send(:class_name_of_active_record_descendant, taggable.class).to_s + end + + def self.find_taggable(tagged_class, tagged_id) + tagged_class.constantize.find(tagged_id) + end +end \ No newline at end of file