1 files changed, 39 insertions, 0 deletions

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 };
+  }
 }