Fix ER diagram & refresh improvements

Author: badsyntax

package.json

@@ -72,22 +72,6 @@
           "dark": "icons/files_dark.svg"
         }
       },
-      {
-        "command": "entityframework.setStartupProject",
-        "title": "Set as Startup Project",
-        "icon": {
-          "light": "icons/files_light.svg",
-          "dark": "icons/files_dark.svg"
-        }
-      },
-      {
-        "command": "entityframework.setProject",
-        "title": "Set as Project",
-        "icon": {
-          "light": "icons/files_light.svg",
-          "dark": "icons/files_dark.svg"
-        }
-      },
       {
         "command": "entityframework.generateERD",
         "title": "Generate ER Diagram",
@@ -112,6 +96,14 @@
           "dark": "icons/refresh_dark.svg"
         }
       },
+      {
+        "command": "entityframework.refreshTreeNoCache",
+        "title": "Refresh",
+        "icon": {
+          "light": "icons/refresh_light.svg",
+          "dark": "icons/refresh_dark.svg"
+        }
+      },
       {
         "command": "entityframework.refreshDbContextTree",
         "title": "Refresh",
@@ -175,6 +167,10 @@
           "command": "entityframework.refreshTree",
           "when": "false"
         },
+        {
+          "command": "entityframework.refreshTreeNoCache",
+          "when": "false"
+        },
         {
           "command": "entityframework.refreshProjectTree",
           "when": "false"
@@ -227,7 +223,7 @@
           "group": "navigation"
         },
         {
-          "command": "entityframework.refreshTree",
+          "command": "entityframework.refreshTreeNoCache",
           "when": "view == entityframeworkTree",
           "group": "navigation"
         }
@@ -308,16 +304,6 @@
           "command": "entityframework.refreshProjectTree",
           "when": "viewItem == project",
           "group": "context@2"
-        },
-        {
-          "command": "entityframework.setProject",
-          "when": "viewItem == project",
-          "group": "context@3"
-        },
-        {
-          "command": "entityframework.setStartupProject",
-          "when": "viewItem == project",
-          "group": "context@4"
         }
       ]
     },

sample_dotnet/src/Infrastructure/blogging.db

@@ -74,10 +74,7 @@ export class AddMigrationAction extends TerminalAction {
           'vscode.open',
           vscode.Uri.file(parsedData.migrationFile),
         );
-        await vscode.commands.executeCommand(
-          CommandProvider.getCommandName(RefreshTreeCommand.commandName),
-          false,
-        );
+
         return output;
       },
     );

src/actions/AddMigrationAction.ts

@@ -2,19 +2,10 @@ import * as vscode from 'vscode';
 import { getProjectsConfig } from '../config/config';
 import { EXTENSION_NAMESPACE } from '../constants/constants';
 import { ProjectFilesProvider } from '../solution/ProjectFilesProvider';
-
 import { InputWizard } from '../util/InputWizard';
-
 import type { IAction } from './IAction';
 
