@effect/language-service 0.16.2 → 0.16.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@effect/language-service",
-  "version": "0.16.2",
+  "version": "0.16.4",
   "description": "A Language-Service Plugin to Refactor and Diagnostic effect-ts projects",
   "main": "index.cjs",
   "repository": {

package/transform.js

@@ -720,6 +720,7 @@ var isNone2 = isNone;
 var isSome2 = isSome;
 
 // node_modules/.pnpm/effect@3.12.5/node_modules/effect/dist/esm/Array.js
+var fromIterable = (collection) => Array.isArray(collection) ? collection : Array.from(collection);
 var isArray = Array.isArray;
 var isEmptyArray = (self) => self.length === 0;
 var isEmptyReadonlyArray = isEmptyArray;
@@ -743,6 +744,16 @@ var flatMap = /* @__PURE__ */ dual(2, (self, f) => {
   return out;
 });
 var flatten = /* @__PURE__ */ flatMap(identity);
+var filter = /* @__PURE__ */ dual(2, (self, predicate) => {
+  const as = fromIterable(self);
+  const out = [];
+  for (let i = 0; i < as.length; i++) {
+    if (predicate(as[i], i)) {
+      out.push(as[i]);
+    }
+  }
+  return out;
+});
 
 // src/core/Nano.ts
 var NanoInternalSuccessProto = {
@@ -1200,9 +1211,9 @@ var expectedAndRealType = fn("TypeCheckerApi.expectedAndRealType")(function* (so
             realType
           ]);
         });
-        ts.forEachChild(node, appendNodeToVisit);
-        continue;
       }
+      ts.forEachChild(node, appendNodeToVisit);
+      continue;
     } else if (ts.isIdentifier(node) || ts.isStringLiteral(node) || ts.isNumericLiteral(node) || ts.isNoSubstitutionTemplateLiteral(node)) {
       const parent = node.parent;
       if (ts.isObjectLiteralElement(parent)) {
@@ -1214,12 +1225,12 @@ var expectedAndRealType = fn("TypeCheckerApi.expectedAndRealType")(function* (so
               const expectedType = typeChecker.getTypeOfSymbolAtLocation(symbol3, node);
               const realType = typeChecker.getTypeAtLocation(node);
               result.push([node, expectedType, node, realType]);
-              ts.forEachChild(node, appendNodeToVisit);
-              continue;
             }
           }
         }
       }
+      ts.forEachChild(node, appendNodeToVisit);
+      continue;
     } else if (ts.isBinaryExpression(node) && node.operatorToken.kind === ts.SyntaxKind.EqualsToken) {
       const expectedType = typeChecker.getTypeAtLocation(node.left);
       const realType = typeChecker.getTypeAtLocation(node.right);
@@ -1233,19 +1244,28 @@ var expectedAndRealType = fn("TypeCheckerApi.expectedAndRealType")(function* (so
         const realType = typeChecker.getTypeAtLocation(node.expression);
         if (isSome2(expectedType)) {
           result.push([node, expectedType.value, node, realType]);
-          ts.forEachChild(node, appendNodeToVisit);
-          continue;
         }
       }
-    } else if (ts.isArrowFunction(node) && ts.isExpression(node.body)) {
+      ts.forEachChild(node, appendNodeToVisit);
+      continue;
+    } else if (ts.isArrowFunction(node) && (node.typeParameters || []).length === 0 && ts.isExpression(node.body)) {
       const body = node.body;
       const expectedType = typeChecker.getContextualType(body);
       const realType = typeChecker.getTypeAtLocation(body);
       if (expectedType) {
         result.push([body, expectedType, body, realType]);
-        appendNodeToVisit(body);
-        continue;
       }
+      ts.forEachChild(node, appendNodeToVisit);
+      continue;
+    } else if (ts.isArrowFunction(node) && (node.typeParameters || []).length > 0 && ts.isExpression(node.body)) {
+      const body = node.body;
+      const expectedType = yield* option(getInferredReturnType(node));
+      const realType = typeChecker.getTypeAtLocation(body);
+      if (isSome2(expectedType)) {
+        result.push([body, expectedType.value, body, realType]);
+      }
+      ts.forEachChild(node, appendNodeToVisit);
+      continue;
     } else if (ts.isSatisfiesExpression(node)) {
       const expectedType = typeChecker.getTypeAtLocation(node.type);
       const realType = typeChecker.getTypeAtLocation(node.expression);
@@ -1899,8 +1919,14 @@ function transform_default(program, pluginConfig, { addDiagnostic, ts: tsInstanc
         provideService(PluginOptions, parsePluginOptions(pluginConfig)),
         run,
         map((_2) => _2.diagnostics),
-        getOrElse(() => [])
-      ).map(addDiagnostic);
+        map(
+          filter(
+            (_2) => _2.category === tsInstance.DiagnosticCategory.Error || _2.category === tsInstance.DiagnosticCategory.Warning
+          )
+        ),
+        getOrElse(() => []),
+        map2(addDiagnostic)
+      );
       return sourceFile;
     };
   };