2024-04-16 20:49:37 +02:00
|
|
|
# Documented in https://zulip.readthedocs.io/en/latest/subsystems/queuing.html
|
|
|
|
import logging
|
2024-07-12 02:30:17 +02:00
|
|
|
from typing import Any
|
2024-04-16 20:49:37 +02:00
|
|
|
|
|
|
|
from typing_extensions import override
|
|
|
|
|
|
|
|
from zerver.lib.outgoing_webhook import do_rest_call, get_outgoing_webhook_service_handler
|
|
|
|
from zerver.models.bots import get_bot_services
|
|
|
|
from zerver.worker.base import QueueProcessingWorker, assign_queue
|
|
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
|
|
|
@assign_queue("outgoing_webhooks")
|
|
|
|
class OutgoingWebhookWorker(QueueProcessingWorker):
|
|
|
|
@override
|
2024-07-12 02:30:17 +02:00
|
|
|
def consume(self, event: dict[str, Any]) -> None:
|
2024-04-16 20:49:37 +02:00
|
|
|
message = event["message"]
|
|
|
|
event["command"] = message["content"]
|
|
|
|
|
|
|
|
services = get_bot_services(event["user_profile_id"])
|
|
|
|
for service in services:
|
|
|
|
event["service_name"] = str(service.name)
|
|
|
|
service_handler = get_outgoing_webhook_service_handler(service)
|
|
|
|
do_rest_call(service.base_url, event, service_handler)
|