Add User.get_domain/1

2 files changed, 9 insertions, 0 deletions

diff --git a/lib/pleroma/user.ex b/lib/pleroma/user.ex
index ed1db04c9..cc1a8ebac 100644
--- a/lib/pleroma/user.ex
+++ b/lib/pleroma/user.ex
@@ -2293,4 +2293,8 @@ defmodule Pleroma.User do
     |> Map.put(:bio, HTML.filter_tags(user.bio, filter))
     |> Map.put(:fields, fields)
   end
+
+  def get_domain(%User{ap_id: ap_id} = user) do
+    URI.parse(ap_id).host
+  end
 end
diff --git a/test/user_test.exs b/test/user_test.exs
index d087e9101..3a624924a 100644
--- a/test/user_test.exs
+++ b/test/user_test.exs
@@ -1885,4 +1885,9 @@ defmodule Pleroma.UserTest do
 
     assert User.avatar_url(user, no_default: true) == nil
   end
+
+  test "get_domain/1" do
+    user = insert(:user, ap_id: "https://lain.com/users/lain", nickname: "lain")
+    assert User.get_domain(user) == "lain.com"
+  end
 end