jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

hide saved search functionality if not enabled

Browse Source
This commit is contained in:
Albert Yi
2016-12-21 11:54:43 -08:00
parent ee4ebce4d7
commit 62956be384
14 changed files with 198 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
app/controllers/saved_search_category_changes_controller.rb
View File

@@ -1,5 +1,6 @@
class SavedSearchCategoryChangesController < ApplicationController
before_filter :member_only
before_filter :check_availabililty
respond_to :html
def new
@@ -11,4 +12,22 @@ class SavedSearchCategoryChangesController < ApplicationController
flash[:notice] = "Saved searches will be renamed"
redirect_to saved_searches_path
end
private
def check_availabililty
if !SavedSearch.enabled?
respond_to do |format|
format.html do
flash[:notice] = "Listbooru service is not configured. Saved searches are not available."
redirect_to :back
end
format.json do
render json: {success: false, reason: "Listbooru service is not configured"}.to_json, status: 501
end
end
return false
end
end
end

17
app/controllers/saved_searches_controller.rb
View File

@@ -1,5 +1,6 @@
class SavedSearchesController < ApplicationController
before_filter :member_only
before_fitler :check_availability
respond_to :html, :xml, :json, :js
def index
@@ -47,6 +48,22 @@ class SavedSearchesController < ApplicationController
private
def check_availabililty
if !SavedSearch.enabled?
respond_to do |format|
format.html do
flash[:notice] = "Listbooru service is not configured. Saved searches are not available."
redirect_to :back
end
format.json do
render json: {success: false, reason: "Listbooru service is not configured"}.to_json, status: 501
end
end
return false
end
end
def saved_searches
CurrentUser.user.saved_searches
end

9
app/logical/moderator/tag_batch_change.rb
View File

@@ -23,11 +23,12 @@ module Moderator
tags = Tag.scan_tags(antecedent, :strip_metatags => true)
conds = tags.map {|x| "tag_query like ?"}.join(" AND ")
conds = [conds, *tags.map {|x| "%#{x}%"}]
SavedSearch.where(*conds).find_each do |ss|
ss.tag_query = (ss.tag_query_array - tags + antecedent).uniq.join(" ")
ss.save
if SavedSearch.enabled?
SavedSearch.where(*conds).find_each do |ss|
ss.tag_query = (ss.tag_query_array - tags + antecedent).uniq.join(" ")
ss.save
end
end
end
end
end

18
app/logical/post_query_builder.rb
View File

@@ -102,15 +102,17 @@ class PostQueryBuilder
end
def add_saved_search_relation(saved_searches, relation)
saved_searches.each do |saved_search|
if saved_search == "all"
post_ids = SavedSearch.post_ids(CurrentUser.id)
else
post_ids = SavedSearch.post_ids(CurrentUser.id, saved_search)
end
if SavedSearch.enabled?
saved_searches.each do |saved_search|
if saved_search == "all"
post_ids = SavedSearch.post_ids(CurrentUser.id)
else
post_ids = SavedSearch.post_ids(CurrentUser.id, saved_search)
end
post_ids = [0] if post_ids.empty?
relation = relation.where(["posts.id IN (?)", post_ids])
post_ids = [0] if post_ids.empty?
relation = relation.where(["posts.id IN (?)", post_ids])
end
end
relation

104
app/models/saved_search.rb
View File

