fix(migrate): 修复版本升级后数据旧版本保存数据丢失问题

Author: czfadmin

src/commands/bookmark/base.ts

@@ -5,7 +5,6 @@ import {
   QuickPickItem,
   workspace,
   Uri,
-  TextEditorRevealType,
   commands,
   Position,
 } from 'vscode';
@@ -34,7 +33,6 @@ import {
   getBookmarkColorFromCtx,
   toggleBookmarksWithSelections,
   highlightSelection,
-  openDocumentAndGotoLocation,
 } from '../../utils';
 import {range} from 'lodash';
 

src/commands/bookmark/gutter.ts

@@ -163,10 +163,6 @@ export default class BookmarksController implements IController {
   private async _initial() {
     this._needWarning = this.configService.getGlobalValue('_needWarning', true);
 
-    // this.configService.onExtensionConfigChange(configuration => {
-    //   this._configuration = configuration;
-    // });
-
     this.configService.onDidChangeConfiguration(ev => {
       if (!ev.affectsConfiguration(`${EXTENSION_ID}.createJsonFile`)) {
         return;
@@ -203,7 +199,7 @@ export default class BookmarksController implements IController {
           store = this._store;
         }
 
-        applySnapshot(this._store, isProxy(store) ? getSnapshot(store) : store);
+        applySnapshot(this._store, isProxy(store) ? getSnapshot(store) : this._sm.migrateService.migrate(store));
 
         if (!this._store.groups.length) {
           this._store.addGroups([]);
@@ -611,7 +607,9 @@ export default class BookmarksController implements IController {
           });
         }
 
-        const storeContent = JSON.parse(content) as IBookmarkStoreInfo;
+        let storeContent = JSON.parse(content) as IBookmarkStoreInfo;
+
+        storeContent = this._sm.migrateService.migrate(storeContent);
 
         applySnapshot(this._store, {
           ...storeContent,

src/controllers/BookmarksController.ts

@@ -1,3 +1,4 @@
+import {IBookmarkStoreInfo} from '../types';
 import {BaseService} from './BaseService';
 import {ServiceManager} from './ServiceManager';
 
@@ -8,6 +9,29 @@ export class MigrateService extends BaseService {
   constructor(sm: ServiceManager) {
     super(MigrateService.name, sm);
   }
+
+  /**
+   * @zh 对之前版本的书签保存的内容进行迁移工作
+   * @param data 
+   * @returns 
+   */
+  migrate(data: IBookmarkStoreInfo) {
+    if (!data.version ||  data.version <= '0.0.36') {
+      data.bookmarks = data.bookmarks.map(it => {
+        return {
+          ...it,
+          sortedInfo: {
+            ...it.sortedInfo,
+            icon: 0,
+          },
+        };
+      });
+      if (!data.version) {
+        data.version = process.env.version!
+      }
+    }
+    return data;
+  }
   initial(): void {}
   dispose(): void {}
 }

src/services/MigrateService.ts

@@ -11,6 +11,7 @@ import {createGlobalStore, GlobalStore} from '../stores';
 import {IconsService} from './IconsService';
 import ColorsService from './ColorsService';
 import {Instance} from 'mobx-state-tree';
+import { MigrateService } from './MigrateService';
 
 export interface IServiceManager {
   readonly configService: ConfigService;
@@ -22,6 +23,8 @@ export interface IServiceManager {
   readonly iconsService: IconsService;
   readonly colorsService: ColorsService;
   readonly fileService: FileService;
+
+  readonly migrateService: MigrateService
 }
 
 export class ServiceManager implements IServiceManager, IDisposable {
@@ -34,6 +37,8 @@ export class ServiceManager implements IServiceManager, IDisposable {
   readonly iconsService: IconsService;
 
   readonly colorsService: ColorsService;
+
+  readonly migrateService: MigrateService
   private _statusbarService: StatusbarService | undefined;
 
   public readonly fileService: FileService;
@@ -82,6 +87,7 @@ export class ServiceManager implements IServiceManager, IDisposable {
     this.configService = new ConfigService(this);
     this.colorsService = new ColorsService(this);
     this.gutterService = new GutterService(this);
+    this.migrateService = new MigrateService(this)
     this.decorationService = new DecorationService(this);
     this.workspaceService = new WorkspaceService(this);
     this.gitService = new GitService(this);

src/services/ServiceManager.ts

@@ -85,6 +85,7 @@ export const BookmarksStore = types
     groups: types.array(BookmarkGroup),
 
     groupInfo: types.array(BookmarkGroupInfo),
+    version: types.optional(types.string, process.env.version || ""),
   })
   .views(self => {
     return {