grafeo-ogm 1.14.0 → 2.1.0

package/dist/cjs/compilers/mutation.compiler.js

@@ -4,6 +4,7 @@ exports.MutationCompiler = void 0;
 const errors_1 = require("../errors");
 const utils_1 = require("../schema/utils");
 const validation_1 = require("../utils/validation");
+const write_coercion_1 = require("../utils/write-coercion");
 const where_compiler_1 = require("./where.compiler");
 /**
  * Relationship filter suffixes that require EXISTS subquery patterns.
@@ -182,9 +183,11 @@ class MutationCompiler {
                 if (value === undefined)
                     continue;
                 (0, validation_1.assertSafePropertyName)(key, 'update property');
+                const propDef = nodeDef.properties.get(key);
                 const paramName = `update_${key}`;
-                setClauses.push(`n.${(0, validation_1.escapeIdentifier)(key)} = $${paramName}`);
-                params[paramName] = value;
+                const valueExpr = (0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, value, propDef);
+                setClauses.push(`n.${(0, validation_1.escapeIdentifier)(key)} = ${valueExpr}`);
+                params[paramName] = (0, write_coercion_1.coerceWriteValue)(value, propDef);
             }
             if (setClauses.length > 0)
                 lines.push(`SET ${setClauses.join(', ')}`);
@@ -262,10 +265,12 @@ class MutationCompiler {
             if (key.includes('_') && !nodeDef.properties.has(key))
                 continue;
             (0, validation_1.assertSafePropertyName)(key, 'merge key');
+            const propDef = nodeDef.properties.get(key);
             const paramName = `merge_${key}`;
-            mergeProps.push(`${(0, validation_1.escapeIdentifier)(key)}: $${paramName}`);
+            const valueExpr = (0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, value, propDef);
+            mergeProps.push(`${(0, validation_1.escapeIdentifier)(key)}: ${valueExpr}`);
             mergeKeyNames.push(key);
-            params[paramName] = value;
+            params[paramName] = (0, write_coercion_1.coerceWriteValue)(value, propDef);
         }
         if (mergeProps.length === 0)
             throw new errors_1.OGMError('upsert requires at least one scalar property in "where" for MERGE key');
@@ -301,9 +306,29 @@ class MutationCompiler {
             if (nodeDef.relationships.has(key))
                 continue;
             (0, validation_1.assertSafePropertyName)(key, 'create property');
+            const propDef = nodeDef.properties.get(key);
             const paramName = `onCreate_${key}`;
-            createSets.push(`n.${(0, validation_1.escapeIdentifier)(key)} = $${paramName}`);
-            params[paramName] = value;
+            const valueExpr = (0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, value, propDef);
+            createSets.push(`n.${(0, validation_1.escapeIdentifier)(key)} = ${valueExpr}`);
+            params[paramName] = (0, write_coercion_1.coerceWriteValue)(value, propDef);
+        }
+        // `@default` values apply to the ON CREATE branch — a defaulted
+        // property not supplied in `create` and not already fixed by the
+        // MERGE key gets its default, mirroring create().
+        for (const [, propDef] of nodeDef.properties) {
+            if (propDef.defaultValue === undefined)
+                continue;
+            if (propDef.isGenerated || propDef.isCypher)
+                continue;
+            if (create[propDef.name] !== undefined)
+                continue;
+            if (mergeKeyNames.includes(propDef.name))
+                continue;
+            const paramName = `onCreate_${propDef.name}`;
+            const bound = (0, write_coercion_1.resolveDefaultWriteValue)(propDef);
+            const valueExpr = (0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, bound, propDef);
+            createSets.push(`n.${(0, validation_1.escapeIdentifier)(propDef.name)} = ${valueExpr}`);
+            params[paramName] = bound;
         }
         if (createSets.length > 0)
             lines.push(`ON CREATE SET ${createSets.join(', ')}`);
@@ -315,9 +340,11 @@ class MutationCompiler {
             if (nodeDef.relationships.has(key))
                 continue;
             (0, validation_1.assertSafePropertyName)(key, 'update property');
+            const propDef = nodeDef.properties.get(key);
             const paramName = `onMatch_${key}`;
-            updateSets.push(`n.${(0, validation_1.escapeIdentifier)(key)} = $${paramName}`);
-            params[paramName] = value;
+            const valueExpr = (0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, value, propDef);
+            updateSets.push(`n.${(0, validation_1.escapeIdentifier)(key)} = ${valueExpr}`);
+            params[paramName] = (0, write_coercion_1.coerceWriteValue)(value, propDef);
         }
         if (updateSets.length > 0)
             lines.push(`ON MATCH SET ${updateSets.join(', ')}`);
@@ -389,12 +416,24 @@ class MutationCompiler {
         for (const [, propDef] of nodeDef.properties)
             if (propDef.isGenerated && !scalarKeys.includes(propDef.name))
                 generatedFields.push(propDef.name);
+        // Whole-column `@default`s: applied only when NO item supplies the
+        // field. Per-item gaps inside a provided column stay null — one
+        // createMany batch shares a single Cypher text, and a per-item
+        // default would need coalesce(), which would also override explicit
+        // nulls.
+        const defaultedProps = [];
+        for (const [, propDef] of nodeDef.properties)
+            if (propDef.defaultValue !== undefined &&
+                !propDef.isGenerated &&
+                !propDef.isCypher &&
+                !scalarKeys.includes(propDef.name))
+                defaultedProps.push(propDef);
         // Build sanitized items (strip undefined values)
         const sanitizedItems = data.map((item) => {
             const sanitized = {};
             for (const key of scalarKeys)
                 if (item[key] !== undefined)
-                    sanitized[key] = item[key];
+                    sanitized[key] = (0, write_coercion_1.coerceWriteValue)(item[key], nodeDef.properties.get(key));
             return sanitized;
         });
         params.items = sanitizedItems;
@@ -411,16 +450,22 @@ class MutationCompiler {
                 throw new errors_1.OGMError('createMany with skipDuplicates requires at least one @id or @unique field in data');
             // MERGE on unique key(s)
             const mergeProps = mergeKeys
-                .map((k) => `${(0, validation_1.escapeIdentifier)(k)}: item.${(0, validation_1.escapeIdentifier)(k)}`)
+                .map((k) => `${(0, validation_1.escapeIdentifier)(k)}: ${(0, write_coercion_1.wrapTemporalListItemExpr)(`item.${(0, validation_1.escapeIdentifier)(k)}`, data.map((d) => d[k]), nodeDef.properties.get(k))}`)
                 .join(', ');
             lines.push(`MERGE (n:${labelStr} { ${mergeProps} })`);
             // ON CREATE SET — remaining fields + generated IDs
             const onCreateSets = [];
             for (const key of scalarKeys)
                 if (!mergeKeys.includes(key))
-                    onCreateSets.push(`n.${(0, validation_1.escapeIdentifier)(key)} = item.${(0, validation_1.escapeIdentifier)(key)}`);
+                    onCreateSets.push(`n.${(0, validation_1.escapeIdentifier)(key)} = ${(0, write_coercion_1.wrapTemporalListItemExpr)(`item.${(0, validation_1.escapeIdentifier)(key)}`, data.map((d) => d[key]), nodeDef.properties.get(key))}`);
             for (const field of generatedFields)
                 onCreateSets.push(`n.${(0, validation_1.escapeIdentifier)(field)} = randomUUID()`);
+            for (const propDef of defaultedProps) {
+                const paramName = `default_${propDef.name}`;
+                const bound = (0, write_coercion_1.resolveDefaultWriteValue)(propDef);
+                onCreateSets.push(`n.${(0, validation_1.escapeIdentifier)(propDef.name)} = ${(0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, bound, propDef)}`);
+                params[paramName] = bound;
+            }
             if (onCreateSets.length > 0)
                 lines.push(`ON CREATE SET ${onCreateSets.join(', ')}`);
         }
@@ -430,7 +475,13 @@ class MutationCompiler {
             for (const field of generatedFields)
                 propParts.push(`${(0, validation_1.escapeIdentifier)(field)}: randomUUID()`);
             for (const key of scalarKeys)
-                propParts.push(`${(0, validation_1.escapeIdentifier)(key)}: item.${(0, validation_1.escapeIdentifier)(key)}`);
+                propParts.push(`${(0, validation_1.escapeIdentifier)(key)}: ${(0, write_coercion_1.wrapTemporalListItemExpr)(`item.${(0, validation_1.escapeIdentifier)(key)}`, data.map((d) => d[key]), nodeDef.properties.get(key))}`);
+            for (const propDef of defaultedProps) {
+                const paramName = `default_${propDef.name}`;
+                const bound = (0, write_coercion_1.resolveDefaultWriteValue)(propDef);
+                propParts.push(`${(0, validation_1.escapeIdentifier)(propDef.name)}: ${(0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, bound, propDef)}`);
+                params[paramName] = bound;
+            }
             lines.push(`CREATE (n:${labelStr} { ${propParts.join(', ')} })`);
         }
         // Extra labels
@@ -442,6 +493,19 @@ class MutationCompiler {
         return { cypher: lines.join('\n'), params };
     }
     // ─── Private helpers ──────────────────────────────────────────────
+    /**
+     * Resolve the PropertyDefinition of an edge (relationship) property so
+     * Int/BigInt edge writes get the same Integer coercion as node writes.
+     * Returns undefined when the relationship declares no properties type
+     * (the coercion then passes the value through untouched).
+     */
+    getEdgePropDef(relDef, propName) {
+        if (!relDef.properties)
+            return undefined;
+        return this.schema.relationshipProperties
+            .get(relDef.properties)
+            ?.properties.get(propName);
+    }
     buildCreateProperties(input, nodeDef, prefix) {
         const parts = [];
         const propParams = {};
@@ -456,9 +520,28 @@ class MutationCompiler {
             if (value === undefined)
                 continue;
             (0, validation_1.assertSafePropertyName)(key, 'create property');
+            const propDef = nodeDef.properties.get(key);
             const paramName = `${prefix}_${key}`;
-            parts.push(`${(0, validation_1.escapeIdentifier)(key)}: $${paramName}`);
-            propParams[paramName] = value;
+            const valueExpr = (0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, value, propDef);
+            parts.push(`${(0, validation_1.escapeIdentifier)(key)}: ${valueExpr}`);
+            propParams[paramName] = (0, write_coercion_1.coerceWriteValue)(value, propDef);
+        }
+        // `@default` values — applied for any property with a default that the
+        // input leaves undefined. Explicit null is respected as null: the user
+        // asked for null, a default would override intent. Generated (@id) and
+        // @cypher (computed, not stored) properties never take defaults.
+        for (const [, propDef] of nodeDef.properties) {
+            if (propDef.defaultValue === undefined)
+                continue;
+            if (propDef.isGenerated || propDef.isCypher)
+                continue;
+            if (input[propDef.name] !== undefined)
+                continue;
+            const paramName = `${prefix}_${propDef.name}`;
+            const bound = (0, write_coercion_1.resolveDefaultWriteValue)(propDef);
+            const valueExpr = (0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, bound, propDef);
+            parts.push(`${(0, validation_1.escapeIdentifier)(propDef.name)}: ${valueExpr}`);
+            propParams[paramName] = bound;
         }
         return { propString: parts.join(', '), propParams };
     }
