refactor: move image handling to shadow class

- lightbox module no longer handles original image visibility
- lightbox module no longer handles shadow placement
- lightbox module now only fetch and use shadow object

1 files changed, 8 insertions, 0 deletions

diff --git a/src/shadow.js b/src/shadow.js
index 19703de..462b3af 100644
--- a/src/shadow.js
+++ b/src/shadow.js
@@ -11,6 +11,7 @@ export default class Shadow {
    */
 
   constructor(image) {
+    this.original = image;
     this.create(image);
 
     this.startingState = {
@@ -49,6 +50,10 @@ export default class Shadow {
 
   open() {
     this.isOpen = true;
+
+    this.original.style.visibility = 'hidden';
+    if (!document.body.contains(this.element)) this.placeItself();
+
     Animation.transformMatrix(
       this.element,
       this.startingState,
@@ -58,6 +63,9 @@ export default class Shadow {
 
   close() {
     this.isOpen = false;
+
+    this.original.style.visibility = 'visible';
+
     Animation.resetTransformMatrix(this.element);
     setTimeout(() => this.destroyItself(), 300);
   }