refactor: simplify int type typing

Author: jankuca

src/notebooks/deepnote/integrations/integrationDetector.ts

@@ -6,7 +6,8 @@ import {
     DATAFRAME_SQL_INTEGRATION_ID,
     DEEPNOTE_TO_INTEGRATION_TYPE,
     IntegrationStatus,
-    IntegrationWithStatus
+    IntegrationWithStatus,
+    RawIntegrationType
 } from '../../../platform/notebooks/deepnote/integrationTypes';
 import { IIntegrationDetector, IIntegrationStorage } from './types';
 
@@ -53,8 +54,7 @@ export class IntegrationDetector implements IIntegrationDetector {
             logger.debug(`IntegrationDetector: Found integration: ${integrationId} (${projectIntegration.type})`);
 
             // Map the Deepnote integration type to our IntegrationType
-            const integrationType =
-                DEEPNOTE_TO_INTEGRATION_TYPE[projectIntegration.type as keyof typeof DEEPNOTE_TO_INTEGRATION_TYPE];
+            const integrationType = DEEPNOTE_TO_INTEGRATION_TYPE[projectIntegration.type as RawIntegrationType];
 
             // Skip unknown integration types
             if (!integrationType) {

src/notebooks/deepnote/sqlCellStatusBarProvider.ts

@@ -25,7 +25,8 @@ import { Commands } from '../../platform/common/constants';
 import {
     DATAFRAME_SQL_INTEGRATION_ID,
     DEEPNOTE_TO_INTEGRATION_TYPE,
-    IntegrationType
+    IntegrationType,
+    RawIntegrationType
 } from '../../platform/notebooks/deepnote/integrationTypes';
 import { IDeepnoteNotebookManager } from '../types';
 
@@ -334,8 +335,7 @@ export class SqlCellStatusBarProvider implements NotebookCellStatusBarItemProvid
                 continue;
             }
 
-            const integrationType =
-                DEEPNOTE_TO_INTEGRATION_TYPE[projectIntegration.type as keyof typeof DEEPNOTE_TO_INTEGRATION_TYPE];
+            const integrationType = DEEPNOTE_TO_INTEGRATION_TYPE[projectIntegration.type as RawIntegrationType];
             const typeLabel = integrationType ? this.getIntegrationTypeLabel(integrationType) : projectIntegration.type;
 
             const item: LocalQuickPickItem = {

src/platform/notebooks/deepnote/integrationTypes.ts

@@ -20,7 +20,7 @@ export const INTEGRATION_TYPE_TO_DEEPNOTE = {
     [IntegrationType.BigQuery]: 'big-query'
 } as const satisfies { [type in IntegrationType]: string };
 
-type RawIntegrationType = (typeof INTEGRATION_TYPE_TO_DEEPNOTE)[keyof typeof INTEGRATION_TYPE_TO_DEEPNOTE];
+export type RawIntegrationType = (typeof INTEGRATION_TYPE_TO_DEEPNOTE)[keyof typeof INTEGRATION_TYPE_TO_DEEPNOTE];
 
 /**
  * Map Deepnote integration type strings to our IntegrationType enum