Console: Preserve connection template and connection set during Postgres source edits

PR-URL: hasura/graphql-engine-mono#11190
Co-authored-by: Matthew Goodwin <[email protected]>
GitOrigin-RevId: 1bc98e4b8a2da3d55f71b9331a8f8822eee3f8c8

Author: 2 people

frontend/libs/console/legacy-ce/src/lib/features/ConnectDBRedesign/components/ConnectPostgresWidget/ConnectPostgresWidget.tsx

@@ -1,16 +1,34 @@
+import { useEffect, useState } from 'react';
+import { Button } from '../../../../new-components/Button';
 import { useConsoleForm } from '../../../../new-components/Form';
-import { getDefaultValues, PostgresConnectionSchema, schema } from './schema';
-import { useManageDatabaseConnection } from '../../hooks/useManageDatabaseConnection';
+import { Tabs } from '../../../../new-components/Tabs';
 import { hasuraToast } from '../../../../new-components/Toasts';
 import { useMetadata } from '../../../hasura-metadata-api';
-import { generatePostgresRequestPayload } from './utils/generateRequests';
-import { useEffect, useState } from 'react';
-import { LimitedFeatureWrapper } from '../LimitedFeatureWrapper/LimitedFeatureWrapper';
-import { DynamicDBRouting } from './parts/DynamicDBRouting';
-import { Tabs } from '../../../../new-components/Tabs';
+import { Source } from '../../../hasura-metadata-types/source/source';
+import { useManageDatabaseConnection } from '../../hooks/useManageDatabaseConnection';
 import { DisplayToastErrorMessage } from '../Common/DisplayToastErrorMessage';
+import { LimitedFeatureWrapper } from '../LimitedFeatureWrapper/LimitedFeatureWrapper';
 import { ConnectPostgresForm } from './parts/ConnectPostgresForm';
-import { Button } from '../../../../new-components/Button';
+import { DynamicDBRouting } from './parts/DynamicDBRouting';
+import { getDefaultValues, PostgresConnectionSchema, schema } from './schema';
+import { generatePostgresRequestPayload } from './utils/generateRequests';
+
+type PostgresLikeSource = Extract<
+  Source,
+  {
+    kind: 'postgres' | 'citus' | 'cockroach';
+  }
+>;
+
+const isPostgresLikeSource = (
+  source: Source | undefined
+): source is PostgresLikeSource => {
+  return (
+    source?.kind === 'postgres' ||
+    source?.kind === 'citus' ||
+    source?.kind === 'cockroach'
+  );
+};
 
 interface ConnectPostgresWidgetProps {
   dataSourceName?: string;
@@ -53,6 +71,12 @@ export const ConnectPostgresWidget = (props: ConnectPostgresWidgetProps) => {
     const payload = generatePostgresRequestPayload({
       driver: overrideDriver ?? 'postgres',
       values: formValues,
+      connectionTemplate: isPostgresLikeSource(metadataSource)
+        ? metadataSource.configuration.connection_template
+        : undefined,
+      connectionSet: isPostgresLikeSource(metadataSource)
+        ? metadataSource.configuration.connection_set
+        : undefined,
     });
 
     if (isEditMode) {

frontend/libs/console/legacy-ce/src/lib/features/ConnectDBRedesign/components/ConnectPostgresWidget/utils/generateRequests.ts

@@ -1,3 +1,4 @@
+import { PostgresConfiguration } from '../../../../hasura-metadata-types';
 import { DatabaseConnection } from '../../../types';
 import { generateGraphQLCustomizationInfo } from '../../GraphQLCustomization/utils/generateRequest';
 import { PostgresConnectionSchema } from '../schema';
@@ -52,9 +53,13 @@ export const generateConnectionInfo = (
 export const generatePostgresRequestPayload = ({
   driver,
   values,
+  connectionTemplate,
+  connectionSet,
 }: {
   driver: string;
   values: PostgresConnectionSchema;
+  connectionTemplate?: PostgresConfiguration['connection_template'];
+  connectionSet?: PostgresConfiguration['connection_set'];
 }): DatabaseConnection => {
   const payload = {
     driver,
@@ -68,6 +73,8 @@ export const generatePostgresRequestPayload = ({
           generateConnectionInfo(readReplica)
         ),
         extensions_schema: values.configuration.extensionSchema,
+        connection_template: connectionTemplate,
+        connection_set: connectionSet,
       },
       customization: generateGraphQLCustomizationInfo(
         values.customization ?? {}