From 4244563b67e0277ea0d9547a1e91ea53e81efc15 Mon Sep 17 00:00:00 2001
From: Wenhuan Li <wenhuan@fb.com>
Date: Fri, 17 Sep 2021 11:13:38 -0700
Subject: [PATCH] fix take screenshot button state

Summary: Take screenshot button state is not reset after screenshot taken. So Flipper can only take 1 screenshot per session.

Reviewed By: mweststrate

Differential Revision: D31018526

fbshipit-source-id: dc7b00f26c77ec94a8a15047f5ae835acdb710c7
---
 desktop/app/src/chrome/ScreenCaptureButtons.tsx | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/desktop/app/src/chrome/ScreenCaptureButtons.tsx b/desktop/app/src/chrome/ScreenCaptureButtons.tsx
index 2f014fcb7..a5724c5ac 100644
--- a/desktop/app/src/chrome/ScreenCaptureButtons.tsx
+++ b/desktop/app/src/chrome/ScreenCaptureButtons.tsx
@@ -71,8 +71,11 @@ export default function ScreenCaptureButtons() {
         console.error('Taking screenshot failed:', e);
         message.error('Taking screenshot failed:' + e);
       })
-      .finally(() => {});
+      .finally(() => {
+        setIsTakingScreenshot(false);
+      });
   }, [selectedDevice]);
+
   const handleRecording = useCallback(() => {
     if (!selectedDevice) {
       return;