add kind to refactor definition (#31)

Effect-TS · Dec 18, 2022 · ddb5b66 · ddb5b66
1 parent 8126efd
commit ddb5b66
Show file tree

Hide file tree

Showing 9 changed files with 14 additions and 1 deletion.
diff --git a/.changeset/hungry-ghosts-wave.md b/.changeset/hungry-ghosts-wave.md
@@ -0,0 +1,5 @@
+---
+"@effect/language-service": patch
+---
+
+Add kind to each refactor definition
diff --git a/src/index.ts b/src/index.ts
@@ -38,7 +38,8 @@ function init(modules: { typescript: typeof import("typescript/lib/tsserverlibra
                     description: refactor.description,
                     actions: [{
                       name: refactor.name,
-                      description: _.description
+                      description: _.description,
+                      kind: _.kind
                     }]
                   }))
                 )

diff --git a/src/refactors/asyncAwaitToGen.ts b/src/refactors/asyncAwaitToGen.ts
@@ -16,6 +16,7 @@ export default createRefactor({
         Ch.filter((node) => !!(ts.getCombinedModifierFlags(node) & ts.ModifierFlags.Async)),
         Ch.head,
         O.map((node) => ({
+          kind: "refactor.rewrite.effect.asyncAwaitToGen",
           description: "Rewrite to Effect.gen",
           apply: (changeTracker) => {
             const effectName = AST.getEffectModuleIdentifier(ts)(sourceFile)

diff --git a/src/refactors/asyncAwaitToGenTryPromise.ts b/src/refactors/asyncAwaitToGenTryPromise.ts
@@ -16,6 +16,7 @@ export default createRefactor({
         Ch.filter((node) => !!(ts.getCombinedModifierFlags(node) & ts.ModifierFlags.Async)),
         Ch.head,
         O.map((node) => ({
+          kind: "refactor.rewrite.effect.asyncAwaitToGenTryPromise",
           description: "Rewrite to Effect.gen with failures",
           apply: (changeTracker) => {
             const effectName = AST.getEffectModuleIdentifier(ts)(sourceFile)

diff --git a/src/refactors/definition.ts b/src/refactors/definition.ts
@@ -12,6 +12,7 @@ export interface RefactorDefinition {
 }
 
 export interface ApplicableRefactorDefinition {
+  kind: string
   description: string
   apply: (changeTracker: ts.textChanges.ChangeTracker) => void
 }

diff --git a/src/refactors/functionToArrow.ts b/src/refactors/functionToArrow.ts
@@ -17,6 +17,7 @@ export default createRefactor({
         Ch.head,
         O.map(
           (node) => ({
+            kind: "refactor.rewrite.effect.functionToArrow",
             description: "Convert to arrow",
             apply: (changeTracker) => {
               const body = node.body!

diff --git a/src/refactors/toggleReturnTypeAnnotation.ts b/src/refactors/toggleReturnTypeAnnotation.ts
@@ -34,6 +34,7 @@ export default createRefactor({
         Ch.head,
         O.map(
           (node) => ({
+            kind: "refactor.rewrite.effect.toggleReturnTypeAnnotation",
             description: "Toggle return type annotation",
             apply: (changeTracker) => {
               const typeChecker = program.getTypeChecker()

diff --git a/src/refactors/toggleTypeAnnotation.ts b/src/refactors/toggleTypeAnnotation.ts
@@ -17,6 +17,7 @@ export default createRefactor({
         Ch.head,
         O.map(
           (node) => ({
+            kind: "refactor.rewrite.effect.toggleTypeAnnotation",
             description: "Toggle type annotation",
             apply: (changeTracker) => {
               const typeChecker = program.getTypeChecker()

diff --git a/src/refactors/wrapWithPipe.ts b/src/refactors/wrapWithPipe.ts
@@ -9,6 +9,7 @@ export default createRefactor({
       if (textRange.end - textRange.pos === 0) return O.none
 
       return O.some({
+        kind: "refactor.rewrite.effect.wrapWithPipe",
         description: `Wrap with pipe(...)`,
         apply: (changeTracker) => {
           changeTracker.insertText(sourceFile, textRange.pos, "pipe(")