From 82b457f52b1c191842fe83bf88623d4fbf1a34f9 Mon Sep 17 00:00:00 2001
From: andreiggr <andrei.ggr@gmail.com>
Date: Mon, 24 Oct 2022 16:08:46 +0300
Subject: [PATCH] Enable variations on new Embed block

---
 src/Blocks/EmbedEEAVisualization/Edit.jsx | 18 +++++++++++++++---
 src/Blocks/EmbedEEAVisualization/index.js | 21 +++++++++++++++------
 2 files changed, 30 insertions(+), 9 deletions(-)

diff --git a/src/Blocks/EmbedEEAVisualization/Edit.jsx b/src/Blocks/EmbedEEAVisualization/Edit.jsx
index 405f74a6..1e10991c 100644
--- a/src/Blocks/EmbedEEAVisualization/Edit.jsx
+++ b/src/Blocks/EmbedEEAVisualization/Edit.jsx
@@ -1,13 +1,13 @@
 import React from 'react';
 import { SidebarPortal } from '@plone/volto/components';
-import InlineForm from '@plone/volto/components/manage/Form/InlineForm';
+import BlockDataForm from '@plone/volto/components/manage/Form/BlockDataForm';
 import ConnectedChart from '@eeacms/volto-plotlycharts/ConnectedChart';
 import schema from './schema';
 
 import '@eeacms/volto-plotlycharts/less/visualization.less';
 
 const Edit = (props) => {
-  const { data } = props;
+  const { data, block } = props;
   return (
     <>
       <ConnectedChart
@@ -28,7 +28,7 @@ const Edit = (props) => {
       />
 
       <SidebarPortal selected={props.selected}>
-        <InlineForm
+        {/* <InlineForm
           schema={schema}
           title={schema.title}
           onChangeField={(id, value) => {
@@ -38,6 +38,18 @@ const Edit = (props) => {
             });
           }}
           formData={props.data}
+        /> */}
+        <BlockDataForm
+          block={block}
+          title={schema.title}
+          schema={schema}
+          onChangeField={(id, value) => {
+            props.onChangeBlock(block, {
+              ...data,
+              [id]: value,
+            });
+          }}
+          formData={data}
         />
       </SidebarPortal>
     </>
diff --git a/src/Blocks/EmbedEEAVisualization/index.js b/src/Blocks/EmbedEEAVisualization/index.js
index c7bfaaa5..586da5f4 100644
--- a/src/Blocks/EmbedEEAVisualization/index.js
+++ b/src/Blocks/EmbedEEAVisualization/index.js
@@ -18,14 +18,23 @@ export default (config) => {
       addPermission: [],
       view: [],
     },
+    variations: [
+      {
+        id: 'default',
+        title: 'Default',
+        isDefault: true,
+        view: View,
+      },
+      {
+        id: 'extra',
+        title: 'Extra (expand if needed)',
+        isDefault: false,
+        view: View,
+      },
+    ],
   };
 
   config.blocks.blocksConfig.embed_eea_visualization = visualizationBlockConfig;
-  // This is required for compatibility with previous version
-  // TODO: script for migration
-  // config.blocks.blocksConfig.embed_chart = {
-  //   ...visualizationBlockConfig,
-  //   id: 'embed_chart',
-  // };
+
   return config;
 };