fix(website): scope docs search by locale and channel

5 files changed, 114 insertions, 14 deletions

diff --git a/website/src/components/Head.astro b/website/src/components/Head.astro
new file mode 100644
index 00000000..b1214cb1
--- /dev/null
+++ b/website/src/components/Head.astro
@@ -0,0 +1,9 @@
+---
+import { docsChannel } from '../docs-path';
+
+const { head } = Astro.locals.starlightRoute;
+const channel = docsChannel(Astro.url.pathname);
+---
+
+{head.map(({ tag: Tag, attrs, content }) => <Tag {...attrs} set:html={content} />)}
+<meta data-pagefind-filter="channel[content]" content={channel} />
diff --git a/website/src/components/Search.astro b/website/src/components/Search.astro
new file mode 100644
index 00000000..2a31ea24
--- /dev/null
+++ b/website/src/components/Search.astro
@@ -0,0 +1,52 @@
+---
+import StarlightSearch from '@astrojs/starlight/components/Search.astro';
+import { docsChannel } from '../docs-path';
+
+const channel = docsChannel(Astro.url.pathname);
+---
+
+<channel-search data-channel={channel}>
+  <StarlightSearch />
+</channel-search>
+
+<script>
+  class ChannelSearch extends HTMLElement {
+    observer?: MutationObserver;
+
+    connectedCallback() {
+      const channel = this.dataset.channel;
+      if (!channel) return;
+
+      const selectChannel = () => {
+        const checkbox = this.querySelector<HTMLInputElement>(
+          `.pagefind-ui__filter-checkbox[name="channel"][value="${channel}"]`,
+        );
+        if (!checkbox) return false;
+        if (!checkbox.checked) checkbox.click();
+        this.observer?.disconnect();
+        return true;
+      };
+
+      if (selectChannel()) return;
+      // Starlight does not expose its PagefindUI instance, so wait for Pagefind's filter control.
+      this.observer = new MutationObserver(selectChannel);
+      this.observer.observe(this, { childList: true, subtree: true });
+    }
+
+    disconnectedCallback() {
+      this.observer?.disconnect();
+    }
+  }
+
+  customElements.define('channel-search', ChannelSearch);
+</script>
+
+<style>
+  channel-search {
+    display: contents;
+  }
+
+  channel-search :global(.pagefind-ui__filter-panel) {
+    display: none;
+  }
+</style>
diff --git a/website/src/content.config.ts b/website/src/content.config.ts
index e84e3781..f4993be8 100644
--- a/website/src/content.config.ts
+++ b/website/src/content.config.ts
@@ -2,20 +2,7 @@ import { defineCollection, z } from 'astro:content';
 import { glob } from 'astro/loaders';
 import { docsLoader } from '@astrojs/starlight/loaders';
 import { docsSchema } from '@astrojs/starlight/schema';
-
-const docsLocales = ['ja', 'zh-cn'];
-
-function docsPath({ entry }: { entry: string }) {
-  const slug = entry.replace(/\.(md|mdx|markdown|mdown|mkdn|mkd|mdwn)$/i, '');
-  const normalized = slug.replace(/\/index$/, '');
-  for (const locale of docsLocales) {
-    if (normalized === locale) return `${locale}/docs`;
-    if (normalized.startsWith(`${locale}/`)) {
-      return `${locale}/docs/${normalized.slice(locale.length + 1)}`;
-    }
-  }
-  return normalized === 'index' ? 'docs' : `docs/${normalized}`;
-}
+import { docsPath } from './docs-path';
 
 export const collections = {
   docs: defineCollection({ loader: docsLoader({ generateId: docsPath }), schema: docsSchema() }),
diff --git a/website/src/docs-path.test.ts b/website/src/docs-path.test.ts
new file mode 100644
index 00000000..26f9a701
--- /dev/null
+++ b/website/src/docs-path.test.ts
@@ -0,0 +1,32 @@
+import { describe, expect, test } from 'bun:test';
+import { docsChannel, docsPath } from './docs-path';
+
+describe('docsChannel', () => {
+  test.each([
+    ['/docs/', 'stable'],
+    ['/ja/docs/install/', 'stable'],
+    ['/zh-cn/docs/', 'stable'],
+    ['/docs/preview/', 'preview'],
+    ['/ja/docs/preview/install/', 'preview'],
+    ['/zh-cn/docs/preview/', 'preview'],
+  ])('maps %s to %s', (pathname, expected) => {
+    expect(docsChannel(pathname)).toBe(expected);
+  });
+});
+
+describe('docsPath', () => {
+  test.each([
+    ['index.mdx', 'docs'],
+    ['install.mdx', 'docs/install'],
+    ['ja/index.mdx', 'ja/docs'],
+    ['ja/install.mdx', 'ja/docs/install'],
+    ['zh-cn/install.mdx', 'zh-cn/docs/install'],
+    ['preview/index.mdx', 'docs/preview'],
+    ['preview/install.mdx', 'docs/preview/install'],
+    ['preview/ja/index.mdx', 'ja/docs/preview'],
+    ['preview/ja/install.mdx', 'ja/docs/preview/install'],
+    ['preview/zh-cn/install.mdx', 'zh-cn/docs/preview/install'],
+  ])('maps %s to %s', (entry, expected) => {
+    expect(docsPath({ entry })).toBe(expected);
+  });
+});
diff --git a/website/src/docs-path.ts b/website/src/docs-path.ts
new file mode 100644
index 00000000..be022a4b
--- /dev/null
+++ b/website/src/docs-path.ts
@@ -0,0 +1,20 @@
+const docsLocales = new Set(['ja', 'zh-cn']);
+
+export function docsChannel(pathname: string) {
+  return /^\/(?:ja\/|zh-cn\/)?docs\/preview(?:\/|$)/.test(pathname) ? 'preview' : 'stable';
+}
+
+export function docsPath({ entry }: { entry: string }) {
+  const slug = entry.replace(/\.(md|mdx|markdown|mdown|mkdn|mkd|mdwn)$/i, '');
+  const normalized = slug.replace(/\/index$/, '');
+  const segments = normalized.split('/');
+  const localeIndex = segments[0] === 'preview' ? 1 : 0;
+  const locale = segments[localeIndex];
+
+  if (locale && docsLocales.has(locale)) {
+    segments.splice(localeIndex, 1);
+    return `${locale}/docs${segments.length > 0 ? `/${segments.join('/')}` : ''}`;
+  }
+
+  return normalized === 'index' ? 'docs' : `docs/${normalized}`;
+}