logo

pleroma

My custom branche(s) on git.pleroma.social/pleroma/pleroma git clone https://hacktivis.me/git/pleroma.git

import.ex (2436B)


  1. # Pleroma: A lightweight social networking server
  2. # Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
  3. # SPDX-License-Identifier: AGPL-3.0-only
  4. defmodule Pleroma.User.Import do
  5. use Ecto.Schema
  6. alias Pleroma.User
  7. alias Pleroma.Web.CommonAPI
  8. alias Pleroma.Workers.BackgroundWorker
  9. require Logger
  10. @spec perform(atom(), User.t(), list()) :: :ok | list() | {:error, any()}
  11. def perform(:mutes_import, %User{} = user, [_ | _] = identifiers) do
  12. Enum.map(
  13. identifiers,
  14. fn identifier ->
  15. with {:ok, %User{} = muted_user} <- User.get_or_fetch(identifier),
  16. {:ok, _} <- User.mute(user, muted_user) do
  17. muted_user
  18. else
  19. error -> handle_error(:mutes_import, identifier, error)
  20. end
  21. end
  22. )
  23. end
  24. def perform(:blocks_import, %User{} = blocker, [_ | _] = identifiers) do
  25. Enum.map(
  26. identifiers,
  27. fn identifier ->
  28. with {:ok, %User{} = blocked} <- User.get_or_fetch(identifier),
  29. {:ok, _block} <- CommonAPI.block(blocker, blocked) do
  30. blocked
  31. else
  32. error -> handle_error(:blocks_import, identifier, error)
  33. end
  34. end
  35. )
  36. end
  37. def perform(:follow_import, %User{} = follower, [_ | _] = identifiers) do
  38. Enum.map(
  39. identifiers,
  40. fn identifier ->
  41. with {:ok, %User{} = followed} <- User.get_or_fetch(identifier),
  42. {:ok, follower, followed} <- User.maybe_direct_follow(follower, followed),
  43. {:ok, _, _, _} <- CommonAPI.follow(follower, followed) do
  44. followed
  45. else
  46. error -> handle_error(:follow_import, identifier, error)
  47. end
  48. end
  49. )
  50. end
  51. def perform(_, _, _), do: :ok
  52. defp handle_error(op, user_id, error) do
  53. Logger.debug("#{op} failed for #{user_id} with: #{inspect(error)}")
  54. error
  55. end
  56. def blocks_import(%User{} = blocker, [_ | _] = identifiers) do
  57. BackgroundWorker.enqueue(
  58. "blocks_import",
  59. %{"user_id" => blocker.id, "identifiers" => identifiers}
  60. )
  61. end
  62. def follow_import(%User{} = follower, [_ | _] = identifiers) do
  63. BackgroundWorker.enqueue(
  64. "follow_import",
  65. %{"user_id" => follower.id, "identifiers" => identifiers}
  66. )
  67. end
  68. def mutes_import(%User{} = user, [_ | _] = identifiers) do
  69. BackgroundWorker.enqueue(
  70. "mutes_import",
  71. %{"user_id" => user.id, "identifiers" => identifiers}
  72. )
  73. end
  74. end