@cratis/screenplay-language 4.53.0 → 4.55.0

package/dist/bundles/index.js

@@ -23,6 +23,7 @@ import {
   importablePaths,
   indentOf,
   isFileImportLine,
+  isQuickFixDiagnostic,
   knownEventNames,
   knownTriggerNames,
   knownTypeNames,
@@ -41,7 +42,7 @@ import {
   typeReferenceSymbol,
   typeReferenceText,
   validateLines
-} from "./chunk-7WJEZ72S.js";
+} from "./chunk-GJX4NJYC.js";
 
 // tokens.ts
 var embeddedLanguages = {
@@ -999,13 +1000,14 @@ function createCodeActionProvider(placement) {
   return {
     provideCodeActions(model, range, context, token) {
       const migrationRequested = context.only !== void 0 && containsKind(context.only, migrateOptional);
+      const migrationOnly = context.only !== void 0 && migrationRequested && !containsKind(context.only, "quickfix");
       if (token.isCancellationRequested || context.only !== void 0 && !migrationRequested && !containsKind(context.only, "quickfix")) return { actions: [], dispose() {
       } };
-      const diagnostic = context.markers.find((marker) => {
+      const diagnostics = context.markers.filter((marker) => {
         const code = typeof marker.code === "object" ? marker.code.value : marker.code;
-        return code === DiagnosticCodes.LegacyOptionalSuffix && intersects(marker, range);
+        return isQuickFixDiagnostic(code) && intersects(marker, range);
       });
-      if (!migrationRequested && diagnostic === void 0) return { actions: [], dispose() {
+      if (!migrationRequested && diagnostics.length === 0) return { actions: [], dispose() {
       } };
       const version = model.getVersionId();
       const placementKey = JSON.stringify(placement) ?? "";
@@ -1014,7 +1016,12 @@ function createCodeActionProvider(placement) {
         analysis = { version, placement: placementKey, fixes: prepareQuickFixes(model.getValue(), { placement }) };
         cache.set(model, analysis);
       }
-      const fixes = analysis.fixes(diagnostic?.startLineNumber).filter((fix) => context.only === void 0 || containsKind(context.only, fix.scope === "document" ? migrateOptional : "quickfix"));
+      const occurrences = migrationOnly ? [] : analysis.fixes(diagnostics.map((diagnostic) => ({
+        line: diagnostic.startLineNumber,
+        diagnosticCode: String(typeof diagnostic.code === "object" ? diagnostic.code.value : diagnostic.code)
+      }))).filter((fix) => fix.scope === "occurrence");
+      const requested = [...occurrences, ...context.only === void 0 || migrationRequested ? analysis.fixes(void 0, DiagnosticCodes.LegacyOptionalSuffix) : []];
+      const fixes = requested.filter((fix) => context.only === void 0 || containsKind(context.only, fix.scope === "document" ? migrateOptional : "quickfix"));
       if (token.isCancellationRequested || model.getVersionId() !== version) return { actions: [], dispose() {
       } };
       return {