chore: adjust

Author: Saul-Mirone

Diff for: packages/components/src/image-inline/components/image-inline.tsx

@@ -14,7 +14,7 @@ type Attrs = {
 export type MilkdownImageInlineProps = {
   selected: Ref<boolean>
   readonly: Ref<boolean>
-  setLink: (link: string) => void
+  setAttr: <T extends keyof Attrs>(attr: T, value: Attrs[T]) => void
   config: InlineImageConfig
 } & {
   [P in keyof Attrs]: Ref<Attrs[P] | undefined>
@@ -42,7 +42,7 @@ export const MilkdownImageInline = defineComponent<MilkdownImageInlineProps>({
       type: Object,
       required: true,
     },
-    setLink: {
+    setAttr: {
       type: Function,
       required: true,
     },
@@ -60,7 +60,7 @@ export const MilkdownImageInline = defineComponent<MilkdownImageInlineProps>({
             src={props.src}
             selected={props.selected}
             readonly={props.readonly}
-            setLink={props.setLink}
+            setLink={(link) => props.setAttr('src', link)}
             imageIcon={props.config.imageIcon()}
             uploadButton={props.config.uploadButton()}
             confirmButton={props.config.confirmButton()}

Diff for: packages/components/src/image-inline/view.ts

@@ -18,11 +18,16 @@ export const inlineImageView = $view(
       const title = ref(initialNode.attrs.title)
       const selected = ref(false)
       const readonly = ref(!view.editable)
-      const setLink = (link: string) => {
+      const setAttr = (attr: string, value: unknown) => {
         const pos = getPos()
         if (pos == null) return
-        const url = DOMPurify.sanitize(link)
-        view.dispatch(view.state.tr.setNodeAttribute(pos, 'src', url))
+        view.dispatch(
+          view.state.tr.setNodeAttribute(
+            pos,
+            attr,
+            attr === 'src' ? DOMPurify.sanitize(value as string) : value
+          )
+        )
       }
 
       const config = ctx.get(inlineImageConfig.key)
@@ -32,7 +37,7 @@ export const inlineImageView = $view(
         title,
         selected,
         readonly,
-        setLink,
+        setAttr,
         config,
       })
       const dom = document.createElement('span')