internetee-registry/app/models/concerns/versions.rb

# Papertrail concerns is mainly tested at country spec
module Versions
  extend ActiveSupport::Concern

  included do
    has_paper_trail class_name: "#{model_name}Version"

    # add creator and updator
    before_create :add_creator
    before_create :add_updator
    before_update :add_updator

    def add_creator
      self.creator_str = ::PaperTrail.whodunnit
      true
    end

    def add_updator
      self.updator_str = ::PaperTrail.whodunnit
      true
    end

    # needs refactoring
    # TODO: optimization work
    # belongs_to :api_creator, class_name: 'ApiUser', foreign_key: :creator_str
    # belongs_to :creator, class_name: 'User', foreign_key: :creator_str
    def creator
      return nil if creator_str.blank?

      if creator_str =~ /^\d-api-/
        ApiUser.find_by(id: creator_str)
      else
        AdminUser.find_by(id: creator_str)
      end
    end

    def updator
      return nil if updator_str.blank?

      if updator_str =~ /^\d-api-/
        ApiUser.find_by(id: updator_str)
      else
        AdminUser.find_by(id: updator_str)
      end
    end

    # callbacks
    def touch_domain_version
      domain.try(:touch_with_version)
    end

    def touch_domains_version
      domains.each(&:touch_with_version)
    end
  end
end