Add support for newId returned by onMutateBlock

eea · Apr 5, 2021 · 4030d3c · 4030d3c
1 parent d53cfb5
commit 4030d3c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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}