From 6bfa4894a89fbd5627a0b3aa4b3137a2a5ed0985 Mon Sep 17 00:00:00 2001
From: blam <ben@blam.sh>
Date: Mon, 2 Oct 2023 10:41:26 +0200
Subject: [PATCH] chore: fixing some missing features that got accidentally
 replaced

Signed-off-by: blam <ben@blam.sh>
---
 .../Stepper/createAsyncValidators.ts          |  1 +
 .../src/next/components/Stepper/index.ts      |  1 +
 .../CustomFieldExplorer.tsx                   |  8 +++--
 .../TemplateEditorPage/TemplateEditorForm.tsx |  2 +-
 .../TemplateFormPreviewer.tsx                 |  3 ++
 .../TemplateEditorPage/TemplateEditorForm.tsx | 36 ++++++++++++++-----
 6 files changed, 40 insertions(+), 11 deletions(-)

diff --git a/plugins/scaffolder-react/src/next/components/Stepper/createAsyncValidators.ts b/plugins/scaffolder-react/src/next/components/Stepper/createAsyncValidators.ts
index a606f01d6b..8eb665911d 100644
--- a/plugins/scaffolder-react/src/next/components/Stepper/createAsyncValidators.ts
+++ b/plugins/scaffolder-react/src/next/components/Stepper/createAsyncValidators.ts
@@ -32,6 +32,7 @@ export type FormValidation = {
   [name: string]: FieldValidation | FormValidation;
 };
 
