refactor: righten integration type mappings

Author: jankuca

src/notebooks/deepnote/integrations/integrationDetector.ts

@@ -53,7 +53,8 @@ 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];
+            const integrationType =
+                DEEPNOTE_TO_INTEGRATION_TYPE[projectIntegration.type as keyof typeof DEEPNOTE_TO_INTEGRATION_TYPE];
 
             // Skip unknown integration types
             if (!integrationType) {

src/notebooks/deepnote/sqlCellStatusBarProvider.ts

@@ -334,7 +334,8 @@ export class SqlCellStatusBarProvider implements NotebookCellStatusBarItemProvid
                 continue;
             }
 
-            const integrationType = DEEPNOTE_TO_INTEGRATION_TYPE[projectIntegration.type];
+            const integrationType =
+                DEEPNOTE_TO_INTEGRATION_TYPE[projectIntegration.type as keyof typeof DEEPNOTE_TO_INTEGRATION_TYPE];
             const typeLabel = integrationType ? this.getIntegrationTypeLabel(integrationType) : projectIntegration.type;
 
             const item: LocalQuickPickItem = {

src/platform/notebooks/deepnote/integrationTypes.ts

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