@@ -5,63 +5,96 @@ class SavedSearch < ActiveRecord::Base
extend ActiveSupport::Concern
module ClassMethods
def enabled?
Danbooru.config.aws_sqs_saved_search_url.present?
end
def posts_search_available?
Danbooru.config.listbooru_server.present? && CurrentUser.is_gold?
enabled? && CurrentUser.is_gold?
end
def sqs_service
SqsService.new(Danbooru.config.aws_sqs_saved_search_url)
end
def refresh_listbooru(user_id)
return false unless Danbooru.config.listbooru_enabled?
return false unless enabled?
sqs = SqsService.new(Danbooru.config.aws_sqs_queue_url)
sqs.send_message("refresh\n#{user_id}")
sqs_service.send_message("refresh\n#{user_id}")
end
def reset_listbooru(user_id)
return false unless Danbooru.config.listbooru_enabled?
return false unless enabled?
sqs = SqsService.new(Danbooru.config.aws_sqs_queue_url)
user = User.find(user_id)
sqs.send_message("delete\n#{user_id}\nall\n")
sqs_service.send_message("delete\n#{user_id}\nall\n")
user.saved_searches.each do |saved_search|
sqs.send_message("create\n#{user_id}\n#{saved_search.category}\n#{saved_search.tag_query}", :delay_seconds => 30)
sqs_service.send_message("create\n#{user_id}\n#{saved_search.category}\n#{saved_search.tag_query}", :delay_seconds => 30)
end
true
end
def rename_listbooru(user_id, old_category, new_category)
return false unless Danbooru.config.listbooru_enabled?
return false unless enabled?
sqs = SqsService.new(Danbooru.config.aws_sqs_queue_url)
sqs.send_message("rename\n#{user_id}\n#{old_category}\n#{new_category}\n")
sqs_service.send_message("rename\n#{user_id}\n#{old_category}\n#{new_category}\n")
true
end
def post_ids(user_id, name = nil)
return [] unless enabled?
if name
hash_name = Cache.hash(name)
else
hash_name = nil
end
body = Cache.get("ss-pids-#{user_id}-#{hash_name}", 60) do
params = {
"key" => Danbooru.config.listbooru_auth_key,
"user_id" => user_id,
"name" => name
}
uri = URI.parse("#{Danbooru.config.listbooru_server}/users")
uri.query = URI.encode_www_form(params)
Net::HTTP.start(uri.host, uri.port, :use_ssl => uri.is_a?(URI::HTTPS)) do |http|
resp = http.request_get(uri.request_uri)
if resp.is_a?(Net::HTTPSuccess)
resp.body
else
raise "HTTP error code: #{resp.code} #{resp.message}"
end
end
end
body.to_s.scan(/\d+/).map(&:to_i)
end
end
def update_listbooru_on_create
return unless Danbooru.config.listbooru_enabled?
return unless SavedSearch.enabled?
return unless user.is_gold?
sqs = SqsService.new(Danbooru.config.aws_sqs_queue_url)
sqs.send_message("create\n#{user_id}\n#{category}\n#{tag_query}")
SavedSearch.sqs_service.send_message("create\n#{user_id}\n#{category}\n#{tag_query}")
end
def update_listbooru_on_destroy
return unless Danbooru.config.listbooru_enabled?
return unless SavedSearch.enabled?
sqs = SqsService.new(Danbooru.config.aws_sqs_queue_url)
sqs.send_message("delete\n#{user_id}\n#{category}\n#{tag_query}")
SavedSearch.sqs_service.send_message("delete\n#{user_id}\n#{category}\n#{tag_query}")
end
def update_listbooru_on_update
return unless Danbooru.config.listbooru_enabled?
return unless SavedSearch.enabled?
return unless user.is_gold?
sqs = SqsService.new(Danbooru.config.aws_sqs_queue_url)
sqs.send_message("update\n#{user_id}\n#{category_was}\n#{tag_query_was}\n#{category}\n#{tag_query}")
SavedSearch.sqs_service.send_message("update\n#{user_id}\n#{category_was}\n#{tag_query_was}\n#{category}\n#{tag_query}")
end
end
@@ -99,37 +132,6 @@ class SavedSearch < ActiveRecord::Base
rename_listbooru(user_id, old_category, new_category)
end
def self.post_ids(user_id, name = nil)
return [] unless Danbooru.config.listbooru_enabled?
if name
hash_name = Cache.hash(name)
else
hash_name = nil
end
body = Cache.get("ss-pids-#{user_id}-#{hash_name}", 60) do
params = {
"key" => Danbooru.config.listbooru_auth_key,
"user_id" => user_id,
"name" => name
}
uri = URI.parse("#{Danbooru.config.listbooru_server}/users")
uri.query = URI.encode_www_form(params)
Net::HTTP.start(uri.host, uri.port, :use_ssl => uri.is_a?(URI::HTTPS)) do |http|
resp = http.request_get(uri.request_uri)
if resp.is_a?(Net::HTTPSuccess)
resp.body
else
raise "HTTP error code: #{resp.code} #{resp.message}"
end
end
end
body.to_s.scan(/\d+/).map(&:to_i)
end
def normalize
self.category = SavedSearch.normalize_category(category) if category
self.tag_query = SavedSearch.normalize(tag_query)

