@effect/language-service 0.69.1 → 0.69.2

package/cli.js

@@ -30214,7 +30214,7 @@ var runMain3 = runMain2;
 // package.json
 var package_default = {
   name: "@effect/language-service",
-  version: "0.69.1",
+  version: "0.69.2",
   packageManager: "pnpm@8.11.0",
   publishConfig: {
     access: "public",
@@ -32039,6 +32039,7 @@ var defaults = {
   }],
   extendedKeyDetection: false,
   pipeableMinArgCount: 2,
+  effectFn: ["span"],
   layerGraphFollowDepth: 0,
   mermaidProvider: "mermaid.live"
 };
@@ -32078,6 +32079,7 @@ function parse4(config2) {
     keyPatterns: isObject(config2) && hasProperty(config2, "keyPatterns") && isArray(config2.keyPatterns) ? parseKeyPatterns(config2.keyPatterns) : defaults.keyPatterns,
     extendedKeyDetection: isObject(config2) && hasProperty(config2, "extendedKeyDetection") && isBoolean(config2.extendedKeyDetection) ? config2.extendedKeyDetection : defaults.extendedKeyDetection,
     pipeableMinArgCount: isObject(config2) && hasProperty(config2, "pipeableMinArgCount") && isNumber(config2.pipeableMinArgCount) ? config2.pipeableMinArgCount : defaults.pipeableMinArgCount,
+    effectFn: isObject(config2) && hasProperty(config2, "effectFn") && isArray(config2.effectFn) && config2.effectFn.every(isString) ? config2.effectFn.map((_) => _.toLowerCase()) : defaults.effectFn,
     layerGraphFollowDepth: isObject(config2) && hasProperty(config2, "layerGraphFollowDepth") && isNumber(config2.layerGraphFollowDepth) ? config2.layerGraphFollowDepth : defaults.layerGraphFollowDepth,
     mermaidProvider: isObject(config2) && hasProperty(config2, "mermaidProvider") && isString(config2.mermaidProvider) ? config2.mermaidProvider : defaults.mermaidProvider
   };
@@ -33302,6 +33304,7 @@ function make64(ts, tsUtils, typeChecker, typeCheckerUtils, program) {
           node
         );
       }
+      const traceExpression = ts.isCallExpression(node.expression) && node.expression.arguments.length > 0 ? node.expression.arguments[0] : void 0;
       const propertyAccess = expressionToTest;
       const pipeArguments2 = node.arguments.slice(1);
       return pipe(
@@ -33311,7 +33314,8 @@ function make64(ts, tsUtils, typeChecker, typeCheckerUtils, program) {
           generatorFunction,
           effectModule: propertyAccess.expression,
           body: generatorFunction.body,
-          pipeArguments: pipeArguments2
+          pipeArguments: pipeArguments2,
+          traceExpression
         }))
       );
     },
@@ -33394,6 +33398,7 @@ function make64(ts, tsUtils, typeChecker, typeCheckerUtils, program) {
       if (!ts.isPropertyAccessExpression(expressionToTest)) {
         return typeParserIssue("Node is not a property access expression", void 0, node);
       }
+      const traceExpression = ts.isCallExpression(node.expression) && node.expression.arguments.length > 0 ? node.expression.arguments[0] : void 0;
       const propertyAccess = expressionToTest;
       const pipeArguments2 = node.arguments.slice(1);
       return pipe(
@@ -33402,7 +33407,8 @@ function make64(ts, tsUtils, typeChecker, typeCheckerUtils, program) {
           node,
           effectModule: propertyAccess.expression,
           regularFunction,
-          pipeArguments: pipeArguments2
+          pipeArguments: pipeArguments2,
+          traceExpression
         }))
       );
     },
@@ -36299,7 +36305,8 @@ var effectFnIife = createDiagnostic({
           kind: "fn",
           effectModule: result.effectModule,
           generatorFunction: result.generatorFunction,
-          pipeArguments: result.pipeArguments
+          pipeArguments: result.pipeArguments,
+          traceExpression: result.traceExpression
         })),
         orElse15(
           () => pipe(
@@ -36308,7 +36315,8 @@ var effectFnIife = createDiagnostic({
               kind: "fnUntraced",
               effectModule: result.effectModule,
               generatorFunction: result.generatorFunction,
-              pipeArguments: result.pipeArguments
+              pipeArguments: result.pipeArguments,
+              traceExpression: void 0
             }))
           )
         ),
