fix(compass-user-data): remove divergent user data types from atlas implementation CLOUDP-334901

Author: nbbeeken

packages/atlas-service/src/atlas-service.ts

@@ -9,6 +9,7 @@ import {
 import type { Logger } from '@mongodb-js/compass-logging';
 import type { PreferencesAccess } from 'compass-preferences-model';
 import type { AtlasClusterMetadata } from '@mongodb-js/connection-info';
+import { type UserDataType } from '@mongodb-js/compass-user-data';
 
 export type AtlasServiceOptions = {
   defaultHeaders?: Record<string, string>;
@@ -81,12 +82,7 @@ export class AtlasService {
   userDataEndpoint(
     orgId: string,
     groupId: string,
-    type:
-      | 'favoriteQueries'
-      | 'recentQueries'
-      | 'favoriteAggregations'
-      | 'savedWorkspaces'
-      | 'dataModelDescriptions',
+    type: UserDataType,
     id?: string
   ): string {
     const encodedOrgId = encodeURIComponent(orgId);

packages/compass-data-modeling/src/services/data-model-storage-web.tsx

@@ -21,7 +21,7 @@ class DataModelStorageAtlas implements DataModelStorage {
   constructor(orgId: string, projectId: string, atlasService: AtlasService) {
     this.userData = new AtlasUserData(
       MongoDBDataModelDescriptionSchema,
-      'dataModelDescriptions',
+      'DataModelDescriptions',
       {
         orgId,
         projectId,
@@ -37,7 +37,7 @@ class DataModelStorageAtlas implements DataModelStorage {
             !type ||
             !pathOrgId ||
             !pathProjectId ||
-            type !== 'dataModelDescriptions'
+            type !== 'DataModelDescriptions'
           ) {
             throw new Error(
               'DataModelStorageAtlas is used outside of Atlas Cloud context'

packages/compass-shell/src/modules/history-storage.spec.ts

@@ -6,14 +6,14 @@ import fs from 'fs/promises';
 import { HistoryStorage } from './history-storage';
 
 describe('HistoryStorage', function () {
-  let tmpDir;
-  let historyFilePath;
+  let tmpDir: string;
+  let historyFilePath: string;
 
-  let historyStorage;
+  let historyStorage: HistoryStorage;
 
   beforeEach(async function () {
     tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), 'compass-shell-test'));
-    historyFilePath = path.join(tmpDir, 'shell-history.json');
+    historyFilePath = path.join(tmpDir, 'ShellHistory', 'shell-history.json');
 
     historyStorage = new HistoryStorage(tmpDir);
   });

packages/compass-shell/src/modules/history-storage.ts

@@ -1,17 +1,34 @@
-import { getAppName } from '@mongodb-js/compass-utils';
 import { FileUserData, z } from '@mongodb-js/compass-user-data';
+import { getAppName } from '@mongodb-js/compass-utils';
 
 export class HistoryStorage {
   fileName = 'shell-history';
   userData;
+  private migrationChecked = false;
 
   constructor(basePath?: string) {
     // TODO: https://jira.mongodb.org/browse/COMPASS-7080
-    this.userData = new FileUserData(z.string().array(), getAppName() ?? '', {
+    this.userData = new FileUserData(z.string().array(), 'ShellHistory', {
       basePath,
     });
   }
 
+  /**
+   * Migrates history from old app-name-based folder to new ShellHistory folder.
+   * Only runs once per instance and only if old folder exists.
+   */
+  private async migrateIfNeeded(): Promise<void> {
+    if (this.migrationChecked) {
+      return;
+    }
+    this.migrationChecked = true;
+
+    const oldAppName = getAppName();
+    if (oldAppName && oldAppName !== 'ShellHistory') {
+      await this.userData.migrateFromOldFolder(oldAppName);
+    }
+  }
+
   /**
    * Saves the history to disk, it creates the directory and the file if
    * not existing and replaces the file content.
@@ -20,6 +37,7 @@ export class HistoryStorage {
    * newest to oldest.
    */
   async save(history: string[]) {
+    await this.migrateIfNeeded();
     await this.userData.write(this.fileName, history);
   }
 
@@ -31,6 +49,7 @@ export class HistoryStorage {
    * newest to oldest.
    */
   async load(): Promise<string[]> {
+    await this.migrateIfNeeded();
     try {
       return (await this.userData.readOne(this.fileName)) ?? [];
     } catch {

packages/compass-user-data/src/index.ts

@@ -1,3 +1,8 @@
-export type { ReadAllResult } from './user-data';
-export { IUserData, FileUserData, AtlasUserData } from './user-data';
+export type { ReadAllResult, UserDataType } from './user-data';
+export {
+  IUserData,
+  FileUserData,
+  AtlasUserData,
+  assertsUserDataType,
+} from './user-data';
 export { z } from 'zod';