Rename 'open database' button and only show if source available

Author: charisk

extensions/ql-vscode/src/model-editor/model-editor-view.ts

@@ -5,6 +5,7 @@ import {
   Uri,
   ViewColumn,
   window,
+  workspace,
 } from "vscode";
 import {
   AbstractWebview,
@@ -353,6 +354,11 @@ export class ModelEditorView extends AbstractWebview<
     const showLlmButton =
       this.databaseItem.language === "java" && this.modelConfig.llmGeneration;
 
+    const sourceArchiveAvailable = workspace.workspaceFolders?.some(
+      (f) =>
+        f.uri.fsPath === this.databaseItem.getSourceArchiveExplorerUri().fsPath,
+    );
+
     await this.postMessage({
       t: "setModelEditorViewState",
       viewState: {
@@ -361,6 +367,7 @@ export class ModelEditorView extends AbstractWebview<
         showLlmButton,
         showMultipleModels: this.modelConfig.showMultipleModels,
         mode: this.mode,
+        sourceArchiveAvailable: !!sourceArchiveAvailable,
       },
     });
   }

extensions/ql-vscode/src/model-editor/shared/view-state.ts

@@ -7,6 +7,7 @@ export interface ModelEditorViewState {
   showLlmButton: boolean;
   showMultipleModels: boolean;
   mode: Mode;
+  sourceArchiveAvailable: boolean;
 }
 
 export interface MethodModelingPanelViewState {

extensions/ql-vscode/src/view/model-editor/ModelEditor.tsx

@@ -282,10 +282,12 @@ export function ModelEditor({
             <>{viewState.extensionPack.name}</>
           </HeaderRow>
           <HeaderRow>
-            <LinkIconButton onClick={onOpenDatabaseClick}>
-              <span slot="start" className="codicon codicon-package"></span>
-              Open database
-            </LinkIconButton>
+            {viewState.sourceArchiveAvailable && (
+              <LinkIconButton onClick={onOpenDatabaseClick}>
+                <span slot="start" className="codicon codicon-package"></span>
+                Open source
+              </LinkIconButton>
+            )}
             <LinkIconButton onClick={onOpenExtensionPackClick}>
               <span slot="start" className="codicon codicon-package"></span>
               Open extension pack

extensions/ql-vscode/src/view/model-editor/__tests__/LibraryRow.spec.tsx

@@ -22,6 +22,7 @@ describe(LibraryRow.name, () => {
     showLlmButton: false,
     showMultipleModels: false,
     extensionPack: createMockExtensionPack(),
+    sourceArchiveAvailable: true,
   };
 
   const render = (props: Partial<LibraryRowProps> = {}) =>

extensions/ql-vscode/src/view/model-editor/__tests__/MethodRow.spec.tsx

@@ -39,6 +39,7 @@ describe(MethodRow.name, () => {
     showLlmButton: false,
     showMultipleModels: false,
     extensionPack: createMockExtensionPack(),
+    sourceArchiveAvailable: true,
   };
 
   const render = (props: Partial<MethodRowProps> = {}) =>

extensions/ql-vscode/src/view/model-editor/__tests__/ModeledMethodDataGrid.spec.tsx

@@ -49,6 +49,7 @@ describe(ModeledMethodDataGrid.name, () => {
     showLlmButton: false,
     showMultipleModels: false,
     extensionPack: createMockExtensionPack(),
+    sourceArchiveAvailable: true,
   };
 
   const render = (props: Partial<ModeledMethodDataGridProps> = {}) =>

extensions/ql-vscode/src/view/model-editor/__tests__/ModeledMethodsList.spec.tsx

@@ -50,6 +50,7 @@ describe(ModeledMethodsList.name, () => {
     showLlmButton: false,
     showMultipleModels: false,
     extensionPack: createMockExtensionPack(),
+    sourceArchiveAvailable: true,
   };
 
   const render = (props: Partial<ModeledMethodsListProps> = {}) =>