From 3f0e06d8958350902ade3653996ab5dd755ae6af Mon Sep 17 00:00:00 2001
From: Alin Voinea <contact@avoinea.com>
Date: Wed, 7 Apr 2021 18:08:39 +0300
Subject: [PATCH] Add support for onInsertBlock with BlockChooser

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

diff --git a/src/ColumnsBlock/EditBlockWrapper.jsx b/src/ColumnsBlock/EditBlockWrapper.jsx
index 8e9ae2e..8d3ef1e 100644
--- a/src/ColumnsBlock/EditBlockWrapper.jsx
+++ b/src/ColumnsBlock/EditBlockWrapper.jsx
@@ -66,7 +66,9 @@ class EditBlockWrapper extends React.Component {
       block,
       data,
       onDeleteBlock,
+      onInsertBlock,
       onMutateBlock,
+      onSelectBlock,
       selected,
     } = blockProps;
     const type = data['@type'];
@@ -139,8 +141,12 @@ class EditBlockWrapper extends React.Component {
               {this.state.addNewBlockOpened && (
                 <BlockChooser
                   onMutateBlock={(id, value) => {
-                    this.setState({ addNewBlockOpened: false });
                     onMutateBlock(id, value);
+                    this.setState({ addNewBlockOpened: false });
+                  }}
+                  onInsertBlock={(id, value) => {
+                    onSelectBlock(onInsertBlock(id, value));
+                    this.setState({ addNewBlockOpened: false });
                   }}
                   currentBlock={block}
                   allowedBlocks={allowedBlocks}