@@ -563,9 +646,10 @@ class MutationCompiler {
                         if (val === undefined)
                             continue;
                         (0, validation_1.assertSafePropertyName)(prop, 'edge property');
+                        const edgePropDef = this.getEdgePropDef(relDef, prop);
                         const paramName = `${prefix}_create${ci}_edge_${prop}`;
-                        setItems.push(`${relVar}.${(0, validation_1.escapeIdentifier)(prop)} = $${paramName}`);
-                        params[paramName] = val;
+                        setItems.push(`${relVar}.${(0, validation_1.escapeIdentifier)(prop)} = ${(0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, val, edgePropDef)}`);
+                        params[paramName] = (0, write_coercion_1.coerceWriteValue)(val, edgePropDef);
                     }
                     if (setItems.length > 0)
                         lines.push(`SET ${setItems.join(', ')}`);
@@ -615,9 +699,10 @@ class MutationCompiler {
                         if (val === undefined)
                             continue;
                         (0, validation_1.assertSafePropertyName)(prop, 'edge property');
+                        const edgePropDef = this.getEdgePropDef(relDef, prop);
                         const paramName = `${prefix}_conn${ci}_edge_${prop}`;
-                        setItems.push(`r_conn_${startCounter + counter - 1}.${(0, validation_1.escapeIdentifier)(prop)} = $${paramName}`);
-                        params[paramName] = val;
+                        setItems.push(`r_conn_${startCounter + counter - 1}.${(0, validation_1.escapeIdentifier)(prop)} = ${(0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, val, edgePropDef)}`);
+                        params[paramName] = (0, write_coercion_1.coerceWriteValue)(val, edgePropDef);
                     }
                     if (setItems.length > 0)
                         lines.push(`SET ${setItems.join(', ')}`);
