can shape edit + when add shape is default selected

src/pages/editor/components/canvas/useDrawingHandlers.ts

@@ -10,7 +10,7 @@ export const useDrawingHandlers = () => {
     const [tempObject, setTempObject] = useState<EditorObject | null>(null);
     const transformerRef = useRef<Konva.Transformer>(null);
 
-    const { tool, addObject, updateObject, setSelectedObjectId, selectedObjectId } = useEditorStore();
+    const { tool, addObject, updateObject, setSelectedObjectId, selectedObjectId, setTool } = useEditorStore();
     const { defaults } = useTextDefaultsStore();
 
     useEffect(() => {
@@ -71,6 +71,7 @@ export const useDrawingHandlers = () => {
             };
             addObject(newText);
             setSelectedObjectId(newText.id);
+            setTool("select");
         }
     };
 
@@ -104,8 +105,13 @@ export const useDrawingHandlers = () => {
     };
 
     const handleStageMouseUp = () => {
+        const lastObjectId = tempObject?.id;
         setIsDrawing(false);
         setTempObject(null);
+        if (lastObjectId) {
+            setSelectedObjectId(lastObjectId);
+            setTool("select");
+        }
     };
 
     return {