feat: connect model generator to UI

5 files changed, 130 insertions, 1 deletions

diff --git a/subprojects/frontend/src/xtext/ModelGenerationService.ts b/subprojects/frontend/src/xtext/ModelGenerationService.ts
new file mode 100644
index 00000000..1e9f837a
--- /dev/null
+++ b/subprojects/frontend/src/xtext/ModelGenerationService.ts
@@ -0,0 +1,46 @@
+/*
+ * SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ */
+
+import type EditorStore from '../editor/EditorStore';
+
+import type UpdateService from './UpdateService';
+import { ModelGenerationResult } from './xtextServiceResults';
+
+export default class ModelGenerationService {
+  constructor(
+    private readonly store: EditorStore,
+    private readonly updateService: UpdateService,
+  ) {}
+
+  onPush(push: unknown): void {
+    const result = ModelGenerationResult.parse(push);
+    if ('status' in result) {
+      this.store.setGeneratedModelMessage(result.uuid, result.status);
+    } else if ('error' in result) {
+      this.store.setGeneratedModelError(result.uuid, result.error);
+    } else {
+      this.store.setGeneratedModelSemantics(result.uuid, result);
+    }
+  }
+
+  onDisconnect(): void {
+    this.store.modelGenerationCancelled();
+  }
+
+  async start(): Promise<void> {
+    const result = await this.updateService.startModelGeneration();
+    if (!result.cancelled) {
+      this.store.addGeneratedModel(result.data.uuid);
+    }
+  }
+
+  async cancel(): Promise<void> {
+    const result = await this.updateService.cancelModelGeneration();
+    if (!result.cancelled) {
+      this.store.modelGenerationCancelled();
+    }
+  }
+}
diff --git a/subprojects/frontend/src/xtext/UpdateService.ts b/subprojects/frontend/src/xtext/UpdateService.ts
index 1ac722e1..d1246d5e 100644
--- a/subprojects/frontend/src/xtext/UpdateService.ts
+++ b/subprojects/frontend/src/xtext/UpdateService.ts
@@ -22,6 +22,7 @@ import {
   FormattingResult,
   isConflictResult,
   OccurrencesResult,
+  ModelGenerationStartedResult,
 } from './xtextServiceResults';
 
 const UPDATE_TIMEOUT_MS = 500;
@@ -341,4 +342,42 @@ export default class UpdateService {
     }
     return { cancelled: false, data: parsedOccurrencesResult };
   }
+
+  async startModelGeneration(): Promise<
+    CancellableResult<ModelGenerationStartedResult>
+  > {
+    try {
+      await this.updateOrThrow();
+    } catch (error) {
+      if (error instanceof CancelledError || error instanceof TimeoutError) {
+        return { cancelled: true };
+      }
+      throw error;
+    }
+    log.debug('Starting model generation');
+    const data = await this.webSocketClient.send({
+      resource: this.resourceName,
+      serviceType: 'modelGeneration',
+      requiredStateId: this.xtextStateId,
+      start: true,
+    });
+    if (isConflictResult(data)) {
+      return { cancelled: true };
+    }
+    const parsedResult = ModelGenerationStartedResult.parse(data);
+    return { cancelled: false, data: parsedResult };
+  }
+
+  async cancelModelGeneration(): Promise<CancellableResult<unknown>> {
+    log.debug('Cancelling model generation');
+    const data = await this.webSocketClient.send({
+      resource: this.resourceName,
+      serviceType: 'modelGeneration',
+      cancel: true,
+    });
+    if (isConflictResult(data)) {
+      return { cancelled: true };
+    }
+    return { cancelled: false, data };
+  }
 }
diff --git a/subprojects/frontend/src/xtext/XtextClient.ts b/subprojects/frontend/src/xtext/XtextClient.ts
index 87778084..77980d35 100644
--- a/subprojects/frontend/src/xtext/XtextClient.ts
+++ b/subprojects/frontend/src/xtext/XtextClient.ts
@@ -16,6 +16,7 @@ import getLogger from '../utils/getLogger';
 
 import ContentAssistService from './ContentAssistService';
 import HighlightingService from './HighlightingService';
