dash: Fix the indexes pane state for tables being added to the schema (#41611)

GitOrigin-RevId: 3f54b431aa596528bcee6fcd743de80462ff9b9c

Author: Nicolapps

npm-packages/dashboard-common/src/features/data/components/DataToolbar/DataToolbar.test.tsx

@@ -127,6 +127,7 @@ describe("DataToolbar", () => {
           tableSchemaStatus={{
             tableName,
             isDefined: false,
+            isDefinedInInProgressSchema: false,
             referencedByTable: undefined,
             isValidationRunning: false,
           }}
@@ -420,6 +421,7 @@ describe("DataToolbar", () => {
         tableSchemaStatus: {
           tableName: "messages",
           isDefined: false,
+          isDefinedInInProgressSchema: false,
           isValidationRunning: true,
           referencedByTable: undefined,
         },
@@ -435,6 +437,7 @@ describe("DataToolbar", () => {
         tableSchemaStatus: {
           tableName: "messages",
           isDefined: true,
+          isDefinedInInProgressSchema: false,
           isValidationRunning: false,
           referencedByTable: undefined,
         },
@@ -450,6 +453,7 @@ describe("DataToolbar", () => {
         tableSchemaStatus: {
           tableName: "messages",
           isDefined: false,
+          isDefinedInInProgressSchema: false,
           isValidationRunning: false,
           referencedByTable: "users",
         },
@@ -468,6 +472,7 @@ describe("DataToolbar", () => {
         tableSchemaStatus: {
           tableName: "messages",
           isDefined: false,
+          isDefinedInInProgressSchema: false,
           isValidationRunning: false,
           referencedByTable: undefined,
         },
@@ -502,6 +507,7 @@ describe("DataToolbar", () => {
         tableSchemaStatus: {
           tableName: "messages",
           isDefined: false,
+          isDefinedInInProgressSchema: false,
           isValidationRunning: false,
           referencedByTable: undefined,
         },

npm-packages/dashboard-common/src/features/data/components/TableIndexesPanel.tsx

@@ -34,7 +34,8 @@ function IndexBody({ tableName }: { tableName: string }) {
 
   return (
     <div className="grow p-4 pt-0 sm:p-6">
-      {tableSchemaStatus.isDefined ? (
+      {tableSchemaStatus.isDefined ||
+      tableSchemaStatus.isDefinedInInProgressSchema ? (
         <IndexList tableName={tableName} />
       ) : (
         <>

npm-packages/dashboard-common/src/features/data/components/TableSchema.tsx

@@ -12,6 +12,7 @@ import { ShowSchema } from "@common/features/data/components/ShowSchema";
 export interface TableSchemaStatus {
   tableName: string;
   isDefined: boolean;
+  isDefinedInInProgressSchema: boolean;
   referencedByTable: string | undefined;
   isValidationRunning: boolean;
 }
@@ -95,11 +96,23 @@ export function useSingleTableSchemaStatus(
     };
   }, [schemas?.active, tableName]);
 
+  const isDefinedInInProgressSchema = useMemo(() => {
+    const inProgress: Schema | undefined = schemas?.inProgress
+      ? JSON.parse(schemas.inProgress)
+      : undefined;
+
+    return (
+      inProgress?.tables.find((table) => table.tableName === tableName) !==
+      undefined
+    );
+  }, [schemas?.inProgress, tableName]);
+
   const isValidationRunning = schemas?.inProgress !== undefined;
   return schemas
     ? {
         tableName,
         isDefined,
+        isDefinedInInProgressSchema,
         referencedByTable,
         isValidationRunning,
       }