From 94740c671faf84fd00adb228c0407ea5602d0739 Mon Sep 17 00:00:00 2001
From: Alin Voinea <contact@avoinea.com>
Date: Tue, 6 Apr 2021 00:33:04 +0300
Subject: [PATCH] Add support for newId returned by onMutateBlock

---
 src/components/manage/Blocks/Group/EditBlockWrapper.jsx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/components/manage/Blocks/Group/EditBlockWrapper.jsx b/src/components/manage/Blocks/Group/EditBlockWrapper.jsx
index e3754c2..37485d8 100644
--- a/src/components/manage/Blocks/Group/EditBlockWrapper.jsx
+++ b/src/components/manage/Blocks/Group/EditBlockWrapper.jsx
@@ -71,6 +71,7 @@ class EditBlockWrapper extends React.Component {
       allowedBlocks,
       block,
       data,
+      onSelectBlock,
       onDeleteBlock,
       onMutateBlock,
       selected,
@@ -149,7 +150,8 @@ class EditBlockWrapper extends React.Component {
                     <BlockChooser
                       onMutateBlock={(id, value) => {
                         this.setState({ addNewBlockOpened: false });
-                        onMutateBlock(id, value);
+                        const newId = onMutateBlock(id, value);
+                        newId && onSelectBlock(newId);
                       }}
                       currentBlock={block}
                       allowedBlocks={allowedBlocks}