highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-01 00:18:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

cors update (#2686)

Browse Source
This commit is contained in:
pablodanswer 2024-10-05 16:08:28 -07:00 committed by GitHub
parent 28e65669b4
commit e56fd43ba6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
backend/shared_configs/configs.py
View File

@ -1,4 +1,5 @@
import os
from typing import List
from urllib.parse import urlparse
# Used for logging
@ -76,16 +77,32 @@ PRESERVED_SEARCH_FIELDS = [
]
# CORS
def validate_cors_origin(origin: str) -> None:
parsed = urlparse(origin)
if parsed.scheme not in ["http", "https"] or not parsed.netloc:
raise ValueError(f"Invalid CORS origin: '{origin}'")
CORS_ALLOWED_ORIGIN = os.environ.get("CORS_ALLOWED_ORIGIN", "*").split(",") or ["*"]
# Examples of valid values for the environment variable:
# - "" (allow all origins)
# - "http://example.com" (single origin)
# - "http://example.com,https://example.org" (multiple origins)
# - "*" (allow all origins)
CORS_ALLOWED_ORIGIN_ENV = os.environ.get("CORS_ALLOWED_ORIGIN", "")
# Validate non-wildcard origins
for origin in CORS_ALLOWED_ORIGIN:
if origin != "*" and (stripped_origin := origin.strip()):
validate_cors_origin(stripped_origin)
# Explicitly declare the type of CORS_ALLOWED_ORIGIN
CORS_ALLOWED_ORIGIN: List[str]
if CORS_ALLOWED_ORIGIN_ENV:
# Split the environment variable into a list of origins
CORS_ALLOWED_ORIGIN = [
origin.strip()
for origin in CORS_ALLOWED_ORIGIN_ENV.split(",")
if origin.strip()
]
# Validate each origin in the list
for origin in CORS_ALLOWED_ORIGIN:
validate_cors_origin(origin)
else:
# If the environment variable is empty, allow all origins
CORS_ALLOWED_ORIGIN = ["*"]