chore: f

Milkdown · Dec 15, 2024 · cf7d4f1 · cf7d4f1
1 parent 3f3e8c5
commit cf7d4f1
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 8 deletions.
diff --git a/packages/components/src/image-block/config.ts b/packages/components/src/image-block/config.ts
@@ -10,7 +10,7 @@ export interface ImageBlockConfig {
   uploadPlaceholderText: string
   captionPlaceholderText: string
   onUpload: (file: File) => Promise<string>
-  proxyDomURL?: (url: string) => Promise<string>
+  proxyDomURL?: (url: string) => Promise<string> | string
 }
 
 export const defaultImageBlockConfig: ImageBlockConfig = {

diff --git a/packages/components/src/image-block/view/index.ts b/packages/components/src/image-block/view/index.ts
@@ -22,9 +22,14 @@ export const imageBlockView = $view(
         if (!proxyDomURL) {
           dom.src = node.attrs.src
         } else {
-          proxyDomURL(node.attrs.src).then((url) => {
-            dom.src = url
-          })
+          const proxiedURL = proxyDomURL(node.attrs.src)
+          if (typeof proxiedURL === 'string') {
+            dom.src = proxiedURL
+          } else {
+            proxiedURL.then((url) => {
+              dom.src = url
+            })
+          }
         }
         dom.ratio = node.attrs.ratio
         dom.caption = node.attrs.caption

diff --git a/packages/components/src/image-inline/config.ts b/packages/components/src/image-inline/config.ts
@@ -8,7 +8,7 @@ export interface InlineImageConfig {
   confirmButton: () => ReturnType<typeof html>
   uploadPlaceholderText: string
   onUpload: (file: File) => Promise<string>
-  proxyDomURL?: (url: string) => Promise<string>
+  proxyDomURL?: (url: string) => Promise<string> | string
 }
 
 export const defaultInlineImageConfig: InlineImageConfig = {

diff --git a/packages/components/src/image-inline/view.ts b/packages/components/src/image-inline/view.ts
@@ -22,9 +22,14 @@ export const inlineImageView = $view(
         if (!proxyDomURL) {
           dom.src = node.attrs.src
         } else {
-          proxyDomURL(node.attrs.src).then((url) => {
-            dom.src = url
-          })
+          const proxiedURL = proxyDomURL(node.attrs.src)
+          if (typeof proxiedURL === 'string') {
+            dom.src = proxiedURL
+          } else {
+            proxiedURL.then((url) => {
+              dom.src = url
+            })
+          }
         }
         dom.alt = node.attrs.alt
         dom.title = node.attrs.title