@@ -686,9 +771,10 @@ class MutationCompiler {
                                 if (val === undefined)
                                     continue;
                                 (0, validation_1.assertSafePropertyName)(prop, 'edge property');
+                                const edgePropDef = this.getEdgePropDef(relDef, prop);
                                 const paramName = `connect_${fieldName}_${ci}_edge_${prop}`;
-                                setItems.push(`${relVar}.${(0, validation_1.escapeIdentifier)(prop)} = $${paramName}`);
-                                params[paramName] = val;
+                                setItems.push(`${relVar}.${(0, validation_1.escapeIdentifier)(prop)} = ${(0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, val, edgePropDef)}`);
+                                params[paramName] = (0, write_coercion_1.coerceWriteValue)(val, edgePropDef);
                             }
                             if (setItems.length > 0)
                                 lines.push(`SET ${setItems.join(', ')}`);
@@ -716,7 +802,7 @@ class MutationCompiler {
                         const cleaned = {};
                         for (const [k, v] of Object.entries(edge))
                             if (v !== undefined)
-                                cleaned[k] = v;
+                                cleaned[k] = (0, write_coercion_1.coerceWriteValue)(v, this.getEdgePropDef(relDef, k));
                         return { ...item, edge: cleaned };
                     });
                     lines.push('WITH n');
