pleroma/lib/pleroma/web/activity_pub/mrf/force_bot_unlisted_policy.ex

74 lines
1.9 KiB
Elixir

# Pleroma: A lightweight social networking server
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only
defmodule Pleroma.Web.ActivityPub.MRF.ForceBotUnlistedPolicy do
alias Pleroma.User
@behaviour Pleroma.Web.ActivityPub.MRF
@moduledoc "Remove bot posts from federated timeline"
require Pleroma.Constants
defp check_by_actor_type(user) do
if user.actor_type in ["Application", "Service"], do: 1.0, else: 0.0
end
defp check_by_nickname(user) do
if Regex.match?(~r/bot@|ebooks@/i, user.nickname), do: 1.0, else: 0.0
end
#defp check_by_displayname(user) do
# if Regex.match?(~r/bot|ebooks/i, user.name), do: 0.5, else: 0.0
#end
defp botness_score(user), do: check_by_actor_type(user) + check_by_nickname(user)
# + check_by_displayname(user)
defp filter_bot_posts(
%{
"type" => "Create",
"to" => to,
"cc" => cc,
"actor" => actor,
"object" => object
} = message
) do
user = User.get_cached_by_ap_id(actor)
isbot = 0.8 < botness_score(user)
#IO.puts user.nickname
#IO.puts isbot
#IO.puts botness_score(user)
if isbot and Enum.member?(to, Pleroma.Constants.as_public()) do
to = List.delete(to, Pleroma.Constants.as_public()) ++ [user.follower_address]
cc = List.delete(cc, user.follower_address) ++ [Pleroma.Constants.as_public()]
object =
object
|> Map.put("to", to)
|> Map.put("cc", cc)
message =
message
|> Map.put("to", to)
|> Map.put("cc", cc)
|> Map.put("object", object)
{:ok, message}
else
{:ok, message}
end
end
@impl true
def filter(%{"type" => "Create"} = message),
do: filter_bot_posts(message)
@impl true
def filter(message), do: {:ok, message}
@impl true
def describe, do: {:ok, %{}}
end