Merge pull request #3479 from BrokenEagle/feat-add-iqdb-page

Initial add of IQDB standalone page
This commit is contained in:
Albert Yi
2018-01-02 10:29:27 -08:00
committed by GitHub
5 changed files with 78 additions and 13 deletions

View File

@@ -2,31 +2,49 @@
class IqdbQueriesController < ApplicationController
respond_to :html, :json, :xml
def index
if !Danbooru.config.iqdbs_server
raise NotImplementedError.new("the IQDBs service isn't configured. Similarity searches are not available.")
end
def show
server_check
if params[:url]
create_by_url
respond_with(@results) do |fmt|
fmt.html { render :layout => false, :action => "create_by_url" }
end
elsif params[:post_id]
create_by_post
respond_with(@results) do |fmt|
fmt.js { render :layout => false, :action => "create_by_post" }
end
else
render :nothing => true, :status => 422
end
end
def check
server_check
if params[:url].present?
create_by_url
elsif params[:post_id].present?
create_by_post
else
@results = []
end
respond_with(@results)
end
# Support both POST /iqdb_queries and GET /iqdb_queries.
alias_method :create, :index
alias_method :create, :show
protected
def server_check
if !Danbooru.config.iqdbs_server
raise NotImplementedError.new("the IQDBs service isn't configured. Similarity searches are not available.")
end
end
def create_by_url
@download = Iqdb::Download.new(params[:url])
@download.find_similar
@results = @download.matches
respond_with(@results) do |fmt|
fmt.html { render :layout => false, :action => "create_by_url" }
end
end
def create_by_post
@@ -34,8 +52,5 @@ protected
@download = Iqdb::Download.new(@post.complete_preview_file_url)
@download.find_similar
@results = @download.matches
respond_with(@results) do |fmt|
fmt.js { render :layout => false, :action => "create_by_post" }
end
end
end