fix: add app webUrl for self link parser

Signed-off-by: Innei <[email protected]>

Author: Innei

src/atoms/url.ts

@@ -1,3 +1,4 @@
+import { queryClient } from '~/providers/root/react-query-provider'
 import { atom, useAtomValue } from 'jotai'
 
 import { getToken } from '~/lib/cookie'
@@ -12,15 +13,23 @@ export interface UrlConfig {
 }
 
 const adminUrlAtom = atom<string | null>(null)
+const webUrlAtom = atom<string | null>(null)
 
 export const fetchAppUrl = async () => {
-  const { data } = await apiClient.proxy.options.url.get<{
-    data: UrlConfig
-  }>()
-
-  jotaiStore.set(adminUrlAtom, data.adminUrl)
+  const { data } = await queryClient.fetchQuery({
+    queryKey: ['app.url'],
+    queryFn: () =>
+      apiClient.proxy.options.url.get<{
+        data: UrlConfig
+      }>(),
+  })
+
+  if (data.adminUrl) jotaiStore.set(adminUrlAtom, data.adminUrl)
+  jotaiStore.set(webUrlAtom, data.webUrl)
 }
 
+export const getWebUrl = () => jotaiStore.get(webUrlAtom)
+export const setWebUrl = (url: string) => jotaiStore.set(webUrlAtom, url)
 export const getAdminUrl = () => jotaiStore.get(adminUrlAtom)
 export const useAppUrl = () => {
   const url = useAggregationSelector((a) => a.url)

src/lib/link-parser.ts

@@ -1,3 +1,5 @@
+import { getWebUrl } from '~/atoms'
+
 import { isClientSide, isDev } from './env'
 
 export const getTweetId = (url: URL) => {
@@ -74,9 +76,13 @@ export const isBilibiliUrl = (url: URL) => {
 
 export const isSelfArticleUrl = (url: URL) => {
   if (!isClientSide) return false
+
+  const webUrl = getWebUrl()
+  const webHost = webUrl ? new URL(webUrl).hostname : ''
+
   if (isDev && url.hostname === 'innei.in') return true
   return (
-    url.hostname === location.hostname &&
+    (url.hostname === location.hostname || webHost === location.hostname) &&
     ['/posts/', '/notes/'].some((path) => url.pathname.startsWith(path))
   )
 }

src/providers/root/aggregation-data-provider.tsx

@@ -7,7 +7,7 @@ import type { AggregateRoot } from '@mx-space/api-client'
 import type { AppConfig } from '~/app/config'
 import type { FC, PropsWithChildren } from 'react'
 
-import { fetchAppUrl } from '~/atoms'
+import { fetchAppUrl, setWebUrl } from '~/atoms'
 import { login } from '~/atoms/owner'
 import { useBeforeMounted } from '~/hooks/common/use-before-mounted'
 import { jotaiStore } from '~/lib/store'
@@ -24,6 +24,7 @@ export const AggregationProvider: FC<
   useBeforeMounted(() => {
     if (!aggregationData) return
     jotaiStore.set(aggregationDataAtom, aggregationData)
+    setWebUrl(aggregationData.url.webUrl)
   })
   useBeforeMounted(() => {
     if (!appConfig) return
@@ -38,9 +39,11 @@ export const AggregationProvider: FC<
   useEffect(() => {
     if (!aggregationData) return
     jotaiStore.set(aggregationDataAtom, aggregationData)
+    setWebUrl(aggregationData.url.webUrl)
   }, [aggregationData])
 
   const callOnceRef = useRef(false)
+
   useEffect(() => {
     if (callOnceRef.current) return
     if (!aggregationData?.user) return