module Drafting::BaseClassMethods

Constants

ALLOWED_DRAFT_OPTION_KEYS

Public Class Methods

child_drafts(user) click to toggle source
# File lib/drafting/base_class_methods.rb, line 25
def self.child_drafts(user)
  Draft.where(
    user: user,
    user_type: user.try(:class).try(:name),
    parent_type: self.base_class.name
  )
end

Public Instance Methods

drafts(user) click to toggle source
# File lib/drafting/base_class_methods.rb, line 17
def drafts(user)
  Draft.where(
    user: user,
    user_type: user.try(:class).try(:name),
    parent: self
  )
end
has_drafts(options={}) click to toggle source
# File lib/drafting/base_class_methods.rb, line 5
def has_drafts(options={})
  raise ArgumentError unless options.is_a?(Hash)
  raise ArgumentError unless options.keys.all? { |k| ALLOWED_DRAFT_OPTION_KEYS.include?(k) }

  class_attribute :draft_parent

  if options[:parent]
    parent_class = self.reflect_on_all_associations(:belongs_to).find { |a| a.name == options[:parent] }.try(:klass)
    raise ArgumentError unless parent_class

    unless parent_class.method_defined? :drafts
      parent_class.class_eval do
        def drafts(user)
          Draft.where(
            user: user,
            user_type: user.try(:class).try(:name),
            parent: self
          )
        end

        def self.child_drafts(user)
          Draft.where(
            user: user,
            user_type: user.try(:class).try(:name),
            parent_type: self.base_class.name
          )
        end
      end
    end

    self.draft_parent = options[:parent]
  end

  include Drafting::InstanceMethods
  extend Drafting::ClassMethods

  attr_accessor :draft_id
  after_create :clear_draft
end