Add follower / friend fetching to user.

1 files changed, 17 insertions, 0 deletions

diff --git a/lib/pleroma/user.ex b/lib/pleroma/user.ex
index af0e569a4..d5befa67b 100644
--- a/lib/pleroma/user.ex
+++ b/lib/pleroma/user.ex
@@ -186,4 +186,21 @@ defmodule Pleroma.User do
       end
     end
   end
+
+  # TODO: these queries could be more efficient if the type in postgresql wasn't map, but array.
+  def get_followers(%User{id: id, follower_address: follower_address}) do
+    q = from u in User,
+      where: fragment("? @> ?", u.following, ^follower_address ),
+      where: u.id != ^id
+
+    {:ok, Repo.all(q)}
+  end
+
+  def get_friends(%User{id: id, following: following}) do
+    q = from u in User,
+      where: u.follower_address in ^following,
+      where: u.id != ^id
+
+    {:ok, Repo.all(q)}
+  end
 end