def audited(options = {})
return if included_modules.include?(Audited::Auditor::AuditedInstanceMethods)
extend Audited::Auditor::AuditedClassMethods
include Audited::Auditor::AuditedInstanceMethods
class_attribute :audit_associated_with, instance_writer: false
class_attribute :audited_options, instance_writer: false
attr_accessor :audit_version, :audit_comment
self.audited_options = options
normalize_audited_options
self.audit_associated_with = audited_options[:associated_with]
if audited_options[:comment_required]
validate :presence_of_audit_comment
before_destroy :require_comment if audited_options[:on].include?(:destroy)
end
has_many :audits, -> { order(version: :asc) }, as: :auditable, class_name: Audited.audit_class.name, inverse_of: :auditable
Audited.audit_class.audited_class_names << to_s
after_create :audit_create if audited_options[:on].include?(:create)
before_update :audit_update if audited_options[:on].include?(:update)
before_destroy :audit_destroy if audited_options[:on].include?(:destroy)
define_callbacks :audit
set_callback :audit, :after, :after_audit, if: lambda { respond_to?(:after_audit, true) }
set_callback :audit, :around, :around_audit, if: lambda { respond_to?(:around_audit, true) }
enable_auditing
end
def has_associated_audits
has_many :associated_audits, as: :associated, class_name: Audited.audit_class.name
end
end
module AuditedInstanceMethods
def method_missing(method_name, *args, &block)
if method_name == :version
ActiveSupport::Deprecation.warn("`version` attribute has been changed to `audit_version`. This attribute will be removed.")
audit_version
else
super
end
end
def save_without_auditing
without_auditing { save }
end
def without_auditing(&block)
self.class.without_auditing(&block)
end
def revisions(from_version = 1)
return [] unless audits.from_version(from_version).exists?
all_audits = audits.select([:audited_changes, :version]).to_a
targeted_audits = all_audits.select { |audit| audit.version >= from_version }
previous_attributes = reconstruct_attributes(all_audits - targeted_audits)
targeted_audits.map do |audit|
previous_attributes.merge!(audit.new_attributes)
revision_with(previous_attributes.merge!(version: audit.version))
end
end
def revision(version)
if version == :previous || self.audits.last.version >= version
revision_with Audited.audit_class.reconstruct_attributes(audits_to(version))
end
end
def revision_at(date_or_time)
audits = self.audits.up_until(date_or_time)
revision_with Audited.audit_class.reconstruct_attributes(audits) unless audits.empty?
end
def audited_attributes
attributes.except(*non_audited_columns)
end
def own_and_associated_audits
Audited.audit_class.unscoped
.where('(auditable_type = :type AND auditable_id = :id) OR (associated_type = :type AND associated_id = :id)',
type: self.class.name, id: id)
.order(created_at: :desc)
end
def combine_audits(audits_to_combine)
combine_target = audits_to_combine.last
combine_target.audited_changes = audits_to_combine.pluck(:audited_changes).reduce(&:merge)
combine_target.comment = "#{combine_target.comment}\nThis audit is the result of multiple audits being combined."
transaction do
combine_target.save!
audits_to_combine.unscope(:limit).where("version < ?", combine_target.version).delete_all
end
end
protected
def non_audited_columns
self.class.non_audited_columns
end
def audited_columns
self.class.audited_columns
end
def revision_with(attributes)
dup.tap do |revision|
revision.id = id
revision.send :instance_variable_set, '@attributes', self.attributes if rails_below?('4.2.0')
revision.send :instance_variable_set, '@new_record', destroyed?
revision.send :instance_variable_set, '@persisted', !destroyed?
revision.send :instance_variable_set, '@readonly', false
revision.send :instance_variable_set, '@destroyed', false
revision.send :instance_variable_set, '@_destroyed', false
revision.send :instance_variable_set, '@marked_for_destruction', false
Audited.audit_class.assign_revision_attributes(revision, attributes)
revision.instance_variables.each do |ivar|
proxy = revision.instance_variable_get ivar
if !proxy.nil? && proxy.respond_to?(:proxy_respond_to?)
revision.instance_variable_set ivar, nil
end
end
end
end
def rails_below?(rails_version)
Gem::Version.new(Rails::VERSION::STRING) < Gem::Version.new(rails_version)
end
private
def audited_changes
all_changes = respond_to?(:changes_to_save) ? changes_to_save : changes
if audited_options[:only].present?
all_changes.slice(*audited_columns)
else
all_changes.except(*non_audited_columns)
end
end
def audits_to(version = nil)
if version == :previous
version = if self.audit_version
self.audit_version - 1
else
previous = audits.descending.offset(1).first
previous ? previous.version : 1
end
end
audits.to_version(version)
end
def audit_create
write_audit(action: 'create', audited_changes: audited_attributes,
comment: audit_comment)
end
def audit_update
unless (changes = audited_changes).empty? && audit_comment.blank?
write_audit(action: 'update', audited_changes: changes,
comment: audit_comment)
end
end
def audit_destroy
write_audit(action: 'destroy', audited_changes: audited_attributes,
comment: audit_comment) unless new_record?
end
def write_audit(attrs)
attrs[:associated] = send(audit_associated_with) unless audit_associated_with.nil?
self.audit_comment = nil
if auditing_enabled
run_callbacks(:audit) {
audit = audits.create(attrs)
combine_audits_if_needed if attrs[:action] != 'create'
audit
}
end
end
def presence_of_audit_comment
if comment_required_state?
errors.add(:audit_comment, "Comment can't be blank!") unless audit_comment.present?
end
end
def comment_required_state?
auditing_enabled &&
((audited_options[:on].include?(:create) && self.new_record?) ||
(audited_options[:on].include?(:update) && self.persisted? && self.changed?))
end
def combine_audits_if_needed
max_audits = audited_options[:max_audits]
if max_audits && (extra_count = audits.count - max_audits) > 0
audits_to_combine = audits.limit(extra_count + 1)
combine_audits(audits_to_combine)
end
end
def require_comment
if auditing_enabled && audit_comment.blank?
errors.add(:audit_comment, "Comment can't be blank!")
return false if Rails.version.start_with?('4.')
throw(:abort)
end
end
CALLBACKS.each do |attr_name|
alias_method "#{attr_name}_callback".to_sym, attr_name
end
def auditing_enabled
return run_conditional_check(audited_options[:if]) &&
run_conditional_check(audited_options[:unless], matching: false) &&
self.class.auditing_enabled
end
def run_conditional_check(condition, matching: true)
return true if condition.blank?
return condition.call(self) == matching if condition.respond_to?(:call)
return send(condition) == matching if respond_to?(condition.to_sym, true)
true
end
def auditing_enabled=(val)
self.class.auditing_enabled = val
end
def reconstruct_attributes(audits)
attributes = {}
audits.each { |audit| attributes.merge!(audit.new_attributes) }
attributes
end
end
module AuditedClassMethods
def audited_columns
@audited_columns ||= column_names - non_audited_columns
end
def non_audited_columns
@non_audited_columns ||= calculate_non_audited_columns
end
def non_audited_columns=(columns)
@audited_columns = nil
@non_audited_columns = columns.map(&:to_s)
end
def without_auditing
auditing_was_enabled = auditing_enabled
disable_auditing
yield
ensure
enable_auditing if auditing_was_enabled
end
def disable_auditing
self.auditing_enabled = false
end
def enable_auditing
self.auditing_enabled = true
end
def audit_as(user, &block)
Audited.audit_class.as_user(user, &block)
end
def auditing_enabled
Audited.store.fetch("#{table_name}_auditing_enabled", true) && Audited.auditing_enabled
end
def auditing_enabled=(val)
Audited.store["#{table_name}_auditing_enabled"] = val
end
def default_ignored_attributes
[primary_key, inheritance_column] | Audited.ignored_attributes
end
protected
def normalize_audited_options
audited_options[:on] = Array.wrap(audited_options[:on])
audited_options[:on] = [:create, :update, :destroy] if audited_options[:on].empty?
audited_options[:only] = Array.wrap(audited_options[:only]).map(&:to_s)
audited_options[:except] = Array.wrap(audited_options[:except]).map(&:to_s)
max_audits = audited_options[:max_audits] || Audited.max_audits
audited_options[:max_audits] = Integer(max_audits).abs if max_audits
end
def calculate_non_audited_columns
if audited_options[:only].present?
(column_names | default_ignored_attributes) - audited_options[:only]
elsif audited_options[:except].present?
default_ignored_attributes | audited_options[:except]
else
default_ignored_attributes
end
end
end
end