From a0d732ec55fcbce8cf345344d1456dc77bb0f3bf Mon Sep 17 00:00:00 2001
From: Karen Konou <konoukaren@gmail.com>
Date: Sat, 9 Feb 2019 17:47:57 +0100
Subject: [PATCH] it works!!

---
 lib/pleroma/notification.ex    |  2 ++
 lib/pleroma/web/thread_mute.ex | 28 +++++++++++++++++++++++++---
 test/web/thread_mute_test.exs  | 25 +++++++++++++------------
 3 files changed, 40 insertions(+), 15 deletions(-)

diff --git a/lib/pleroma/notification.ex b/lib/pleroma/notification.ex
index 2364d36da..d05708e41 100644
--- a/lib/pleroma/notification.ex
+++ b/lib/pleroma/notification.ex
@@ -6,6 +6,7 @@ defmodule Pleroma.Notification do
   use Ecto.Schema
   alias Pleroma.{User, Activity, Notification, Repo}
   alias Pleroma.Web.CommonAPI.Utils
+  alias Pleroma.Web.ThreadMute
   import Ecto.Query
 
   schema "notifications" do
@@ -112,6 +113,7 @@ def create_notifications(_), do: {:ok, []}
   # TODO move to sql, too.
   def create_notification(%Activity{} = activity, %User{} = user) do
     unless User.blocks?(user, %{ap_id: activity.data["actor"]}) or
+             ThreadMute.muted?(user, activity) or
              user.ap_id == activity.data["actor"] or
              (activity.data["type"] == "Follow" and
                 Enum.any?(Notification.for_user(user), fn notif ->
diff --git a/lib/pleroma/web/thread_mute.ex b/lib/pleroma/web/thread_mute.ex
index 3a950f474..146de0d80 100644
--- a/lib/pleroma/web/thread_mute.ex
+++ b/lib/pleroma/web/thread_mute.ex
@@ -4,6 +4,7 @@
 
 defmodule Pleroma.Web.ThreadMute do
   use Ecto.Schema
+  alias Pleroma.Web.ThreadMute
   alias Pleroma.{Activity, Repo, User}
   require Ecto.Query
 
@@ -13,16 +14,37 @@ defmodule Pleroma.Web.ThreadMute do
   end
 
   def add_mute(user, id) do
-    %{data: %{"context" => context}} = Activity.get_by_id(id)
+    activity = Activity.get_by_id(id)
+    context = activity.data["context"]
     mute = %Pleroma.Web.ThreadMute{user_id: user.id, context: context}
     Repo.insert(mute)
+    {:ok, activity}
   end
 
   def remove_mute(user, id) do
     user_id = Pleroma.FlakeId.from_string(user.id)
-    %{data: %{"context" => context}} = Activity.get_by_id(id)
+    activity = Activity.get_by_id(id)
+    context = activity.data["context"]
 
-    Ecto.Query.from(m in "thread_mutes", where: m.user_id == ^user_id and m.context == ^context)
+    Ecto.Query.from(m in ThreadMute, where: m.user_id == ^user_id and m.context == ^context)
     |> Repo.delete_all()
+
+    {:ok, activity}
+  end
+
+  def muted?(user, activity) do
+    user_id = Pleroma.FlakeId.from_string(user.id)
+    context = activity.data["context"]
+
+    result =
+      Ecto.Query.from(m in ThreadMute,
+        where: m.user_id == ^user_id and m.context == ^context
+      )
+      |> Repo.all()
+
+    case result do
+      [] -> false
+      _ -> true
+    end
   end
 end
diff --git a/test/web/thread_mute_test.exs b/test/web/thread_mute_test.exs
index 75277ef62..212cae860 100644
--- a/test/web/thread_mute_test.exs
+++ b/test/web/thread_mute_test.exs
@@ -10,31 +10,32 @@ defmodule Pleroma.Web.ThreadMuteTest do
 
   describe "mute tests" do
     setup do
-      user = insert(:user, %{id: "1"})
+      user = insert(:user)
 
-      activity =
-        insert(:note_activity, %{
-          data: %{
-            "context" => "http://localhost:4000/contexts/361ca23e-ffa7-4773-b981-a355a18dc592"
-          }
-        })
+      activity = insert(:note_activity)
 
       [user: user, activity: activity]
     end
 
     test "add mute", %{user: user, activity: activity} do
       id = activity.id
-      {:ok, mute} = add_mute(user, id)
-
-      assert mute.user_id == "1"
-      assert mute.context == "http://localhost:4000/contexts/361ca23e-ffa7-4773-b981-a355a18dc592"
+      {:ok, _activity} = add_mute(user, id)
     end
 
     test "remove mute", %{user: user, activity: activity} do
       id = activity.id
 
       add_mute(user, id)
-      {1, nil} = remove_mute(user, id)
+      {:ok, _activity} = remove_mute(user, id)
+    end
+
+    test "check mute", %{user: user, activity: activity} do
+      id = activity.id
+
+      add_mute(user, id)
+      assert muted?(user, activity)
+      remove_mute(user, id)
+      refute muted?(user, activity)
     end
   end
 end