diff --git a/app/services/fetch_remote_resource_service.rb b/app/services/fetch_remote_resource_service.rb index 2185ceb202..a0c270673a 100644 --- a/app/services/fetch_remote_resource_service.rb +++ b/app/services/fetch_remote_resource_service.rb @@ -1,18 +1,41 @@ # frozen_string_literal: true class FetchRemoteResourceService < BaseService - def call(url) - atom_url, body = FetchAtomService.new.call(url) + attr_reader :url - return nil if atom_url.nil? + def call(url) + @url = url + process_url unless atom_url.nil? + end - xml = Nokogiri::XML(body) - xml.encoding = 'utf-8' + private - if xml.root.name == 'feed' + def process_url + case xml_root + when 'feed' FetchRemoteAccountService.new.call(atom_url, body) - elsif xml.root.name == 'entry' + when 'entry' FetchRemoteStatusService.new.call(atom_url, body) end end + + def fetched_atom_feed + @_fetched_atom_feed ||= FetchAtomService.new.call(url) + end + + def atom_url + fetched_atom_feed.first + end + + def body + fetched_atom_feed.last + end + + def xml_root + xml_data.root.name + end + + def xml_data + @_xml_data ||= Nokogiri::XML(body, nil, 'utf-8') + end end diff --git a/app/services/search_service.rb b/app/services/search_service.rb index e9745010b0..1ed3f0032f 100644 --- a/app/services/search_service.rb +++ b/app/services/search_service.rb @@ -1,21 +1,38 @@ # frozen_string_literal: true class SearchService < BaseService + attr_accessor :query + def call(query, limit, resolve = false, account = nil) - results = { accounts: [], hashtags: [], statuses: [] } + @query = query - return results if query.blank? + default_results.tap do |results| + if url_query? + results.merge!(remote_resource_results) unless remote_resource.nil? + elsif query.present? + results[:accounts] = AccountSearchService.new.call(query, limit, resolve, account) + results[:hashtags] = Tag.search_for(query.gsub(/\A#/, ''), limit) unless query.start_with?('@') + end + end + end - if query =~ /\Ahttps?:\/\// - resource = FetchRemoteResourceService.new.call(query) + def default_results + { accounts: [], hashtags: [], statuses: [] } + end - results[:accounts] << resource if resource.is_a?(Account) - results[:statuses] << resource if resource.is_a?(Status) - else - results[:accounts] = AccountSearchService.new.call(query, limit, resolve, account) - results[:hashtags] = Tag.search_for(query.gsub(/\A#/, ''), limit) unless query.start_with?('@') - end + def url_query? + query =~ /\Ahttps?:\/\// + end + + def remote_resource_results + { remote_resource_symbol => [remote_resource] } + end + + def remote_resource + @_remote_resource ||= FetchRemoteResourceService.new.call(query) + end - results + def remote_resource_symbol + remote_resource.class.name.downcase.pluralize.to_sym end end diff --git a/spec/services/fetch_remote_resource_service_spec.rb b/spec/services/fetch_remote_resource_service_spec.rb new file mode 100644 index 0000000000..79834842a3 --- /dev/null +++ b/spec/services/fetch_remote_resource_service_spec.rb @@ -0,0 +1,53 @@ +# frozen_string_literal: true + +require 'rails_helper' + +describe FetchRemoteResourceService do + subject { described_class.new } + + describe '#call' do + it 'returns nil when there is no atom url' do + url = 'http://example.com/missing-atom' + service = double + allow(FetchAtomService).to receive(:new).and_return service + allow(service).to receive(:call).with(url).and_return([nil, 'body']) + + result = subject.call(url) + expect(result).to be_nil + end + + it 'fetches remote accounts for feed types' do + url = 'http://example.com/atom-feed' + service = double + allow(FetchAtomService).to receive(:new).and_return service + feed_url = 'http://feed-url' + feed_content = 'contents' + allow(service).to receive(:call).with(url).and_return([feed_url, feed_content]) + + account_service = double + allow(FetchRemoteAccountService).to receive(:new).and_return(account_service) + allow(account_service).to receive(:call) + + _result = subject.call(url) + + expect(account_service).to have_received(:call).with(feed_url, feed_content) + end + + it 'fetches remote statuses for entry types' do + url = 'http://example.com/atom-entry' + service = double + allow(FetchAtomService).to receive(:new).and_return service + feed_url = 'http://feed-url' + feed_content = 'contents' + allow(service).to receive(:call).with(url).and_return([feed_url, feed_content]) + + account_service = double + allow(FetchRemoteStatusService).to receive(:new).and_return(account_service) + allow(account_service).to receive(:call) + + _result = subject.call(url) + + expect(account_service).to have_received(:call).with(feed_url, feed_content) + end + end +end diff --git a/spec/services/search_service_spec.rb b/spec/services/search_service_spec.rb new file mode 100644 index 0000000000..00475c6995 --- /dev/null +++ b/spec/services/search_service_spec.rb @@ -0,0 +1,101 @@ +# frozen_string_literal: true + +require 'rails_helper' + +describe SearchService do + subject { described_class.new } + + describe '#call' do + describe 'with a blank query' do + it 'returns empty results without searching' do + allow(AccountSearchService).to receive(:new) + allow(Tag).to receive(:search_for) + results = subject.call('', 10) + + expect(results).to eq(empty_results) + expect(AccountSearchService).not_to have_received(:new) + expect(Tag).not_to have_received(:search_for) + end + end + + describe 'with an url query' do + before do + @query = 'http://test.host/query' + end + + context 'that does not find anything' do + it 'returns the empty results' do + service = double(call: nil) + allow(FetchRemoteResourceService).to receive(:new).and_return(service) + results = subject.call(@query, 10) + + expect(service).to have_received(:call).with(@query) + expect(results).to eq empty_results + end + end + + context 'that finds an account' do + it 'includes the account in the results' do + account = Account.new + service = double(call: account) + allow(FetchRemoteResourceService).to receive(:new).and_return(service) + + results = subject.call(@query, 10) + expect(service).to have_received(:call).with(@query) + expect(results).to eq empty_results.merge(accounts: [account]) + end + end + + context 'that finds a status' do + it 'includes the status in the results' do + status = Status.new + service = double(call: status) + allow(FetchRemoteResourceService).to receive(:new).and_return(service) + + results = subject.call(@query, 10) + expect(service).to have_received(:call).with(@query) + expect(results).to eq empty_results.merge(statuses: [status]) + end + end + end + + describe 'with a non-url query' do + context 'that matches an account' do + it 'includes the account in the results' do + query = 'username' + account = Account.new + service = double(call: [account]) + allow(AccountSearchService).to receive(:new).and_return(service) + + results = subject.call(query, 10) + expect(service).to have_received(:call).with(query, 10, false, nil) + expect(results).to eq empty_results.merge(accounts: [account]) + end + end + + context 'that matches a tag' do + it 'includes the tag in the results' do + query = '#tag' + tag = Tag.new + allow(Tag).to receive(:search_for).with('tag', 10).and_return([tag]) + + results = subject.call(query, 10) + expect(Tag).to have_received(:search_for).with('tag', 10) + expect(results).to eq empty_results.merge(hashtags: [tag]) + end + it 'does not include tag when starts with @ character' do + query = '@username' + allow(Tag).to receive(:search_for) + + results = subject.call(query, 10) + expect(Tag).not_to have_received(:search_for) + expect(results).to eq empty_results + end + end + end + end + + def empty_results + { accounts: [], hashtags: [], statuses: [] } + end +end