From 86085a3e5bf29afceb1f49d5c5bfeedf4f80cc62 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=BD=A0=E6=89=8B=E6=8D=A7=E9=B2=9C=E8=8A=B1?=
 <157215725@qq.com>
Date: Tue, 21 Mar 2023 23:24:33 +0800
Subject: [PATCH] fix: svg+xml image decoding in Safari and Firefox (close #15)

---
 index.html                                |  9 ++---
 src/context.ts                            |  6 +--
 src/converts/dom-to-canvas.ts             |  2 +-
 src/converts/dom-to-foreign-object-svg.ts |  4 +-
 src/create-context.ts                     |  8 ++--
 src/destroy-context.ts                    |  1 -
 src/embed-css-style-image.ts              |  4 ++
 src/embed-image-element.ts                | 45 ++++++++++++++---------
 src/fetch.ts                              |  6 +--
 src/image-to-canvas.ts                    | 26 +++++++------
 src/utils.ts                              |  1 +
 11 files changed, 63 insertions(+), 49 deletions(-)
diff --git a/index.html b/index.html
index a740842..cdd5180 100644
--- a/index.html
+++ b/index.html
@@ -13,16 +13,15 @@
     }
 
     #root {
-      display: inline-flex;
-      font-size: .875rem;
-      font-family: "Helvetica Neue",Arial,"Noto Sans","Liberation Sans",sans-serif,"Apple Color Emoji","Segoe UI Emoji","Segoe UI Symbol","Noto Color Emoji";
-      line-height: 1.5;
+      width: 300px;
+      height: 300px;
+      background-image: url('data:image/svg+xml,%3Csvg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 1080 1080"%3E%3Cpath fill="none" d="M0,657C0,438,0,219,0,0c360,0,720,0,1080,0c0,360,0,720,0,1080c-186.7,0-373.3,0-560,0c0.3-1.5,0.4-3.2,0.9-4.6 c19.7-58.2,24.6-117.7,13.7-178.2c-13.7-76.2-48.2-141.9-104.9-194.5c-90.3-83.8-196.9-113-317.7-90.1C71.9,620.2,34.8,635.9,0,657z"%3E%3C/path%3E%3Cpath fill="%23182430" d="M0,657c34.8-21.1,71.9-36.8,112-44.4c120.8-22.9,227.4,6.3,317.7,90.1C486.5,755.4,521,821,534.6,897.2 c10.9,60.5,6,119.9-13.7,178.2c-0.5,1.5-0.6,3.1-0.9,4.6c-173.2,0-346.5,0-520,0C0,939,0,798,0,657z"%3E%3C/path%3E%3C/svg%3E');
     }
   </style>
 </head>
 <body>
   <div>
-    <div id="root">前缀 <!---->after</div>
+    <div id="root"></div>
   </div>
 
   <script type="module" async>
diff --git a/src/context.ts b/src/context.ts
index 0ed6f3c..866d548 100644
--- a/src/context.ts
+++ b/src/context.ts
@@ -85,9 +85,9 @@ export interface InternalContext<T extends Node> {
   requests: Map<string, Request>
 
   /**
-   * All request images count
+   * Canvas multiple draw image fix svg+xml image decoding in Safari and Firefox
    */
-  requestImagesCount: number
+  drawImageCount: number
 
   /**
    * Wait for all tasks embedded in
@@ -97,7 +97,7 @@ export interface InternalContext<T extends Node> {
   /**
    * Automatically destroy context
    */
-  autodestruct: boolean
+  autoDestruct: boolean
 }
 
 export type Context<T extends Node = Node> = InternalContext<T> & Required<Options>
