Set sourceCatalogId when creating a new connection

Author: malikdiarra

airbyte-webapp/src/components/CreateConnectionContent/CreateConnectionContent.tsx

@@ -51,15 +51,16 @@ const CreateConnectionContent: React.FC<IProps> = ({
   const { mutateAsync: createConnection } = useCreateConnection();
   const analyticsService = useAnalyticsService();
 
-  const { schema, isLoading, schemaErrorStatus, onDiscoverSchema } = useDiscoverSchema(source.sourceId);
+  const { schema, isLoading, schemaErrorStatus, catalogId, onDiscoverSchema } = useDiscoverSchema(source.sourceId);
 
   const connection = useMemo(
     () => ({
       syncCatalog: schema,
       destination,
       source,
+      catalogId,
     }),
-    [schema, destination, source]
+    [schema, destination, source, catalogId]
   );
 
   const onSubmitConnectionStep = async (values: ValuesProps) => {
@@ -75,6 +76,7 @@ const CreateConnectionContent: React.FC<IProps> = ({
         name: destination?.name ?? "",
         destinationDefinitionId: destination?.destinationDefinitionId ?? "",
       },
+      sourceCatalogId: catalogId,
     });
 
     if (afterSubmitConnection) {

airbyte-webapp/src/core/domain/catalog/api.ts

@@ -5,6 +5,7 @@ import { JobInfo } from "../job";
 export interface SourceDiscoverSchemaRead {
   catalog: SyncSchema;
   jobInfo?: JobInfo;
+  catalogId: string;
 }
 
 export type SchemaFields = JSONSchema7;

airbyte-webapp/src/core/domain/connector/SourceService.ts

@@ -94,6 +94,7 @@ class SourceService extends AirbyteRequestService {
     return {
       catalog: result.catalog,
       jobInfo: result.jobInfo,
+      catalogId: result.catalogId,
       id: sourceId,
     };
   }

airbyte-webapp/src/hooks/services/useConnectionHook.tsx

@@ -43,6 +43,7 @@ type CreateConnectionProps = {
   destination?: Destination;
   sourceDefinition?: SourceDefinition | { name: string; sourceDefinitionId: string };
   destinationDefinition?: { name: string; destinationDefinitionId: string };
+  sourceCatalogId: string;
 };
 
 type UpdateConnection = {
@@ -139,12 +140,13 @@ const useCreateConnection = () => {
 
   return useMutation(
     async (conn: CreateConnectionProps) => {
-      const { values, source, destination, sourceDefinition, destinationDefinition } = conn;
+      const { values, source, destination, sourceDefinition, destinationDefinition, sourceCatalogId } = conn;
       const response = await service.create({
         sourceId: source?.sourceId,
         destinationId: destination?.destinationId,
         ...values,
         status: "active",
+        sourceCatalogId: sourceCatalogId,
       });
 
       const frequencyData = FrequencyConfig.find((item) => equal(item.config, values.schedule));

airbyte-webapp/src/hooks/services/useSourceHook.tsx

@@ -175,10 +175,12 @@ const useDiscoverSchema = (
   isLoading: boolean;
   schema: SyncSchema;
   schemaErrorStatus: { status: number; response: JobInfo } | null;
+  catalogId: string;
   onDiscoverSchema: () => Promise<void>;
 } => {
   const service = useSourceService();
   const [schema, setSchema] = useState<SyncSchema>({ streams: [] });
+  const [catalogId, setCatalogId] = useState<string>("");
   const [isLoading, setIsLoading] = useState(false);
   const [schemaErrorStatus, setSchemaErrorStatus] = useState<{
     status: number;
@@ -191,6 +193,7 @@ const useDiscoverSchema = (
     try {
       const data = await service.discoverSchema(sourceId || "");
       setSchema(data.catalog);
+      setCatalogId(data.catalogId);
     } catch (e) {
       setSchemaErrorStatus(e);
     } finally {
@@ -207,7 +210,7 @@ const useDiscoverSchema = (
     })();
   }, [onDiscoverSchema, sourceId]);
 
-  return { schemaErrorStatus, isLoading, schema, onDiscoverSchema };
+  return { schemaErrorStatus, isLoading, schema, catalogId, onDiscoverSchema };
 };
 
 export { useSourceList, useGetSource, useCreateSource, useDeleteSource, useUpdateSource, useDiscoverSchema };