+/** @alpha */
 export const createAsyncValidators = (
   rootSchema: JsonObject,
   validators: Record<
diff --git a/plugins/scaffolder-react/src/next/components/Stepper/index.ts b/plugins/scaffolder-react/src/next/components/Stepper/index.ts
index daa47f078e..ebf0f7e902 100644
--- a/plugins/scaffolder-react/src/next/components/Stepper/index.ts
+++ b/plugins/scaffolder-react/src/next/components/Stepper/index.ts
@@ -14,3 +14,4 @@
  * limitations under the License.
  */
 export { Stepper, type StepperProps } from './Stepper';
+export { createAsyncValidators } from './createAsyncValidators';
diff --git a/plugins/scaffolder/src/legacy/TemplateEditorPage/CustomFieldExplorer.tsx b/plugins/scaffolder/src/legacy/TemplateEditorPage/CustomFieldExplorer.tsx
index c790dbd72c..de0c243e42 100644
--- a/plugins/scaffolder/src/legacy/TemplateEditorPage/CustomFieldExplorer.tsx
+++ b/plugins/scaffolder/src/legacy/TemplateEditorPage/CustomFieldExplorer.tsx
@@ -74,6 +74,7 @@ export const CustomFieldExplorer = ({
   const [selectedField, setSelectedField] = useState(fieldOptions[0]);
   const [fieldFormState, setFieldFormState] = useState({});
   const [refreshKey, setRefreshKey] = useState(Date.now());
+  const [formState, setFormState] = useState({});
   const sampleFieldTemplate = useMemo(
     () =>
       yaml.stringify({
@@ -103,8 +104,9 @@ export const CustomFieldExplorer = ({
     selection => {
       setSelectedField(selection);
       setFieldFormState({});
+      setFormState({});
     },
-    [setFieldFormState, setSelectedField],
+    [setFieldFormState, setSelectedField, setFormState],
   );
 
   const handleFieldConfigChange = useCallback(
@@ -149,7 +151,7 @@ export const CustomFieldExplorer = ({
             <Form
               showErrorList={false}
               // @ts-ignore
-              fields={{ ...fieldComponents }}
+              fields={{ ...fieldOverrides, ...fieldComponents }}
               noHtml5Validate
               formData={fieldFormState}
               formContext={{ fieldFormState }}
@@ -183,6 +185,8 @@ export const CustomFieldExplorer = ({
           </CardContent>
         </Card>
         <TemplateEditorForm
+          data={formState}
+          onUpdate={setFormState}
           key={refreshKey}
           content={sampleFieldTemplate}
           contentIsSpec
diff --git a/plugins/scaffolder/src/legacy/TemplateEditorPage/TemplateEditorForm.tsx b/plugins/scaffolder/src/legacy/TemplateEditorPage/TemplateEditorForm.tsx
index 766581e311..34e1b05c98 100644
--- a/plugins/scaffolder/src/legacy/TemplateEditorPage/TemplateEditorForm.tsx
+++ b/plugins/scaffolder/src/legacy/TemplateEditorPage/TemplateEditorForm.tsx
@@ -196,7 +196,7 @@ export function TemplateEditorForm(props: TemplateEditorFormProps) {
             formData={data}
             onChange={e => onUpdate(e.formData)}
             onReset={() => onUpdate({})}
-            finishButtonLabel={onDryRun && 'Try It'}
+            finishButtonLabel={onDryRun && 'Try iIt'}
             onFinish={onDryRun && (() => onDryRun(data))}
             layouts={layouts}
           />
diff --git a/plugins/scaffolder/src/legacy/TemplateEditorPage/TemplateFormPreviewer.tsx b/plugins/scaffolder/src/legacy/TemplateEditorPage/TemplateFormPreviewer.tsx
index 21df43cf8c..b83acf85a5 100644
--- a/plugins/scaffolder/src/legacy/TemplateEditorPage/TemplateFormPreviewer.tsx
+++ b/plugins/scaffolder/src/legacy/TemplateEditorPage/TemplateFormPreviewer.tsx
@@ -125,6 +125,7 @@ export const TemplateFormPreviewer = ({
   const [errorText, setErrorText] = useState<string>();
   const [templateOptions, setTemplateOptions] = useState<TemplateOption[]>([]);
   const [templateYaml, setTemplateYaml] = useState(defaultPreviewTemplate);
+  const [formState, setFormState] = useState({});
 
   const { loading } = useAsync(
     () =>
@@ -204,6 +205,8 @@ export const TemplateFormPreviewer = ({
         </div>
         <div className={classes.preview}>
           <TemplateEditorForm
+            data={formState}
+            onUpdate={setFormState}
             content={templateYaml}
             contentIsSpec
             fieldExtensions={customFieldExtensions}
diff --git a/plugins/scaffolder/src/next/TemplateEditorPage/TemplateEditorForm.tsx b/plugins/scaffolder/src/next/TemplateEditorPage/TemplateEditorForm.tsx
index 15c53d6d81..4b5a00b7ba 100644
--- a/plugins/scaffolder/src/next/TemplateEditorPage/TemplateEditorForm.tsx
+++ b/plugins/scaffolder/src/next/TemplateEditorPage/TemplateEditorForm.tsx
@@ -16,7 +16,7 @@
 import { useApiHolder } from '@backstage/core-plugin-api';
 import { JsonObject, JsonValue } from '@backstage/types';
 import { makeStyles } from '@material-ui/core/styles';
-import React, { Component, ReactNode, useState } from 'react';
+import React, { Component, ReactNode, useMemo, useState } from 'react';
 import useDebounce from 'react-use/lib/useDebounce';
 import yaml from 'yaml';
 import {
@@ -24,7 +24,10 @@ import {
   TemplateParameterSchema,
   FieldExtensionOptions,
 } from '@backstage/plugin-scaffolder-react';
-import { Stepper } from '@backstage/plugin-scaffolder-react/alpha';
+import {
+  Stepper,
+  createAsyncValidators,
+} from '@backstage/plugin-scaffolder-react/alpha';
 import { useDryRun } from './DryRunContext';
 import { useDirectoryEditor } from './DirectoryEditorContext';
 
@@ -80,6 +83,7 @@ interface TemplateEditorFormProps {
   /** Setting this to true will cause the content to be parsed as if it is the template entity spec */
   contentIsSpec?: boolean;
   setErrorText: (errorText?: string) => void;
+
   onDryRun?: (data: JsonObject) => Promise<void>;
   fieldExtensions?: FieldExtensionOptions<any, any>[];
   layouts?: LayoutOptions[];
@@ -104,6 +108,12 @@ export function TemplateEditorForm(props: TemplateEditorFormProps) {
 
   const [steps, setSteps] = useState<TemplateParameterSchema['steps']>();
 
+  const fields = useMemo(() => {
+    return Object.fromEntries(
+      fieldExtensions.map(({ name, component }) => [name, component]),
+    );
+  }, [fieldExtensions]);
+
   useDebounce(
     () => {
       try {
@@ -111,7 +121,10 @@ export function TemplateEditorForm(props: TemplateEditorFormProps) {
           setSteps(undefined);
           return;
         }
-        const parsed: JsonValue = yaml.parse(content);
+        const parsed: JsonValue = yaml
+          .parseAllDocuments(content)
+          .filter(c => c)
+          .map(c => c.toJSON())[0];
 
         if (!isJsonObject(parsed)) {
           setSteps(undefined);
@@ -138,6 +151,10 @@ export function TemplateEditorForm(props: TemplateEditorFormProps) {
           return;
         }
 
+        const fieldValidators = Object.fromEntries(
+          fieldExtensions.map(({ name, validation }) => [name, validation]),
+        );
+
         setErrorText();
         setSteps(
           parameters.flatMap(param =>
@@ -146,6 +163,9 @@ export function TemplateEditorForm(props: TemplateEditorFormProps) {
                   {
                     title: String(param.title),
                     schema: param,
+                    validate: createAsyncValidators(param, fieldValidators, {
+                      apiHolder,
+                    }),
                   },
                 ]
               : [],
@@ -170,11 +190,11 @@ export function TemplateEditorForm(props: TemplateEditorFormProps) {
           <Stepper
             manifest={{ steps, title: 'Template Editor' }}
             extensions={fieldExtensions}
-            onCreate={async data => {
-              await onDryRun?.(data);
+            components={fields}
+            onCreate={async options => {
+              await onDryRun?.(options);
             }}
             layouts={layouts}
-            components={{ createButtonText: onDryRun && 'Try It' }}
           />
         </ErrorBoundary>
       </div>
@@ -195,7 +215,7 @@ export function TemplateEditorFormDirectoryEditorDryRun(
   const directoryEditor = useDirectoryEditor();
   const { selectedFile } = directoryEditor;
 
-  const handleDryRun = async (values: JsonObject) => {
+  const handleDryRun = async (data: JsonObject) => {
     if (!selectedFile) {
       return;
     }
@@ -203,7 +223,7 @@ export function TemplateEditorFormDirectoryEditorDryRun(
     try {
       await dryRun.execute({
         templateContent: selectedFile.content,
-        values,
+        values: data,
         files: directoryEditor.files,
       });
       setErrorText();