feat: add image proxy for crepe

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

Hide file tree

Showing 4 changed files with 18 additions and 2 deletions.
diff --git a/packages/components/src/image-block/config.ts b/packages/components/src/image-block/config.ts
@@ -10,6 +10,7 @@ export interface ImageBlockConfig {
   uploadPlaceholderText: string
   captionPlaceholderText: string
   onUpload: (file: File) => Promise<string>
+  proxyDomURL?: (url: string) => Promise<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
@@ -17,8 +17,15 @@ export const imageBlockView = $view(
         'milkdown-image-block'
       ) as HTMLElement & ImageComponentProps
       const config = ctx.get(imageBlockConfig.key)
+      const proxyDomURL = config.proxyDomURL
       const bindAttrs = (node: Node) => {
-        dom.src = node.attrs.src
+        if (!proxyDomURL) {
+          dom.src = node.attrs.src
+        } else {
+          proxyDomURL(node.attrs.src).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,6 +8,7 @@ export interface InlineImageConfig {
   confirmButton: () => ReturnType<typeof html>
   uploadPlaceholderText: string
   onUpload: (file: File) => Promise<string>
+  proxyDomURL?: (url: string) => Promise<string>
 }
 
 export const defaultInlineImageConfig: InlineImageConfig = {

diff --git a/packages/components/src/image-inline/view.ts b/packages/components/src/image-inline/view.ts
@@ -17,8 +17,15 @@ export const inlineImageView = $view(
         'milkdown-image-inline'
       ) as HTMLElement & InlineImageComponentProps
       const config = ctx.get(inlineImageConfig.key)
+      const proxyDomURL = config.proxyDomURL
       const bindAttrs = (node: Node) => {
-        dom.src = node.attrs.src
+        if (!proxyDomURL) {
+          dom.src = node.attrs.src
+        } else {
+          proxyDomURL(node.attrs.src).then((url) => {
+            dom.src = url
+          })
+        }
         dom.alt = node.attrs.alt
         dom.title = node.attrs.title
       }