teagles / comfortable-mexican-sofa

module ComfortableMexicanSofa::IsCategorized
  
  def self.included(base)
    base.send :extend, ClassMethods
  end
  
  module ClassMethods
    def cms_is_categorized
      include ComfortableMexicanSofa::IsCategorized::InstanceMethods
      
      has_many :categorizations,
        :as         => :categorized,
        :class_name => 'Cms::Categorization',
        :dependent  => :destroy
      has_many :categories,
        :through    => :categorizations,
        :class_name => 'Cms::Category'
        
      attr_accessor :category_ids
      
      after_save :sync_categories
      
      scope :for_category, lambda { |*categories|
        if (categories = [categories].flatten.compact).present?
          select("DISTINCT #{table_name}.*").
            joins(:categorizations => :category).
            where('cms_categories.label' => categories)
        end
      }
    end
  end
  
  module InstanceMethods
    def sync_categories
      (self.category_ids || {}).each do |category_id, flag|
        case flag.to_i
        when 1
          if category = Cms::Category.find_by_id(category_id)
            category.categorizations.create(:categorized => self)
          end
        when 0
          self.categorizations.where(:category_id => category_id).destroy_all
        end
      end
    end
  end
end

ActiveRecord::Base.send :include, ComfortableMexicanSofa::IsCategorized