Implemented restore_on_error

ishida123 · Apr 10, 2022 · 1f2b4da · 1f2b4da
1 parent 4ffd5bf
commit 1f2b4da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/defaults.js b/src/defaults.js
@@ -30,7 +30,8 @@ const defaultSettings = {
     callback_error: null,
     callback_finish: null,
     callback_cancel: null,
-    use_native: false
+    use_native: false,
+    restore_on_error: false,
 };
 
 export const getExtendedSettings = (customSettings) => {

diff --git a/src/event.js b/src/event.js
@@ -10,6 +10,7 @@ import {
     haveElementsToLoad,
     isSomethingLoading
 } from "./counters";
+import { attrsSrcSrcsetSizes, restoreOriginalAttrs } from "./originalAttributes";
 
 const elementsWithLoadEvent = ["IMG", "IFRAME", "VIDEO", "OBJECT"];
 export const hasLoadEvent = (element) => elementsWithLoadEvent.indexOf(element.tagName) > -1;
@@ -77,6 +78,7 @@ export const errorHandler = (event, element, settings, instance) => {
     addClass(element, settings.class_error);
     setStatus(element, statusError);
     safeCallback(settings.callback_error, element, instance);
+    if (settings.restore_on_error) restoreOriginalAttrs(element, attrsSrcSrcsetSizes);
     if (!goingNative) checkFinish(settings, instance);
 };