add reset schemaAnalysis and fix typing

Author: ncarbon

packages/compass-collection/src/modules/collection-tab.ts

@@ -21,7 +21,7 @@ import {
   type SchemaAnalysisState,
 } from '../schema-analysis-types';
 import { calculateSchemaDepth } from '../calculate-schema-depth';
-import type { MongoError } from 'mongodb';
+import type { Document, MongoError } from 'mongodb';
 
 const DEFAULT_SAMPLE_SIZE = 100;
 
@@ -79,13 +79,18 @@ enum CollectionActions {
   SchemaAnalysisStarted = 'compass-collection/SchemaAnalysisStarted',
   SchemaAnalysisFinished = 'compass-collection/SchemaAnalysisFinished',
   SchemaAnalysisFailed = 'compass-collection/SchemaAnalysisFailed',
+  SchemaAnalysisReset = 'compass-collection/SchemaAnalysisReset',
 }
 
 interface CollectionMetadataFetchedAction {
   type: CollectionActions.CollectionMetadataFetched;
   metadata: CollectionMetadata;
 }
 
+interface SchemaAnalysisResetAction {
+  type: CollectionActions.SchemaAnalysisReset;
+}
+
 interface SchemaAnalysisStartedAction {
   type: CollectionActions.SchemaAnalysisStarted;
 }
@@ -129,6 +134,20 @@ const reducer: Reducer<CollectionState, Action> = (
     };
   }
 
+  if (
+    isAction<SchemaAnalysisResetAction>(
+      action,
+      CollectionActions.SchemaAnalysisReset
+    )
+  ) {
+    return {
+      ...state,
+      schemaAnalysis: {
+        status: SCHEMA_ANALYSIS_STATE_INITIAL,
+      },
+    };
+  }
+
   if (
     isAction<SchemaAnalysisStartedAction>(
       action,
@@ -139,10 +158,6 @@ const reducer: Reducer<CollectionState, Action> = (
       ...state,
       schemaAnalysis: {
         status: SCHEMA_ANALYSIS_STATE_ANALYZING,
-        error: null,
-        schema: null,
-        sampleDocument: null,
-        schemaMetadata: null,
       },
     };
   }

packages/compass-collection/src/schema-analysis-types.ts

@@ -1,3 +1,4 @@
+import type { Document } from 'mongodb';
 import { type Schema } from 'mongodb-schema';
 
 export const SCHEMA_ANALYSIS_STATE_INITIAL = 'initial';
@@ -40,7 +41,7 @@ export type SchemaAnalysisCompletedState = {
 };
 
 export type SchemaAnalysisState =
-  | SchemaAnalysisErrorState
   | SchemaAnalysisInitialState
   | SchemaAnalysisStartedState
-  | SchemaAnalysisCompletedState;
+  | SchemaAnalysisCompletedState
+  | SchemaAnalysisErrorState;