From 6c0996b0a3d0a96d0d4cd6a26944b08b08e81323 Mon Sep 17 00:00:00 2001
From: Valentin Mihai <Valentin-Gabriel.Mihai.ext@knecon.com>
Date: Mon, 11 Sep 2023 18:03:02 +0300
Subject: [PATCH] DM-426 - Enable layout grid button needs two clicks in new
 file

---
 .../file-preview/file-preview-screen.component.ts        | 1 +
 .../app/modules/pdf-viewer/services/layers.service.ts    | 9 +++++++++
 .../modules/pdf-viewer/services/viewer-header.service.ts | 4 ++++
 3 files changed, 14 insertions(+)

diff --git a/apps/red-ui/src/app/modules/file-preview/file-preview-screen.component.ts b/apps/red-ui/src/app/modules/file-preview/file-preview-screen.component.ts
index 7b93ef488..cf56904e0 100644
--- a/apps/red-ui/src/app/modules/file-preview/file-preview-screen.component.ts
+++ b/apps/red-ui/src/app/modules/file-preview/file-preview-screen.component.ts
@@ -338,6 +338,7 @@ export class FilePreviewScreenComponent
         document.documentElement.addEventListener('fullscreenchange', this.fullscreenListener);
 
         this.#openRssDialogIfDefault();
+        this._viewerHeaderService.resetLayers();
     }
 
     ngAfterViewInit() {
diff --git a/apps/red-ui/src/app/modules/pdf-viewer/services/layers.service.ts b/apps/red-ui/src/app/modules/pdf-viewer/services/layers.service.ts
index 18e2f0019..98ec02af1 100644
--- a/apps/red-ui/src/app/modules/pdf-viewer/services/layers.service.ts
+++ b/apps/red-ui/src/app/modules/pdf-viewer/services/layers.service.ts
@@ -33,6 +33,11 @@ export class LayersService {
         return tooltipsDisabled ? this.#enableIcon : this.#disableIcon;
     }
 
+    resetLayers() {
+        this._active = false;
+        this.#updateButton();
+    }
+
     async toggleLayers(): Promise<void> {
         const layers = await this._documentViewer.document.getLayersArray();
         layers.forEach(layer => {
@@ -45,6 +50,10 @@ export class LayersService {
         this._documentViewer.refreshAndUpdateView();
 
         this._active = !this._active;
+        this.#updateButton();
+    }
+
+    #updateButton() {
         this._pdf.instance.UI.updateElement(HeaderElements.TOGGLE_LAYERS, {
             title: this.toggleLayersBtnTitle,
             img: this.toggleLayersBtnIcon,
diff --git a/apps/red-ui/src/app/modules/pdf-viewer/services/viewer-header.service.ts b/apps/red-ui/src/app/modules/pdf-viewer/services/viewer-header.service.ts
index aaeab41d8..967a125c3 100644
--- a/apps/red-ui/src/app/modules/pdf-viewer/services/viewer-header.service.ts
+++ b/apps/red-ui/src/app/modules/pdf-viewer/services/viewer-header.service.ts
@@ -377,4 +377,8 @@ export class ViewerHeaderService {
             this.disable(ROTATION_ACTION_BUTTONS);
         }
     }
+
+    resetLayers() {
+        this._layersService.resetLayers();
+    }
 }