feat: correctly close lightbox

1 files changed, 13 insertions, 0 deletions

diff --git a/src/lightbox.js b/src/lightbox.js
index 579f14b..5523fc7 100644
--- a/src/lightbox.js
+++ b/src/lightbox.js
@@ -55,9 +55,13 @@ export default class LightBox {
    */
 
   openLightBox(image) {
+    image.style.visibility = 'hidden';
     const shadow = this.shadows[image.id] || new Shadow(image);
     if (!this.shadows[image.id]) {
       this.shadows[image.id] = shadow;
+      shadow.element.addEventListener('click', () => {
+        if (shadow.isOpen) this.closeLightBox(image);
+      });
     }
 
     shadow.open();
@@ -65,4 +69,13 @@ export default class LightBox {
     this.overlay.add(shadow);
     this.overlay.show();
   }
+
+  closeLightBox(image) {
+    image.style.visibility = 'visible';
+    const shadow = this.shadows[image.id];
+    if (!shadow) return;
+
+    shadow.close();
+    this.overlay.hide();
+  }
 }