commit: 44aef2183e0f44dbb8619ad3438e80cca67fb5cb
parent: 140f72725dd3d2840b356107e24542ba2896e4e1
Author: Roger Braun <roger@rogerbraun.net>
Date: Thu, 20 Jul 2017 19:37:41 +0200
Add follower / friend fetching to user.
Diffstat:
2 files changed, 47 insertions(+), 0 deletions(-)
diff --git 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
diff --git a/test/user_test.exs b/test/user_test.exs
@@ -197,5 +197,35 @@ defmodule Pleroma.UserTest do
end)
end
end
+
+ describe "followers and friends" do
+ test "gets all followers for a given user" do
+ user = insert(:user)
+ follower_one = insert(:user)
+ follower_two = insert(:user)
+ not_follower = insert(:user)
+
+ {:ok, follower_one} = User.follow(follower_one, user)
+ {:ok, follower_two} = User.follow(follower_two, user)
+
+ {:ok, res} = User.get_followers(user)
+
+ assert res == [follower_one, follower_two]
+ end
+
+ test "gets all friends (followed users) for a given user" do
+ user = insert(:user)
+ followed_one = insert(:user)
+ followed_two = insert(:user)
+ not_followed = insert(:user)
+
+ {:ok, user} = User.follow(user, followed_one)
+ {:ok, user} = User.follow(user, followed_two)
+
+ {:ok, res} = User.get_friends(user)
+
+ assert res == [followed_one, followed_two]
+ end
+ end
end