mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 14:03:30 +00:00
2bd9c8cb42
This commit introduces a simple field where the user can now specify custom HTTP headers. This commit does not introduce an improved system for storing HTTP headers as fixtures - such a change would modify both the existing unit tests as well as this devtool.
79 lines
3.1 KiB
Python
79 lines
3.1 KiB
Python
import os
|
|
import ujson
|
|
from typing import List
|
|
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.shortcuts import render
|
|
from django.test import Client
|
|
|
|
from zerver.lib.integrations import WEBHOOK_INTEGRATIONS
|
|
from zerver.lib.request import has_request_variables, REQ
|
|
from zerver.lib.response import json_success, json_error
|
|
from zerver.models import UserProfile, get_realm
|
|
from zerver.management.commands.send_webhook_fixture_message import parse_headers
|
|
|
|
|
|
ZULIP_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../../../')
|
|
|
|
|
|
def get_webhook_integrations() -> List[str]:
|
|
return [integration.name for integration in WEBHOOK_INTEGRATIONS]
|
|
|
|
|
|
def dev_panel(request: HttpRequest) -> HttpResponse:
|
|
integrations = get_webhook_integrations()
|
|
bots = UserProfile.objects.filter(is_bot=True, bot_type=UserProfile.INCOMING_WEBHOOK_BOT)
|
|
context = {"integrations": integrations, "bots": bots}
|
|
return render(request, "zerver/integrations/development/dev_panel.html", context)
|
|
|
|
|
|
@has_request_variables
|
|
def get_fixtures(request: HttpResponse,
|
|
integration_name: str=REQ()) -> HttpResponse:
|
|
integrations = get_webhook_integrations()
|
|
if integration_name not in integrations:
|
|
return json_error("\"{integration_name}\" is not a valid webhook integration.".format(
|
|
integration_name=integration_name), status=404)
|
|
|
|
fixtures = {}
|
|
fixtures_dir = os.path.join(ZULIP_PATH, "zerver/webhooks/{integration_name}/fixtures".format(
|
|
integration_name=integration_name))
|
|
if not os.path.exists(fixtures_dir):
|
|
msg = ("The integration \"{integration_name}\" does not have fixtures.").format(
|
|
integration_name=integration_name)
|
|
return json_error(msg, status=404)
|
|
|
|
for fixture in os.listdir(fixtures_dir):
|
|
fixture_path = os.path.join(fixtures_dir, fixture)
|
|
try:
|
|
json_data = ujson.loads(open(fixture_path).read())
|
|
except ValueError:
|
|
msg = ("The integration \"{integration_name}\" has non-JSON fixtures.").format(
|
|
integration_name=integration_name)
|
|
return json_error(msg)
|
|
fixtures[fixture] = json_data
|
|
|
|
return json_success({"fixtures": fixtures})
|
|
|
|
|
|
@has_request_variables
|
|
def check_send_webhook_fixture_message(request: HttpRequest,
|
|
url: str=REQ(),
|
|
body: str=REQ(),
|
|
custom_headers: str=REQ()) -> HttpResponse:
|
|
client = Client()
|
|
realm = get_realm("zulip")
|
|
try:
|
|
headers = parse_headers(custom_headers)
|
|
except ValueError as ve:
|
|
return json_error("Custom HTTP headers are not in a valid JSON format. {}".format(ve)) # nolint
|
|
if not headers:
|
|
headers = {}
|
|
http_host = headers.pop("HTTP_HOST", realm.host)
|
|
content_type = headers.pop("HTTP_CONTENT_TYPE", "application/json")
|
|
response = client.post(url, body, content_type=content_type, HTTP_HOST=http_host, **headers)
|
|
if response.status_code == 200:
|
|
return json_success()
|
|
else:
|
|
return response
|