@@ -740,7 +826,7 @@ class MutationCompiler {
                                 `silently dropped. Split into separate connect calls, or normalise the items ` +
                                 `to share the same shape.`);
                     }
-                    const whereConditions = this.extractConnectWhereConditions(firstItem, 'target', 'connItem');
+                    const whereConditions = this.extractConnectWhereConditions(firstItem, 'target', 'connItem', targetNodeDef);
                     const policyPredicate = this.buildTargetPolicyPredicate(targetNodeDef, 'target', params, policyContext, paramCounter);
                     const unwindConditions = policyPredicate
                         ? [...whereConditions, policyPredicate]
@@ -765,7 +851,7 @@ class MutationCompiler {
                         // Replace MERGE line with one that has a variable
                         lines[lines.length - 1] = `MERGE ${mergeWithVar}`;
                         // r.prop is relationship property (escape), connItem.edge.prop is parameter map (no escape)
-                        const setItems = edgeProps.map((p) => `r.${(0, validation_1.escapeIdentifier)(p)} = connItem.edge.${p}`);
+                        const setItems = edgeProps.map((p) => `r.${(0, validation_1.escapeIdentifier)(p)} = ${(0, write_coercion_1.wrapTemporalListItemExpr)(`connItem.edge.${p}`, spec.map((it) => it.edge?.[p]), this.getEdgePropDef(relDef, p))}`);
                         lines.push(`SET ${setItems.join(', ')}`);
                     }
                 }
