Contents

module StorageUnit
  module Core
    extend ActiveSupport::Concern

    included do
      default_scope { where(with_deleted_scope_sql) }
      define_model_callbacks :trash
      define_model_callbacks :recover
    end

    module ClassMethods
      def with_deleted_scope_sql
        all.table[storage_unit_opts[:column]].eq(nil).to_sql
      end

      # lifted from acts_as_paranoid, works around https://github.com/rails/rails/issues/4306
      # with this in place Post.limit(10).with_deleted, will work as expected
      def with_deleted
        scope = self.all
        scope.where_values.delete(with_deleted_scope_sql)
        scope
      end
    end

    def trashed?
      send(storage_unit_opts[:column]).present?
    end

    def trash!
      run_callbacks :trash do
        update_columns trash_hash(DateTime.now)
        trash_dependents
      end
    end

    def trash_dependents
      Array(storage_unit_opts[:cascade]).each do |x|
        send(x).update_all trash_hash(DateTime.now)
      end
    end

    def recover!(opts = {})
      run_callbacks :recover do
        update_columns trash_hash(nil)
        recover_dependents
      end
    end

    def recover_dependents
      Array(storage_unit_opts[:cascade]).each do |x|
        send(x).with_deleted.update_all trash_hash(nil)
      end
    end

    def trash_hash(value)
      {}.tap { |h| h[storage_unit_opts[:column]] = value }
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
storage_unit-0.0.2	lib/storage_unit/core.rb