Replace domains in content during compilation

Author: antalike

app/composables/useVariableReplacement.ts

@@ -4,9 +4,23 @@ export const useVariableReplacement = () => {
   const replaceVariables = (text: string) => {
     if (!text) return text
 
-    return text.replace(/{{(\w+)}}/g, (match, key) => {
-      return (config.public as unknown as Record<string, string>)[key] || match
+    let newText = text
+
+    const domains: Record<string, string> = {
+      'https://cdn.memtensor.com.cn': config.public.cdnUrl as string,
+      'https://statics.memtensor.com.cn': config.public.staticCdnUrl as string,
+      'https://memos-playground.openmem.net': config.public.playgroundUrl as string,
+      'https://memos-dashboard.openmem.net': config.public.dashboardUrl as string,
+      'https://memos.openmem.net': config.public.homeDomain as string
+    }
+
+    Object.keys(domains).forEach((key) => {
+      if (domains[key]) {
+        newText = newText.split(key).join(domains[key])
+      }
     })
+
+    return newText
   }
 
   return {

envConfig/config.intl.ts

@@ -1,7 +1,7 @@
 export default {
   env: 'intl',
-  homeDomain: 'https://memos.memtensor.net/',
-  openMemUrl: 'https://openmem.net/',
+  homeDomain: 'https://memos.memtensor.net',
+  openMemUrl: 'https://openmem.net',
   githubMemosUrl: 'https://github.com/MemTensor/MemOS',
   dashboardUrl: 'https://memos-dashboard.memtensor.net',
   playgroundUrl: 'https://memos-playground.memtensor.net',

envConfig/config.prod.ts

@@ -1,7 +1,7 @@
 export default {
   env: 'prod',
   homeDomain: 'https://memos.openmem.net',
-  openMemUrl: 'https://openmem.net/',
+  openMemUrl: 'https://openmem.net',
   githubMemosUrl: 'https://github.com/MemTensor/MemOS',
   dashboardUrl: 'https://memos-dashboard.openmem.net',
   playgroundUrl: 'https://memos-playground.openmem.net'

nuxt.config.ts

@@ -2,7 +2,7 @@ import yaml from '@rollup/plugin-yaml'
 import type { NuxtConfig } from '@nuxt/schema'
 import pkg from './package.json'
 import { getCnRoutes, getCnApiReferenceRoutes } from './scripts/extract-routes.mjs'
-import remarkVariables from './scripts/remark-variables.mjs'
+import remarkReplaceDomains from './scripts/remark-replace-domains.mjs'
 
 const cnRoutes = getCnRoutes()
 const cnApiRoutes = getCnApiReferenceRoutes()
@@ -18,9 +18,6 @@ const envConfig = await import(`./envConfig/config.${env}.ts`).then(m => m.defau
 
 const staticCdnUrl = envConfig.staticCdnUrl || 'https://statics.memtensor.com.cn'
 const cdnUrl = envConfig.cdnUrl || 'https://cdn.memtensor.com.cn'
-const dashboardUrl = envConfig.dashboardUrl || 'https://memos-dashboard.openmem.net'
-const playgroundUrl = envConfig.playgroundUrl || 'https://memos-playground.openmem.net'
-const landingUrl = envConfig.homeDomain || 'https://memos.openmem.net'
 
 const config: NuxtConfig = {
   app: {
@@ -46,10 +43,7 @@ const config: NuxtConfig = {
       version: pkg.version,
       apiBase: 'https://apigw.memtensor.cn',
       staticCdnUrl,
-      cdnUrl,
-      dashboardUrl,
-      playgroundUrl,
-      landingUrl
+      cdnUrl
     }
   },
 
@@ -87,7 +81,7 @@ const config: NuxtConfig = {
     },
     build: {
       rollupOptions: {
-        external: ['remark-variables']
+        external: ['remark-replace-domains']
       }
     }
   },
@@ -105,15 +99,17 @@ const config: NuxtConfig = {
     build: {
       markdown: {
         remarkPlugins: {
-          'remark-variables': {
-            instance: remarkVariables,
+          'remark-replace-domains': {
+            instance: remarkReplaceDomains,
             options: {
-              variables: {
-                staticCdnUrl,
-                cdnUrl,
-                dashboardUrl,
-                playgroundUrl,
-                landingUrl
+              imageDomains: {
+                'https://cdn.memtensor.com.cn': cdnUrl,
+                'https://statics.memtensor.com.cn': staticCdnUrl
+              },
+              linkDomains: {
+                'https://memos-playground.openmem.net': envConfig.playgroundUrl,
+                'https://memos-dashboard.openmem.net': envConfig.dashboardUrl,
+                'https://memos.openmem.net': envConfig.homeDomain
               }
             }
           }

scripts/remark-replace-domains.mjs

@@ -0,0 +1,55 @@
+export default function remarkReplaceDomains(options = {}) {
+  const imageDomains = options.imageDomains || {}
+  const linkDomains = options.linkDomains || {}
+
+  function escapeRegExp(string) {
+    return string.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')
+  }
+
+  return (tree) => {
+    function visit(node) {
+      // Handle standard markdown images: ![alt](url)
+      if (node.type === 'image' && node.url && typeof node.url === 'string') {
+        Object.keys(imageDomains).forEach((key) => {
+          if (node.url.startsWith(key)) {
+            node.url = node.url.replace(key, imageDomains[key])
+          }
+        })
+      }
+
+      // Handle standard markdown links: [text](url)
+      if (node.type === 'link' && node.url && typeof node.url === 'string') {
+        Object.keys(linkDomains).forEach((key) => {
+          if (node.url.startsWith(key)) {
+            node.url = node.url.replace(key, linkDomains[key])
+          }
+        })
+      }
+
+      // Handle HTML nodes for both images and links
+      if (node.type === 'html' && node.value && typeof node.value === 'string') {
+        // Replace image sources
+        Object.keys(imageDomains).forEach((key) => {
+          const regex = new RegExp(`(<img\\s+(?:[^>]*?\\s+)?src=["'])(${escapeRegExp(key)})([^"']*["'][^>]*>)`, 'gi')
+          node.value = node.value.replace(regex, (match, p1, p2, p3) => {
+            return `${p1}${imageDomains[key]}${p3}`
+          })
+        })
+
+        // Replace link hrefs
+        Object.keys(linkDomains).forEach((key) => {
+          const regex = new RegExp(`(<a\\s+(?:[^>]*?\\s+)?href=["'])(${escapeRegExp(key)})([^"']*["'][^>]*>)`, 'gi')
+          node.value = node.value.replace(regex, (match, p1, p2, p3) => {
+            return `${p1}${linkDomains[key]}${p3}`
+          })
+        })
+      }
+
+      if (node.children && Array.isArray(node.children)) {
+        node.children.forEach(visit)
+      }
+    }
+
+    visit(tree)
+  }
+}