akkoma/lib/pleroma/web/activity_pub/mrf/keyword_policy.ex

133 lines
3.6 KiB
Elixir
Raw Normal View History

2019-02-08 09:38:24 +00:00
# Pleroma: A lightweight social networking server
2020-03-03 22:44:49 +00:00
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
2019-02-08 09:38:24 +00:00
# SPDX-License-Identifier: AGPL-3.0-only
defmodule Pleroma.Web.ActivityPub.MRF.KeywordPolicy do
require Pleroma.Constants
@moduledoc "Reject or Word-Replace messages with a keyword or regex"
2019-02-08 09:38:24 +00:00
@behaviour Pleroma.Web.ActivityPub.MRF
defp string_matches?(string, _) when not is_binary(string) do
false
end
2019-02-08 09:38:24 +00:00
defp string_matches?(string, pattern) when is_binary(pattern) do
String.contains?(string, pattern)
end
defp string_matches?(string, pattern) do
String.match?(string, pattern)
end
2019-02-11 18:35:40 +00:00
defp check_reject(%{"object" => %{"content" => content, "summary" => summary}} = message) do
2019-02-08 09:38:24 +00:00
if Enum.any?(Pleroma.Config.get([:mrf_keyword, :reject]), fn pattern ->
2019-02-11 18:35:40 +00:00
string_matches?(content, pattern) or string_matches?(summary, pattern)
2019-02-08 09:38:24 +00:00
end) do
{:reject, "[KeywordPolicy] Matches with rejected keyword"}
2019-02-08 09:38:24 +00:00
else
{:ok, message}
end
end
2019-02-11 18:35:40 +00:00
defp check_ftl_removal(
%{"to" => to, "object" => %{"content" => content, "summary" => summary}} = message
) do
if Pleroma.Constants.as_public() in to and
Enum.any?(Pleroma.Config.get([:mrf_keyword, :federated_timeline_removal]), fn pattern ->
2019-02-11 18:35:40 +00:00
string_matches?(content, pattern) or string_matches?(summary, pattern)
2019-02-08 09:38:24 +00:00
end) do
to = List.delete(to, Pleroma.Constants.as_public())
cc = [Pleroma.Constants.as_public() | message["cc"] || []]
2019-02-08 09:38:24 +00:00
message =
message
|> Map.put("to", to)
|> Map.put("cc", cc)
{:ok, message}
else
{:ok, message}
end
end
2019-02-11 18:35:40 +00:00
defp check_replace(%{"object" => %{"content" => content, "summary" => summary}} = message) do
content =
if is_binary(content) do
content
else
""
end
summary =
if is_binary(summary) do
summary
else
""
end
2019-02-11 18:35:40 +00:00
{content, summary} =
Enum.reduce(
Pleroma.Config.get([:mrf_keyword, :replace]),
{content, summary},
fn {pattern, replacement}, {content_acc, summary_acc} ->
{String.replace(content_acc, pattern, replacement),
String.replace(summary_acc, pattern, replacement)}
end
)
2019-02-08 09:38:24 +00:00
2019-02-11 18:35:40 +00:00
{:ok,
message
|> put_in(["object", "content"], content)
|> put_in(["object", "summary"], summary)}
2019-02-08 09:38:24 +00:00
end
@impl true
def filter(%{"type" => "Create", "object" => %{"content" => _content}} = message) do
with {:ok, message} <- check_reject(message),
{:ok, message} <- check_ftl_removal(message),
{:ok, message} <- check_replace(message) do
{:ok, message}
else
{:reject, nil} -> {:reject, "[KeywordPolicy] "}
{:reject, _} = e -> e
_e -> {:reject, "[KeywordPolicy] "}
2019-02-08 09:38:24 +00:00
end
end
@impl true
def filter(message), do: {:ok, message}
@impl true
2019-08-13 22:36:24 +00:00
def describe do
# This horror is needed to convert regex sigils to strings
mrf_keyword =
Pleroma.Config.get(:mrf_keyword, [])
|> Enum.map(fn {key, value} ->
{key,
Enum.map(value, fn
{pattern, replacement} ->
%{
"pattern" =>
if not is_binary(pattern) do
inspect(pattern)
else
pattern
end,
"replacement" => replacement
}
pattern ->
if not is_binary(pattern) do
inspect(pattern)
else
pattern
end
end)}
end)
|> Enum.into(%{})
{:ok, %{mrf_keyword: mrf_keyword}}
end
2019-02-08 09:38:24 +00:00
end