From 828259fb6517d35b5f950e07601bab0bdc5b5efd Mon Sep 17 00:00:00 2001
From: Phil Hagelberg <phil@hagelb.org>
Date: Mon, 4 Nov 2019 22:56:51 -0800
Subject: [PATCH] Catch 404s.

---
 .../web/static_fe/static_fe_controller.ex     | 55 ++++++++++++-------
 1 file changed, 34 insertions(+), 21 deletions(-)

diff --git a/lib/pleroma/web/static_fe/static_fe_controller.ex b/lib/pleroma/web/static_fe/static_fe_controller.ex
index 9f4eeaa36..4798cad24 100644
--- a/lib/pleroma/web/static_fe/static_fe_controller.ex
+++ b/lib/pleroma/web/static_fe/static_fe_controller.ex
@@ -61,35 +61,48 @@ def represent(%Activity{object: %Object{data: data}} = activity, selected) do
   end
 
   def show(%{assigns: %{notice_id: notice_id}} = conn, _params) do
-    activity = Activity.get_by_id_with_object(notice_id)
-    timeline =
-      activity.object.data["context"]
-      |> ActivityPub.fetch_activities_for_context(%{})
-      |> Enum.reverse()
-      |> Enum.map(&represent(&1, &1.object.id == activity.object.id))
+    case Activity.get_by_id_with_object(notice_id) do
+      %Activity{} = activity ->
+        timeline =
+          activity.object.data["context"]
+          |> ActivityPub.fetch_activities_for_context(%{})
+          |> Enum.reverse()
+          |> Enum.map(&represent(&1, &1.object.id == activity.object.id))
 
-    render(conn, "conversation.html", %{activities: timeline})
+        render(conn, "conversation.html", %{activities: timeline})
+
+      _ ->
+        conn
+        |> put_status(404)
+        |> render_error(:not_found, "Notice not found")
+    end
   end
 
   def show(%{assigns: %{username_or_id: username_or_id}} = conn, params) do
-    %User{} = user = User.get_cached_by_nickname_or_id(username_or_id)
+    case User.get_cached_by_nickname_or_id(username_or_id) do
+      %User{} = user ->
+        timeline =
+          ActivityPub.fetch_user_activities(user, nil, Map.take(params, @page_keys))
+          |> Enum.map(&represent/1)
 
-    timeline =
-      ActivityPub.fetch_user_activities(user, nil, Map.take(params, @page_keys))
-      |> Enum.map(&represent/1)
+        prev_page_id =
+          (params["min_id"] || params["max_id"]) &&
+            List.first(timeline) && List.first(timeline).id
 
-    prev_page_id =
-      (params["min_id"] || params["max_id"]) &&
-        List.first(timeline) && List.first(timeline).id
+        next_page_id = List.last(timeline) && List.last(timeline).id
 
-    next_page_id = List.last(timeline) && List.last(timeline).id
+        render(conn, "profile.html", %{
+          user: user,
+          timeline: timeline,
+          prev_page_id: prev_page_id,
+          next_page_id: next_page_id
+        })
 
-    render(conn, "profile.html", %{
-      user: user,
-      timeline: timeline,
-      prev_page_id: prev_page_id,
-      next_page_id: next_page_id
-    })
+      _ ->
+        conn
+        |> put_status(404)
+        |> render_error(:not_found, "User not found")
+    end
   end
 
   def assign_id(%{path_info: ["notice", notice_id]} = conn, _opts),