140 lines
3.8 KiB
Elixir
140 lines
3.8 KiB
Elixir
# Pleroma: A lightweight social networking server
|
|
# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
defmodule Pleroma.Web.ActivityPub.MRF.KeywordPolicy do
|
|
@behaviour Pleroma.Web.ActivityPub.MRF
|
|
|
|
defp string_matches?(string, _) when not is_binary(string) do
|
|
false
|
|
end
|
|
|
|
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
|
|
|
|
defp check_reject(%{"object" => %{"content" => content, "summary" => summary}} = message) do
|
|
if Enum.any?(Pleroma.Config.get([:mrf_keyword, :reject]), fn pattern ->
|
|
string_matches?(content, pattern) or string_matches?(summary, pattern)
|
|
end) do
|
|
{:reject, nil}
|
|
else
|
|
{:ok, message}
|
|
end
|
|
end
|
|
|
|
defp check_ftl_removal(
|
|
%{"to" => to, "object" => %{"content" => content, "summary" => summary}} = message
|
|
) do
|
|
if "https://www.w3.org/ns/activitystreams#Public" in to and
|
|
Enum.any?(Pleroma.Config.get([:mrf_keyword, :federated_timeline_removal]), fn pattern ->
|
|
string_matches?(content, pattern) or string_matches?(summary, pattern)
|
|
end) do
|
|
to = List.delete(to, "https://www.w3.org/ns/activitystreams#Public")
|
|
cc = ["https://www.w3.org/ns/activitystreams#Public" | message["cc"] || []]
|
|
|
|
message =
|
|
message
|
|
|> Map.put("to", to)
|
|
|> Map.put("cc", cc)
|
|
|
|
{:ok, message}
|
|
else
|
|
{:ok, message}
|
|
end
|
|
end
|
|
|
|
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
|
|
|
|
{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
|
|
)
|
|
|
|
{:ok,
|
|
message
|
|
|> put_in(["object", "content"], content)
|
|
|> put_in(["object", "summary"], summary)}
|
|
end
|
|
|
|
def save(%{
|
|
federated_timeline_removal: ftr,
|
|
reject: reject,
|
|
replace: replace
|
|
}) do
|
|
with true <- Enum.all?(ftr, &String.valid?(&1)),
|
|
true <- Enum.all?(reject, &String.valid?(&1)),
|
|
true <- Enum.all?(Map.keys(replace), &String.valid?(&1)),
|
|
true <- Enum.all?(Map.values(replace), &String.valid?(&1)) do
|
|
Pleroma.Config.put(:mrf_keyword, %{
|
|
federated_timeline_removal: ftr,
|
|
reject: reject,
|
|
replace: replace
|
|
})
|
|
|
|
:ok
|
|
else
|
|
false -> {:error, "All elements must be valid strings"}
|
|
end
|
|
end
|
|
|
|
def list do
|
|
config = Pleroma.Config.get(:mrf_keyword, %{})
|
|
|
|
Enum.reduce(config, %{}, fn {key, value}, acc ->
|
|
base = if is_map(value), do: %{}, else: []
|
|
|
|
value =
|
|
Enum.reduce(value, base, fn
|
|
{pattern, replacement}, acc ->
|
|
pattern = unless is_binary(pattern), do: inspect(pattern), else: pattern
|
|
Map.put(acc, pattern, replacement)
|
|
|
|
pattern, acc ->
|
|
pattern = unless is_binary(pattern), do: inspect(pattern), else: pattern
|
|
|
|
acc ++ [pattern]
|
|
end)
|
|
|
|
Map.put(acc, key, value)
|
|
end)
|
|
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
|
|
_e ->
|
|
{:reject, nil}
|
|
end
|
|
end
|
|
|
|
@impl true
|
|
def filter(message), do: {:ok, message}
|
|
end
|