mirror of
https://gitlab.com/chicken-riders/RcGcDw.git
synced 2025-02-23 00:24:09 +00:00
209 lines
9.5 KiB
Python
209 lines
9.5 KiB
Python
# This file is part of Recent changes Goat compatible Discord webhook (RcGcDw).
|
|
#
|
|
# RcGcDw is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# RcGcDw is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with RcGcDw. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import ipaddress
|
|
from src.api.hook import post_hook
|
|
from src.configloader import settings
|
|
|
|
# {
|
|
# "hooks": {
|
|
# "edit_alerts": [
|
|
# {
|
|
# "content": "DISCORD MARKDOWN TEXT",
|
|
# "allowed_mentions": {
|
|
# "users": ["USERID"],
|
|
# "roles": ["ROLEID"]
|
|
# },
|
|
# "requirements": [
|
|
# {
|
|
# "feed": [
|
|
# "recentchanges",
|
|
# "abuselog",
|
|
# "discussion"
|
|
# ],
|
|
# "action": [
|
|
# "edit",
|
|
# "delete/delete",
|
|
# "delete"
|
|
# ],
|
|
# "user": [
|
|
# "USERNAME",
|
|
# "@__anon__",
|
|
# "@__user__"
|
|
# ],
|
|
# "title": [
|
|
# "PAGETITLE"
|
|
# ],
|
|
# "forum": [
|
|
# "FORUMNAME",
|
|
# null
|
|
# ],
|
|
# "is_reply": null,
|
|
# "namespace": [
|
|
# 0
|
|
# ],
|
|
# "tags": [
|
|
# ["EDIT TAG", "AND EDIT TAG"],
|
|
# ["OR EDIT TAG"]
|
|
# ],
|
|
# "categories": [
|
|
# {
|
|
# "added": [
|
|
# ["CATEGORY", "AND CATEGORY"],
|
|
# ["OR CATEGORY"]
|
|
# ],
|
|
# "removed": [
|
|
# ["CATEGORY", "AND CATEGORY"],
|
|
# ["OR CATEGORY"]
|
|
# ]
|
|
# }
|
|
# ],
|
|
# "filter": [
|
|
# "Section removal",
|
|
# "1"
|
|
# ],
|
|
# "af_action": [
|
|
# "edit"
|
|
# ],
|
|
# "result": [
|
|
# "disallow"
|
|
# ]
|
|
# }
|
|
# ]
|
|
# }
|
|
# ]
|
|
# }
|
|
# }
|
|
edit_alerts = settings.get("hooks", {}).get("edit_alerts", [])
|
|
|
|
|
|
class RequirementNotMet(Exception):
|
|
"""Exception raised when the requirement is not met and another requirement must be processed"""
|
|
pass
|
|
|
|
|
|
def check_group_requirements(change_data: list, settings_data: list):
|
|
"""This function resolves group discussions and raises RequirementNotMet when requirement is not met"""
|
|
if settings_data:
|
|
for required_group in settings_data:
|
|
# test all items in required_group are in change_data (one group fulfills the requirement) return the function
|
|
for required_item in required_group:
|
|
if required_item not in change_data:
|
|
break
|
|
else:
|
|
break
|
|
else:
|
|
raise RequirementNotMet
|
|
|
|
|
|
|
|
@post_hook
|
|
def edit_alerts_hook(message, metadata, context, change):
|
|
# For every alert in edit_alerts, they can have different functions and so on
|
|
for alert in edit_alerts:
|
|
# For every requirement, if one of the requirements passes the alert gets executed
|
|
for requirement in alert.get("requirements", []):
|
|
try:
|
|
req_feed = requirement.get("feed", [])
|
|
if req_feed and context.feed_type not in req_feed:
|
|
raise RequirementNotMet
|
|
req_action = requirement.get("action", [])
|
|
# If current action isn't in config for this requirement AND current event type is not in the requirements in settings skip this requirement
|
|
if req_action and context.event not in req_action and context.event.split('/', 1)[0] not in req_action:
|
|
raise RequirementNotMet
|
|
req_user = requirement.get("user", [])
|
|
change_user = None
|
|
change_anon = False
|
|
if context.feed_type == "discussion":
|
|
if change["creatorIp"]:
|
|
change_user = change["creatorIp"][1:]
|
|
change_anon = True
|
|
elif change["createdBy"]["name"]:
|
|
change_user = change["createdBy"]["name"]
|
|
change_anon = False
|
|
else:
|
|
change_user = change["user"]
|
|
if context.feed_type == "recentchanges":
|
|
change_anon = "anon" in change
|
|
else:
|
|
try:
|
|
ipaddress.ip_address(change_user)
|
|
except ValueError:
|
|
change_anon = False
|
|
else:
|
|
change_anon = True
|
|
# If current user is not in config AND checkings for anon and user fail
|
|
if req_user and change_user and change_user not in req_user and ("@__anon__" if change_anon else "@__user__") not in req_user:
|
|
raise RequirementNotMet
|
|
req_title = requirement.get("title", [])
|
|
change_title = change["title"]
|
|
if context.feed_type == "discussion" and change_title is None:
|
|
change_title = change["_embedded"]["thread"][0]["title"]
|
|
if change_title is None and context.comment_page is not None:
|
|
change_title = context.comment_page["title"]
|
|
if req_title and change_title not in req_title:
|
|
raise RequirementNotMet
|
|
if context.feed_type == "discussion":
|
|
req_forum = requirement.get("forum", [])
|
|
if req_forum and change["forumName"] not in req_forum:
|
|
raise RequirementNotMet
|
|
req_reply = requirement.get("is_reply", None)
|
|
if req_reply is not None and change["isReply"] == req_reply:
|
|
raise RequirementNotMet
|
|
else:
|
|
req_namespace = requirement.get("namespace", [])
|
|
if req_namespace and change["ns"] not in req_namespace:
|
|
raise RequirementNotMet
|
|
if context.feed_type == "recentchanges":
|
|
check_group_requirements(change.get("tags", []), requirement.get("tags", []))
|
|
if requirement.get("categories", []):
|
|
for req_cats in requirement.get("categories", []):
|
|
try:
|
|
check_group_requirements(context.categories.new, req_cats.get("added", []))
|
|
check_group_requirements(context.categories.removed, req_cats.get("removed", []))
|
|
except RequirementNotMet:
|
|
continue
|
|
else:
|
|
break
|
|
else:
|
|
raise RequirementNotMet
|
|
elif context.feed_type == "abuselog":
|
|
req_filter = requirement.get("filter", [])
|
|
# Allow both filter id and name as id might be hidden when logged out
|
|
if req_filter and change["filter"] not in req_filter and change["filter_id"] not in req_filter:
|
|
raise RequirementNotMet
|
|
af_action = requirement.get("af_action", [])
|
|
if af_action and change["action"] not in af_action:
|
|
raise RequirementNotMet
|
|
req_result = requirement.get("result", [])
|
|
if req_result and not list(set(change["result"].split(",")) & set(req_result)):
|
|
raise RequirementNotMet
|
|
except RequirementNotMet:
|
|
continue
|
|
else:
|
|
break
|
|
else:
|
|
continue
|
|
message.webhook_object["content"] = (message.webhook_object.get("content", "") or "") + alert["content"]
|
|
allowed_mentions = message.webhook_object["allowed_mentions"]
|
|
if alert.get("allowed_mentions", {}).get("users", []):
|
|
if not allowed_mentions.get("users", []):
|
|
allowed_mentions["users"] = []
|
|
allowed_mentions["users"].extend(alert["allowed_mentions"]["users"])
|
|
if alert.get("allowed_mentions", {}).get("roles", []):
|
|
if not allowed_mentions.get("roles", []):
|
|
allowed_mentions["roles"] = []
|
|
allowed_mentions["roles"].extend(alert["allowed_mentions"]["roles"])
|