handle ExitFromSelf, Exit, CauseFromSelf, Cause

Effect-TS · Jul 23, 2024 · 423dd35 · 423dd35
1 parent 0a01c5e
commit 423dd35
Show file tree

Hide file tree

Showing 2 changed files with 130 additions and 0 deletions.
diff --git a/public/codemods/schema-0.69.ts b/public/codemods/schema-0.69.ts
@@ -267,6 +267,47 @@ export default function transformer(file: cs.FileInfo, api: cs.API) {
       "CauseDefectUnknown",
       "Defect",
     )
+
+    // ---------------------------------------------------------------
+    // ExitFromSelf, Exit, CauseFromSelf, Cause
+    // ---------------------------------------------------------------
+
+    const addDefect = (api: string) =>
+      root.find(j.CallExpression, {
+        callee: {
+          type: "MemberExpression",
+          object: { name: schemaNamespace },
+          property: { name: api },
+        },
+      }).forEach(path => {
+        const args = path.value.arguments
+        if (args.length === 1 && args[0].type === "ObjectExpression") {
+          const properties = args[0].properties
+          if (
+            !properties.some(prop => {
+              return prop.type === "ObjectProperty"
+                && prop.key.type === "Identifier"
+                && prop.key.name === "defect"
+            })
+          ) {
+            properties.push(
+              j.property(
+                "init",
+                j.identifier("defect"),
+                j.memberExpression(
+                  j.identifier(schemaNamespace),
+                  j.identifier("Defect"),
+                ),
+              ),
+            )
+          }
+        }
+      })
+
+    addDefect("ExitFromSelf")
+    addDefect("Exit")
+    addDefect("CauseFromSelf")
+    addDefect("Cause")
   }
 
   return root.toSource()

diff --git a/test/schema-0.69.test.ts b/test/schema-0.69.test.ts
@@ -259,3 +259,92 @@ const schema = Schema.CauseDefectUnknown`,
 const schema = Schema.Defect`,
   )
 })
+
+describe("ExitFromSelf", () => {
+  expectTransformation(
+    "explicit defect",
+    `import { Schema as S } from "@effect/schema"
+const schema = S.ExitFromSelf({ failure: S.String, success: S.Number, defect: S.Unknown })`,
+    `import { Schema as S } from "@effect/schema"
+const schema = S.ExitFromSelf({ failure: S.String, success: S.Number, defect: S.Unknown })`,
+  )
+  expectTransformation(
+    "explicit defect with CauseDefectUnknown",
+    `import { Schema as S } from "@effect/schema"
+const schema = S.ExitFromSelf({ failure: S.String, success: S.Number, defect: S.CauseDefectUnknown })`,
+    `import { Schema as S } from "@effect/schema"
+const schema = S.ExitFromSelf({ failure: S.String, success: S.Number, defect: S.Defect })`,
+  )
+  expectTransformation(
+    "implicit defect",
+    `import { Schema as S } from "@effect/schema"
+const schema = S.ExitFromSelf({ failure: S.String, success: S.Number })`,
+    `import { Schema as S } from "@effect/schema"
+const schema = S.ExitFromSelf({
+  failure: S.String,
+  success: S.Number,
+  defect: S.Defect
+})`,
+  )
+})
+
+describe("Exit", () => {
+  expectTransformation(
+    "explicit defect",
+    `import { Schema as S } from "@effect/schema"
+const schema = S.Exit({ failure: S.String, success: S.Number, defect: S.Unknown })`,
+    `import { Schema as S } from "@effect/schema"
+const schema = S.Exit({ failure: S.String, success: S.Number, defect: S.Unknown })`,
+  )
+  expectTransformation(
+    "implicit defect",
+    `import { Schema as S } from "@effect/schema"
+const schema = S.Exit({ failure: S.String, success: S.Number })`,
+    `import { Schema as S } from "@effect/schema"
+const schema = S.Exit({
+  failure: S.String,
+  success: S.Number,
+  defect: S.Defect
+})`,
+  )
+})
+
+describe("CauseFromSelf", () => {
+  expectTransformation(
+    "explicit defect",
+    `import { Schema as S } from "@effect/schema"
+const schema = S.CauseFromSelf({ error: S.NumberFromString, defect: S.Unknown })`,
+    `import { Schema as S } from "@effect/schema"
+const schema = S.CauseFromSelf({ error: S.NumberFromString, defect: S.Unknown })`,
+  )
+  expectTransformation(
+    "implicit defect",
+    `import { Schema as S } from "@effect/schema"
+const schema = S.CauseFromSelf({ error: S.NumberFromString })`,
+    `import { Schema as S } from "@effect/schema"
+const schema = S.CauseFromSelf({
+  error: S.NumberFromString,
+  defect: S.Defect
+})`,
+  )
+})
+
+describe("Cause", () => {
+  expectTransformation(
+    "explicit defect",
+    `import { Schema as S } from "@effect/schema"
+const schema = S.Cause({ error: S.NumberFromString, defect: S.Unknown })`,
+    `import { Schema as S } from "@effect/schema"
+const schema = S.Cause({ error: S.NumberFromString, defect: S.Unknown })`,
+  )
+  expectTransformation(
+    "implicit defect",
+    `import { Schema as S } from "@effect/schema"
+const schema = S.Cause({ error: S.NumberFromString })`,
+    `import { Schema as S } from "@effect/schema"
+const schema = S.Cause({
+  error: S.NumberFromString,
+  defect: S.Defect
+})`,
+  )
+})