Skip to content

Commit edbc658

Browse files
committed
Use tryGetQueryLanguage function
1 parent 322b376 commit edbc658

File tree

1 file changed

+9
-11
lines changed

1 file changed

+9
-11
lines changed

extensions/ql-vscode/src/model-editor/method-modeling/method-modeling-view-provider.ts

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,10 @@ import { ModelEditorViewTracker } from "../model-editor-view-tracker";
1515
import { ModelConfigListener } from "../../config";
1616
import { DatabaseItem } from "../../databases/local-databases";
1717
import { ModelingEvents } from "../modeling-events";
18-
import { isQueryLanguage, QueryLanguage } from "../../common/query-language";
18+
import {
19+
QueryLanguage,
20+
tryGetQueryLanguage,
21+
} from "../../common/query-language";
1922

2023
export class MethodModelingViewProvider extends AbstractWebviewViewProvider<
2124
ToMethodModelingMessage,
@@ -59,10 +62,7 @@ export class MethodModelingViewProvider extends AbstractWebviewViewProvider<
5962
): Promise<void> {
6063
this.method = method;
6164
this.databaseItem = databaseItem;
62-
this.language =
63-
databaseItem && isQueryLanguage(databaseItem?.language)
64-
? databaseItem.language
65-
: undefined;
65+
this.language = databaseItem && tryGetQueryLanguage(databaseItem.language);
6666

6767
if (this.isShowingView) {
6868
await this.postMessage({
@@ -77,9 +77,9 @@ export class MethodModelingViewProvider extends AbstractWebviewViewProvider<
7777
const selectedMethod = this.modelingStore.getSelectedMethodDetails();
7878
if (selectedMethod) {
7979
this.databaseItem = selectedMethod.databaseItem;
80-
this.language = isQueryLanguage(selectedMethod.databaseItem.language)
81-
? selectedMethod.databaseItem.language
82-
: undefined;
80+
this.language = tryGetQueryLanguage(
81+
selectedMethod.databaseItem.language,
82+
);
8383
this.method = selectedMethod.method;
8484

8585
await this.postMessage({
@@ -195,9 +195,7 @@ export class MethodModelingViewProvider extends AbstractWebviewViewProvider<
195195
if (this.webviewView) {
196196
this.method = e.method;
197197
this.databaseItem = e.databaseItem;
198-
this.language = isQueryLanguage(e.databaseItem.language)
199-
? e.databaseItem.language
200-
: undefined;
198+
this.language = tryGetQueryLanguage(e.databaseItem.language);
201199

202200
await this.postMessage({
203201
t: "setSelectedMethod",

0 commit comments

Comments
 (0)