ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-13 21:05:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: PWA share_target implementation

Browse source 
Co-Authored-By: gjveld <19951982+gjveld@users.noreply.github.com>
This commit is contained in:
Timothy Jaeryang Baek 2025-10-01 19:57:19 -05:00
parent 6f88db1be2
commit 6c8c3257fd
1 changed files with 29 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
backend/open_webui/main.py
View file

@ -8,6 +8,7 @@ import shutil
import sys import sys
import time import time
import random import random
import re
from uuid import uuid4 from uuid import uuid4
@ -1170,12 +1171,32 @@ class RedirectMiddleware(BaseHTTPMiddleware):
path = request.url.path path = request.url.path
query_params = dict(parse_qs(urlparse(str(request.url)).query)) query_params = dict(parse_qs(urlparse(str(request.url)).query))
redirect_params = {}
# Check for the specific watch path and the presence of 'v' parameter # Check for the specific watch path and the presence of 'v' parameter
if path.endswith("/watch") and "v" in query_params: if path.endswith("/watch") and "v" in query_params:
# Extract the first 'v' parameter # Extract the first 'v' parameter
video_id = query_params["v"][0] youtube_video_id = query_params["v"][0]
encoded_video_id = urlencode({"youtube": video_id}) redirect_params["youtube"] = youtube_video_id
redirect_url = f"/?{encoded_video_id}"
if "shared" in query_params and len(query_params["shared"]) > 0:
# PWA share_target support
text = query_params["shared"][0]
if text:
urls = re.match(r"https://\S+", text)
if urls:
from open_webui.retrieval.loaders.youtube import _parse_video_id
if youtube_video_id := _parse_video_id(urls[0]):
redirect_params["youtube"] = youtube_video_id
else:
redirect_params["load-url"] = urls[0]
else:
redirect_params["q"] = text
if redirect_params:
redirect_url = f"/?{urlencode(redirect_params)}"
return RedirectResponse(url=redirect_url) return RedirectResponse(url=redirect_url)
# Proceed with the normal flow of other requests # Proceed with the normal flow of other requests
@ -2004,6 +2025,11 @@ async def get_manifest_json():
"purpose": "maskable", "purpose": "maskable",
}, },
], ],
"share_target": {
"action": "/",
"method": "GET",
"params": {"text": "shared"},
},
} }