@crvouga/mockingbird-service-postgres 3.9.0 → 3.10.0

package/dist/wire/index.js

@@ -13452,6 +13452,45 @@ function validateCaseSubExpr(ctx, scope, e) {
       return;
   }
 }
+function caseBranchType(ctx, scope, e) {
+  switch (e.type) {
+    case "null_lit":
+    case "string_lit":
+      return UNKNOWN;
+    case "number_lit":
+      return numberLitValue(e.raw).t;
+    case "bool_lit":
+      return "bool";
+    case "colref":
+      return scope.lookupColumn(e.parts)?.t ?? null;
+    case "cast":
+      return resolveTypeName(ctx.state, e.target).column.id;
+    case "collate":
+      return caseBranchType(ctx, scope, e.expr);
+    case "binop": {
+      const left = caseBranchType(ctx, scope, e.left);
+      if (left !== "json" && left !== "jsonb") return null;
+      if (e.op === "->" || e.op === "#>") return left;
+      if (e.op === "->>" || e.op === "#>>") return "text";
+      return null;
+    }
+    case "case": {
+      let type = UNKNOWN;
+      const branches = [...e.whens.map((w) => w.then), ...e.elseExpr ? [e.elseExpr] : []];
+      for (const branch of branches) {
+        const next = caseBranchType(ctx, scope, branch);
+        if (next === null) return null;
+        if (next === UNKNOWN) continue;
+        const unified = unifyTypes(type, next);
+        if (unified === null) return null;
+        type = unified;
+      }
+      return type;
+    }
+    default:
+      return null;
+  }
+}
 function evalCase(ctx, scope, e) {
   validateSearchedCaseTree(ctx, scope, e);
   let chosen = null;
@@ -13475,8 +13514,10 @@ function evalCase(ctx, scope, e) {
     }
   }
   const resultExpr = chosen ?? e.elseExpr;
-  if (resultExpr === null) return tv(UNKNOWN, null);
-  return evalExpr(ctx, scope, resultExpr);
+  const result = resultExpr === null ? tv(UNKNOWN, null) : evalExpr(ctx, scope, resultExpr);
+  if (result.t !== UNKNOWN) return result;
+  const type = caseBranchType(ctx, scope, e);
+  return type !== null && type !== UNKNOWN ? castTo(ctx, result, type, {}) : result;
 }
 function evalIn(ctx, scope, e) {
   const left = evalExpr(ctx, scope, e.left);