refac: get_discovery_urls

This commit is contained in:
Timothy Jaeryang Baek 2025-09-26 14:34:26 -05:00
parent f8a3ed2d18
commit 0431ad9cc4

View file

@ -198,13 +198,25 @@ def get_parsed_and_base_url(server_url) -> tuple[urllib.parse.ParseResult, str]:
def get_discovery_urls(server_url) -> list[str]: def get_discovery_urls(server_url) -> list[str]:
urls = []
parsed, base_url = get_parsed_and_base_url(server_url) parsed, base_url = get_parsed_and_base_url(server_url)
urls.append( urls = [
urllib.parse.urljoin(base_url, "/.well-known/oauth-authorization-server") urllib.parse.urljoin(base_url, "/.well-known/oauth-authorization-server"),
) urllib.parse.urljoin(base_url, "/.well-known/openid-configuration"),
urls.append(urllib.parse.urljoin(base_url, "/.well-known/openid-configuration")) ]
if parsed.path and parsed.path != "/":
urls.append(
urllib.parse.urljoin(
base_url,
f"/.well-known/oauth-authorization-server{parsed.path.rstrip('/')}",
)
)
urls.append(
urllib.parse.urljoin(
base_url, f"/.well-known/openid-configuration{parsed.path.rstrip('/')}"
)
)
return urls return urls