X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=lib%2Fpleroma%2Fworkers%2Fbackground_worker.ex;h=ac2fe6946f549d3cbe867dac3002e2481bafa56c;hb=f917285b72dbc770be40478e1a55973f29d5db7d;hp=3ab2b6bcc81e91af1cd06ce94f9691548a578d72;hpb=0e1c481a94392b69833fbe6afc184ebbd90e1330;p=akkoma

diff --git a/lib/pleroma/workers/background_worker.ex b/lib/pleroma/workers/background_worker.ex
index 3ab2b6bcc..e24b9c175 100644
--- a/lib/pleroma/workers/background_worker.ex
+++ b/lib/pleroma/workers/background_worker.ex
@@ -1,66 +1,41 @@
 # Pleroma: A lightweight social networking server
-# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
+# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
 # SPDX-License-Identifier: AGPL-3.0-only
 
 defmodule Pleroma.Workers.BackgroundWorker do
-  alias Pleroma.Activity
   alias Pleroma.User
-  alias Pleroma.Web.ActivityPub.MRF.MediaProxyWarmingPolicy
-  alias Pleroma.Web.OAuth.Token.CleanWorker
 
-  # Note: `max_attempts` is intended to be overridden in `new/1` call
-  use Oban.Worker,
-    queue: "background",
-    max_attempts: Pleroma.Config.get([:workers, :retries, :compile_time_default])
+  use Pleroma.Workers.WorkerHelper, queue: "background"
 
   @impl Oban.Worker
-  def perform(%{"op" => "fetch_initial_posts", "user_id" => user_id}) do
-    user = User.get_by_id(user_id)
-    User.perform(:fetch_initial_posts, user)
-  end
 
-  def perform(%{"op" => "deactivate_user", "user_id" => user_id, "status" => status}) do
-    user = User.get_by_id(user_id)
+  def perform(%Job{args: %{"op" => "deactivate_user", "user_id" => user_id, "status" => status}}) do
+    user = User.get_cached_by_id(user_id)
     User.perform(:deactivate_async, user, status)
   end
 
-  def perform(%{"op" => "delete_user", "user_id" => user_id}) do
-    user = User.get_by_id(user_id)
+  def perform(%Job{args: %{"op" => "delete_user", "user_id" => user_id}}) do
+    user = User.get_cached_by_id(user_id)
     User.perform(:delete, user)
   end
 
-  def perform(%{
-        "op" => "blocks_import",
-        "blocker_id" => blocker_id,
-        "blocked_identifiers" => blocked_identifiers
-      }) do
-    blocker = User.get_by_id(blocker_id)
-    User.perform(:blocks_import, blocker, blocked_identifiers)
-  end
-
-  def perform(%{
-        "op" => "follow_import",
-        "follower_id" => follower_id,
-        "followed_identifiers" => followed_identifiers
-      }) do
-    follower = User.get_by_id(follower_id)
-    User.perform(:follow_import, follower, followed_identifiers)
+  def perform(%Job{args: %{"op" => "force_password_reset", "user_id" => user_id}}) do
+    user = User.get_cached_by_id(user_id)
+    User.perform(:force_password_reset, user)
   end
 
-  def perform(%{"op" => "clean_expired_tokens"}) do
-    CleanWorker.perform(:clean)
+  def perform(%Job{args: %{"op" => op, "user_id" => user_id, "identifiers" => identifiers}})
+      when op in ["blocks_import", "follow_import", "mutes_import"] do
+    user = User.get_cached_by_id(user_id)
+    {:ok, User.Import.perform(String.to_atom(op), user, identifiers)}
   end
 
-  def perform(%{"op" => "media_proxy_preload", "message" => message}) do
-    MediaProxyWarmingPolicy.perform(:preload, message)
-  end
-
-  def perform(%{"op" => "media_proxy_prefetch", "url" => url}) do
-    MediaProxyWarmingPolicy.perform(:prefetch, url)
-  end
+  def perform(%Job{
+        args: %{"op" => "move_following", "origin_id" => origin_id, "target_id" => target_id}
+      }) do
+    origin = User.get_cached_by_id(origin_id)
+    target = User.get_cached_by_id(target_id)
 
-  def perform(%{"op" => "fetch_data_for_activity", "activity_id" => activity_id}) do
-    activity = Activity.get_by_id(activity_id)
-    Pleroma.Web.RichMedia.Helpers.perform(:fetch, activity)
+    Pleroma.FollowingRelationship.move_following(origin, target)
   end
 end