diff options
| -rw-r--r-- | lib/pleroma/activity.ex | 4 | ||||
| -rw-r--r-- | lib/pleroma/web/activity_pub/activity_pub.ex | 5 | ||||
| -rw-r--r-- | lib/pleroma/web/api_spec/operations/status_operation.ex | 3 | ||||
| -rw-r--r-- | lib/pleroma/web/controller_helper.ex | 49 | ||||
| -rw-r--r-- | test/web/mastodon_api/controllers/status_controller_test.exs | 43 | 
5 files changed, 71 insertions, 33 deletions
| diff --git a/lib/pleroma/activity.ex b/lib/pleroma/activity.ex index da1be20b3..c3cea8d2a 100644 --- a/lib/pleroma/activity.ex +++ b/lib/pleroma/activity.ex @@ -31,6 +31,10 @@ defmodule Pleroma.Activity do      field(:recipients, {:array, :string}, default: [])      field(:thread_muted?, :boolean, virtual: true) +    # A field that can be used if you need to join some kind of other +    # id to order / paginate this field by +    field(:pagination_id, :string, virtual: true) +      # This is a fake relation,      # do not use outside of with_preloaded_user_actor/with_joined_user_actor      has_one(:user_actor, User, on_delete: :nothing, foreign_key: :id) diff --git a/lib/pleroma/web/activity_pub/activity_pub.ex b/lib/pleroma/web/activity_pub/activity_pub.ex index e7d0a9caa..c9dc6135c 100644 --- a/lib/pleroma/web/activity_pub/activity_pub.ex +++ b/lib/pleroma/web/activity_pub/activity_pub.ex @@ -1172,12 +1172,11 @@ defmodule Pleroma.Web.ActivityPub.ActivityPub do      |> Activity.Queries.by_type("Like")      |> Activity.with_joined_object()      |> Object.with_joined_activity() -    |> select([_like, object, activity], %{activity | object: object}) +    |> select([like, object, activity], %{activity | object: object, pagination_id: like.id})      |> order_by([like, _, _], desc_nulls_last: like.id)      |> Pagination.fetch_paginated(        Map.merge(params, %{skip_order: true}), -      pagination, -      :object_activity +      pagination      )    end diff --git a/lib/pleroma/web/api_spec/operations/status_operation.ex b/lib/pleroma/web/api_spec/operations/status_operation.ex index ca9db01e5..0b7fad793 100644 --- a/lib/pleroma/web/api_spec/operations/status_operation.ex +++ b/lib/pleroma/web/api_spec/operations/status_operation.ex @@ -333,7 +333,8 @@ defmodule Pleroma.Web.ApiSpec.StatusOperation do      %Operation{        tags: ["Statuses"],        summary: "Favourited statuses", -      description: "Statuses the user has favourited", +      description: +        "Statuses the user has favourited. Please note that you have to use the link headers to paginate this. You can not build the query parameters yourself.",        operationId: "StatusController.favourites",        parameters: pagination_params(),        security: [%{"oAuth" => ["read:favourites"]}], diff --git a/lib/pleroma/web/controller_helper.ex b/lib/pleroma/web/controller_helper.ex index 5d67d75b5..69946fb81 100644 --- a/lib/pleroma/web/controller_helper.ex +++ b/lib/pleroma/web/controller_helper.ex @@ -57,35 +57,36 @@ defmodule Pleroma.Web.ControllerHelper do      end    end +  @id_keys Pagination.page_keys() -- ["limit", "order"] +  defp build_pagination_fields(conn, min_id, max_id, extra_params) do +    params = +      conn.params +      |> Map.drop(Map.keys(conn.path_params)) +      |> Map.merge(extra_params) +      |> Map.drop(@id_keys) + +    %{ +      "next" => current_url(conn, Map.put(params, :max_id, max_id)), +      "prev" => current_url(conn, Map.put(params, :min_id, min_id)), +      "id" => current_url(conn) +    } +  end +    def get_pagination_fields(conn, activities, extra_params \\ %{}) do      case List.last(activities) do -      %{id: max_id} -> -        params = -          conn.params -          |> Map.drop(Map.keys(conn.path_params)) -          |> Map.merge(extra_params) -          |> Map.drop(Pagination.page_keys() -- ["limit", "order"]) +      %{pagination_id: max_id} when not is_nil(max_id) -> +        %{pagination_id: min_id} = +          activities +          |> List.first() + +        build_pagination_fields(conn, min_id, max_id, extra_params) -        min_id = +      %{id: max_id} -> +        %{id: min_id} =            activities            |> List.first() -          |> Map.get(:id) - -        fields = %{ -          "next" => current_url(conn, Map.put(params, :max_id, max_id)), -          "prev" => current_url(conn, Map.put(params, :min_id, min_id)) -        } - -        #  Generating an `id` without already present pagination keys would -        # need a query-restriction with an `q.id >= ^id` or `q.id <= ^id` -        # instead of the `q.id > ^min_id` and `q.id < ^max_id`. -        #  This is because we only have ids present inside of the page, while -        # `min_id`, `since_id` and `max_id` requires to know one outside of it. -        if Map.take(conn.params, Pagination.page_keys() -- ["limit", "order"]) != [] do -          Map.put(fields, "id", current_url(conn, conn.params)) -        else -          fields -        end + +        build_pagination_fields(conn, min_id, max_id, extra_params)        _ ->          %{} diff --git a/test/web/mastodon_api/controllers/status_controller_test.exs b/test/web/mastodon_api/controllers/status_controller_test.exs index 700c82e4f..648e6f2ce 100644 --- a/test/web/mastodon_api/controllers/status_controller_test.exs +++ b/test/web/mastodon_api/controllers/status_controller_test.exs @@ -1541,14 +1541,49 @@ defmodule Pleroma.Web.MastodonAPI.StatusControllerTest do             } = response    end +  test "favorites paginate correctly" do +    %{user: user, conn: conn} = oauth_access(["read:favourites"]) +    other_user = insert(:user) +    {:ok, first_post} = CommonAPI.post(other_user, %{status: "bla"}) +    {:ok, second_post} = CommonAPI.post(other_user, %{status: "bla"}) +    {:ok, third_post} = CommonAPI.post(other_user, %{status: "bla"}) + +    {:ok, _first_favorite} = CommonAPI.favorite(user, third_post.id) +    {:ok, _second_favorite} = CommonAPI.favorite(user, first_post.id) +    {:ok, third_favorite} = CommonAPI.favorite(user, second_post.id) + +    result = +      conn +      |> get("/api/v1/favourites?limit=1") + +    assert [%{"id" => post_id}] = json_response_and_validate_schema(result, 200) +    assert post_id == second_post.id + +    # Using the header for pagination works correctly +    [next, _] = get_resp_header(result, "link") |> hd() |> String.split(", ") +    [_, max_id] = Regex.run(~r/max_id=(.*)>;/, next) + +    assert max_id == third_favorite.id + +    result = +      conn +      |> get("/api/v1/favourites?max_id=#{max_id}") + +    assert [%{"id" => first_post_id}, %{"id" => third_post_id}] = +             json_response_and_validate_schema(result, 200) + +    assert first_post_id == first_post.id +    assert third_post_id == third_post.id +  end +    test "returns the favorites of a user" do      %{user: user, conn: conn} = oauth_access(["read:favourites"])      other_user = insert(:user)      {:ok, _} = CommonAPI.post(other_user, %{status: "bla"}) -    {:ok, activity} = CommonAPI.post(other_user, %{status: "traps are happy"}) +    {:ok, activity} = CommonAPI.post(other_user, %{status: "trees are happy"}) -    {:ok, _} = CommonAPI.favorite(user, activity.id) +    {:ok, last_like} = CommonAPI.favorite(user, activity.id)      first_conn = get(conn, "/api/v1/favourites") @@ -1566,9 +1601,7 @@ defmodule Pleroma.Web.MastodonAPI.StatusControllerTest do      {:ok, _} = CommonAPI.favorite(user, second_activity.id) -    last_like = status["id"] - -    second_conn = get(conn, "/api/v1/favourites?since_id=#{last_like}") +    second_conn = get(conn, "/api/v1/favourites?since_id=#{last_like.id}")      assert [second_status] = json_response_and_validate_schema(second_conn, 200)      assert second_status["id"] == to_string(second_activity.id) | 
