136 lines
3.6 KiB
Ruby
136 lines
3.6 KiB
Ruby
class PoolVersion < ActiveRecord::Base
|
|
class Error < Exception ; end
|
|
|
|
validates_presence_of :updater_id, :updater_ip_addr
|
|
belongs_to :pool
|
|
belongs_to :updater, :class_name => "User"
|
|
before_validation :initialize_updater
|
|
attr_accessible :pool_id, :is_deleted, :name, :description, :post_ids, :post_id_array, :post_count, :is_active, :category, :updater_id, :updater_ip_addr
|
|
|
|
module SearchMethods
|
|
def for_user(user_id)
|
|
where("updater_id = ?", user_id)
|
|
end
|
|
|
|
def search(params)
|
|
q = where("true")
|
|
return q if params.blank?
|
|
|
|
if params[:updater_id].present?
|
|
q = q.for_user(params[:updater_id].to_i)
|
|
end
|
|
|
|
if params[:updater_name].present?
|
|
q = q.where("updater_id = (select _.id from users _ where lower(_.name) = ?)", params[:updater_name].mb_chars.downcase)
|
|
end
|
|
|
|
if params[:pool_id].present?
|
|
q = q.where("pool_id = ?", params[:pool_id].to_i)
|
|
end
|
|
|
|
q
|
|
end
|
|
end
|
|
|
|
extend SearchMethods
|
|
|
|
def self.export_to_archives(starting_version_id = 0)
|
|
raise "SQS URL not setup" if Danbooru.config.aws_sqs_archives_url.nil?
|
|
|
|
credentials = Aws::Credentials.new(
|
|
Danbooru.config.aws_access_key_id,
|
|
Danbooru.config.aws_secret_access_key
|
|
)
|
|
sqs = Aws::SQS::Client.new(
|
|
credentials: credentials,
|
|
region: Danbooru.config.aws_sqs_region
|
|
)
|
|
last_version_id = 0
|
|
|
|
where("id > ?", starting_version_id).find_each do |version|
|
|
last_version_id = version.id
|
|
|
|
json = {
|
|
id: version.id,
|
|
pool_id: version.pool_id,
|
|
post_ids: version.post_ids.scan(/\d+/).map(&:to_i),
|
|
updater_id: version.updater_id,
|
|
updater_ip_addr: version.updater_ip_addr.to_s,
|
|
created_at: version.created_at.try(:iso8601),
|
|
updated_at: version.updated_at.try(:iso8601),
|
|
description: version.pool.description,
|
|
name: version.name,
|
|
is_active: version.pool.is_active?,
|
|
is_deleted: version.pool.is_deleted?,
|
|
category: version.pool.category
|
|
}
|
|
msg = "add pool version\n#{json.to_json}"
|
|
sqs.send_message(
|
|
message_body: msg,
|
|
queue_url: Danbooru.config.aws_sqs_archives_url
|
|
)
|
|
end
|
|
|
|
puts "last version id: #{last_version_id}"
|
|
end
|
|
|
|
def updater_name
|
|
User.id_to_name(updater_id)
|
|
end
|
|
|
|
def pretty_name
|
|
name.tr("_", " ")
|
|
end
|
|
|
|
def initialize_updater
|
|
self.updater_id = CurrentUser.id
|
|
self.updater_ip_addr = CurrentUser.ip_addr
|
|
end
|
|
|
|
def post_id_array
|
|
@post_id_array ||= post_ids.scan(/\d+/).map(&:to_i)
|
|
end
|
|
|
|
def diff(version)
|
|
new_posts = post_id_array
|
|
old_posts = version.present? ? version.post_id_array : []
|
|
|
|
return {
|
|
:added_posts => array_difference_with_duplicates(new_posts, old_posts),
|
|
:removed_posts => array_difference_with_duplicates(old_posts, new_posts),
|
|
:unchanged_posts => array_intersect_with_duplicates(new_posts, old_posts)
|
|
}
|
|
end
|
|
|
|
def array_difference_with_duplicates(array, other_array)
|
|
new_array = array.dup
|
|
other_array.each do |id|
|
|
index = new_array.index(id)
|
|
if index
|
|
new_array.delete_at(index)
|
|
end
|
|
end
|
|
new_array
|
|
end
|
|
|
|
def array_intersect_with_duplicates(array, other_array)
|
|
other_array = other_array.dup
|
|
array.inject([]) do |intersect, id|
|
|
index = other_array.index(id)
|
|
if index
|
|
intersect << id
|
|
other_array.delete_at(index)
|
|
end
|
|
intersect
|
|
end
|
|
end
|
|
|
|
def changes
|
|
@changes ||= diff(previous)
|
|
end
|
|
|
|
def previous
|
|
PoolVersion.where(["pool_id = ? and updated_at < ?", pool_id, updated_at]).order("updated_at desc").first
|
|
end
|
|
end
|