@@ -796,9 +882,10 @@ class MutationCompiler {
                     for (const [prop, val] of Object.entries(edgeInput)) {
                         if (val === undefined)
                             continue;
+                        const edgePropDef = this.getEdgePropDef(relDef, prop);
                         const paramName = `connect_${fieldName}_edge_${prop}`;
-                        setItems.push(`r.${(0, validation_1.escapeIdentifier)(prop)} = $${paramName}`);
-                        params[paramName] = val;
+                        setItems.push(`r.${(0, validation_1.escapeIdentifier)(prop)} = ${(0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, val, edgePropDef)}`);
+                        params[paramName] = (0, write_coercion_1.coerceWriteValue)(val, edgePropDef);
                     }
                     if (setItems.length > 0)
                         lines.push(`SET ${setItems.join(', ')}`);
@@ -1111,9 +1198,10 @@ class MutationCompiler {
                                 if (val === undefined)
                                     continue;
                                 (0, validation_1.assertSafePropertyName)(prop, 'edge property');
+                                const edgePropDef = this.getEdgePropDef(relDef, prop);
                                 const paramName = `${itemPrefix}_conn${ci}_edge_${prop}`;
-                                setItems.push(`r_conn_${key}_${ci}.${(0, validation_1.escapeIdentifier)(prop)} = $${paramName}`);
-                                params[paramName] = val;
+                                setItems.push(`r_conn_${key}_${ci}.${(0, validation_1.escapeIdentifier)(prop)} = ${(0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, val, edgePropDef)}`);
+                                params[paramName] = (0, write_coercion_1.coerceWriteValue)(val, edgePropDef);
                             }
                             if (setItems.length > 0)
                                 lines.push(`SET ${setItems.join(', ')}`);
@@ -1149,9 +1237,10 @@ class MutationCompiler {
                         if (val === undefined)
                             continue;
                         (0, validation_1.assertSafePropertyName)(prop, 'nested update property');
+                        const nodePropDef = targetNodeDef.properties.get(prop);
                         const paramName = `${itemPrefix}_set_${prop}`;
-                        setClauses.push(`${updateVar}.${(0, validation_1.escapeIdentifier)(prop)} = $${paramName}`);
-                        setParams[paramName] = val;
+                        setClauses.push(`${updateVar}.${(0, validation_1.escapeIdentifier)(prop)} = ${(0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, val, nodePropDef)}`);
+                        setParams[paramName] = (0, write_coercion_1.coerceWriteValue)(val, nodePropDef);
                     }
                     // Pre-compute SET clauses for edge (relationship) properties
                     const edgeClauses = [];
@@ -1161,9 +1250,10 @@ class MutationCompiler {
                             if (val === undefined)
                                 continue;
                             (0, validation_1.assertSafePropertyName)(prop, 'edge update property');
+                            const edgePropDef = this.getEdgePropDef(relDef, prop);
                             const paramName = `${itemPrefix}_edge_${prop}`;
-                            edgeClauses.push(`${relVar}.${(0, validation_1.escapeIdentifier)(prop)} = $${paramName}`);
-                            edgeParams[paramName] = val;
+                            edgeClauses.push(`${relVar}.${(0, validation_1.escapeIdentifier)(prop)} = ${(0, write_coercion_1.wrapTemporalWriteExpr)(`$${paramName}`, val, edgePropDef)}`);
+                            edgeParams[paramName] = (0, write_coercion_1.coerceWriteValue)(val, edgePropDef);
                         }
                     // Pre-compute nested relationship operations
                     const nestedParams = {};
@@ -1249,15 +1339,16 @@ class MutationCompiler {
                     (0, validation_1.assertSafePropertyName)(baseProp, 'where property');
                     const paramName = `${prefix}_NOT_${notProp}`;
                     const escapedProp = (0, validation_1.escapeIdentifier)(baseProp);
+                    const paramRef = (0, write_coercion_1.wrapTemporalWhereParam)(`$${paramName}`, notVal, targetNodeDef?.properties.get(baseProp), notProp.slice(baseProp.length));
                     let expr;
                     if (template === '%v = $%p')
                         // Simple equality → negate to <>
-                        expr = `${targetVar}.${escapedProp} <> $${paramName}`;
+                        expr = `${targetVar}.${escapedProp} <> ${paramRef}`;
                     else {
                         // Operator suffix — wrap with NOT if not already negated
                         const resolved = template
                             .replace('%v', `${targetVar}.${escapedProp}`)
-                            .replace('$%p', `$${paramName}`);
+                            .replace('$%p', paramRef);
                         expr = resolved.startsWith('NOT ') ? resolved : `NOT ${resolved}`;
                     }
                     conditions.push(expr);
@@ -1274,9 +1365,10 @@ class MutationCompiler {
                 const { baseProp, template } = this.parseOperatorSuffix(prop);
                 (0, validation_1.assertSafePropertyName)(baseProp, 'where property');
                 const paramName = `${prefix}_${prop}`;
+                const paramRef = (0, write_coercion_1.wrapTemporalWhereParam)(`$${paramName}`, val, targetNodeDef?.properties.get(baseProp), prop.slice(baseProp.length));
                 conditions.push(template
                     .replace('%v', `${targetVar}.${(0, validation_1.escapeIdentifier)(baseProp)}`)
-                    .replace('$%p', `$${paramName}`));
+                    .replace('$%p', paramRef));
                 params[paramName] = val;
             }
         return conditions;
@@ -1441,7 +1533,7 @@ class MutationCompiler {
         }
         return false;
     }
-    extractConnectWhereConditions(item, targetVar, itemVar) {
+    extractConnectWhereConditions(item, targetVar, itemVar, targetNodeDef) {
         const conditions = [];
         const whereSpec = item.where;
         if (!whereSpec)
@@ -1452,7 +1544,9 @@ class MutationCompiler {
             (0, validation_1.assertSafePropertyName)(baseProp, 'connect where property');
             // Replace template placeholders with dynamic UNWIND references
             // Note: itemVar map access is NOT escaped (parameter map keys, not Cypher identifiers)
-            const valueRef = `${itemVar}.where.node.${key}`;
+            // Temporal wrap keys off the FIRST item's value type — the UNWIND
+            // fast path already requires shape-homogeneous items.
+            const valueRef = (0, write_coercion_1.wrapTemporalWhereParam)(`${itemVar}.where.node.${key}`, nodeWhere[key], targetNodeDef?.properties.get(baseProp), key.slice(baseProp.length));
             conditions.push(template
                 .replace('%v', `${targetVar}.${(0, validation_1.escapeIdentifier)(baseProp)}`)
                 .replace('$%p', valueRef));