diff --git a/app/controllers/api/v1/mutes_controller.rb b/app/controllers/api/v1/mutes_controller.rb index 53eb2554b6..b9ac741765 100644 --- a/app/controllers/api/v1/mutes_controller.rb +++ b/app/controllers/api/v1/mutes_controller.rb @@ -3,23 +3,59 @@ class Api::V1::MutesController < ApiController before_action -> { doorkeeper_authorize! :follow } before_action :require_user! + after_action :insert_pagination_headers respond_to :json def index - @accounts = Account.includes(:muted_by) - .references(:muted_by) - .merge(Mute.where(account: current_account) - .paginate_by_max_id(limit_param(DEFAULT_ACCOUNTS_LIMIT), params[:max_id], params[:since_id])) - .to_a + @accounts = load_accounts + end - next_path = api_v1_mutes_url(pagination_params(max_id: @accounts.last.muted_by_ids.last)) if @accounts.size == limit_param(DEFAULT_ACCOUNTS_LIMIT) - prev_path = api_v1_mutes_url(pagination_params(since_id: @accounts.first.muted_by_ids.first)) unless @accounts.empty? + private + def load_accounts + default_accounts.merge(paginated_mutes).to_a + end + + def default_accounts + Account.includes(:muted_by).references(:muted_by) + end + + def paginated_mutes + Mute.where(account: current_account).paginate_by_max_id( + limit_param(DEFAULT_ACCOUNTS_LIMIT), + params[:max_id], + params[:since_id] + ) + end + + def insert_pagination_headers set_pagination_headers(next_path, prev_path) end - private + def next_path + if records_continue? + api_v1_mutes_url pagination_params(max_id: pagination_max_id) + end + end + + def prev_path + unless @accounts.empty? + api_v1_mutes_url pagination_params(since_id: pagination_since_id) + end + end + + def pagination_max_id + @accounts.last.muted_by_ids.last + end + + def pagination_since_id + @accounts.first.muted_by_ids.first + end + + def records_continue? + @accounts.size == limit_param(DEFAULT_ACCOUNTS_LIMIT) + end def pagination_params(core_params) params.permit(:limit).merge(core_params) diff --git a/spec/controllers/api/v1/mutes_controller_spec.rb b/spec/controllers/api/v1/mutes_controller_spec.rb index be8a5e7dd2..85aad43841 100644 --- a/spec/controllers/api/v1/mutes_controller_spec.rb +++ b/spec/controllers/api/v1/mutes_controller_spec.rb @@ -7,12 +7,14 @@ RSpec.describe Api::V1::MutesController, type: :controller do let(:token) { double acceptable?: true, resource_owner_id: user.id } before do + Fabricate(:mute, account: user.account) allow(controller).to receive(:doorkeeper_token) { token } end describe 'GET #index' do it 'returns http success' do - get :index + get :index, params: { limit: 1 } + expect(response).to have_http_status(:success) end end