class PoolVersion < ApplicationRecord belongs_to :updater, :class_name => "User" belongs_to :pool def self.enabled? Danbooru.config.aws_sqs_archives_url.present? end establish_connection (ENV["ARCHIVE_DATABASE_URL"] || "archive_#{Rails.env}".to_sym) if enabled? module SearchMethods def default_order order(updated_at: :desc) end def for_user(user_id) where("updater_id = ?", user_id) end def for_post_id(post_id) where_array_includes_any(:added_post_ids, [post_id]).or(where_array_includes_any(:removed_post_ids, [post_id])) end def name_matches(name) name = normalize_name_for_search(name) name = "*#{name}*" unless name =~ /\*/ where_ilike(:name, name) end def search(params) q = super q = q.search_attributes(params, :pool_id, :post_ids, :added_post_ids, :removed_post_ids, :updater_id, :description, :description_changed, :name, :name_changed, :version, :is_active, :is_deleted, :category) if params[:post_id] q = q.for_post_id(params[:post_id].to_i) end if params[:name_matches].present? q = q.name_matches(params[:name_matches]) end if params[:updater_name].present? q = q.where(updater_id: User.name_to_id(params[:updater_name])) end if params[:is_new].to_s.truthy? q = q.where(version: 1) elsif params[:is_new].to_s.falsy? q = q.where("version != 1") end q.apply_default_order(params) end end extend SearchMethods def self.sqs_service SqsService.new(Danbooru.config.aws_sqs_archives_url) end def self.queue(pool, updater, updater_ip_addr) # queue updates to sqs so that if archives goes down for whatever reason it won't # block pool updates raise NotImplementedError.new("Archive service is not configured.") if !enabled? json = { pool_id: pool.id, post_ids: pool.post_ids, updater_id: updater.id, updater_ip_addr: updater_ip_addr.to_s, created_at: pool.created_at.try(:iso8601), updated_at: pool.updated_at.try(:iso8601), description: pool.description, name: pool.name, is_active: pool.is_active?, is_deleted: pool.is_deleted?, category: pool.category } msg = "add pool version\n#{json.to_json}" sqs_service.send_message(msg, message_group_id: "pool:#{pool.id}") end def self.normalize_name(name) name.gsub(/[_[:space:]]+/, "_").gsub(/\A_|_\z/, "") end def self.normalize_name_for_search(name) normalize_name(name).mb_chars.downcase end def build_diff(other = previous) diff = {} if other.nil? diff[:added_post_ids] = added_post_ids diff[:removed_post_ids] = removed_post_ids diff[:added_desc] = description else diff[:added_post_ids] = post_ids - other.post_ids diff[:removed_post_ids] = other.post_ids - post_ids diff[:added_desc] = description diff[:removed_desc] = other.description end diff end def previous @previous ||= begin PoolVersion.where("pool_id = ? and version < ?", pool_id, version).order("version desc").limit(1).to_a end @previous.first end def self.status_fields { name: "Renamed", description: "Description", was_deleted: "Deleted", was_undeleted: "Undeleted", was_activated: "Activated", was_deactivated: "Deactivated", } end def was_deleted is_deleted && !previous.is_deleted end def was_undeleted !is_deleted && previous.is_deleted end def was_activated is_active && !previous.is_active end def was_deactivated !is_active && previous.is_active end def text_field_changed previous.present? && (name_changed || description_changed) end def pretty_name name.tr("_", " ") end def self.available_includes [:updater, :pool] end end