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

---
 src/ColumnsBlock/EditBlockWrapper.jsx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/ColumnsBlock/EditBlockWrapper.jsx b/src/ColumnsBlock/EditBlockWrapper.jsx
index 8e9ae2e..feb41d4 100644
--- a/src/ColumnsBlock/EditBlockWrapper.jsx
+++ b/src/ColumnsBlock/EditBlockWrapper.jsx
@@ -65,6 +65,7 @@ class EditBlockWrapper extends React.Component {
       allowedBlocks,
       block,
       data,
+      onSelectBlock,
       onDeleteBlock,
       onMutateBlock,
       selected,
@@ -140,7 +141,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}