diff options
Diffstat (limited to 'lib/pleroma/user.ex')
| -rw-r--r-- | lib/pleroma/user.ex | 260 |
1 files changed, 157 insertions, 103 deletions
diff --git a/lib/pleroma/user.ex b/lib/pleroma/user.ex index 0594afb38..55b290309 100644 --- a/lib/pleroma/user.ex +++ b/lib/pleroma/user.ex @@ -8,20 +8,20 @@ defmodule Pleroma.User do alias Pleroma.Web.ActivityPub.{Utils, ActivityPub} schema "users" do - field :bio, :string - field :email, :string - field :name, :string - field :nickname, :string - field :password_hash, :string - field :password, :string, virtual: true - field :password_confirmation, :string, virtual: true - field :following, {:array, :string}, default: [] - field :ap_id, :string - field :avatar, :map - field :local, :boolean, default: true - field :info, :map, default: %{} - field :follower_address, :string - has_many :notifications, Notification + field(:bio, :string) + field(:email, :string) + field(:name, :string) + field(:nickname, :string) + field(:password_hash, :string) + field(:password, :string, virtual: true) + field(:password_confirmation, :string, virtual: true) + field(:following, {:array, :string}, default: []) + field(:ap_id, :string) + field(:avatar, :map) + field(:local, :boolean, default: true) + field(:info, :map, default: %{}) + field(:follower_address, :string) + has_many(:notifications, Notification) timestamps() end @@ -41,7 +41,7 @@ defmodule Pleroma.User do end def ap_id(%User{nickname: nickname}) do - "#{Web.base_url}/users/#{nickname}" + "#{Web.base_url()}/users/#{nickname}" end def ap_followers(%User{} = user) do @@ -62,6 +62,7 @@ defmodule Pleroma.User do def user_info(%User{} = user) do oneself = if user.local, do: 1, else: 0 + %{ following_count: length(user.following) - oneself, note_count: user.info["note_count"] || 0, @@ -71,21 +72,25 @@ defmodule Pleroma.User do @email_regex ~r/^[a-zA-Z0-9.!#$%&'*+\/=?^_`{|}~-]+@[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(?:\.[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)*$/ def remote_user_creation(params) do - changes = %User{} - |> cast(params, [:bio, :name, :ap_id, :nickname, :info, :avatar]) - |> validate_required([:name, :ap_id, :nickname]) - |> unique_constraint(:nickname) - |> validate_format(:nickname, @email_regex) - |> validate_length(:bio, max: 5000) - |> validate_length(:name, max: 100) - |> put_change(:local, false) + changes = + %User{} + |> cast(params, [:bio, :name, :ap_id, :nickname, :info, :avatar]) + |> validate_required([:name, :ap_id, :nickname]) + |> unique_constraint(:nickname) + |> validate_format(:nickname, @email_regex) + |> validate_length(:bio, max: 5000) + |> validate_length(:name, max: 100) + |> put_change(:local, false) + if changes.valid? do case changes.changes[:info]["source_data"] do %{"followers" => followers} -> changes |> put_change(:follower_address, followers) + _ -> followers = User.ap_followers(%User{nickname: changes.changes[:nickname]}) + changes |> put_change(:follower_address, followers) end @@ -113,13 +118,15 @@ defmodule Pleroma.User do end def password_update_changeset(struct, params) do - changeset = struct - |> cast(params, [:password, :password_confirmation]) - |> validate_required([:password, :password_confirmation]) - |> validate_confirmation(:password) + changeset = + struct + |> cast(params, [:password, :password_confirmation]) + |> validate_required([:password, :password_confirmation]) + |> validate_confirmation(:password) if changeset.valid? do hashed = Pbkdf2.hashpwsalt(changeset.changes[:password]) + changeset |> put_change(:password_hash, hashed) else @@ -132,21 +139,23 @@ defmodule Pleroma.User do end def register_changeset(struct, params \\ %{}) do - changeset = struct - |> cast(params, [:bio, :email, :name, :nickname, :password, :password_confirmation]) - |> validate_required([:email, :name, :nickname, :password, :password_confirmation]) - |> validate_confirmation(:password) - |> unique_constraint(:email) - |> unique_constraint(:nickname) - |> validate_format(:nickname, ~r/^[a-zA-Z\d]+$/) - |> validate_format(:email, @email_regex) - |> validate_length(:bio, max: 1000) - |> validate_length(:name, min: 1, max: 100) + changeset = + struct + |> cast(params, [:bio, :email, :name, :nickname, :password, :password_confirmation]) + |> validate_required([:email, :name, :nickname, :password, :password_confirmation]) + |> validate_confirmation(:password) + |> unique_constraint(:email) + |> unique_constraint(:nickname) + |> validate_format(:nickname, ~r/^[a-zA-Z\d]+$/) + |> validate_format(:email, @email_regex) + |> validate_length(:bio, max: 1000) + |> validate_length(:name, min: 1, max: 100) if changeset.valid? do hashed = Pbkdf2.hashpwsalt(changeset.changes[:password]) ap_id = User.ap_id(%User{nickname: changeset.changes[:nickname]}) followers = User.ap_followers(%User{nickname: changeset.changes[:nickname]}) + changeset |> put_change(:password_hash, hashed) |> put_change(:ap_id, ap_id) @@ -161,19 +170,20 @@ defmodule Pleroma.User do ap_followers = followed.follower_address if following?(follower, followed) or info["deactivated"] do - {:error, - "Could not follow user: #{followed.nickname} is already on your list."} + {:error, "Could not follow user: #{followed.nickname} is already on your list."} else if !followed.local && follower.local && !ap_enabled?(followed) do Websub.subscribe(follower, followed) end - following = [ap_followers | follower.following] - |> Enum.uniq + following = + [ap_followers | follower.following] + |> Enum.uniq() - follower = follower - |> follow_changeset(%{following: following}) - |> update_and_set_cache + follower = + follower + |> follow_changeset(%{following: following}) + |> update_and_set_cache {:ok, _} = update_follower_count(followed) @@ -183,13 +193,16 @@ defmodule Pleroma.User do def unfollow(%User{} = follower, %User{} = followed) do ap_followers = followed.follower_address + if following?(follower, followed) and follower.ap_id != followed.ap_id do - following = follower.following - |> List.delete(ap_followers) + following = + follower.following + |> List.delete(ap_followers) - { :ok, follower } = follower - |> follow_changeset(%{following: following}) - |> update_and_set_cache + {:ok, follower} = + follower + |> follow_changeset(%{following: following}) + |> update_and_set_cache {:ok, followed} = update_follower_count(followed) @@ -225,12 +238,12 @@ defmodule Pleroma.User do def get_cached_by_ap_id(ap_id) do key = "ap_id:#{ap_id}" - Cachex.get!(:user_cache, key, fallback: fn(_) -> get_by_ap_id(ap_id) end) + Cachex.get!(:user_cache, key, fallback: fn _ -> get_by_ap_id(ap_id) end) end def get_cached_by_nickname(nickname) do key = "nickname:#{nickname}" - Cachex.get!(:user_cache, key, fallback: fn(_) -> get_or_fetch_by_nickname(nickname) end) + Cachex.get!(:user_cache, key, fallback: fn _ -> get_or_fetch_by_nickname(nickname) end) end def get_by_nickname(nickname) do @@ -239,7 +252,7 @@ defmodule Pleroma.User do def get_cached_user_info(user) do key = "user_info:#{user.id}" - Cachex.get!(:user_cache, key, fallback: fn(_) -> user_info(user) end) + Cachex.get!(:user_cache, key, fallback: fn _ -> user_info(user) end) end def fetch_by_nickname(nickname) do @@ -252,29 +265,37 @@ defmodule Pleroma.User do end def get_or_fetch_by_nickname(nickname) do - with %User{} = user <- get_by_nickname(nickname) do + with %User{} = user <- get_by_nickname(nickname) do user - else _e -> - with [_nick, _domain] <- String.split(nickname, "@"), - {:ok, user} <- fetch_by_nickname(nickname) do - user - else _e -> nil - end + else + _e -> + with [_nick, _domain] <- String.split(nickname, "@"), + {:ok, user} <- fetch_by_nickname(nickname) do + user + else + _e -> nil + end end end def get_followers(%User{id: id, follower_address: follower_address}) do - q = from u in User, - where: fragment("? <@ ?", ^[follower_address], u.following), - where: u.id != ^id + q = + from( + u in User, + where: fragment("? <@ ?", ^[follower_address], u.following), + 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 + q = + from( + u in User, + where: u.follower_address in ^following, + where: u.id != ^id + ) {:ok, Repo.all(q)} end @@ -289,9 +310,12 @@ defmodule Pleroma.User do end def update_note_count(%User{} = user) do - note_count_query = from a in Object, - where: fragment("?->>'actor' = ? and ?->>'type' = 'Note'", a.data, ^user.ap_id, a.data), - select: count(a.id) + note_count_query = + from( + a in Object, + where: fragment("?->>'actor' = ? and ?->>'type' = 'Note'", a.data, ^user.ap_id, a.data), + select: count(a.id) + ) note_count = Repo.one(note_count_query) @@ -303,10 +327,13 @@ defmodule Pleroma.User do end def update_follower_count(%User{} = user) do - follower_count_query = from u in User, - where: ^user.follower_address in u.following, - where: u.id != ^user.id, - select: count(u.id) + follower_count_query = + from( + u in User, + where: ^user.follower_address in u.following, + where: u.id != ^user.id, + select: count(u.id) + ) follower_count = Repo.one(follower_count_query) @@ -318,20 +345,25 @@ defmodule Pleroma.User do end def get_notified_from_activity(%Activity{recipients: to}) do - query = from u in User, - where: u.ap_id in ^to, - where: u.local == true + query = + from( + u in User, + where: u.ap_id in ^to, + where: u.local == true + ) Repo.all(query) end def get_recipients_from_activity(%Activity{recipients: to}) do - query = from u in User, - where: u.ap_id in ^to, - or_where: fragment("? && ?", u.following, ^to) + query = + from( + u in User, + where: u.ap_id in ^to, + or_where: fragment("? && ?", u.following, ^to) + ) - query = from u in query, - where: u.local == true + query = from(u in query, where: u.local == true) Repo.all(query) end @@ -340,9 +372,20 @@ defmodule Pleroma.User do if resolve do User.get_or_fetch_by_nickname(query) end - q = from u in User, - where: fragment("(to_tsvector('english', ?) || to_tsvector('english', ?)) @@ plainto_tsquery('english', ?)", u.nickname, u.name, ^query), - limit: 20 + + q = + from( + u in User, + where: + fragment( + "(to_tsvector('english', ?) || to_tsvector('english', ?)) @@ plainto_tsquery('english', ?)", + u.nickname, + u.name, + ^query + ), + limit: 20 + ) + Repo.all(q) end @@ -370,36 +413,40 @@ defmodule Pleroma.User do end def local_user_query() do - from u in User, - where: u.local == true + from(u in User, where: u.local == true) end - def deactivate (%User{} = user) do + def deactivate(%User{} = user) do new_info = Map.put(user.info, "deactivated", true) cs = User.info_changeset(user, %{info: new_info}) update_and_set_cache(cs) end - def delete (%User{} = user) do + def delete(%User{} = user) do {:ok, user} = User.deactivate(user) # Remove all relationships - {:ok, followers } = User.get_followers(user) + {:ok, followers} = User.get_followers(user) + followers - |> Enum.each(fn (follower) -> User.unfollow(follower, user) end) + |> Enum.each(fn follower -> User.unfollow(follower, user) end) {:ok, friends} = User.get_friends(user) + friends - |> Enum.each(fn (followed) -> User.unfollow(user, followed) end) + |> Enum.each(fn followed -> User.unfollow(user, followed) end) - query = from a in Activity, - where: a.actor == ^user.ap_id + query = from(a in Activity, where: a.actor == ^user.ap_id) Repo.all(query) - |> Enum.each(fn (activity) -> + |> Enum.each(fn activity -> case activity.data["type"] do - "Create" -> ActivityPub.delete(Object.get_by_ap_id(activity.data["object"]["id"])) - _ -> "Doing nothing" # TODO: Do something with likes, follows, repeats. + "Create" -> + ActivityPub.delete(Object.get_by_ap_id(activity.data["object"]["id"])) + + # TODO: Do something with likes, follows, repeats. + _ -> + "Doing nothing" end end) @@ -413,7 +460,9 @@ defmodule Pleroma.User do ap_try = ActivityPub.make_user_from_ap_id(ap_id) case ap_try do - {:ok, user} -> user + {:ok, user} -> + user + _ -> case OStatus.make_user(ap_id) do {:ok, user} -> user @@ -424,12 +473,15 @@ defmodule Pleroma.User do end # AP style - def public_key_from_info(%{"source_data" => %{"publicKey" => %{"publicKeyPem" => public_key_pem}}}) do - key = :public_key.pem_decode(public_key_pem) - |> hd() - |> :public_key.pem_entry_decode() + def public_key_from_info(%{ + "source_data" => %{"publicKey" => %{"publicKeyPem" => public_key_pem}} + }) do + key = + :public_key.pem_decode(public_key_pem) + |> hd() + |> :public_key.pem_entry_decode() - {:ok, key} + {:ok, key} end # OStatus Magic Key @@ -450,8 +502,10 @@ defmodule Pleroma.User do defp blank?(n), do: n def insert_or_update_user(data) do - data = data - |> Map.put(:name, blank?(data[:name]) || data[:nickname]) + data = + data + |> Map.put(:name, blank?(data[:name]) || data[:nickname]) + cs = User.remote_user_creation(data) Repo.insert(cs, on_conflict: :replace_all, conflict_target: :nickname) end |