add toggle definition site lazy (#43)

.changeset/serious-pens-rescue.md

@@ -0,0 +1,5 @@
+---
+"@effect/language-service": patch
+---
+
+Add toggle lazy const initializer

examples/refactors/toggleLazyConst.ts

@@ -0,0 +1,9 @@
+// 3:10,5:10,7:10
+
+const test1 = 1
+
+const test2 = () => 1
+
+const test3 = () => {
+    console.log("Hello")
+}

src/refactors/index.ts

@@ -2,6 +2,7 @@ import asyncAwaitToGen from "@effect/language-service/refactors/asyncAwaitToGen"
 import asyncAwaitToGenTryPromise from "@effect/language-service/refactors/asyncAwaitToGenTryPromise"
 import functionToArrow from "@effect/language-service/refactors/functionToArrow"
 import pipeableToDatafirst from "@effect/language-service/refactors/pipeableToDatafirst"
+import toggleLazyConst from "@effect/language-service/refactors/toggleLazyConst"
 import toggleReturnTypeAnnotation from "@effect/language-service/refactors/toggleReturnTypeAnnotation"
 import toggleTypeAnnotation from "@effect/language-service/refactors/toggleTypeAnnotation"
 import wrapWithPipe from "@effect/language-service/refactors/wrapWithPipe"
@@ -13,5 +14,6 @@ export default {
   toggleTypeAnnotation,
   toggleReturnTypeAnnotation,
   wrapWithPipe,
-  pipeableToDatafirst
+  pipeableToDatafirst,
+  toggleLazyConst
 }

src/refactors/toggleLazyConst.ts

@@ -0,0 +1,41 @@
+import { createRefactor } from "@effect/language-service/refactors/definition"
+import * as AST from "@effect/language-service/utils/AST"
+import { pipe } from "@effect/language-service/utils/Function"
+import * as O from "@effect/language-service/utils/Option"
+import * as Ch from "@effect/language-service/utils/ReadonlyArray"
+
+export default createRefactor({
+  name: "effect/toggleLazyConst",
+  description: "Toggle type annotation",
+  apply: (ts) =>
+    (sourceFile, textRange) =>
+      pipe(
+        AST.getNodesContainingRange(ts)(sourceFile, textRange),
+        Ch.filter(ts.isVariableDeclaration),
+        Ch.filter((node) => AST.isNodeInRange(textRange)(node.name)),
+        Ch.filter((node) =>
+          !!node.initializer && !(ts.isArrowFunction(node.initializer) && ts.isBlock(node.initializer.body))
+        ),
+        Ch.head,
+        O.map(
+          (node) => ({
+            kind: "refactor.rewrite.effect.toggleLazyConst",
+            description: "Toggle lazy const",
+            apply: (changeTracker) => {
+              const initializer = node.initializer!
+
+              if (ts.isArrowFunction(initializer) && initializer.parameters.length === 0) {
+                // delete eventual closing bracked
+                changeTracker.deleteRange(sourceFile, { pos: initializer.body.end, end: initializer.end })
+                // remove () => {
+                changeTracker.deleteRange(sourceFile, { pos: initializer.pos, end: initializer.body.pos })
+                return
+              }
+
+              // adds () => before
+              changeTracker.insertText(sourceFile, initializer.pos, " () =>")
+            }
+          })
+        )
+      )
+})

test/__snapshots__/refactors.test.ts.snap

@@ -188,6 +188,45 @@ const test3 = pipe(T.flatMap(T.succeed(\\"Hello\\"), _ => T.log(_)), noDataFirst
 "
 `;
 
+exports[`toggleLazyConst.ts > toggleLazyConst.ts at 3:10 1`] = `
+"// Result of running refactor effect/toggleLazyConst at position 3:10
+
+const test1 = () => 1
+
+const test2 = () => 1
+
+const test3 = () => {
+    console.log(\\"Hello\\")
+}
+"
+`;
+
+exports[`toggleLazyConst.ts > toggleLazyConst.ts at 5:10 1`] = `
+"// Result of running refactor effect/toggleLazyConst at position 5:10
+
+const test1 = 1
+
+const test2 = 1
+
+const test3 = () => {
+    console.log(\\"Hello\\")
+}
+"
+`;
+
+exports[`toggleLazyConst.ts > toggleLazyConst.ts at 7:10 1`] = `
+"// Result of running refactor effect/toggleLazyConst at position 7:10
+
+const test1 = 1
+
+const test2 = () => 1
+
+const test3 = () => {
+    console.log(\\"Hello\\")
+}
+"
+`;
+
 exports[`toggleReturnTypeAnnotation.ts > toggleReturnTypeAnnotation.ts at 4:32 1`] = `
 "// Result of running refactor effect/toggleReturnTypeAnnotation at position 4:32
 import * as T from \\"@effect/io/Effect\\"