[D&D] Refactor and cleanup embeddables

src/plugins/wizard/public/embeddable/wizard_component.tsx

@@ -3,66 +3,32 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import React, { useEffect, useState } from 'react';
+import React from 'react';
 
 import { SavedObjectEmbeddableInput, withEmbeddableSubscription } from '../../../embeddable/public';
 import { WizardEmbeddable, WizardOutput } from './wizard_embeddable';
-import { validateSchemaState } from '../application/utils/validate_schema_state';
+import { getReactExpressionRenderer } from '../plugin_services';
 
 interface Props {
   embeddable: WizardEmbeddable;
   input: SavedObjectEmbeddableInput;
   output: WizardOutput;
 }
 
-function WizardEmbeddableComponentInner({
-  embeddable,
-  input: {},
-  output: { savedAttributes },
-}: Props) {
-  const { ReactExpressionRenderer, toasts, types, indexPatterns, aggs } = embeddable;
-  const [expression, setExpression] = useState<string>();
-
-  useEffect(() => {
-    const { visualizationState: visualization, styleState: style } = savedAttributes || {};
-    if (savedAttributes === undefined || visualization === undefined || style === undefined) {
-      return;
-    }
-
-    const rootState = {
-      visualization: JSON.parse(visualization),
-      style: JSON.parse(style),
-    };
-
-    const visualizationType = types.get(rootState.visualization?.activeVisualization?.name ?? '');
-    if (!visualizationType) {
-      throw new Error(`Invalid visualization type ${visualizationType}`);
-    }
-    const { toExpression, ui } = visualizationType;
-
-    async function loadExpression() {
-      const schemas = ui.containerConfig.data.schemas;
-      const [valid, errorMsg] = validateSchemaState(schemas, rootState);
-
-      if (!valid) {
-        if (errorMsg) {
-          toasts.addWarning(errorMsg);
-        }
-        setExpression(undefined);
-        return;
-      }
-      const exp = await toExpression(rootState, indexPatterns, aggs);
-      setExpression(exp);
-    }
-
-    if (savedAttributes !== undefined) {
-      loadExpression();
-    }
-  }, [aggs, indexPatterns, savedAttributes, toasts, types]);
-
-  return <ReactExpressionRenderer expression={expression ?? ''} />;
-
-  // TODO: add correct loading and error states
+function WizardEmbeddableComponentInner({ embeddable, input: {}, output: { error } }: Props) {
+  const { expression } = embeddable;
+  const ReactExpressionRenderer = getReactExpressionRenderer();
+
+  return (
+    <>
+      {error?.message ? (
+        // TODO: add correct loading and error states
+        <div>{error.message}</div>
+      ) : (
+        <ReactExpressionRenderer expression={expression ?? ''} />
+      )}
+    </>
+  );
 }
 
 export const WizardEmbeddableComponent = withEmbeddableSubscription<