mirror of
https://git.pleroma.social/pleroma/pleroma.git
synced 2024-06-02 11:20:51 +00:00
125 lines
3.4 KiB
Elixir
125 lines
3.4 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.AutoSubjectPolicy do
|
|
@moduledoc "Apply Subject to local posts matching certain keywords."
|
|
|
|
@behaviour Pleroma.Web.ActivityPub.MRF
|
|
|
|
alias Pleroma.User
|
|
|
|
require Pleroma.Constants
|
|
require Logger
|
|
|
|
@trim_regex Regex.compile!("[.?!:;]+$")
|
|
|
|
@impl true
|
|
def filter(%{"type" => "Create", "actor" => actor, "object" => _object} = message) do
|
|
with {:ok, %User{local: true}} <- User.get_or_fetch_by_ap_id(actor),
|
|
{:ok, message} <- check_subject(message),
|
|
{:ok, message} <- check_match(message) do
|
|
{:ok, message}
|
|
else
|
|
{:ok, %User{local: false}} ->
|
|
{:ok, message}
|
|
|
|
{:error, :has_subject} ->
|
|
{:ok, message}
|
|
|
|
{:error, _} ->
|
|
{:reject, "[AutoSubjectPolicy] Failed to get or fetch user by ap_id"}
|
|
|
|
e ->
|
|
{:reject, "[AutoSubjectPolicy] Unhandled error #{inspect(e)}"}
|
|
end
|
|
end
|
|
|
|
@impl true
|
|
def filter(message), do: {:ok, message}
|
|
|
|
defp check_subject(%{"object" => %{"summary" => subject}} = message) do
|
|
subject = String.trim(subject)
|
|
|
|
if String.length(subject) == 0 do
|
|
{:ok, message}
|
|
else
|
|
{:error, :has_subject}
|
|
end
|
|
end
|
|
|
|
defp check_subject(message), do: {:ok, message}
|
|
|
|
defp string_matches?(content, keywords) when is_list(keywords) do
|
|
wordlist = content |> make_wordlist |> trim_punct
|
|
Enum.any?(keywords, fn match -> String.downcase(match) in wordlist end)
|
|
end
|
|
|
|
defp string_matches?(content, keyword) when is_binary(keyword) do
|
|
wordlist = content |> make_wordlist |> trim_punct
|
|
String.downcase(keyword) in wordlist
|
|
end
|
|
|
|
defp check_match(%{"object" => %{} = object} = message) do
|
|
match_settings = Pleroma.Config.get([:mrf_auto_subject, :match])
|
|
|
|
auto_summary =
|
|
Enum.reduce(match_settings, [], fn {keywords, subject}, acc ->
|
|
if string_matches?(object["content"], keywords) do
|
|
[subject | acc]
|
|
else
|
|
acc
|
|
end
|
|
end)
|
|
|> Enum.join(", ")
|
|
|
|
message = put_in(message["object"]["summary"], auto_summary)
|
|
|
|
{:ok, message}
|
|
end
|
|
|
|
defp make_wordlist(content),
|
|
do:
|
|
content
|
|
|> String.downcase()
|
|
|> String.split(" ", trim: true)
|
|
|> Enum.uniq()
|
|
|
|
defp trim_punct(wordlist) when is_list(wordlist),
|
|
do: wordlist |> Enum.map(fn word -> String.replace(word, @trim_regex, "") end)
|
|
|
|
@impl true
|
|
def describe do
|
|
mrf_autosubject =
|
|
:mrf_auto_subject
|
|
|> Pleroma.Config.get()
|
|
|> Enum.into(%{})
|
|
|
|
{:ok, %{mrf_auto_subject: mrf_autosubject}}
|
|
end
|
|
|
|
@impl true
|
|
def config_description do
|
|
%{
|
|
key: :mrf_auto_subject,
|
|
related_policy: "Pleroma.Web.ActivityPub.MRF.AutoSubjectPolicy",
|
|
label: "MRF AutoSubject",
|
|
description:
|
|
"Adds subject to messages matching a keyword or list of keywords if no subject is defined.",
|
|
children: [
|
|
%{
|
|
key: :match,
|
|
type: {:keyword, :string},
|
|
description: """
|
|
**Keyword**: a string or list of keywords. E.g., ["cat", "dog"] to match on both "cat" and "dog".
|
|
|
|
**Subject**: a string to insert into the subject field.
|
|
|
|
Note: the keyword matching is case-insensitive and matches only the whole word.
|
|
"""
|
|
}
|
|
]
|
|
}
|
|
end
|
|
end
|