Fix redirect location to handle / (#2204)

b/236404667
youtube · Jan 12, 2024 · 9028632 · 9028632
1 parent e9cde16
commit 9028632
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/cobalt/site/deploy/main.py b/cobalt/site/deploy/main.py
@@ -25,11 +25,14 @@ def app(environ, start_response):
 
   Changing the file or function name requires configuring a matching entrypoint.
   """
-  base_url = 'https://developers.google.com'
-  path_prefix = '/youtube/cobalt/docs'
+  base_url = 'https://developers.google.com/youtube/cobalt'
+  doc_prefix = '/docs'
   # Remove extension from filename if present.
   doc_path = os.path.splitext(environ['RAW_URI'])[0]
+  if not doc_path or doc_path == '/':
+    location = base_url
+  else:
+    location = f'{base_url}{doc_prefix}{doc_path}'
 
-  start_response('301 Moved Permanently',
-                 [('Location', f'{base_url}{path_prefix}{doc_path}')])
+  start_response('301 Moved Permanently', [('Location', location)])
   return []