From 3a4e5b205a42a34dba07414b2f66fab4eec19aa1 Mon Sep 17 00:00:00 2001
From: Kent Dong <ch3cho@qq.com>
Date: Tue, 8 Oct 2024 13:41:18 +0800
Subject: [PATCH] feat: Improve search function (#324)

---
 .../common/Header/searchComp/index.astro      | 44 +++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/src/components/common/Header/searchComp/index.astro b/src/components/common/Header/searchComp/index.astro
index 4a6619e07b..ae050db565 100644
--- a/src/components/common/Header/searchComp/index.astro
+++ b/src/components/common/Header/searchComp/index.astro
@@ -114,6 +114,19 @@ const docsearchTranslations: DocSearchTranslationProps = {
 
 <script>
 	import { ALGOLIA } from "@/constant";
+
+	const getRootUrl = (url: string) => {
+		if (!url) {
+			return url;
+		}
+		const schemeIndex = url.indexOf("://");
+		const firstSlashIndex = url.indexOf(
+			"/",
+			schemeIndex === -1 ? 0 : schemeIndex + "://".length,
+		);
+		return firstSlashIndex === -1 ? url : url.substring(0, firstSlashIndex);
+	};
+
 	let url = "";
 	class StarlightDocSearch extends HTMLElement {
 		constructor() {
@@ -132,6 +145,37 @@ const docsearchTranslations: DocSearchTranslationProps = {
 			const options: Parameters<typeof docsearch>[0] = {
 				...ALGOLIA,
 				container: "sl-doc-search",
+				transformItems(items) {
+					if (!items || !items.length) {
+						return items;
+					}
+					const filteredItems = [];
+					const pageRootUrl = getRootUrl(url);
+					const isPageEnglish = url
+						.substring(pageRootUrl.length)
+						.startsWith("/en");
+					for (const item of items) {
+						const itemUrl = item.url;
+						if (!itemUrl) {
+							continue;
+						}
+						const itemRootUrl = getRootUrl(itemUrl);
+						const itemPath = itemUrl.substring(itemRootUrl.length);
+						if (itemPath.startsWith("/en") !== isPageEnglish) {
+							continue;
+						}
+						if (itemRootUrl === pageRootUrl) {
+							filteredItems.push(item);
+						} else {
+							filteredItems.push(
+								Object.assign({}, item, {
+									url: pageRootUrl + itemPath,
+								}),
+							);
+						}
+					}
+					return filteredItems;
+				},
 			};
 			try {
 				const translations = JSON.parse(