diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/pleroma/activity.ex | 2 | ||||
| -rw-r--r-- | lib/pleroma/plugs/authentication_plug.ex | 1 | ||||
| -rw-r--r-- | lib/pleroma/plugs/oauth_plug.ex | 3 | ||||
| -rw-r--r-- | lib/pleroma/user.ex | 64 | ||||
| -rw-r--r-- | lib/pleroma/web/activity_pub/utils.ex | 7 | ||||
| -rw-r--r-- | lib/pleroma/web/federator/federator.ex | 29 | 
6 files changed, 82 insertions, 24 deletions
| diff --git a/lib/pleroma/activity.ex b/lib/pleroma/activity.ex index a35ccc9b4..afd09982f 100644 --- a/lib/pleroma/activity.ex +++ b/lib/pleroma/activity.ex @@ -7,7 +7,7 @@ defmodule Pleroma.Activity do      field :data, :map      field :local, :boolean, default: true      field :actor, :string -    has_many :notifications, Notification +    has_many :notifications, Notification, on_delete: :delete_all      timestamps()    end diff --git a/lib/pleroma/plugs/authentication_plug.ex b/lib/pleroma/plugs/authentication_plug.ex index beb02eb88..60f6faf49 100644 --- a/lib/pleroma/plugs/authentication_plug.ex +++ b/lib/pleroma/plugs/authentication_plug.ex @@ -12,6 +12,7 @@ defmodule Pleroma.Plugs.AuthenticationPlug do    def call(conn, opts) do      with {:ok, username, password} <- decode_header(conn),           {:ok, user} <- opts[:fetcher].(username), +         false <- !!user.info["deactivated"],           saved_user_id <- get_session(conn, :user_id),           {:ok, verified_user} <- verify(user, password, saved_user_id)      do diff --git a/lib/pleroma/plugs/oauth_plug.ex b/lib/pleroma/plugs/oauth_plug.ex index 775423bb1..be737dc9a 100644 --- a/lib/pleroma/plugs/oauth_plug.ex +++ b/lib/pleroma/plugs/oauth_plug.ex @@ -16,7 +16,8 @@ defmodule Pleroma.Plugs.OAuthPlug do              end      with token when not is_nil(token) <- token,           %Token{user_id: user_id} <- Repo.get_by(Token, token: token), -         %User{} = user <- Repo.get(User, user_id) do +         %User{} = user <- Repo.get(User, user_id), +         false <- !!user.info["deactivated"] do        conn        |> assign(:user, user)      else diff --git a/lib/pleroma/user.ex b/lib/pleroma/user.ex index afc62f265..09bcf0cb4 100644 --- a/lib/pleroma/user.ex +++ b/lib/pleroma/user.ex @@ -5,7 +5,7 @@ defmodule Pleroma.User do    alias Pleroma.{Repo, User, Object, Web, Activity, Notification}    alias Comeonin.Pbkdf2    alias Pleroma.Web.{OStatus, Websub} -  alias Pleroma.Web.ActivityPub.Utils +  alias Pleroma.Web.ActivityPub.{Utils, ActivityPub}    schema "users" do      field :bio, :string @@ -113,7 +113,7 @@ defmodule Pleroma.User do    end    def reset_password(user, data) do -    Repo.update(password_update_changeset(user, data)) +    update_and_set_cache(password_update_changeset(user, data))    end    def register_changeset(struct, params \\ %{}) do @@ -142,9 +142,9 @@ defmodule Pleroma.User do      end    end -  def follow(%User{} = follower, %User{} = followed) do +  def follow(%User{} = follower, %User{info: info} = followed) do      ap_followers = followed.follower_address -    if following?(follower, followed) do +    if following?(follower, followed) or info["deactivated"] do        {:error,         "Could not follow user: #{followed.nickname} is already on your list."}      else @@ -157,7 +157,7 @@ defmodule Pleroma.User do        follower = follower        |> follow_changeset(%{following: following}) -      |> Repo.update +      |> update_and_set_cache        {:ok, _} = update_follower_count(followed) @@ -173,7 +173,7 @@ defmodule Pleroma.User do        { :ok, follower } = follower        |> follow_changeset(%{following: following}) -      |> Repo.update +      |> update_and_set_cache        {:ok, followed} = update_follower_count(followed) @@ -191,6 +191,17 @@ defmodule Pleroma.User do      Repo.get_by(User, ap_id: ap_id)    end +  def update_and_set_cache(changeset) do +    with {:ok, user} <- Repo.update(changeset) do +      Cachex.set(:user_cache, "ap_id:#{user.ap_id}", user) +      Cachex.set(:user_cache, "nickname:#{user.nickname}", user) +      Cachex.set(:user_cache, "user_info:#{user.id}", user_info(user)) +      {:ok, user} +    else +      e -> e +    end +  end +    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) @@ -245,7 +256,7 @@ defmodule Pleroma.User do      cs = info_changeset(user, %{info: new_info}) -    Repo.update(cs) +    update_and_set_cache(cs)    end    def update_note_count(%User{} = user) do @@ -259,7 +270,7 @@ defmodule Pleroma.User do      cs = info_changeset(user, %{info: new_info}) -    Repo.update(cs) +    update_and_set_cache(cs)    end    def update_follower_count(%User{} = user) do @@ -274,7 +285,7 @@ defmodule Pleroma.User do      cs = info_changeset(user, %{info: new_info}) -    Repo.update(cs) +    update_and_set_cache(cs)    end    def get_notified_from_activity(%Activity{data: %{"to" => to}}) do @@ -312,7 +323,7 @@ defmodule Pleroma.User do      new_info = Map.put(user.info, "blocks", new_blocks)      cs = User.info_changeset(user, %{info: new_info}) -    Repo.update(cs) +    update_and_set_cache(cs)    end    def unblock(user, %{ap_id: ap_id}) do @@ -321,7 +332,7 @@ defmodule Pleroma.User do      new_info = Map.put(user.info, "blocks", new_blocks)      cs = User.info_changeset(user, %{info: new_info}) -    Repo.update(cs) +    update_and_set_cache(cs)    end    def blocks?(user, %{ap_id: ap_id}) do @@ -334,4 +345,35 @@ defmodule Pleroma.User do        where: u.local == true    end +  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 +    {:ok, user} = User.deactivate(user) + +    # Remove all relationships +    {:ok, followers } = User.get_followers(user) +    followers +    |> 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) + +    query = from a in Activity, +      where: a.actor == ^user.ap_id + +    Repo.all(query) +    |> 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. +      end +    end) + +    :ok +  end  end diff --git a/lib/pleroma/web/activity_pub/utils.ex b/lib/pleroma/web/activity_pub/utils.ex index 51fac6fe2..ac20a2822 100644 --- a/lib/pleroma/web/activity_pub/utils.ex +++ b/lib/pleroma/web/activity_pub/utils.ex @@ -29,7 +29,12 @@ defmodule Pleroma.Web.ActivityPub.Utils do    Enqueues an activity for federation if it's local    """    def maybe_federate(%Activity{local: true} = activity) do -    Pleroma.Web.Federator.enqueue(:publish, activity) +    priority = case activity.data["type"] do +                 "Delete" -> 10 +                 "Create" -> 1 +                 _ -> 5 +               end +    Pleroma.Web.Federator.enqueue(:publish, activity, priority)      :ok    end    def maybe_federate(_), do: :ok diff --git a/lib/pleroma/web/federator/federator.ex b/lib/pleroma/web/federator/federator.ex index 9f6f983aa..b23ed5fcc 100644 --- a/lib/pleroma/web/federator/federator.ex +++ b/lib/pleroma/web/federator/federator.ex @@ -15,8 +15,8 @@ defmodule Pleroma.Web.Federator do        enqueue(:refresh_subscriptions, nil)      end)      GenServer.start_link(__MODULE__, %{ -          in: {:sets.new(), :queue.new()}, -          out: {:sets.new(), :queue.new()} +          in: {:sets.new(), []}, +          out: {:sets.new(), []}                           }, name: __MODULE__)    end @@ -79,17 +79,17 @@ defmodule Pleroma.Web.Federator do      {:error, "Don't know what do do with this"}    end -  def enqueue(type, payload) do +  def enqueue(type, payload, priority \\ 1) do      if Mix.env == :test do        handle(type, payload)      else -      GenServer.cast(__MODULE__, {:enqueue, type, payload}) +      GenServer.cast(__MODULE__, {:enqueue, type, payload, priority})      end    end    def maybe_start_job(running_jobs, queue) do -    if (:sets.size(running_jobs) < @max_jobs) && !:queue.is_empty(queue) do -      {{:value, {type, payload}}, queue} = :queue.out(queue) +    if (:sets.size(running_jobs) < @max_jobs) && queue != [] do +      {{type, payload}, queue} = queue_pop(queue)        {:ok, pid} = Task.start(fn -> handle(type, payload) end)        mref = Process.monitor(pid)        {:sets.add_element(mref, running_jobs), queue} @@ -98,16 +98,16 @@ defmodule Pleroma.Web.Federator do      end    end -  def handle_cast({:enqueue, type, payload}, state) when type in [:incoming_doc] do +  def handle_cast({:enqueue, type, payload, priority}, state) when type in [:incoming_doc] do      %{in: {i_running_jobs, i_queue}, out: {o_running_jobs, o_queue}} = state -    i_queue = :queue.in({type, payload}, i_queue) +    i_queue = enqueue_sorted(i_queue, {type, payload}, 1)      {i_running_jobs, i_queue} = maybe_start_job(i_running_jobs, i_queue)      {:noreply, %{in: {i_running_jobs, i_queue}, out: {o_running_jobs, o_queue}}}    end -  def handle_cast({:enqueue, type, payload}, state) do +  def handle_cast({:enqueue, type, payload, priority}, state) do      %{in: {i_running_jobs, i_queue}, out: {o_running_jobs, o_queue}} = state -    o_queue = :queue.in({type, payload}, o_queue) +    o_queue = enqueue_sorted(o_queue, {type, payload}, 1)      {o_running_jobs, o_queue} = maybe_start_job(o_running_jobs, o_queue)      {:noreply, %{in: {i_running_jobs, i_queue}, out: {o_running_jobs, o_queue}}}    end @@ -126,4 +126,13 @@ defmodule Pleroma.Web.Federator do      {:noreply, %{in: {i_running_jobs, i_queue}, out: {o_running_jobs, o_queue}}}    end + +  def enqueue_sorted(queue, element, priority) do +    [%{item: element, priority: priority} | queue] +    |> Enum.sort_by(fn (%{priority: priority}) -> priority end) +  end + +  def queue_pop([%{item: element} | queue]) do +    {element, queue} +  end  end | 
