@effect/language-service 0.9.1 → 0.10.0

package/index.js

@@ -167,7 +167,7 @@ var let_ = (map8) => dual(3, (self, name, f) => map8(self, (a) => Object.assign(
 var bindTo = (map8) => dual(2, (self, name) => map8(self, (a) => ({
   [name]: a
 })));
-var bind = (map8, flatMap5) => dual(3, (self, name, f) => flatMap5(self, (a) => map8(f(a), (b) => Object.assign({}, a, {
+var bind = (map8, flatMap6) => dual(3, (self, name, f) => flatMap6(self, (a) => map8(f(a), (b) => Object.assign({}, a, {
   [name]: b
 }))));
 
@@ -930,6 +930,8 @@ var getOrUndefined2 = /* @__PURE__ */ getOrElse2(constUndefined);
 var fromIterable = (collection) => Array.isArray(collection) ? collection : Array.from(collection);
 var append = /* @__PURE__ */ dual(2, (self, last) => [...self, last]);
 var isArray = Array.isArray;
+var isEmptyArray = (self) => self.length === 0;
+var isEmptyReadonlyArray = isEmptyArray;
 var isNonEmptyReadonlyArray = isNonEmptyArray;
 var isOutOfBound = (i, as) => i < 0 || i >= as.length;
 var get = /* @__PURE__ */ dual(2, (self, index) => {
@@ -954,18 +956,20 @@ var sort = /* @__PURE__ */ dual(2, (self, O) => {
 });
 var empty = () => [];
 var map2 = /* @__PURE__ */ dual(2, (self, f) => self.map(f));
-var filterMap = /* @__PURE__ */ dual(2, (self, f) => {
-  const as = fromIterable(self);
+var flatMap2 = /* @__PURE__ */ dual(2, (self, f) => {
+  if (isEmptyReadonlyArray(self)) {
+    return [];
+  }
   const out = [];
-  for (let i = 0; i < as.length; i++) {
-    const o = f(as[i], i);
-    if (isSome2(o)) {
-      out.push(o.value);
+  for (let i = 0; i < self.length; i++) {
+    const inner = f(self[i], i);
+    for (let j = 0; j < inner.length; j++) {
+      out.push(inner[j]);
     }
   }
   return out;
 });
-var getSomes = /* @__PURE__ */ filterMap(identity);
+var flatten = /* @__PURE__ */ flatMap2(identity);
 var filter = /* @__PURE__ */ dual(2, (self, predicate) => {
   const as = fromIterable(self);
   const out = [];
@@ -2719,7 +2723,7 @@ var run = (fa) => {
 var succeed = (value) => new Nano(() => ({ _tag: "Right", right: value }));
 var fail3 = (value) => new Nano(() => ({ _tag: "Left", left: value }));
 var sync = (value) => new Nano(() => ({ _tag: "Right", right: value() }));
-var flatMap4 = dual(2, (fa, f) => new Nano((ctx) => {
+var flatMap5 = dual(2, (fa, f) => new Nano((ctx) => {
   const result = fa.run(ctx);
   if (result._tag !== "Right") return result;
   return f(result.right).run(ctx);
@@ -2782,102 +2786,11 @@ var option = (fa) => new Nano((ctx) => {
   }
 });
 
-// src/core/LSP.ts
-var RefactorNotApplicableError = class extends TaggedError("RefactorNotApplicableError") {
-};
-function createRefactor(definition) {
-  return definition;
-}
-function createDiagnostic(definition) {
-  return definition;
-}
-function createCompletion(definition) {
-  return definition;
-}
-var PluginOptions = Tag("PluginOptions");
-var SourceFileNotFoundError = class extends TaggedError("SourceFileNotFoundError") {
-};
-var getSemanticDiagnostics = fn("LSP.getSemanticDiagnostics")(function* (diagnostics2, sourceFile) {
-  const effectDiagnostics = [];
-  for (const diagnostic of diagnostics2) {
-    const result = yield* option(diagnostic.apply(sourceFile));
-    if (isSome2(result)) {
-      effectDiagnostics.push(...result.value.map((_) => ({
-        file: sourceFile,
-        start: _.node.getStart(sourceFile),
-        length: _.node.getEnd() - _.node.getStart(sourceFile),
-        messageText: _.messageText,
-        category: _.category,
-        code: diagnostic.code,
-        source: "effect"
-      })));
-    }
-  }
-  return effectDiagnostics;
-});
-var getCodeFixesAtPosition = fn("LSP.getCodeFixesAtPosition")(function* (diagnostics2, sourceFile, start, end, errorCodes) {
-  const runnableDiagnostics = diagnostics2.filter((_) => errorCodes.indexOf(_.code) > -1);
-  const applicableFixes = [];
-  for (const diagnostic of runnableDiagnostics) {
-    const result = yield* option(diagnostic.apply(sourceFile));
-    if (isSome2(result)) {
-      applicableFixes.push(
-        ...pipe(
-          result.value,
-          filter(
-            (_) => _.node.getStart(sourceFile) === start && _.node.getEnd() === end
-          ),
-          map2((_) => _.fix),
-          getSomes
-        )
-      );
-    }
-  }
-  return applicableFixes;
-});
-var getApplicableRefactors = fn("LSP.getApplicableRefactors")(function* (refactors2, sourceFile, positionOrRange) {
-  const textRange = typeof positionOrRange === "number" ? { pos: positionOrRange, end: positionOrRange } : positionOrRange;
-  const effectRefactors = [];
-  for (const refactor of refactors2) {
-    const result = yield* option(refactor.apply(sourceFile, textRange));
-    if (isSome2(result)) {
-      effectRefactors.push({
-        name: refactor.name,
-        description: refactor.description,
-        actions: [{
-          name: refactor.name,
-          description: result.value.description,
-          kind: result.value.kind
-        }]
-      });
-    }
-  }
-  return effectRefactors;
-});
-var getEditsForRefactor = fn("LSP.getEditsForRefactor")(function* (refactors2, sourceFile, positionOrRange, refactorName) {
-  const refactor = refactors2.find((refactor2) => refactor2.name === refactorName);
-  if (!refactor) {
-    return yield* fail3(new RefactorNotApplicableError());
-  }
-  const textRange = typeof positionOrRange === "number" ? { pos: positionOrRange, end: positionOrRange } : positionOrRange;
-  return yield* refactor.apply(sourceFile, textRange);
-});
-var getCompletionsAtPosition = fn("LSP.getCompletionsAtPosition")(function* (completions2, sourceFile, position, options, formatCodeSettings) {
-  const effectCompletions = [];
-  for (const completion of completions2) {
-    const result = yield* completion.apply(sourceFile, position, options, formatCodeSettings);
-    effectCompletions.push(
-      ...result.map((_) => ({ sortText: "11", ..._ }))
-    );
-  }
-  return effectCompletions;
-});
-
-// src/utils/TypeScriptApi.ts
+// src/core/TypeScriptApi.ts
 var TypeScriptApi = Tag("TypeScriptApi");
 var ChangeTracker = Tag("ChangeTracker");
 
-// src/utils/AST.ts
+// src/core/AST.ts
 function collectSelfAndAncestorNodesInRange(node, textRange) {
   return sync(() => {
     let result = empty();
@@ -2892,10 +2805,9 @@ function collectSelfAndAncestorNodesInRange(node, textRange) {
   });
 }
 var getAncestorNodesInRange = fn("AST.getAncestorNodesInRange")(function* (sourceFile, textRange) {
-  const ts = yield* service(TypeScriptApi);
-  const precedingToken = ts.findPrecedingToken(textRange.pos, sourceFile);
-  if (!precedingToken) return empty();
-  return yield* collectSelfAndAncestorNodesInRange(precedingToken, textRange);
+  const nodeAtPosition = yield* option(findNodeAtPosition(sourceFile, textRange.pos));
+  if (isNone2(nodeAtPosition)) return empty();
+  return yield* collectSelfAndAncestorNodesInRange(nodeAtPosition.value, textRange);
 });
 var NodeNotFoundError = class extends TaggedError("@effect/language-service/NodeNotFoundError") {
 };
@@ -2911,13 +2823,21 @@ var findNodeAtPosition = fn("AST.findNodeAtPosition")(function* (sourceFile, pos
   if (!result) return yield* fail3(new NodeNotFoundError());
   return result;
 });
-var collectDescendantsAndAncestorsInRange = fn(
-  "AST.collectDescendantsAndAncestorsInRange"
-)(function* (sourceFile, textRange) {
-  const nodeAtPosition = yield* option(findNodeAtPosition(sourceFile, textRange.pos));
-  if (isNone2(nodeAtPosition)) return empty();
-  return yield* collectSelfAndAncestorNodesInRange(nodeAtPosition.value, textRange);
+var getCommentAtPosition = fn("TypeScriptApi.getCommentAtPosition")(function* (sourceFile, pos) {
+  const ts = yield* service(TypeScriptApi);
+  const token = yield* findNodeAtPosition(sourceFile, pos);
+  if (token === void 0 || token.kind === ts.SyntaxKind.JsxText || pos >= token.end - (ts.tokenToString(token.kind) || "").length) {
+    return yield* fail3(new NodeNotFoundError());
+  }
+  const startPos = token.pos === 0 ? (ts.getShebang(sourceFile.text) || "").length : token.pos;
+  if (startPos === 0) return yield* fail3(new NodeNotFoundError());
+  const result = ts.forEachTrailingCommentRange(sourceFile.text, startPos, isCommentInRange, pos) || ts.forEachLeadingCommentRange(sourceFile.text, startPos, isCommentInRange, pos);
+  if (!result) return yield* fail3(new NodeNotFoundError());
+  return result;
 });
+function isCommentInRange(pos, end, kind, _nl, at) {
+  return at >= pos && at < end ? { pos, end, kind } : void 0;
+}
 function toTextRange(positionOrRange) {
   return typeof positionOrRange === "number" ? { end: positionOrRange, pos: positionOrRange } : positionOrRange;
 }
@@ -3177,6 +3097,180 @@ var parseDataForExtendsClassCompletion = fn(
   );
 });
 
+// src/core/LSP.ts
+var RefactorNotApplicableError = class extends TaggedError("RefactorNotApplicableError") {
+};
+function createRefactor(definition) {
+  return definition;
+}
+function createDiagnostic(definition) {
+  return definition;
+}
+function createCompletion(definition) {
+  return definition;
+}
+var PluginOptions = Tag("PluginOptions");
+var SourceFileNotFoundError = class extends TaggedError("SourceFileNotFoundError") {
+};
+var getSemanticDiagnostics = fn("LSP.getSemanticDiagnostics")(function* (diagnostics2, sourceFile) {
+  const effectDiagnostics = [];
+  const executor = yield* createDiagnosticExecutor(sourceFile);
+  for (const diagnostic of diagnostics2) {
+    const result = yield* option(executor.execute(diagnostic));
+    if (isSome2(result)) {
+      effectDiagnostics.push(
+        ...pipe(
+          result.value,
+          map2((_) => ({
+            file: sourceFile,
+            start: _.node.getStart(sourceFile),
+            length: _.node.getEnd() - _.node.getStart(sourceFile),
+            messageText: _.messageText,
+            category: _.category,
+            code: diagnostic.code,
+            source: "effect"
+          }))
+        )
+      );
+    }
+  }
+  return effectDiagnostics;
+});
+var getCodeFixesAtPosition = fn("LSP.getCodeFixesAtPosition")(function* (diagnostics2, sourceFile, start, end, errorCodes) {
+  const runnableDiagnostics = diagnostics2.filter((_) => errorCodes.indexOf(_.code) > -1);
+  const applicableFixes = [];
+  const executor = yield* createDiagnosticExecutor(sourceFile);
+  for (const diagnostic of runnableDiagnostics) {
+    const result = yield* option(executor.execute(diagnostic));
+    if (isSome2(result)) {
+      applicableFixes.push(
+        ...pipe(
+          result.value,
+          filter(
+            (_) => _.node.getStart(sourceFile) === start && _.node.getEnd() === end
+          ),
+          map2((_) => _.fixes),
+          flatten
+        )
+      );
+    }
+  }
+  return applicableFixes;
+});
+var getApplicableRefactors = fn("LSP.getApplicableRefactors")(function* (refactors2, sourceFile, positionOrRange) {
+  const textRange = typeof positionOrRange === "number" ? { pos: positionOrRange, end: positionOrRange } : positionOrRange;
+  const effectRefactors = [];
+  for (const refactor of refactors2) {
+    const result = yield* option(refactor.apply(sourceFile, textRange));
+    if (isSome2(result)) {
+      effectRefactors.push({
+        name: refactor.name,
+        description: refactor.description,
+        actions: [{
+          name: refactor.name,
+          description: result.value.description,
+          kind: result.value.kind
+        }]
+      });
+    }
+  }
+  return effectRefactors;
+});
+var getEditsForRefactor = fn("LSP.getEditsForRefactor")(function* (refactors2, sourceFile, positionOrRange, refactorName) {
+  const refactor = refactors2.find((refactor2) => refactor2.name === refactorName);
+  if (!refactor) {
+    return yield* fail3(new RefactorNotApplicableError());
+  }
+  const textRange = typeof positionOrRange === "number" ? { pos: positionOrRange, end: positionOrRange } : positionOrRange;
+  return yield* refactor.apply(sourceFile, textRange);
+});
+var getCompletionsAtPosition = fn("LSP.getCompletionsAtPosition")(function* (completions2, sourceFile, position, options, formatCodeSettings) {
+  const effectCompletions = [];
+  for (const completion of completions2) {
+    const result = yield* completion.apply(sourceFile, position, options, formatCodeSettings);
+    effectCompletions.push(
+      ...result.map((_) => ({ sortText: "11", ..._ }))
+    );
+  }
+  return effectCompletions;
+});
+var createDiagnosticExecutor = fn("LSP.createCommentDirectivesProcessor")(
+  function* (sourceFile) {
+    const ts = yield* service(TypeScriptApi);
+    const ruleOverrides = {};
+    const skippedRules = [];
+    const regex = /@effect-diagnostics((?:\s[a-zA-Z0-9/]+:(?:off|warning|error|message|suggestion|skip-file))+)?/gm;
+    let match3;
+    while ((match3 = regex.exec(sourceFile.text)) !== null) {
+      const rulesCaptureGroup = match3[1];
+      if (rulesCaptureGroup) {
+        const trimmedRuleString = rulesCaptureGroup.trim();
+        if (trimmedRuleString) {
+          const individualRules = trimmedRuleString.split(/\s+/);
+          for (const rulePair of individualRules) {
+            const [ruleName, ruleLevel] = rulePair.toLowerCase().split(":");
+            if (ruleName && ruleLevel) {
+              if (ruleLevel === "skip-file") skippedRules.push(ruleName);
+              ruleOverrides[ruleName] = ruleOverrides[ruleName] || [];
+              const newLength = ruleOverrides[ruleName].push({
+                start: match3.index,
+                end: Number.MAX_SAFE_INTEGER,
+                level: ruleLevel
+              });
+              if (newLength > 1) ruleOverrides[ruleName][newLength - 2].end = match3.index;
+            }
+          }
+        }
+      }
+    }
+    const levelToDiagnosticCategory = {
+      error: ts.DiagnosticCategory.Error,
+      warning: ts.DiagnosticCategory.Warning,
+      message: ts.DiagnosticCategory.Message,
+      suggestion: ts.DiagnosticCategory.Suggestion
+    };
+    const execute = fn("LSP.ruleExecutor")(function* (rule) {
+      const ruleNameLowered = rule.name.toLowerCase();
+      if (skippedRules.indexOf(ruleNameLowered) > -1) return [];
+      let modifiedDiagnostics = yield* rule.apply(sourceFile);
+      for (const override of ruleOverrides[ruleNameLowered] || []) {
+        if (override.level === "off") {
+          modifiedDiagnostics = modifiedDiagnostics.filter(
+            (_) => !(_.node.getStart(sourceFile) >= override.start && _.node.getEnd() <= override.end)
+          );
+        } else {
+          for (const message of modifiedDiagnostics.filter(
+            (_) => _.node.getStart(sourceFile) >= override.start && _.node.getEnd() <= override.end
+          )) {
+            message.category = override.level in levelToDiagnosticCategory ? levelToDiagnosticCategory[override.level] : message.category;
+          }
+        }
+      }
+      const fixByDisableEntireFile = {
+        fixName: rule.name + "_skipFile",
+        description: "Disable " + rule.name + " for this file",
+        apply: flatMap5(
+          service(ChangeTracker),
+          (changeTracker) => sync(
+            () => changeTracker.insertText(
+              sourceFile,
+              0,
+              `/** @effect-diagnostics ${rule.name}:skip-file */
+`
+            )
+          )
+        )
+      };
+      const rulesWithDisableFix = modifiedDiagnostics.map((diagnostic) => ({
+        ...diagnostic,
+        fixes: diagnostic.fixes.concat([fixByDisableEntireFile])
+      }));
+      return rulesWithDisableFix;
+    });
+    return { execute };
+  }
+);
+
 // src/completions/contextSelfInClasses.ts
 var contextSelfInClasses = createCompletion({
   name: "effect/contextSelfInClasses",
@@ -3208,6 +3302,43 @@ var contextSelfInClasses = createCompletion({
   })
 });
 
+// src/completions/effectDataClasses.ts
+var effectDataClasses = createCompletion({
+  name: "effect/effectDataClasses",
+  apply: fn("effectDataClasses")(function* (sourceFile, position) {
+    const ts = yield* service(TypeScriptApi);
+    const maybeInfos = yield* parseDataForExtendsClassCompletion(sourceFile, position);
+    if (isNone2(maybeInfos)) return [];
+    const { accessedObject, className, replacementSpan } = maybeInfos.value;
+    const dataName = yield* option(
+      findImportedModuleIdentifierByPackageAndNameOrBarrel(
+        sourceFile,
+        "effect",
+        "Data"
+      )
+    );
+    const effectDataIdentifier = match2(dataName, {
+      onNone: () => "Data",
+      onSome: (_) => _.text
+    });
+    if (effectDataIdentifier !== accessedObject.text) return [];
+    const name = className.text;
+    return [{
+      name: `TaggedError("${name}")`,
+      kind: ts.ScriptElementKind.constElement,
+      insertText: `${effectDataIdentifier}.TaggedError("${name}")<{${"${0}"}}>{}`,
+      replacementSpan,
+      isSnippet: true
+    }, {
+      name: `TaggedClass("${name}")`,
+      kind: ts.ScriptElementKind.constElement,
+      insertText: `${effectDataIdentifier}.TaggedClass("${name}")<{${"${0}"}}>{}`,
+      replacementSpan,
+      isSnippet: true
+    }];
+  })
+});
+
 // src/completions/effectSchemaSelfInClasses.ts
 var effectSchemaSelfInClasses = createCompletion({
   name: "effect/effectSchemaSelfInClasses",
@@ -3289,9 +3420,14 @@ var effectSelfInClasses = createCompletion({
 });
 
 // src/completions.ts
-var completions = [effectSchemaSelfInClasses, effectSelfInClasses, contextSelfInClasses];
+var completions = [
+  effectSchemaSelfInClasses,
+  effectSelfInClasses,
+  contextSelfInClasses,
+  effectDataClasses
+];
 
-// src/utils/TypeCheckerApi.ts
+// src/core/TypeCheckerApi.ts
 var TypeCheckerApi = Tag("TypeChecker");
 var TypeCheckerApiCache = Tag("TypeCheckerApiCache");
 function makeTypeCheckerApiCache() {
@@ -3752,7 +3888,7 @@ var floatingEffect = createDiagnostic({
             node,
             category: ts.DiagnosticCategory.Error,
             messageText: `Effect must be yielded or assigned to a variable.`,
-            fix: none2()
+            fixes: []
           });
         }
       }
@@ -3804,7 +3940,7 @@ var missingEffectContext = createDiagnostic({
             node,
             category: ts.DiagnosticCategory.Error,
             messageText: `Missing '${sortTypes(missingContext).map((_) => typeChecker.typeToString(_)).join(" | ")}' in the expected Effect context.`,
-            fix: none2()
+            fixes: []
           }
         );
       }
@@ -3856,7 +3992,7 @@ var missingEffectError = createDiagnostic({
             node,
             category: ts.DiagnosticCategory.Error,
             messageText: `Missing '${sortTypes(missingContext).map((_) => typeChecker.typeToString(_)).join(" | ")}' in the expected Effect errors.`,
-            fix: none2()
+            fixes: []
           }
         );
       }
@@ -3913,11 +4049,11 @@ var missingStarInYieldEffectGen = createDiagnostic({
         node,
         category: ts.DiagnosticCategory.Error,
         messageText: `Seems like you used yield instead of yield* inside this Effect.gen.`,
-        fix: none2()
+        fixes: []
       })
     );
     brokenYields.forEach((node) => {
-      const fix = node.expression ? some2({
+      const fix = node.expression ? [{
         fixName: "missingStarInYieldEffectGen_fix",
         description: "Replace yield with yield*",
         apply: gen2(function* () {
@@ -3931,12 +4067,12 @@ var missingStarInYieldEffectGen = createDiagnostic({
             )
           );
         })
-      }) : none2();
+      }] : [];
       effectDiagnostics.push({
         node,
         category: ts.DiagnosticCategory.Error,
         messageText: `When yielding Effects inside Effect.gen, you should use yield* instead of yield.`,
-        fix
+        fixes: fix
       });
     });
     return effectDiagnostics;
@@ -3962,7 +4098,7 @@ var unnecessaryEffectGen = createDiagnostic({
       ts.forEachChild(node, appendNodeToVisit);
       const maybeUnnecessaryGen = yield* pipe(
         effectGen(node),
-        flatMap4(({ body }) => returnYieldEffectBlock(body)),
+        flatMap5(({ body }) => returnYieldEffectBlock(body)),
         option
       );
       if (isSome2(maybeUnnecessaryGen)) {
@@ -3974,7 +4110,7 @@ var unnecessaryEffectGen = createDiagnostic({
         node,
         category: ts.DiagnosticCategory.Suggestion,
         messageText: `This Effect.gen is useless here because it only contains a single return statement.`,
-        fix: some2({
+        fixes: [{
           fixName: "unnecessaryEffectGen_fix",
           description: "Remove the Effect.gen, and keep the body",
           apply: gen2(function* () {
@@ -3983,7 +4119,7 @@ var unnecessaryEffectGen = createDiagnostic({
             );
             textChanges.replaceNode(sourceFile, node, body);
           })
-        })
+        }]
       })
     );
     return effectDiagnostics;
@@ -4462,13 +4598,10 @@ var removeUnnecessaryEffectGen = createRefactor({
   name: "effect/removeUnnecessaryEffectGen",
   description: "Remove unnecessary Effect.gen",
   apply: (sourceFile, textRange) => gen2(function* () {
-    for (const nodeToReplace of yield* collectDescendantsAndAncestorsInRange(
-      sourceFile,
-      textRange
-    )) {
+    for (const nodeToReplace of yield* getAncestorNodesInRange(sourceFile, textRange)) {
       const maybeNode = yield* pipe(
         effectGen(nodeToReplace),
-        flatMap4(({ body }) => returnYieldEffectBlock(body)),
+        flatMap5(({ body }) => returnYieldEffectBlock(body)),
         option
       );
       if (isNone2(maybeNode)) continue;
@@ -4489,7 +4622,7 @@ var removeUnnecessaryEffectGen = createRefactor({
 // src/refactors/toggleLazyConst.ts
 var toggleLazyConst = createRefactor({
   name: "effect/toggleLazyConst",
-  description: "Toggle type annotation",
+  description: "Toggle lazy const",
   apply: (sourceFile, textRange) => gen2(function* () {
     const ts = yield* service(TypeScriptApi);
     const maybeNode = pipe(