From 9dfa6c976f0f8c91c916a22bcd3dbea377b9d0f9 Mon Sep 17 00:00:00 2001
From: pengzhanbo <volodymyr@foxmail.com>
Date: Sat, 17 Feb 2024 00:20:48 +0800
Subject: [PATCH] feat(plugin-baidu-tongji): add `useBaiduTongji`

---
 .../src/client/composables/index.ts           | 29 +++++++++++++
 .../plugin-baidu-tongji/src/client/config.ts  | 14 ++++++
 .../plugin-baidu-tongji/src/node/plugin.ts    | 43 ++++++++++++-------
 3 files changed, 71 insertions(+), 15 deletions(-)
 create mode 100644 plugins/plugin-baidu-tongji/src/client/composables/index.ts
 create mode 100644 plugins/plugin-baidu-tongji/src/client/config.ts

diff --git a/plugins/plugin-baidu-tongji/src/client/composables/index.ts b/plugins/plugin-baidu-tongji/src/client/composables/index.ts
new file mode 100644
index 00000000..52ad390c
--- /dev/null
+++ b/plugins/plugin-baidu-tongji/src/client/composables/index.ts
@@ -0,0 +1,29 @@
+import { watch } from 'vue'
+import { usePageData } from 'vuepress/client'
+
+declare global {
+  interface Window {
+    _hmt?: [name: string, options: any][]
+  }
+}
+
+/**
+ * Add baidu analytics to the site
+ *
+ * @see https://tongji.baidu.com/
+ * @see https://tongji.baidu.com/holmes/Analytics/%E7%99%BE%E5%BA%A6%E7%BB%9F%E8%AE%A1%E4%BD%BF%E7%94%A8%E6%89%8B%E5%86%8C
+ * @see https://tongji.baidu.com/holmes/Analytics/%E6%8A%80%E6%9C%AF%E6%8E%A5%E5%85%A5%E6%8C%87%E5%8D%97/JS%20API/JS%20API%E6%8A%80%E6%9C%AF%E6%96%87%E6%A1%A3/_trackPageview
+ */
+export function useBaiduTongji(): void {
+  if (!window._hmt)
+    return
+
+  const page = usePageData()
+
+  watch(
+    () => page.value.path,
+    (newLocation) => {
+      window._hmt?.push(['_trackPageview', newLocation])
+    },
+  )
+}
diff --git a/plugins/plugin-baidu-tongji/src/client/config.ts b/plugins/plugin-baidu-tongji/src/client/config.ts
new file mode 100644
index 00000000..115bf3a5
--- /dev/null
+++ b/plugins/plugin-baidu-tongji/src/client/config.ts
@@ -0,0 +1,14 @@
+import { defineClientConfig } from 'vuepress/client'
+import type { ClientConfig } from 'vuepress/client'
+import { useBaiduTongji } from './composables/index.js'
+
+declare const __VUEPRESS_SSR__: boolean
+
+export default defineClientConfig({
+  enhance() {
+    if (__VUEPRESS_SSR__)
+      return
+
+    useBaiduTongji()
+  },
+}) as ClientConfig
diff --git a/plugins/plugin-baidu-tongji/src/node/plugin.ts b/plugins/plugin-baidu-tongji/src/node/plugin.ts
index 478830fa..a4370bfd 100644
--- a/plugins/plugin-baidu-tongji/src/node/plugin.ts
+++ b/plugins/plugin-baidu-tongji/src/node/plugin.ts
@@ -1,23 +1,36 @@
-import type { Plugin } from 'vuepress/core'
+import type { Plugin, PluginObject } from 'vuepress/core'
+import { getDirname, path } from 'vuepress/utils'
 
 export interface BaiduTongjiOptions {
   key?: string
 }
 
+const __dirname = getDirname(import.meta.url)
+
 export function baiduTongjiPlugin({ key = '' }: BaiduTongjiOptions): Plugin {
-  return {
-    name: '@vuepress-plume/plugin-baidu-tongji',
-    extendsPage: (page) => {
-      page.frontmatter.head ??= []
-      page.frontmatter.head?.push([
-        'script',
-        { type: 'text/javascript' },
-        'var _hmt = _hmt || []',
-      ])
-      page.frontmatter.head?.push([
-        'script',
-        { src: `https://hm.baidu.com/hm.js?${key}` },
-      ])
-    },
+  return (app) => {
+    const plugin: PluginObject = {
+      name: '@vuepress-plume/plugin-baidu-tongji',
+    }
+
+    if (app.env.isDev)
+      return plugin
+
+    return {
+      ...plugin,
+      clientConfigFile: path.resolve(__dirname, '../client/config.js'),
+      extendsPage: (page) => {
+        page.frontmatter.head ??= []
+        page.frontmatter.head?.push([
+          'script',
+          { type: 'text/javascript' },
+          'window._hmt = window._hmt || []',
+        ])
+        page.frontmatter.head?.push([
+          'script',
+          { src: `https://hm.baidu.com/hm.js?${key}`, async: true },
+        ])
+      },
+    }
   }
 }