@@ -36319,14 +36327,15 @@ var effectFnIife = createDiagnostic({
               kind: "fn",
               effectModule: result.effectModule,
               generatorFunction: void 0,
-              pipeArguments: result.pipeArguments
+              pipeArguments: result.pipeArguments,
+              traceExpression: result.traceExpression
             }))
           )
         ),
         option5
       );
       if (isNone2(parsed)) continue;
-      const { effectModule, generatorFunction, kind, pipeArguments: pipeArguments2 } = parsed.value;
+      const { effectModule, generatorFunction, kind, pipeArguments: pipeArguments2, traceExpression } = parsed.value;
       const effectModuleName = ts.isIdentifier(effectModule) ? ts.idText(effectModule) : sourceEffectModuleName;
       const fixes = [];
       if (generatorFunction && generatorFunction.parameters.length === 0) {
@@ -36355,9 +36364,10 @@ var effectFnIife = createDiagnostic({
           })
         });
       }
+      const traceExpressionText = traceExpression ? sourceFile.text.slice(traceExpression.pos, traceExpression.end) : void 0;
       report({
         location: node,
-        messageText: `${effectModuleName}.${kind} returns a reusable function that can take arguments, but here it's called immediately. Use Effect.gen instead (optionally with Effect.withSpan for tracing).`,
+        messageText: `${effectModuleName}.${kind} returns a reusable function that can take arguments, but here it's called immediately. Use Effect.gen instead${traceExpressionText ? ` with Effect.withSpan(${traceExpressionText}) piped in the end to mantain tracing spans` : ``}.`,
         fixes
       });
     }
