@effect/language-service 0.16.7 → 0.17.0

package/README.md

@@ -45,6 +45,7 @@ And you're done! You'll now be able to use a set of refactor and diagnostics tha
 ### Completions
 
 - Autocomplete 'Self' in Effect.Service, Context.Tag, Schema.TaggedClass, Schema.TaggedRequest and family
+- Autocomplete Effect.gen with `function*(){}`
 
 ### Refactors
 

package/index.js

@@ -1372,50 +1372,62 @@ var tryPreserveDeclarationSemantics = fn("AST.tryPreserveDeclarationSemantics")(
     return node;
   }
 );
+var parseAccessedExpressionForCompletion = fn(
+  "AST.parseAccessedExpressionForCompletion"
+)(
+  function* (sourceFile, position) {
+    const ts = yield* service(TypeScriptApi);
+    const precedingToken = ts.findPrecedingToken(position, sourceFile, void 0, true);
+    if (!precedingToken) return yield* fail(new NodeNotFoundError());
+    let accessedObject = precedingToken;
+    let replacementSpan = ts.createTextSpan(position, 0);
+    let outerNode = precedingToken;
+    if (ts.isIdentifier(precedingToken) && precedingToken.parent && ts.isPropertyAccessExpression(precedingToken.parent)) {
+      replacementSpan = ts.createTextSpan(
+        precedingToken.parent.getStart(sourceFile),
+        precedingToken.end - precedingToken.parent.getStart(sourceFile)
+      );
+      accessedObject = precedingToken.parent.expression;
+      outerNode = precedingToken.parent;
+    } else if (ts.isToken(precedingToken) && precedingToken.kind === ts.SyntaxKind.DotToken && ts.isPropertyAccessExpression(precedingToken.parent)) {
+      replacementSpan = ts.createTextSpan(
+        precedingToken.parent.getStart(sourceFile),
+        precedingToken.end - precedingToken.parent.getStart(sourceFile)
+      );
+      accessedObject = precedingToken.parent.expression;
+      outerNode = precedingToken.parent;
+    } else if (ts.isIdentifier(precedingToken) && precedingToken.parent) {
+      replacementSpan = ts.createTextSpan(
+        precedingToken.getStart(sourceFile),
+        precedingToken.end - precedingToken.getStart(sourceFile)
+      );
+      accessedObject = precedingToken;
+      outerNode = precedingToken;
+    } else {
+      return yield* fail(new NodeNotFoundError());
+    }
+    return { accessedObject, outerNode, replacementSpan };
+  }
+);
 var parseDataForExtendsClassCompletion = fn(
   "AST.parseDataForExtendsClassCompletion"
 )(function* (sourceFile, position) {
   const ts = yield* service(TypeScriptApi);
-  const precedingToken = ts.findPrecedingToken(position, sourceFile, void 0, true);
-  if (!precedingToken) return none2();
-  let accessedObject = precedingToken;
-  let replacementSpan = ts.createTextSpan(position, 0);
-  let outerNode = precedingToken;
-  if (ts.isIdentifier(precedingToken) && precedingToken.parent && ts.isPropertyAccessExpression(precedingToken.parent)) {
-    replacementSpan = ts.createTextSpan(
-      precedingToken.parent.getStart(sourceFile),
-      precedingToken.end - precedingToken.parent.getStart(sourceFile)
-    );
-    accessedObject = precedingToken.parent.expression;
-    outerNode = precedingToken.parent;
-  } else if (ts.isToken(precedingToken) && precedingToken.kind === ts.SyntaxKind.DotToken && ts.isPropertyAccessExpression(precedingToken.parent)) {
-    replacementSpan = ts.createTextSpan(
-      precedingToken.parent.getStart(sourceFile),
-      precedingToken.end - precedingToken.parent.getStart(sourceFile)
-    );
-    accessedObject = precedingToken.parent.expression;
-    outerNode = precedingToken.parent;
-  } else if (ts.isIdentifier(precedingToken) && precedingToken.parent) {
-    replacementSpan = ts.createTextSpan(
-      precedingToken.getStart(sourceFile),
-      precedingToken.end - precedingToken.getStart(sourceFile)
-    );
-    accessedObject = precedingToken;
-    outerNode = precedingToken;
-  } else {
-    return none2();
-  }
-  if (!ts.isIdentifier(accessedObject)) return none2();
+  const { accessedObject, outerNode, replacementSpan } = yield* parseAccessedExpressionForCompletion(sourceFile, position);
+  if (!ts.isIdentifier(accessedObject)) return yield* fail(new NodeNotFoundError());
   let classDeclaration = outerNode.parent;
   while (ts.isExpressionWithTypeArguments(classDeclaration) || ts.isHeritageClause(classDeclaration)) {
     if (!classDeclaration.parent) break;
     classDeclaration = classDeclaration.parent;
   }
-  if (!ts.isClassDeclaration(classDeclaration)) return none2();
-  if (!classDeclaration.name) return none2();
-  return some2(
-    { accessedObject, classDeclaration, className: classDeclaration.name, replacementSpan }
-  );
+  if (!ts.isClassDeclaration(classDeclaration)) return yield* fail(new NodeNotFoundError());
+  if (!classDeclaration.name) return yield* fail(new NodeNotFoundError());
+  return {
+    accessedObject,
+    classDeclaration,
+    className: classDeclaration.name,
+    replacementSpan
+  };
 });
 var createEffectGenCallExpression = fn("AST.createEffectGenCallExpression")(function* (effectModuleIdentifierName, node) {
   const ts = yield* service(TypeScriptApi);
@@ -1647,7 +1659,9 @@ var contextSelfInClasses = createCompletion({
   name: "effect/contextSelfInClasses",
   apply: fn("contextSelfInClasses")(function* (sourceFile, position) {
     const ts = yield* service(TypeScriptApi);
-    const maybeInfos = yield* parseDataForExtendsClassCompletion(sourceFile, position);
+    const maybeInfos = yield* option(
+      parseDataForExtendsClassCompletion(sourceFile, position)
+    );
     if (isNone2(maybeInfos)) return [];
     const { accessedObject, className, replacementSpan } = maybeInfos.value;
     const contextName = yield* option(
@@ -1678,7 +1692,9 @@ var effectDataClasses = createCompletion({
   name: "effect/effectDataClasses",
   apply: fn("effectDataClasses")(function* (sourceFile, position) {
     const ts = yield* service(TypeScriptApi);
-    const maybeInfos = yield* parseDataForExtendsClassCompletion(sourceFile, position);
+    const maybeInfos = yield* option(
+      parseDataForExtendsClassCompletion(sourceFile, position)
+    );
     if (isNone2(maybeInfos)) return [];
     const { accessedObject, className, replacementSpan } = maybeInfos.value;
     const dataName = yield* option(
@@ -1715,7 +1731,9 @@ var effectSchemaSelfInClasses = createCompletion({
   name: "effect/effectSchemaSelfInClasses",
   apply: fn("effectSchemaSelfInClasses")(function* (sourceFile, position) {
     const ts = yield* service(TypeScriptApi);
-    const maybeInfos = yield* parseDataForExtendsClassCompletion(sourceFile, position);
+    const maybeInfos = yield* option(
+      parseDataForExtendsClassCompletion(sourceFile, position)
+    );
     if (isNone2(maybeInfos)) return [];
     const { accessedObject, className, replacementSpan } = maybeInfos.value;
     const effectSchemaName = yield* option(
@@ -1764,7 +1782,9 @@ var effectSelfInClasses = createCompletion({
   name: "effect/effectSelfInClasses",
   apply: fn("effectSelfInClasses")(function* (sourceFile, position) {
     const ts = yield* service(TypeScriptApi);
-    const maybeInfos = yield* parseDataForExtendsClassCompletion(sourceFile, position);
+    const maybeInfos = yield* option(
+      parseDataForExtendsClassCompletion(sourceFile, position)
+    );
     if (isNone2(maybeInfos)) return [];
     const { accessedObject, className, replacementSpan } = maybeInfos.value;
     const effectName = yield* option(
@@ -1790,14 +1810,6 @@ var effectSelfInClasses = createCompletion({
   })
 });
 
-// src/completions.ts
-var completions = [
-  effectSchemaSelfInClasses,
-  effectSelfInClasses,
-  contextSelfInClasses,
-  effectDataClasses
-];
-
 // src/core/TypeCheckerApi.ts
 var TypeCheckerApi = Tag("TypeChecker");
 var TypeCheckerApiCache = Tag("TypeCheckerApiCache");
@@ -2004,6 +2016,45 @@ var expectedAndRealType = fn("TypeCheckerApi.expectedAndRealType")(function* (so
   return result;
 });
 
+// src/completions/genFunctionStar.ts
+var genFunctionStar = createCompletion({
+  name: "effect/genFunctionStar",
+  apply: fn("genFunctionStar")(function* (sourceFile, position) {
+    const ts = yield* service(TypeScriptApi);
+    const typeChecker = yield* service(TypeCheckerApi);
+    const maybeInfos = yield* option(
+      parseAccessedExpressionForCompletion(sourceFile, position)
+    );
+    if (isNone2(maybeInfos)) return [];
+    const { accessedObject } = maybeInfos.value;
+    const type = typeChecker.getTypeAtLocation(accessedObject);
+    const genMemberSymbol = type.getProperty("gen");
+    if (!genMemberSymbol) return [];
+    const genType = typeChecker.getTypeOfSymbolAtLocation(genMemberSymbol, accessedObject);
+    if (genType.getCallSignatures().length === 0) return [];
+    const span = ts.createTextSpan(
+      accessedObject.end + 1,
+      Math.max(0, position - accessedObject.end - 1)
+    );
+    return [{
+      name: `gen(function*(){})`,
+      kind: ts.ScriptElementKind.constElement,
+      insertText: `gen(function*(){${"${0}"}})`,
+      replacementSpan: span,
+      isSnippet: true
+    }];
+  })
+});
+
+// src/completions.ts
+var completions = [
+  effectSchemaSelfInClasses,
+  effectSelfInClasses,
+  contextSelfInClasses,
+  genFunctionStar,
+  effectDataClasses
+];
+
 // src/utils/TypeParser.ts
 var TypeParserIssue = class {
   constructor(type, node, message) {