47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
import urllib.parse
|
|
|
|
|
|
def parse_url(url: str, absolute: bool =False):
|
|
"""Return URL parts from this URL.
|
|
|
|
This uses urllib.parse.urlparse to not reinvent the wheel, with a few
|
|
adjustments.
|
|
|
|
First, urllib does not know the Gemini scheme (yet!) so if it
|
|
is specified we strip it to get an absolute netloc.
|
|
|
|
Second, as this function can be used to process arbitrary user input, we
|
|
clean it a bit:
|
|
- strip whitespaces from the URL
|
|
- if "absolute" is True, consider that the URL is meant to be absolute, even
|
|
though it technically is not, e.g. "dece.space" is not absolute as it
|
|
misses either the // delimiter.
|
|
"""
|
|
if url.startswith("gemini://"):
|
|
url = url[7:]
|
|
parts = urllib.parse.urlparse(url, scheme="gemini")
|
|
if not parts.netloc or absolute:
|
|
parts = urllib.parse.urlparse(f"//{url}", scheme="gemini")
|
|
return parts
|
|
|
|
|
|
def sanitize_url(url: str):
|
|
"""Parse and unparse an URL to ensure it has been properly formatted."""
|
|
return urllib.parse.urlunparse(parse_url(url))
|
|
|
|
|
|
def join_url(base_url: str, url: str):
|
|
"""Join a base URL with a relative url."""
|
|
if base_url.startswith("gemini://"):
|
|
base_url = base_url[7:]
|
|
parts = parse_url(urllib.parse.urljoin(base_url, url))
|
|
return urllib.parse.urlunparse(parts)
|
|
|
|
|
|
def set_parameter(url: str, user_input: str):
|
|
"""Return a new URL with the user input escaped (RFC 3986) appended."""
|
|
quoted_input = urllib.parse.quote(user_input)
|
|
if "?" in url:
|
|
url = url.rsplit("?", maxsplit=1)[0]
|
|
return url + "?" + quoted_input
|