-export type ScaffoldResult = {
-  contextFile: string;
-  entityTypeFiles: string[];
-};
-
 export class ConfigureAction implements IAction {
-  constructor() {}
-
   public async run() {
     const { projectFiles } = await ProjectFilesProvider.getProjectFiles();
 
@@ -32,7 +23,7 @@ export class ConfigureAction implements IAction {
         items: projects,
         value: configProject,
         options: {
-          title: 'Select Project (1/2)',
+          title: 'Select Project',
           ignoreFocusOut: true,
         },
         required: true,
@@ -42,7 +33,7 @@ export class ConfigureAction implements IAction {
         items: projects,
         value: configStartupProject,
         options: {
-          title: 'Select Startup Project (2/2)',
+          title: 'Select Startup Project',
           ignoreFocusOut: true,
         },
         required: true,

src/actions/ConfigureAction.ts

@@ -3,7 +3,6 @@ import * as vscode from 'vscode';
 import { EFOutputParser } from '../cli/EFOutputParser';
 import { getCommandsConfig } from '../config/config';
 import { TREE_VIEW_ID } from '../constants/constants';
-
 import type { TerminalProvider } from '../terminal/TerminalProvider';
 import { TextDocumentProvider } from '../util/TextDocumentProvider';
 import { TerminalAction } from './TerminalAction';

src/actions/DBContextInfoCommandAction.ts

@@ -107,6 +107,7 @@ export class GenerateERDAction extends TerminalAction {
         options: {
           title: 'Connection String',
           value: '',
+          prompt: 'EG: Data Source=blogging.db',
         },
         required: true,
       },

src/actions/GenerateERDAction.ts

@@ -4,5 +4,5 @@ export interface IAction {
   run(
     params?: { [key: string]: string },
     execArgs?: Partial<ExecOpts>,
-  ): Promise<string | void>;
+  ): Promise<string | number | void>;
 }

src/actions/IAction.ts

@@ -2,9 +2,12 @@ import type { TreeDataProvider } from '../treeView/TreeDataProvider';
 import type { IAction } from './IAction';
 
 export class RefreshTreeAction implements IAction {
-  constructor(private readonly treeDataProvider: TreeDataProvider) {}
+  constructor(
+    private readonly treeDataProvider: TreeDataProvider,
+    private readonly clearCache: boolean,
+  ) {}
 
   public async run() {
-    this.treeDataProvider.refresh();
+    this.treeDataProvider.refresh(this.clearCache);
   }
 }

src/actions/RefreshTreeAction.ts

@@ -1,22 +1,15 @@
 import * as vscode from 'vscode';
-import { CommandProvider } from '../commands/CommandProvider';
-import { RefreshTreeCommand } from '../commands/RefreshTreeCommand';
 import { getCommandsConfig } from '../config/config';
 import { TREE_VIEW_ID } from '../constants/constants';
 import type { TerminalProvider } from '../terminal/TerminalProvider';
-import {
-  dbContextsCache,
-  DbContextTreeItem,
-} from '../treeView/DbContextTreeItem';
 import { TerminalAction } from './TerminalAction';
 
 export class RemoveMigrationAction extends TerminalAction {
   constructor(
     terminalProvider: TerminalProvider,
-    private readonly workspaceRoot: string,
-    private readonly dbContext: string,
-    private readonly project: string,
-    private readonly refresh?: boolean,
+    readonly workspaceRoot: string,
+    readonly dbContext: string,
+    readonly project: string,
   ) {
     super(
       terminalProvider,
@@ -41,21 +34,7 @@ export class RemoveMigrationAction extends TerminalAction {
           this.cancel();
         });
         await this.start();
-        const output = await this.getOutput();
-        const cacheId = DbContextTreeItem.getCacheId(
-          this.workspaceRoot,
-          this.project,
-          this.dbContext,
-        );
-        dbContextsCache.clear(cacheId);
-        const refresh = this.refresh || this.refresh === undefined;
-        if (refresh) {
-          await vscode.commands.executeCommand(
-            CommandProvider.getCommandName(RefreshTreeCommand.commandName),
-            false,
-          );
-        }
-        return output;
+        return await this.getOutput();
       },
     );
   }

src/actions/RemoveMigrationAction.ts

@@ -1,6 +1,3 @@
-import * as vscode from 'vscode';
-import { CommandProvider } from '../commands/CommandProvider';
-import { RefreshTreeCommand } from '../commands/RefreshTreeCommand';
 import type { TerminalProvider } from '../terminal/TerminalProvider';
 import {
   dbContextsCache,
@@ -38,14 +35,8 @@ export class RemoveMigrationsAction implements IAction {
         this.workspaceRoot,
         this.dbContext,
         this.project,
-        false,
       ).run();
     }
-    if (migrationsToRemove.length > 0) {
-      await vscode.commands.executeCommand(
-        CommandProvider.getCommandName(RefreshTreeCommand.commandName),
-        false,
-      );
-    }
+    return migrationsToRemove.length;
   }
 }