From 70d584e01d30a64054e43ad30afe2b65bef183ea Mon Sep 17 00:00:00 2001
From: Bandini Bhopi <bandinib@amazon.com>
Date: Thu, 29 Feb 2024 19:19:55 +0000
Subject: [PATCH] PoC for passing state variable to child form

Signed-off-by: Bandini Bhopi <bandinib@amazon.com>
---
 .../authentication_methods_registry.ts        |  2 +-
 .../create_form/create_data_source_form.tsx   | 32 ++++++++++++++++---
 2 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/src/plugins/data_source_management/public/auth_registry/authentication_methods_registry.ts b/src/plugins/data_source_management/public/auth_registry/authentication_methods_registry.ts
index 00c3b0dbf0ee..5593b82f7bf6 100644
--- a/src/plugins/data_source_management/public/auth_registry/authentication_methods_registry.ts
+++ b/src/plugins/data_source_management/public/auth_registry/authentication_methods_registry.ts
@@ -9,7 +9,7 @@ import { EuiSuperSelectOption } from '@elastic/eui';
 export interface AuthenticationMethod {
   name: string;
   credentialSourceOption: EuiSuperSelectOption<string>;
-  credentialForm?: React.JSX.Element;
+  credentialForm?: (state?: any, setState?: any) => React.JSX.Element;
   crendentialFormField?: { [key: string]: string };
 }
 
diff --git a/src/plugins/data_source_management/public/components/create_data_source_wizard/components/create_form/create_data_source_form.tsx b/src/plugins/data_source_management/public/components/create_data_source_wizard/components/create_form/create_data_source_form.tsx
index e178ea1bcffa..1c7fbc64d0d0 100644
--- a/src/plugins/data_source_management/public/components/create_data_source_wizard/components/create_form/create_data_source_form.tsx
+++ b/src/plugins/data_source_management/public/components/create_data_source_wizard/components/create_form/create_data_source_form.tsx
@@ -22,6 +22,7 @@ import {
 } from '@elastic/eui';
 import { i18n } from '@osd/i18n';
 import { FormattedMessage } from '@osd/i18n/react';
+import { AuthenticationMethodRegistery } from 'src/plugins/data_source_management/public/auth_registry';
 import { SigV4Content, SigV4ServiceName } from '../../../../../../data_source/common/data_sources';
 import {
   AuthType,
@@ -55,7 +56,11 @@ export interface CreateDataSourceState {
   endpoint: string;
   auth: {
     type: AuthType;
-    credentials: UsernamePasswordTypedContent | SigV4Content | undefined;
+    credentials:
+      | UsernamePasswordTypedContent
+      | SigV4Content
+      | undefined
+      | { [key: string]: string };
   };
 }
 
@@ -68,16 +73,17 @@ export class CreateDataSourceForm extends React.Component<
 
   authOptions: Array<EuiSuperSelectOption<string>> = [];
   isNoAuthOptionEnabled: boolean;
+  authenticationMethodRegistery: AuthenticationMethodRegistery;
 
   constructor(props: CreateDataSourceProps, context: DataSourceManagementContextValue) {
     super(props, context);
 
-    const authenticationMethodRegistery = context.services.authenticationMethodRegistery;
-    const registeredAuthMethods = authenticationMethodRegistery.getAllAuthenticationMethods();
-    const initialSelectedAuthMethod = getDefaultAuthMethod(authenticationMethodRegistery);
+    this.authenticationMethodRegistery = context.services.authenticationMethodRegistery;
+    const registeredAuthMethods = this.authenticationMethodRegistery.getAllAuthenticationMethods();
+    const initialSelectedAuthMethod = getDefaultAuthMethod(this.authenticationMethodRegistery);
 
     this.isNoAuthOptionEnabled =
-      authenticationMethodRegistery.getAuthenticationMethod(AuthType.NoAuth) !== undefined;
+      this.authenticationMethodRegistery.getAuthenticationMethod(AuthType.NoAuth) !== undefined;
 
     this.authOptions = registeredAuthMethods.map((authMethod) => {
       return authMethod.credentialSourceOption;
@@ -152,6 +158,20 @@ export class CreateDataSourceForm extends React.Component<
     });
   };
 
+  // Function to update state
+  handleStateChange = (newState: any) => {
+    this.setState(newState);
+  };
+
+  getAuthUIFromRegistry = (authType: string) => {
+    const authMethodUI = this.authenticationMethodRegistery.getAuthenticationMethod(authType);
+    const form = authMethodUI?.credentialForm;
+    if (form !== undefined) {
+      return form(this.state, this.handleStateChange);
+    }
+    return null;
+  };
+
   onChangeSigV4ServiceName = (service: SigV4ServiceName) => {
     this.setState({
       auth: {
@@ -296,6 +316,7 @@ export class CreateDataSourceForm extends React.Component<
 
   getFormValues = (): DataSourceAttributes => {
     let credentials = this.state.auth.credentials;
+    console.log(`Inside getFormValues credentials = ${JSON.stringify(credentials)}`);
     if (this.state.auth.type === AuthType.UsernamePasswordType) {
       credentials = {
         username: this.state.auth.credentials.username,
@@ -631,6 +652,7 @@ export class CreateDataSourceForm extends React.Component<
               />
             </EuiFormRow>
 
+            {this.getAuthUIFromRegistry(this.state.auth.type)}
             {/* Create New credentials */}
             {this.state.auth.type === AuthType.UsernamePasswordType
               ? this.renderCreateNewCredentialsForm(this.state.auth.type)