From 24d7380333884f1bf1b69b6627236703403752a1 Mon Sep 17 00:00:00 2001
From: Zsolt Viczian <viczian.zsolt@gmail.com>
Date: Sun, 20 Mar 2022 17:07:25 +0100
Subject: [PATCH] penMode

---
 src/components/Actions.tsx | 7 +++----
 src/components/App.tsx     | 6 +-----
 2 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/src/components/Actions.tsx b/src/components/Actions.tsx
index 1cc315789..72669195a 100644
--- a/src/components/Actions.tsx
+++ b/src/components/Actions.tsx
@@ -193,7 +193,7 @@ export const ShapesSwitcher = ({
   onImageAction: (data: { pointerType: PointerType | null }) => void;
   appState: AppState;
 }) => {
-  const deviceType = useDeviceType();
+  const penDetected = useDeviceType().penDetected;
   return (
     <>
       {SHAPES.map(({ value, icon, key }, index) => {
@@ -216,13 +216,12 @@ export const ShapesSwitcher = ({
             aria-keyshortcuts={shortcut}
             data-testid={value}
             onChange={({ pointerType }) => {
-              if (!deviceType.penDetected && pointerType === "pen") {
-                setAppState({ penMode: true });
-              }
               setAppState({
                 elementType: value,
                 multiElement: null,
                 selectedElementIds: {},
+                penMode:
+                  (!penDetected && pointerType === "pen") ?? appState.penMode,
               });
               setCursorForShape(canvas, { ...appState, elementType: value });
               if (value === "image") {
diff --git a/src/components/App.tsx b/src/components/App.tsx
index 0da9fac81..136905a4d 100644
--- a/src/components/App.tsx
+++ b/src/components/App.tsx
@@ -2867,11 +2867,7 @@ class App extends React.Component<AppProps, AppState> {
     //the user can disable this by toggling the penMode button
     if (!this.deviceType.penDetected && event.pointerType === "pen") {
       this.deviceType = updateObject(this.deviceType, { penDetected: true });
-      this.setState((prevState) => {
-        return {
-          penMode: true,
-        };
-      });
+      this.setState({ penMode: true });
     }
 
     if (