9
app/models/tag_alias.rb
View File

@@ -180,9 +180,12 @@ class TagAlias < ActiveRecord::Base
def move_saved_searches
escaped = Regexp.escape(antecedent_name)
SavedSearch.where("tag_query like ?", "%#{antecedent_name}%").find_each do |ss|
ss.tag_query = ss.tag_query.sub(/(?:^| )#{escaped}(?:$| )/, " #{consequent_name} ").strip.gsub(/ /, " ")
ss.save
if SavedSearch.enabled?
SavedSearch.where("tag_query like ?", "%#{antecedent_name}%").find_each do |ss|
ss.tag_query = ss.tag_query.sub(/(?:^| )#{escaped}(?:$| )/, " #{consequent_name} ").strip.gsub(/ /, " ")
ss.save
end
end
end

20
app/models/user.rb
View File

@@ -847,15 +847,19 @@ class User < ActiveRecord::Base
module SavedSearchMethods
def unique_saved_search_categories
categories = saved_searches.pluck(:category)
if categories.any? {|x| x.blank?}
categories.reject! {|x| x.blank?}
categories.unshift(SavedSearch::UNCATEGORIZED_NAME)
if SavedSearch.enabled?
categories = saved_searches.pluck(:category)
if categories.any? {|x| x.blank?}
categories.reject! {|x| x.blank?}
categories.unshift(SavedSearch::UNCATEGORIZED_NAME)
end
categories.uniq!
categories
else
[]
end
categories.uniq!
categories
end
end

4
app/presenters/user_presenter.rb
View File

@@ -44,6 +44,10 @@ class UserPresenter
end
def posts_for_saved_search_category(category)
if !SavedSearch.enabled?
return Post.where("false")
end
if category == SavedSearch::UNCATEGORIZED_NAME
ids = SavedSearch.post_ids(CurrentUser.user.id)
else

26
app/views/saved_searches/_interface.html.erb
View File

@@ -1,17 +1,19 @@
<% if CurrentUser.show_saved_searches? %>
<% if SavedSearch.enabled? && CurrentUser.show_saved_searches? %>
<%= button_tag "Save search", :id => "save-search" %>
<% end %>
<div id="save-search-dialog" title="Save Search" style="display: none;">
<%= form_tag(saved_searches_path, :class => "simple_form", :remote => true) do %>
<%= hidden_field_tag "saved_search_tags", params[:tags] %>
<% if SavedSearch.enabled? %>
<div id="save-search-dialog" title="Save Search" style="display: none;">
<%= form_tag(saved_searches_path, :class => "simple_form", :remote => true) do %>
<%= hidden_field_tag "saved_search_tags", params[:tags] %>
<div class="input">
<label>
<%= text_field_tag "saved_search_category", "", :placeholder => "Category" %>
</label>
</div>
<div class="input">
<label>
<%= text_field_tag "saved_search_category", "", :placeholder => "Category" %>
</label>
</div>
<p><label><%= check_box_tag "saved_search_disable_categories" %> Disable categorization</label></p>
<% end %>
</div>
<p><label><%= check_box_tag "saved_search_disable_categories" %> Disable categorization</label></p>
<% end %>
</div>
<% end %>

14
app/views/users/upgrade_information.html.erb
View File

@@ -43,12 +43,14 @@
<td>2,000</td>
<td>5,000</td>
</tr>
<tr>
<td>Saved Searches</td>
<td>No</td>
<td>Yes</td>
<td>Yes</td>
</tr>
<% if SavedSearch.enabled? %>
<tr>
<td>Saved Searches</td>
<td>No</td>
<td>Yes</td>
<td>Yes</td>
</tr>
<% end %>
<tr>
<td>See Censored Tags</td>
<td>No</td>