revert to the last kind of working code lol

This commit is contained in:
PC-Admin 2023-08-11 01:09:20 +08:00
parent e4f8723b15
commit 129b66dcff

View File

@ -3,6 +3,8 @@ from synapse.module_api import ModuleApi
from twisted.web.http import OK, NO_CONTENT
import json
from twisted.web import http
from twisted.internet import defer
from twisted.internet.defer import inlineCallbacks
logger = logging.getLogger(__name__)
@ -19,45 +21,66 @@ class AbuseLookupModule:
logger.info("AbuseLookupModule initialized.")
class AbuseLookupResource:
isLeaf = True
def __init__(self, module):
self._module = module
async def on_PUT(self, request):
# Extract body from the request
body = await request.content.read()
content = body.decode("utf-8")
def render(self, request):
method = request.method.decode('ascii')
handler = getattr(self, f"on_{method}", None)
# Log the request to Synapse's log
logger.info(f"Received abuse lookup request: {content}")
if handler:
result = handler(request)
return result if isinstance(result, bytes) else self.method_not_allowed(request)
else:
return self.method_not_allowed(request)
# Extract room_id from the content (assuming the content is JSON and valid)
@inlineCallbacks
def on_PUT(self, request):
try:
# Extract body from the request
body = yield request.content.read()
content = body.decode("utf-8")
# Log the request to Synapse's log
logger.info(f"Received abuse lookup request: {content}")
# Extract room_id from the content
data = json.loads(content)
room_id = data["room_id"]
# TODO: Check the room_id against your list/database
# For now, we'll just simulate it
is_abuse = room_id == "!OEedGOAXDBahPyWMSQ:example.com"
if is_abuse:
return (http.OK, json.dumps({
request.setResponseCode(http.OK)
defer.returnValue(json.dumps({
"error": None,
"report_id": "b973d82a-6932-4cad-ac9f-f647a3a9d204",
}).encode("utf-8"))
else:
return (http.NO_CONTENT, b"")
request.setResponseCode(http.NO_CONTENT)
defer.returnValue(b"")
except Exception as e:
logger.error(f"Error processing abuse lookup request: {e}")
return (400, json.dumps({"error": "Bad Request"}).encode("utf-8"))
request.setResponseCode(400)
defer.returnValue(json.dumps({"error": "Bad Request"}).encode("utf-8"))
def __getattr__(self, name):
# This will handle other HTTP methods like GET, POST, etc.
# and return a 405 Method Not Allowed
return self.method_not_allowed
def method_not_allowed(self, _):
return (405, json.dumps({"error": "Method Not Allowed"}).encode("utf-8"))
def on_GET(self, request):
return self.method_not_allowed(request)
def on_POST(self, request):
return self.method_not_allowed(request)
# And similarly for other methods you want to block like DELETE, HEAD, etc.
def method_not_allowed(self, request):
request.setResponseCode(405)
return json.dumps({"error": "Method Not Allowed"}).encode("utf-8")
def parse_config(config: dict) -> dict:
return config