1 files changed, 49 insertions, 31 deletions
diff --git a/app/controllers/api/v1/timelines/home_controller.rb b/app/controllers/api/v1/timelines/home_controller.rb
index 33ff48b39..29e570fa5 100644
--- a/app/controllers/api/v1/timelines/home_controller.rb
+++ b/app/controllers/api/v1/timelines/home_controller.rb
@@ -1,44 +1,62 @@
 # frozen_string_literal: true
 
-module Api::V1::Timelines
-  class HomeController < BaseController
-    before_action -> { doorkeeper_authorize! :read }, only: [:show]
-    before_action :require_user!, only: [:show]
+class Api::V1::Timelines::HomeController < ApiController
+  before_action -> { doorkeeper_authorize! :read }, only: [:show]
+  before_action :require_user!, only: [:show]
+  after_action :insert_pagination_headers, unless: -> { @statuses.empty? }
 
-    def show
-      @statuses = load_statuses
-    end
+  respond_to :json
 
-    private
+  def show
+    @statuses = load_statuses
+    render 'api/v1/timelines/show'
+  end
 
-    def load_statuses
-      cached_home_statuses.tap do |statuses|
-        set_maps(statuses)
-      end
-    end
+  private
 
-    def cached_home_statuses
-      cache_collection home_statuses
+  def load_statuses
+    cached_home_statuses.tap do |statuses|
+      set_maps(statuses)
     end
+  end
 
-    def home_statuses
-      account_home_feed.get(
-        limit_param(DEFAULT_STATUSES_LIMIT),
-        params[:max_id],
-        params[:since_id]
-      )
-    end
+  def cached_home_statuses
+    cache_collection home_statuses, Status
+  end
 
-    def account_home_feed
-      Feed.new(:home, current_account)
-    end
+  def home_statuses
+    account_home_feed.get(
+      limit_param(DEFAULT_STATUSES_LIMIT),
+      params[:max_id],
+      params[:since_id]
+    )
+  end
 
-    def next_path
-      api_v1_timelines_home_url pagination_params(max_id: @statuses.last.id)
-    end
+  def account_home_feed
+    Feed.new(:home, current_account)
+  end
 
-    def prev_path
-      api_v1_timelines_home_url pagination_params(since_id: @statuses.first.id)
-    end
+  def insert_pagination_headers
+    set_pagination_headers(next_path, prev_path)
+  end
+
+  def pagination_params(core_params)
+    params.permit(:local, :limit).merge(core_params)
+  end
+
+  def next_path
+    api_v1_timelines_home_url pagination_params(max_id: pagination_max_id)
+  end
+
+  def prev_path
+    api_v1_timelines_home_url pagination_params(since_id: pagination_since_id)
+  end
+
+  def pagination_max_id
+    @statuses.last.id
+  end
+
+  def pagination_since_id
+    @statuses.first.id
   end
 end