From dd8ee7afa5749f3b5cc9ed2a2d29e383880efeb3 Mon Sep 17 00:00:00 2001
From: Xavi Lee <awxiaoxian2020@163.com>
Date: Wed, 24 Jan 2024 09:30:02 +0800
Subject: [PATCH] sync with
 https://github.com/vuejs/vitepress/commit/ba3c6441be38b0828c67dad35acabab0035fc816

---
 docs/zh/reference/site-config.md | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/docs/zh/reference/site-config.md b/docs/zh/reference/site-config.md
index 9b48f1ed..ef412d98 100644
--- a/docs/zh/reference/site-config.md
+++ b/docs/zh/reference/site-config.md
@@ -645,19 +645,20 @@ export default {
 }
 ```
 
-#### 示例：添加一个典型的 URL `<link>` {#example-adding-a-canonical-url-link} 
+#### 示例：添加 canonical URL `<link>` {#example-adding-a-canonical-url-link} 
 
 ```ts
 export default {
-  transformHead({ page }) {
-    // Skip the 404 page
-    if (page !== '404.md') {
-      const canonicalUrl = `https://example.com/${page}`
-        .replace(/index\.md$/, '')
-        .replace(/\.md$/, '.html')
-
-      return [['link', { rel: 'canonical', href: canonicalUrl }]]
-    }
+  transformPageData(pageData) {
+    const canonicalUrl = `https://example.com/${pageData.relativePath}`
+      .replace(/index\.md$/, '')
+      .replace(/\.md$/, '.html')
+
+    pageData.frontmatter.head ??= []
+    pageData.frontmatter.head.push([
+      'link',
+      { rel: 'canonical', href: canonicalUrl }
+    ])
   }
 }
 ```