refactor: drop utils and use records only

Author: jankuca

src/notebooks/deepnote/integrations/integrationDetector.ts

@@ -4,9 +4,9 @@ import { logger } from '../../../platform/logging';
 import { IDeepnoteNotebookManager } from '../../types';
 import {
     DATAFRAME_SQL_INTEGRATION_ID,
+    DEEPNOTE_TO_INTEGRATION_TYPE,
     IntegrationStatus,
-    IntegrationWithStatus,
-    mapDeepnoteIntegrationType
+    IntegrationWithStatus
 } from '../../../platform/notebooks/deepnote/integrationTypes';
 import { IIntegrationDetector, IIntegrationStorage } from './types';
 
@@ -56,7 +56,7 @@ export class IntegrationDetector implements IIntegrationDetector {
             const config = await this.integrationStorage.getIntegrationConfig(integrationId);
 
             // Map the Deepnote integration type to our IntegrationType
-            const integrationType = mapDeepnoteIntegrationType(projectIntegration.type);
+            const integrationType = DEEPNOTE_TO_INTEGRATION_TYPE[projectIntegration.type];
 
             const status: IntegrationWithStatus = {
                 config: config || null,

src/notebooks/deepnote/integrations/integrationWebview.ts

@@ -8,10 +8,10 @@ import { LocalizedMessages, SharedMessages } from '../../../messageTypes';
 import { IDeepnoteNotebookManager } from '../../types';
 import { IIntegrationStorage, IIntegrationWebviewProvider } from './types';
 import {
+    INTEGRATION_TYPE_TO_DEEPNOTE,
     IntegrationConfig,
     IntegrationStatus,
-    IntegrationWithStatus,
-    mapToDeepnoteIntegrationType
+    IntegrationWithStatus
 } from '../../../platform/notebooks/deepnote/integrationTypes';
 
 /**
@@ -321,7 +321,7 @@ export class IntegrationWebviewProvider implements IIntegrationWebviewProvider {
                 }
 
                 // Map to Deepnote integration type
-                const deepnoteType = mapToDeepnoteIntegrationType(type);
+                const deepnoteType = INTEGRATION_TYPE_TO_DEEPNOTE[type];
                 if (!deepnoteType) {
                     logger.warn(`IntegrationWebviewProvider: Cannot map type ${type} for integration ${id}, skipping`);
                     return null;

src/notebooks/deepnote/sqlCellStatusBarProvider.ts

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

src/platform/notebooks/deepnote/integrationTypes.ts

@@ -15,27 +15,19 @@ export enum IntegrationType {
 /**
  * Map Deepnote integration type strings to our IntegrationType enum
  */
-const DEEPNOTE_TO_INTEGRATION_TYPE: Record<string, IntegrationType> = {
+export const DEEPNOTE_TO_INTEGRATION_TYPE: Record<string, IntegrationType> = {
     pgsql: IntegrationType.Postgres,
     'big-query': IntegrationType.BigQuery
 };
 
-export function mapDeepnoteIntegrationType(deepnoteType: string): IntegrationType | undefined {
-    return DEEPNOTE_TO_INTEGRATION_TYPE[deepnoteType];
-}
-
 /**
  * Map our IntegrationType enum to Deepnote integration type strings
  */
-const INTEGRATION_TYPE_TO_DEEPNOTE: Record<IntegrationType, string> = {
+export const INTEGRATION_TYPE_TO_DEEPNOTE: Record<IntegrationType, string> = {
     [IntegrationType.Postgres]: 'pgsql',
     [IntegrationType.BigQuery]: 'big-query'
 };
 
-export function mapToDeepnoteIntegrationType(type: IntegrationType): string {
-    return INTEGRATION_TYPE_TO_DEEPNOTE[type];
-}
-
 /**
  * Base interface for all integration configurations
  */