chore: 兼容cnb访问

Author: liweijie0812

packages/cherry-markdown/index.html

@@ -77,7 +77,10 @@
 
       let CONFIG ={};
 
-      switch (window.location.pathname) {
+      // 使用注入的原始路径或当前路径
+      const currentPath = window.__ORIGINAL_PATH__ || window.location.pathname;
+
+      switch (currentPath) {
         case '/':
           CONFIG = Object.assign({}, basicConfig, { value: indexMd, ...devCompatibleConfig });
           break;

packages/cherry-markdown/vite.config.ts

@@ -40,6 +40,28 @@ function printLinks() {
   };
 }
 
+function spaFallback() {
+  return {
+    name: 'spa-fallback',
+    transformIndexHtml: (html: string, ctx: { path: string; filename?: string; server?: any }) => {
+      const path = ctx.path;
+      // 如果是配置的路径之一，注入脚本设置原始路径
+      // 优化的路径匹配逻辑
+      const isExactMatch = paths.includes(path);
+      const isPathWithoutHtmlMatch = path.endsWith('.html') && paths.includes(path.slice(0, -5));
+      const matchedPath = isExactMatch || isPathWithoutHtmlMatch ? (isExactMatch ? path : path.slice(0, -5)) : null;
+
+      if (matchedPath && matchedPath !== '/index.html') {
+        // 转义路径中的特殊字符以防止XSS注入
+        const escapedPath = matchedPath.replace(/['"\\]/g, '\\$&');
+        const script = `<script>window.__ORIGINAL_PATH__ = '${escapedPath}';</script>`;
+        return html.replace('</head>', `${script}\n  </head>`);
+      }
+      return html;
+    },
+  };
+}
+
 export default defineConfig({
   root: process.cwd(),
   base: '/',
@@ -69,5 +91,5 @@ export default defineConfig({
     BUILD_ENV: JSON.stringify(process.env.NODE_ENV || 'development'),
     __EXAMPLES_PATH__: JSON.stringify(path.resolve(__dirname, '../../examples').replace(/\\/g, '/')),
   },
-  plugins: [printLinks()],
+  plugins: [spaFallback(), printLinks()],
 });