From e38fb58e1bdde7984c4f20c97711a77b27932e87 Mon Sep 17 00:00:00 2001
From: Misode <Misoloo64@gmail.com>
Date: Tue, 8 Dec 2020 00:45:10 +0100
Subject: [PATCH] Update preview controls when switching previews

---
 src/app/components/panels/PreviewPanel.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/app/components/panels/PreviewPanel.ts b/src/app/components/panels/PreviewPanel.ts
index e47c0072..19441fa7 100644
--- a/src/app/components/panels/PreviewPanel.ts
+++ b/src/app/components/panels/PreviewPanel.ts
@@ -23,20 +23,23 @@ export const PreviewPanel = (view: View, model: DataModel) => {
         App.preview.set(null)
       }
     }
-    view.mount(el.querySelector('.panel-controls')!, `
+    const updatePreview = () => {
+      redraw()
+      view.mount(el.querySelector('.panel-controls')!, `
       ${App.preview.get()?.menu(view, redraw) ?? ''}
       <div class="btn" data-id="${view.onClick(() => {
         Tracker.hidePreview(); App.preview.set(null)
       })}">
         ${Octicon.x}
       </div>`, false)
+    }
     model.addListener({
       invalidated: redraw
     })
     App.preview.watchRun((value) => {
       if (value) {
         value.redraw = redraw
-        redraw()
+        updatePreview()
       }
     }, 'preview-panel')