diff --git a/src/converts/dom-to-canvas.ts b/src/converts/dom-to-canvas.ts
index 78554f7..fdd51fd 100644
--- a/src/converts/dom-to-canvas.ts
+++ b/src/converts/dom-to-canvas.ts
@@ -12,7 +12,7 @@ export async function domToCanvas(node: any, options?: any) {
   const context = await orCreateContext(node, options)
   const svg = await domToForeignObjectSvg(context)
   const dataUrl = svgToDataUrl(svg)
-  if (!context.autodestruct) {
+  if (!context.autoDestruct) {
     context.svgStyleElement = createStyleElement(context.ownerDocument)
   }
   const image = createImage(dataUrl, svg.ownerDocument)
diff --git a/src/converts/dom-to-foreign-object-svg.ts b/src/converts/dom-to-foreign-object-svg.ts
index ef335c2..3b61c3a 100644
--- a/src/converts/dom-to-foreign-object-svg.ts
+++ b/src/converts/dom-to-foreign-object-svg.ts
@@ -25,7 +25,7 @@ export async function domToForeignObjectSvg(node: any, options?: any) {
     svgStyleElement,
     font,
     progress,
-    autodestruct,
+    autoDestruct,
     onCloneNode,
     onEmbedNode,
     onCreateForeignObjectSvg,
@@ -68,7 +68,7 @@ export async function domToForeignObjectSvg(node: any, options?: any) {
   const svg = createForeignObjectSvg(clone, context)
   svgStyleElement && svg.insertBefore(svgStyleElement, svg.children[0])
 
-  autodestruct && destroyContext(context)
+  autoDestruct && destroyContext(context)
 
   onCreateForeignObjectSvg?.(svg)
 
diff --git a/src/create-context.ts b/src/create-context.ts
index fa6bb9a..768fb56 100644
--- a/src/create-context.ts
+++ b/src/create-context.ts
@@ -13,10 +13,10 @@ import type { Options } from './options'
 export async function orCreateContext<T extends Node>(context: Context<T>): Promise<Context<T>>
 export async function orCreateContext<T extends Node>(node: T, options?: Options): Promise<Context<T>>
 export async function orCreateContext(node: any, options?: Options): Promise<Context> {
-  return isContext(node) ? node : createContext(node, { ...options, autodestruct: true })
+  return isContext(node) ? node : createContext(node, { ...options, autoDestruct: true })
 }
 
-export async function createContext<T extends Node>(node: T, options?: Options & { autodestruct?: boolean }): Promise<Context<T>> {
+export async function createContext<T extends Node>(node: T, options?: Options & { autoDestruct?: boolean }): Promise<Context<T>> {
   const { scale = 1, workerUrl, workerNumber = 1 } = options || {}
 
   const debug = Boolean(options?.debug)
@@ -52,7 +52,7 @@ export async function createContext<T extends Node>(node: T, options?: Options &
     onCloneNode: null,
     onEmbedNode: null,
     onCreateForeignObjectSvg: null,
-    autodestruct: false,
+    autoDestruct: false,
     ...options,
 
     // InternalContext
@@ -95,7 +95,7 @@ export async function createContext<T extends Node>(node: T, options?: Options &
       '*/*',
     ].filter(Boolean).join(',') };q=0.8`,
     requests,
-    requestImagesCount: 0,
+    drawImageCount: 0,
     tasks: [],
   }
 
diff --git a/src/destroy-context.ts b/src/destroy-context.ts
index 3c69e23..129135c 100644
--- a/src/destroy-context.ts
+++ b/src/destroy-context.ts
@@ -16,7 +16,6 @@ export function destroyContext(context: Context) {
   context.workers = []
   context.fontFamilies.clear()
   context.fontCssTexts.clear()
-  context.requestImagesCount = 0
   context.requests.clear()
   context.tasks = []
 }
diff --git a/src/embed-css-style-image.ts b/src/embed-css-style-image.ts
index cb95076..3c2dca3 100644
--- a/src/embed-css-style-image.ts
+++ b/src/embed-css-style-image.ts
@@ -1,4 +1,5 @@
 import { replaceCssUrlToDataUrl } from './css-url'
+import { IN_FIREFOX, IN_SAFARI } from './utils'
 import type { Context } from './context'
 
 const properties = [
@@ -19,6 +20,9 @@ export function embedCssStyleImage(
       if (!value) {
         return null
       }
+      if ((IN_SAFARI || IN_FIREFOX) && value.includes('data:image/svg+xml')) {
+        context.drawImageCount++
+      }
       return replaceCssUrlToDataUrl(value, null, context, true).then(newValue => {
         if (!newValue || value === newValue) return
         style.setProperty(
diff --git a/src/embed-image-element.ts b/src/embed-image-element.ts
index 4efd8d0..6981f2e 100644
--- a/src/embed-image-element.ts
+++ b/src/embed-image-element.ts
@@ -1,4 +1,4 @@
-import { isDataUrl, isImageElement, isSVGElementNode } from './utils'
+import { IN_FIREFOX, IN_SAFARI, isDataUrl, isImageElement, isSVGElementNode } from './utils'
 import { contextFetch } from './fetch'
 import type { Context } from './context'
 
@@ -6,30 +6,39 @@ export function embedImageElement<T extends HTMLImageElement | SVGImageElement>(
   clone: T,
   context: Context,
 ): Promise<void>[] {
-  if (isImageElement(clone) && !isDataUrl(clone.currentSrc || clone.src)) {
-    const url = clone.currentSrc || clone.src
-    clone.srcset = ''
-    clone.dataset.originalSrc = url
-    return [
-      contextFetch(context, {
-        url,
-        imageDom: clone,
-        requestType: 'image',
-        responseType: 'dataUrl',
-      }).then(url => {
-        clone.src = url || ''
-      }),
-    ]
+  if (isImageElement(clone)) {
+    const originalSrc = clone.currentSrc || clone.src
+
+    if (!isDataUrl(originalSrc)) {
+      return [
+        contextFetch(context, {
+          url: originalSrc,
+          imageDom: clone,
+          requestType: 'image',
+          responseType: 'dataUrl',
+        }).then(url => {
+          if (!url) return
+          clone.srcset = ''
+          clone.dataset.originalSrc = originalSrc
+          clone.src = url || ''
+        }),
+      ]
+    }
+
+    if ((IN_SAFARI || IN_FIREFOX) && originalSrc.includes('data:image/svg+xml')) {
+      context.drawImageCount++
+    }
   } else if (isSVGElementNode(clone) && !isDataUrl(clone.href.baseVal)) {
-    const url = clone.href.baseVal
-    clone.dataset.originalSrc = url
+    const originalSrc = clone.href.baseVal
     return [
       contextFetch(context, {
-        url,
+        url: originalSrc,
         imageDom: clone,
         requestType: 'image',
         responseType: 'dataUrl',
       }).then(url => {
+        if (!url) return
+        clone.dataset.originalSrc = originalSrc
         clone.href.baseVal = url || ''
       }),
     ]
diff --git a/src/fetch.ts b/src/fetch.ts
index 061e29f..38ec3dd 100644
--- a/src/fetch.ts
+++ b/src/fetch.ts
@@ -1,4 +1,4 @@
-import { blobToDataUrl, consoleWarn } from './utils'
+import { IN_SAFARI, blobToDataUrl, consoleWarn } from './utils'
 import type { Context, Request } from './context'
 
 export type BaseFetchOptions = RequestInit & {
@@ -62,8 +62,8 @@ export function contextFetch(context: Context, options: ContextFetchOptions) {
       url += (/\?/.test(url) ? '&' : '?') + new Date().getTime()
     }
 
-    if (requestType === 'image') {
-      context.requestImagesCount++
+    if (requestType === 'image' && IN_SAFARI) {
+      context.drawImageCount++
     }
 
     const baseFetchOptions: BaseFetchOptions = {
diff --git a/src/image-to-canvas.ts b/src/image-to-canvas.ts
index 2656a69..d436eb9 100644
--- a/src/image-to-canvas.ts
+++ b/src/image-to-canvas.ts
@@ -1,4 +1,4 @@
-import { IN_SAFARI, consoleWarn, loadMedia } from './utils'
+import { consoleWarn, loadMedia } from './utils'
 import type { Context } from './context'
 
 export async function imageToCanvas<T extends HTMLImageElement>(
@@ -7,8 +7,8 @@ export async function imageToCanvas<T extends HTMLImageElement>(
 ): Promise<HTMLCanvasElement> {
   const {
     log,
-    requestImagesCount,
     timeout,
+    drawImageCount,
     drawImageInterval,
   } = context
 
@@ -22,18 +22,20 @@ export async function imageToCanvas<T extends HTMLImageElement>(
       consoleWarn('Failed to drawImage', error)
     }
   }
+
   drawImage()
-  // fix: image not decode when drawImage svg+xml in safari/webkit
-  if (IN_SAFARI) {
-    for (let i = 0; i < requestImagesCount; i++) {
-      await new Promise<void>(resolve => {
-        setTimeout(() => {
-          drawImage()
-          resolve()
-        }, i + drawImageInterval)
-      })
-    }
+
+  for (let i = 0; i < drawImageCount; i++) {
+    await new Promise<void>(resolve => {
+      setTimeout(() => {
+        drawImage()
+        resolve()
+      }, i + drawImageInterval)
+    })
   }
+
+  context.drawImageCount = 0
+
   log.timeEnd('image to canvas')
   return canvas
 }
diff --git a/src/utils.ts b/src/utils.ts
index 9c71ff8..0f8f66c 100644
--- a/src/utils.ts
+++ b/src/utils.ts
@@ -9,6 +9,7 @@ export const SUPPORT_BTOA = IN_BROWSER && 'btoa' in window
 export const USER_AGENT = IN_BROWSER ? window.navigator?.userAgent : ''
 export const IN_CHROME = USER_AGENT.includes('Chrome')
 export const IN_SAFARI = USER_AGENT.includes('AppleWebKit') && !IN_CHROME
+export const IN_FIREFOX = USER_AGENT.includes('Firefox')
 
 // Context
 export const isContext = <T extends Node>(value: any): value is Context<T> => value && '__CONTEXT__' in value