+import ModelGenerationService from './ModelGenerationService';
 import OccurrencesService from './OccurrencesService';
 import SemanticsService from './SemanticsService';
 import UpdateService from './UpdateService';
@@ -40,6 +41,8 @@ export default class XtextClient {
 
   private readonly semanticsService: SemanticsService;
 
+  private readonly modelGenerationService: ModelGenerationService;
+
   constructor(
     private readonly store: EditorStore,
     private readonly pwaStore: PWAStore,
@@ -58,6 +61,10 @@ export default class XtextClient {
     this.validationService = new ValidationService(store, this.updateService);
     this.occurrencesService = new OccurrencesService(store, this.updateService);
     this.semanticsService = new SemanticsService(store, this.validationService);
+    this.modelGenerationService = new ModelGenerationService(
+      store,
+      this.updateService,
+    );
   }
 
   start(): void {
@@ -75,6 +82,7 @@ export default class XtextClient {
     this.highlightingService.onDisconnect();
     this.validationService.onDisconnect();
     this.occurrencesService.onDisconnect();
+    this.modelGenerationService.onDisconnect();
   }
 
   onTransaction(transaction: Transaction): void {
@@ -101,7 +109,7 @@ export default class XtextClient {
       );
       return;
     }
-    if (stateId !== xtextStateId) {
+    if (stateId !== xtextStateId && service !== 'modelGeneration') {
       log.error(
         'Unexpected xtext state id: expected:',
         xtextStateId,
@@ -122,6 +130,9 @@ export default class XtextClient {
       case 'semantics':
         this.semanticsService.onPush(push);
         return;
+      case 'modelGeneration':
+        this.modelGenerationService.onPush(push);
+        return;
       default:
         throw new Error('Unknown service');
     }
@@ -131,6 +142,14 @@ export default class XtextClient {
     return this.contentAssistService.contentAssist(context);
   }
 
+  startModelGeneration(): Promise<void> {
+    return this.modelGenerationService.start();
+  }
+
+  cancelModelGeneration(): Promise<void> {
+    return this.modelGenerationService.cancel();
+  }
+
   formatText(): void {
     this.updateService.formatText().catch((e) => {
       log.error('Error while formatting text', e);
diff --git a/subprojects/frontend/src/xtext/xtextMessages.ts b/subprojects/frontend/src/xtext/xtextMessages.ts
index 971720e1..15831c5a 100644
--- a/subprojects/frontend/src/xtext/xtextMessages.ts
+++ b/subprojects/frontend/src/xtext/xtextMessages.ts
@@ -38,6 +38,7 @@ export const XtextWebPushService = z.enum([
   'highlight',
   'validate',
   'semantics',
+  'modelGeneration',
 ]);
 
 export type XtextWebPushService = z.infer<typeof XtextWebPushService>;
diff --git a/subprojects/frontend/src/xtext/xtextServiceResults.ts b/subprojects/frontend/src/xtext/xtextServiceResults.ts
index caf2cf0b..e473bd48 100644
--- a/subprojects/frontend/src/xtext/xtextServiceResults.ts
+++ b/subprojects/frontend/src/xtext/xtextServiceResults.ts
@@ -126,6 +126,14 @@ export const FormattingResult = DocumentStateResult.extend({
 
 export type FormattingResult = z.infer<typeof FormattingResult>;
 
+export const ModelGenerationStartedResult = z.object({
+  uuid: z.string().nonempty(),
+});
+
+export type ModelGenerationStartedResult = z.infer<
+  typeof ModelGenerationStartedResult
+>;
+
 export const NodeMetadata = z.object({
   name: z.string(),
   simpleName: z.string(),
@@ -171,3 +179,19 @@ export const SemanticsResult = z.union([
 ]);
 
 export type SemanticsResult = z.infer<typeof SemanticsResult>;
+
+export const ModelGenerationResult = z.union([
+  z.object({
+    uuid: z.string().nonempty(),
+    status: z.string(),
+  }),
+  z.object({
+    uuid: z.string().nonempty(),
+    error: z.string(),
+  }),
+  SemanticsSuccessResult.extend({
+    uuid: z.string().nonempty(),
+  }),
+]);
+
+export type ModelGenerationResult = z.infer<typeof ModelGenerationResult>;