@@ -36376,6 +36386,7 @@ var effectFnOpportunity = createDiagnostic({
     const typeCheckerUtils = yield* service2(TypeCheckerUtils);
     const typeParser = yield* service2(TypeParser);
     const tsUtils = yield* service2(TypeScriptUtils);
+    const pluginOptions = yield* service2(LanguageServicePluginOptions);
     const sourceEffectModuleName = tsUtils.findImportedModuleIdentifierByPackageAndNameOrBarrel(
       sourceFile,
       "effect",
@@ -36452,6 +36463,18 @@ var effectFnOpportunity = createDiagnostic({
       }
       return false;
     };
+    const tryExtractWithSpanExpression = (expr) => gen3(function* () {
+      if (!ts.isCallExpression(expr)) return void 0;
+      const callee = expr.expression;
+      const isWithSpan = yield* pipe(
+        typeParser.isNodeReferenceToEffectModuleApi("withSpan")(callee),
+        map34(() => true),
+        orElse15(() => succeed17(false))
+      );
+      if (!isWithSpan) return void 0;
+      if (expr.arguments.length === 0) return void 0;
+      return expr.arguments[0];
+    });
     const tryParseGenOpportunity = (fnNode) => gen3(function* () {
       const bodyExpression = getBodyExpression(fnNode);
       if (!bodyExpression) return yield* TypeParserIssue.issue;
@@ -36462,7 +36485,15 @@ var effectFnOpportunity = createDiagnostic({
       );
       const { effectModule, generatorFunction } = yield* typeParser.effectGen(subject);
       const effectModuleName = ts.isIdentifier(effectModule) ? ts.idText(effectModule) : sourceEffectModuleName;
-      return { effectModuleName, generatorFunction, pipeArguments: pipeArguments2 };
+      let explicitTraceExpression;
+      if (pipeArguments2.length > 0) {
+        const lastArg = pipeArguments2[pipeArguments2.length - 1];
+        const withSpanExpr = yield* tryExtractWithSpanExpression(lastArg);
+        if (withSpanExpr) {
+          explicitTraceExpression = withSpanExpr;
+        }
+      }
+      return { effectModuleName, generatorFunction, pipeArguments: pipeArguments2, explicitTraceExpression };
     });
     const isInsideEffectFn = (fnNode) => {
       const parent = fnNode.parent;
@@ -36490,6 +36521,9 @@ var effectFnOpportunity = createDiagnostic({
       if (ts.isFunctionExpression(node) && node.name) {
         return yield* TypeParserIssue.issue;
       }
+      if (node.type) {
+        return yield* TypeParserIssue.issue;
+      }
       if (yield* isInsideEffectFn(node)) {
         return yield* TypeParserIssue.issue;
       }
@@ -36517,7 +36551,8 @@ var effectFnOpportunity = createDiagnostic({
           return succeed17({
             effectModuleName: sourceEffectModuleName,
             pipeArguments: [],
-            generatorFunction: void 0
+            generatorFunction: void 0,
+            explicitTraceExpression: void 0
           });
         })
       );
@@ -36525,7 +36560,8 @@ var effectFnOpportunity = createDiagnostic({
         node,
         nameIdentifier,
         effectModuleName: opportunity.effectModuleName,
-        traceName,
+        inferredTraceName: traceName,
+        explicitTraceExpression: opportunity.explicitTraceExpression,
         pipeArguments: opportunity.pipeArguments,
         generatorFunction: opportunity.generatorFunction,
         hasParamsInPipeArgs: areParametersReferencedIn(node, opportunity.pipeArguments)
@@ -36544,7 +36580,7 @@ var effectFnOpportunity = createDiagnostic({
       if (ts.isArrowFunction(node)) return false;
       return node.asteriskToken !== void 0;
     };
-    const createEffectFnNode = (originalNode, innerFunction, effectModuleName, traceName, pipeArguments2) => {
+    const createEffectFnNode = (originalNode, innerFunction, effectModuleName, traceNameOrExpression, pipeArguments2) => {
       const isGenerator = isGeneratorFunction2(innerFunction);
       const newFunction = ts.factory.createFunctionExpression(
         void 0,
@@ -36559,11 +36595,12 @@ var effectFnOpportunity = createDiagnostic({
         ts.factory.createIdentifier(effectModuleName),
         "fn"
       );
-      if (traceName) {
+      if (traceNameOrExpression) {
+        const traceArg = typeof traceNameOrExpression === "string" ? ts.factory.createStringLiteral(traceNameOrExpression) : traceNameOrExpression;
         fnExpression = ts.factory.createCallExpression(
           fnExpression,
           void 0,
-          [ts.factory.createStringLiteral(traceName)]
+          [traceArg]
         );
       }
       const effectFnCall = ts.factory.createCallExpression(fnExpression, void 0, [newFunction, ...pipeArguments2]);
@@ -36605,19 +36642,35 @@ var effectFnOpportunity = createDiagnostic({
       const target = yield* pipe(parseEffectFnOpportunityTarget(node), option5);
       if (isNone2(target)) continue;
       if (target.value.hasParamsInPipeArgs) continue;
-      const { effectModuleName, nameIdentifier, node: targetNode, pipeArguments: pipeArguments2, traceName } = target.value;
+      const {
+        effectModuleName,
+        explicitTraceExpression,
+        inferredTraceName,
+        nameIdentifier,
+        node: targetNode,
+        pipeArguments: pipeArguments2
+      } = target.value;
       const innerFunction = target.value.generatorFunction ?? targetNode;
       const fixes = [];
-      fixes.push({
-        fixName: "effectFnOpportunity_toEffectFn",
-        description: traceName ? `Convert to Effect.fn("${traceName}")` : "Convert to Effect.fn",
-        apply: gen3(function* () {
-          const changeTracker = yield* service2(ChangeTracker);
-          const newNode = createEffectFnNode(targetNode, innerFunction, effectModuleName, traceName, pipeArguments2);
-          changeTracker.replaceNode(sourceFile, targetNode, newNode);
-        })
-      });
-      if (target.value.generatorFunction) {
+      if (pluginOptions.effectFn.includes("span") && explicitTraceExpression) {
+        fixes.push({
+          fixName: "effectFnOpportunity_toEffectFnWithSpan",
+          description: "Convert to Effect.fn (with span from withSpan)",
+          apply: gen3(function* () {
+            const changeTracker = yield* service2(ChangeTracker);
+            const finalPipeArguments = pipeArguments2.slice(0, -1);
+            const newNode = createEffectFnNode(
+              targetNode,
+              innerFunction,
+              effectModuleName,
+              explicitTraceExpression,
+              finalPipeArguments
+            );
+            changeTracker.replaceNode(sourceFile, targetNode, newNode);
+          })
+        });
+      }
+      if (pluginOptions.effectFn.includes("untraced") && target.value.generatorFunction) {
         fixes.push({
           fixName: "effectFnOpportunity_toEffectFnUntraced",
           description: "Convert to Effect.fnUntraced",
@@ -36628,12 +36681,67 @@ var effectFnOpportunity = createDiagnostic({
           })
         });
       }
-      const pipeArgsSuffix = pipeArguments2.length > 0 ? ` Effect.fn also accepts the piped transformations as additional arguments.` : ``;
-      const suggestConsultingQuickFixes = ` Your editor quickfixes or the "effect-language-service" cli can show you how to convert to Effect.fn or Effect.fnUntraced.`;
-      const orFnUntraced = target.value.generatorFunction ? `, or Effect.fnUntraced` : ``;
+      if (pluginOptions.effectFn.includes("no-span")) {
+        fixes.push({
+          fixName: "effectFnOpportunity_toEffectFnNoSpan",
+          description: "Convert to Effect.fn (no span)",
+          apply: gen3(function* () {
+            const changeTracker = yield* service2(ChangeTracker);
+            const newNode = createEffectFnNode(targetNode, innerFunction, effectModuleName, void 0, pipeArguments2);
+            changeTracker.replaceNode(sourceFile, targetNode, newNode);
+          })
+        });
+      }
+      if (pluginOptions.effectFn.includes("inferred-span") && inferredTraceName && !explicitTraceExpression) {
+        fixes.push({
+          fixName: "effectFnOpportunity_toEffectFnSpanInferred",
+          description: `Convert to Effect.fn("${inferredTraceName}")`,
+          apply: gen3(function* () {
+            const changeTracker = yield* service2(ChangeTracker);
+            const newNode = createEffectFnNode(
+              targetNode,
+              innerFunction,
+              effectModuleName,
+              inferredTraceName,
+              pipeArguments2
+            );
+            changeTracker.replaceNode(sourceFile, targetNode, newNode);
+          })
+        });
+      }
+      if (fixes.length === 0) continue;
+      const generateExpectedSignature = () => {
+        const firstFix = fixes[0];
+        if (!firstFix) return "Effect.fn(function*() { ... })";
+        const typeParamNames = targetNode.typeParameters ? `<${targetNode.typeParameters.map((tp) => ts.idText(tp.name)).join(", ")}>` : "";
+        const paramNames = targetNode.parameters.map((param) => {
+          if (ts.isIdentifier(param.name)) {
+            return ts.idText(param.name);
+          }
+          return "_";
+        }).join(", ");
+        const fnSignature = `function*${typeParamNames}(${paramNames}) { ... }`;
+        const pipeArgsForWithSpan = pipeArguments2.slice(0, -1);
+        const pipeArgsSuffix = (args3) => args3.length > 0 ? ", ...pipeTransformations" : "";
+        switch (firstFix.fixName) {
+          case "effectFnOpportunity_toEffectFnWithSpan": {
+            const traceName = explicitTraceExpression ? sourceFile.text.slice(explicitTraceExpression.pos, explicitTraceExpression.end).trim() : void 0;
+            return `${effectModuleName}.fn(${traceName})(${fnSignature}${pipeArgsSuffix(pipeArgsForWithSpan)})`;
+          }
+          case "effectFnOpportunity_toEffectFnUntraced":
+            return `${effectModuleName}.fnUntraced(${fnSignature}${pipeArgsSuffix(pipeArguments2)})`;
+          case "effectFnOpportunity_toEffectFnNoSpan":
+            return `${effectModuleName}.fn(${fnSignature}${pipeArgsSuffix(pipeArguments2)})`;
+          case "effectFnOpportunity_toEffectFnSpanInferred":
+            return `${effectModuleName}.fn("${inferredTraceName}")(${fnSignature}${pipeArgsSuffix(pipeArguments2)})`;
+          default:
+            return `${effectModuleName}.fn(${fnSignature})`;
+        }
+      };
+      const expectedSignature = generateExpectedSignature();
       report({
         location: nameIdentifier ?? targetNode,
-        messageText: `This function could benefit from Effect.fn's automatic tracing and concise syntax${orFnUntraced}.${pipeArgsSuffix}${suggestConsultingQuickFixes}`,
+        messageText: `Can be rewritten as a reusable function: ${expectedSignature}`,
         fixes
       });
     }