Skip to content

Commit 0661b98

Browse files
authored
Add support for userAgent in swagger (#2377)
Remove SessionAuthentication from AUTHENTICATION_WHITELIST for swagger Signed-off-by: ziad hany <ziadhany2016@gmail.com>
1 parent a7e9ce8 commit 0661b98

2 files changed

Lines changed: 19 additions & 3 deletions

File tree

‎vulnerabilities/templates/drf_spectacular/swagger_ui.html‎

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,24 @@
2222
<div id="swagger-ui"></div>
2323
<script src="{{ dist }}/swagger-ui-bundle.js"></script>
2424
<script src="{{ dist }}/swagger-ui-standalone-preset.js"></script>
25+
2526
{% if script_url %}
2627
<script src="{{ script_url }}"></script>
2728
{% else %}
2829
<script>
29-
{% include template_name_js %}
30+
const swaggerSettings = {{ settings|safe }};
31+
32+
swaggerSettings.url = "{{ schema_url }}";
33+
swaggerSettings.dom_id = "#swagger-ui";
34+
35+
swaggerSettings.requestInterceptor = (request) => {
36+
const userAgent = swaggerSettings.userAgent || "VCIO_API_AGENT";
37+
request.headers['User-Agent'] = userAgent;
38+
return request;
39+
};
40+
41+
const ui = SwaggerUIBundle(swaggerSettings);
42+
window.ui = ui;
3043
</script>
3144
{% endif %}
32-
{% endblock %}
33-
45+
{% endblock %}

‎vulnerablecode/settings.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,11 @@
329329
"defaultModelsExpandDepth": 1,
330330
"displayRequestDuration": True,
331331
"docExpansion": "list",
332+
"userAgent": VCIO_USER_AGENT,
332333
},
334+
"AUTHENTICATION_WHITELIST": [
335+
"rest_framework.authentication.TokenAuthentication",
336+
],
333337
"SORT_OPERATIONS": False,
334338
"TAGS_SORTER": False,
335339
}

0 commit comments

Comments
 (0)