@constructive-io/graphql-codegen 4.47.9 → 4.47.11

package/core/codegen/index.js

@@ -138,6 +138,7 @@ function generate(options) {
     const mutationHooks = (0, mutations_1.generateAllMutationHooks)(tables, {
         reactQueryEnabled,
         useCentralizedKeys,
+        typeRegistry: customOperations?.typeRegistry,
     });
     for (const hook of mutationHooks) {
         files.push({

package/core/codegen/mutations.d.ts

@@ -1,4 +1,4 @@
-import type { Table } from '../../types/schema';
+import type { Table, TypeRegistry } from '../../types/schema';
 export interface GeneratedMutationFile {
     fileName: string;
     content: string;
@@ -6,6 +6,7 @@ export interface GeneratedMutationFile {
 export interface MutationGeneratorOptions {
     reactQueryEnabled?: boolean;
     useCentralizedKeys?: boolean;
+    typeRegistry?: TypeRegistry;
 }
 export declare function generateCreateMutationHook(table: Table, options?: MutationGeneratorOptions): GeneratedMutationFile | null;
 export declare function generateUpdateMutationHook(table: Table, options?: MutationGeneratorOptions): GeneratedMutationFile | null;

package/core/codegen/mutations.js

@@ -172,7 +172,7 @@ function generateCreateMutationHook(table, options = {}) {
     };
 }
 function generateUpdateMutationHook(table, options = {}) {
-    const { reactQueryEnabled = true, useCentralizedKeys = true } = options;
+    const { reactQueryEnabled = true, useCentralizedKeys = true, typeRegistry } = options;
     if (!reactQueryEnabled)
         return null;
     if (table.query?.update === null)
@@ -191,6 +191,8 @@ function generateUpdateMutationHook(table, options = {}) {
     const pkField = pkFields[0];
     const patchFieldName = table.query?.patchFieldName ?? (0, utils_1.lcFirst)(typeName) + 'Patch';
     const pkTsType = pkField.tsType === 'string' ? t.tsStringKeyword() : t.tsNumberKeyword();
+    const updateInputTypeName = (0, utils_1.getUpdateInputTypeName)(table);
+    const updateExtraKeys = (0, utils_1.getExtraInputKeys)(updateInputTypeName, new Set(pkFields.map((pk) => pk.name)), patchFieldName, typeRegistry);
     const statements = [];
     // Imports
     statements.push((0, hooks_ast_1.createImportDeclaration)('@tanstack/react-query', [
@@ -209,9 +211,12 @@ function generateUpdateMutationHook(table, options = {}) {
     statements.push((0, hooks_ast_1.createImportDeclaration)('../../orm/select-types', ['InferSelectResult', 'HookStrictSelect'], true));
     // Re-exports
     statements.push((0, hooks_ast_1.createTypeReExport)([selectTypeName, relationTypeName, patchTypeName], '../../orm/input-types'));
-    // Variable type: { pkField: type; patchFieldName: PatchType }
+    // Variable type: { pkField: type; extraKeys...; patchFieldName: PatchType }
     const updateVarType = t.tsTypeLiteral([
         t.tsPropertySignature(t.identifier(pkField.name), t.tsTypeAnnotation(pkTsType)),
+        ...updateExtraKeys.map((ek) => t.tsPropertySignature(t.identifier(ek.name), t.tsTypeAnnotation(ek.tsType === 'number' ? t.tsNumberKeyword() :
+            ek.tsType === 'boolean' ? t.tsBooleanKeyword() :
+                t.tsStringKeyword()))),
         t.tsPropertySignature(t.identifier(patchFieldName), t.tsTypeAnnotation((0, hooks_ast_1.typeRef)(patchTypeName))),
     ]);
     const resultType = (sel) => buildMutationResultType(mutationName, singularName, relationTypeName, sel);
@@ -258,11 +263,15 @@ function generateUpdateMutationHook(table, options = {}) {
     //   getClient().singular.update({ where: { pkField }, data: patchFieldName, select: ... }).unwrap()
     const destructParam = t.objectPattern([
         (0, hooks_ast_1.shorthandProp)(pkField.name),
+        ...updateExtraKeys.map((ek) => (0, hooks_ast_1.shorthandProp)(ek.name)),
         (0, hooks_ast_1.shorthandProp)(patchFieldName),
     ]);
     destructParam.typeAnnotation = t.tsTypeAnnotation(updateVarType);
     const mutationFnExpr = t.arrowFunctionExpression([destructParam], (0, hooks_ast_1.getClientCallUnwrap)(singularName, 'update', t.objectExpression([
-        (0, hooks_ast_1.objectProp)('where', t.objectExpression([(0, hooks_ast_1.shorthandProp)(pkField.name)])),
+        (0, hooks_ast_1.objectProp)('where', t.objectExpression([
+            (0, hooks_ast_1.shorthandProp)(pkField.name),
+            ...updateExtraKeys.map((ek) => (0, hooks_ast_1.shorthandProp)(ek.name)),
+        ])),
         (0, hooks_ast_1.objectProp)('data', t.identifier(patchFieldName)),
         (0, hooks_ast_1.objectProp)('select', t.memberExpression(t.identifier('args'), t.identifier('select'))),
     ])));
@@ -299,7 +308,7 @@ function generateUpdateMutationHook(table, options = {}) {
     };
 }
 function generateDeleteMutationHook(table, options = {}) {
-    const { reactQueryEnabled = true, useCentralizedKeys = true } = options;
+    const { reactQueryEnabled = true, useCentralizedKeys = true, typeRegistry } = options;
     if (!reactQueryEnabled)
         return null;
     if (table.query?.delete === null)
@@ -316,6 +325,8 @@ function generateDeleteMutationHook(table, options = {}) {
     const pkFields = (0, utils_1.getPrimaryKeyInfo)(table);
     const pkField = pkFields[0];
     const pkTsType = pkField.tsType === 'string' ? t.tsStringKeyword() : t.tsNumberKeyword();
+    const deleteInputTypeName = (0, utils_1.getDeleteInputTypeName)(table);
+    const deleteExtraKeys = (0, utils_1.getExtraInputKeys)(deleteInputTypeName, new Set(pkFields.map((pk) => pk.name)), null, typeRegistry);
     const statements = [];
     // Imports
     statements.push((0, hooks_ast_1.createImportDeclaration)('@tanstack/react-query', [
@@ -334,9 +345,12 @@ function generateDeleteMutationHook(table, options = {}) {
     statements.push((0, hooks_ast_1.createImportDeclaration)('../../orm/select-types', ['InferSelectResult', 'HookStrictSelect'], true));
     // Re-exports
     statements.push((0, hooks_ast_1.createTypeReExport)([selectTypeName, relationTypeName], '../../orm/input-types'));
-    // Variable type: { pkField: type }
+    // Variable type: { pkField: type; extraKeys... }
     const deleteVarType = t.tsTypeLiteral([
         t.tsPropertySignature(t.identifier(pkField.name), t.tsTypeAnnotation(pkTsType)),
+        ...deleteExtraKeys.map((ek) => t.tsPropertySignature(t.identifier(ek.name), t.tsTypeAnnotation(ek.tsType === 'number' ? t.tsNumberKeyword() :
+            ek.tsType === 'boolean' ? t.tsBooleanKeyword() :
+                t.tsStringKeyword()))),
     ]);
     const resultType = (sel) => buildMutationResultType(mutationName, singularName, relationTypeName, sel);
     // Overload 1: with fields
@@ -380,10 +394,16 @@ function generateDeleteMutationHook(table, options = {}) {
         : undefined;
     // mutationFn: ({ pkField }: { pkField: type }) =>
     //   getClient().singular.delete({ where: { pkField }, select: ... }).unwrap()
-    const destructParam = t.objectPattern([(0, hooks_ast_1.shorthandProp)(pkField.name)]);
+    const destructParam = t.objectPattern([
+        (0, hooks_ast_1.shorthandProp)(pkField.name),
+        ...deleteExtraKeys.map((ek) => (0, hooks_ast_1.shorthandProp)(ek.name)),
+    ]);
     destructParam.typeAnnotation = t.tsTypeAnnotation(deleteVarType);
     const mutationFnExpr = t.arrowFunctionExpression([destructParam], (0, hooks_ast_1.getClientCallUnwrap)(singularName, 'delete', t.objectExpression([
-        (0, hooks_ast_1.objectProp)('where', t.objectExpression([(0, hooks_ast_1.shorthandProp)(pkField.name)])),
+        (0, hooks_ast_1.objectProp)('where', t.objectExpression([
+            (0, hooks_ast_1.shorthandProp)(pkField.name),
+            ...deleteExtraKeys.map((ek) => (0, hooks_ast_1.shorthandProp)(ek.name)),
+        ])),
         (0, hooks_ast_1.objectProp)('select', t.memberExpression(t.identifier('args'), t.identifier('select'))),
     ])));
     // onSuccess: remove detail, invalidate lists

package/core/codegen/orm/index.js

@@ -35,7 +35,7 @@ function generateOrm(options) {
         content: selectTypesFile.content,
     });
     // 2. Generate model files
-    const modelFiles = (0, model_generator_1.generateAllModelFiles)(tables, useSharedTypes);
+    const modelFiles = (0, model_generator_1.generateAllModelFiles)(tables, useSharedTypes, typeRegistry);
     for (const modelFile of modelFiles) {
         files.push({
             path: `models/${modelFile.fileName}`,

package/core/codegen/orm/model-generator.d.ts

@@ -1,9 +1,9 @@
-import type { Table } from '../../../types/schema';
+import type { Table, TypeRegistry } from '../../../types/schema';
 export interface GeneratedModelFile {
     fileName: string;
     content: string;
     modelName: string;
     tableName: string;
 }
-export declare function generateModelFile(table: Table, _useSharedTypes: boolean, options?: Record<string, never>, allTables?: Table[]): GeneratedModelFile;
-export declare function generateAllModelFiles(tables: Table[], useSharedTypes: boolean): GeneratedModelFile[];
+export declare function generateModelFile(table: Table, _useSharedTypes: boolean, options?: Record<string, never>, allTables?: Table[], typeRegistry?: TypeRegistry): GeneratedModelFile;
+export declare function generateAllModelFiles(tables: Table[], useSharedTypes: boolean, typeRegistry?: TypeRegistry): GeneratedModelFile[];

package/core/codegen/orm/model-generator.js

@@ -103,7 +103,8 @@ function strictSelectGuard(selectTypeName) {
         t.tsTypeReference(t.identifier(selectTypeName)),
     ]));
 }
-function generateModelFile(table, _useSharedTypes, options, allTables) {
+// ExtraInputKey type and getExtraInputKeys are imported from ../utils
+function generateModelFile(table, _useSharedTypes, options, allTables, typeRegistry) {
     const { typeName, singularName, pluralName } = (0, utils_1.getTableNames)(table);
     const modelName = `${typeName}Model`;
     const baseFileName = (0, utils_1.lcFirst)(typeName);
@@ -114,7 +115,7 @@ function generateModelFile(table, _useSharedTypes, options, allTables) {
     const whereTypeName = (0, utils_1.getFilterTypeName)(table);
     const orderByTypeName = (0, utils_1.getOrderByTypeName)(table);
     const createInputTypeName = `Create${typeName}Input`;
-    const updateInputTypeName = `Update${typeName}Input`;
+    const updateInputTypeName = (0, utils_1.getUpdateInputTypeName)(table);
     const patchTypeName = `${typeName}Patch`;
     const pkFields = (0, utils_1.getPrimaryKeyInfo)(table);
     const pkField = pkFields[0];
@@ -401,12 +402,19 @@ function generateModelFile(table, _useSharedTypes, options, allTables) {
     }
     // ── update ─────────────────────────────────────────────────────────────
     if (updateMutationName) {
+        const patchFieldName = table.query?.patchFieldName ?? (0, utils_1.lcFirst)(typeName) + 'Patch';
+        const updateExtraKeys = (0, utils_1.getExtraInputKeys)(updateInputTypeName, new Set(pkFields.map((pk) => pk.name)), patchFieldName, typeRegistry);
         const whereLiteral = () => t.tsTypeLiteral([
             (() => {
                 const prop = t.tsPropertySignature(t.identifier(pkField.name), t.tsTypeAnnotation(pkTsType()));
                 prop.optional = false;
                 return prop;
             })(),
+            ...updateExtraKeys.map((ek) => {
+                const prop = t.tsPropertySignature(t.identifier(ek.name), t.tsTypeAnnotation(tsTypeFromPrimitive(ek.tsType)));
+                prop.optional = false;
+                return prop;
+            }),
         ]);
         const argsType = (sel) => t.tsTypeReference(t.identifier('UpdateArgs'), t.tsTypeParameterInstantiation([
             sel,
@@ -430,7 +438,10 @@ function generateModelFile(table, _useSharedTypes, options, allTables) {
             strictSelectGuard(selectTypeName),
         ]));
         const selectExpr = t.memberExpression(t.identifier('args'), t.identifier('select'));
-        const patchFieldName = table.query?.patchFieldName ?? (0, utils_1.lcFirst)(typeName) + 'Patch';
+        // Build extraKeys object for partitioned table fields
+        const extraKeysArg = updateExtraKeys.length > 0
+            ? t.objectExpression(updateExtraKeys.map((ek) => t.objectProperty(t.identifier(ek.name), t.memberExpression(t.memberExpression(t.identifier('args'), t.identifier('where')), t.identifier(ek.name)))))
+            : t.identifier('undefined');
         const bodyArgs = [
             t.stringLiteral(typeName),
             t.stringLiteral(updateMutationName),
@@ -442,17 +453,26 @@ function generateModelFile(table, _useSharedTypes, options, allTables) {
             t.stringLiteral(pkField.name),
             t.stringLiteral(patchFieldName),
             t.identifier('connectionFieldsMap'),
+            extraKeysArg,
         ];
         classBody.push(createClassMethod('update', createTypeParam(selectTypeName), [implParam], retType(sRef()), buildMethodBody('buildUpdateByPkDocument', bodyArgs, 'mutation', typeName, updateMutationName)));
     }
     // ── delete ─────────────────────────────────────────────────────────────
     if (deleteMutationName) {
-        // Build where type with ALL PK fields (supports composite PKs)
-        const whereLiteral = () => t.tsTypeLiteral(pkFields.map((pk) => {
-            const prop = t.tsPropertySignature(t.identifier(pk.name), t.tsTypeAnnotation(tsTypeFromPrimitive(pk.tsType ?? 'string')));
-            prop.optional = false;
-            return prop;
-        }));
+        const deleteExtraKeys = (0, utils_1.getExtraInputKeys)(deleteInputTypeName, new Set(pkFields.map((pk) => pk.name)), null, typeRegistry);
+        // Build where type with ALL PK fields + extra keys (supports composite PKs + partition keys)
+        const whereLiteral = () => t.tsTypeLiteral([
+            ...pkFields.map((pk) => {
+                const prop = t.tsPropertySignature(t.identifier(pk.name), t.tsTypeAnnotation(tsTypeFromPrimitive(pk.tsType ?? 'string')));
+                prop.optional = false;
+                return prop;
+            }),
+            ...deleteExtraKeys.map((ek) => {
+                const prop = t.tsPropertySignature(t.identifier(ek.name), t.tsTypeAnnotation(tsTypeFromPrimitive(ek.tsType)));
+                prop.optional = false;
+                return prop;
+            }),
+        ]);
         const argsType = (sel) => t.tsTypeReference(t.identifier('DeleteArgs'), t.tsTypeParameterInstantiation([whereLiteral(), sel]));
         const retType = (sel) => t.tsTypeAnnotation(t.tsTypeReference(t.identifier('QueryBuilder'), t.tsTypeParameterInstantiation([
             t.tsTypeLiteral([
@@ -471,8 +491,11 @@ function generateModelFile(table, _useSharedTypes, options, allTables) {
             strictSelectGuard(selectTypeName),
         ]));
         const selectExpr = t.memberExpression(t.identifier('args'), t.identifier('select'));
-        // Build keys object: { field1: args.where.field1, field2: args.where.field2, ... }
-        const keysObj = t.objectExpression(pkFields.map((pk) => t.objectProperty(t.identifier(pk.name), t.memberExpression(t.memberExpression(t.identifier('args'), t.identifier('where')), t.identifier(pk.name)))));
+        // Build keys object: { field1: args.where.field1, ... } including extra keys
+        const keysObj = t.objectExpression([
+            ...pkFields.map((pk) => t.objectProperty(t.identifier(pk.name), t.memberExpression(t.memberExpression(t.identifier('args'), t.identifier('where')), t.identifier(pk.name)))),
+            ...deleteExtraKeys.map((ek) => t.objectProperty(t.identifier(ek.name), t.memberExpression(t.memberExpression(t.identifier('args'), t.identifier('where')), t.identifier(ek.name)))),
+        ]);
         const bodyArgs = [
             t.stringLiteral(typeName),
             t.stringLiteral(deleteMutationName),
@@ -748,6 +771,6 @@ function generateModelFile(table, _useSharedTypes, options, allTables) {
         tableName: table.name,
     };
 }
-function generateAllModelFiles(tables, useSharedTypes) {
-    return tables.map((table) => generateModelFile(table, useSharedTypes, undefined, tables));
+function generateAllModelFiles(tables, useSharedTypes, typeRegistry) {
+    return tables.map((table) => generateModelFile(table, useSharedTypes, undefined, tables, typeRegistry));
 }

package/core/codegen/templates/query-builder.ts

@@ -509,6 +509,7 @@ export function buildUpdateByPkDocument<TSelect, TData>(
   idFieldName: string,
   patchFieldName: string,
   connectionFieldsMap?: Record<string, Record<string, string>>,
+  extraKeys?: Record<string, unknown>,
 ): { document: string; variables: Record<string, unknown> } {
   const selections = select
     ? buildSelections(
@@ -533,6 +534,7 @@ export function buildUpdateByPkDocument<TSelect, TData>(
     variables: {
       input: {
         [idFieldName]: id,
+        ...extraKeys,
         [patchFieldName]: data,
       },
     },

package/core/codegen/utils.d.ts

@@ -146,6 +146,17 @@ export declare function getUpdateInputTypeName(table: Table): string;
  * like DeletePostTagByPostIdAndTagIdInput), falls back to Delete${Entity}Input.
  */
 export declare function getDeleteInputTypeName(table: Table): string;
+export interface ExtraInputKey {
+    name: string;
+    gqlType: string;
+    tsType: string;
+}
+/**
+ * Discover extra required fields on a mutation input type beyond the standard
+ * ones (clientMutationId, PK fields, patch field). PostGraphile adds these for
+ * partitioned tables (e.g. databaseId as the partition key).
+ */
+export declare function getExtraInputKeys(inputTypeName: string, pkFieldNames: Set<string>, patchFieldName: string | null, typeRegistry?: TypeRegistry): ExtraInputKey[];
 /**
  * Convert GraphQL type to TypeScript type
  */

package/core/codegen/utils.js

@@ -35,6 +35,7 @@ exports.getCreateInputTypeName = getCreateInputTypeName;
 exports.getPatchTypeName = getPatchTypeName;
 exports.getUpdateInputTypeName = getUpdateInputTypeName;
 exports.getDeleteInputTypeName = getDeleteInputTypeName;
+exports.getExtraInputKeys = getExtraInputKeys;
 exports.gqlTypeToTs = gqlTypeToTs;
 exports.fieldTypeToTs = fieldTypeToTs;
 exports.getScalarFilterType = getScalarFilterType;
@@ -295,6 +296,38 @@ function getDeleteInputTypeName(table) {
     const mutationName = table.query?.delete;
     return mutationName ? (0, inflekt_1.ucFirst)(mutationName) + 'Input' : `Delete${table.name}Input`;
 }
+/**
+ * Discover extra required fields on a mutation input type beyond the standard
+ * ones (clientMutationId, PK fields, patch field). PostGraphile adds these for
+ * partitioned tables (e.g. databaseId as the partition key).
+ */
+function getExtraInputKeys(inputTypeName, pkFieldNames, patchFieldName, typeRegistry) {
+    if (!typeRegistry)
+        return [];
+    const inputType = typeRegistry.get(inputTypeName);
+    if (!inputType || inputType.kind !== 'INPUT_OBJECT' || !inputType.inputFields)
+        return [];
+    const skip = new Set(['clientMutationId', ...(patchFieldName ? [patchFieldName] : [])]);
+    for (const pk of pkFieldNames)
+        skip.add(pk);
+    const extras = [];
+    for (const field of inputType.inputFields) {
+        if (skip.has(field.name))
+            continue;
+        if (field.type.kind !== 'NON_NULL')
+            continue;
+        const innerName = field.type.ofType?.name;
+        if (!innerName)
+            continue;
+        let tsType = 'string';
+        if (innerName === 'Int' || innerName === 'Float' || innerName === 'BigFloat')
+            tsType = 'number';
+        else if (innerName === 'Boolean')
+            tsType = 'boolean';
+        extras.push({ name: field.name, gqlType: innerName, tsType });
+    }
+    return extras;
+}
 // ============================================================================
 // Type mapping: GraphQL → TypeScript
 // ============================================================================

package/esm/core/codegen/index.js

@@ -133,6 +133,7 @@ export function generate(options) {
     const mutationHooks = generateAllMutationHooks(tables, {
         reactQueryEnabled,
         useCentralizedKeys,
+        typeRegistry: customOperations?.typeRegistry,
     });
     for (const hook of mutationHooks) {
         files.push({

package/esm/core/codegen/mutations.d.ts

@@ -1,4 +1,4 @@
-import type { Table } from '../../types/schema';
+import type { Table, TypeRegistry } from '../../types/schema';
 export interface GeneratedMutationFile {
     fileName: string;
     content: string;
@@ -6,6 +6,7 @@ export interface GeneratedMutationFile {
 export interface MutationGeneratorOptions {
     reactQueryEnabled?: boolean;
     useCentralizedKeys?: boolean;
+    typeRegistry?: TypeRegistry;
 }
 export declare function generateCreateMutationHook(table: Table, options?: MutationGeneratorOptions): GeneratedMutationFile | null;
 export declare function generateUpdateMutationHook(table: Table, options?: MutationGeneratorOptions): GeneratedMutationFile | null;

package/esm/core/codegen/mutations.js

@@ -9,7 +9,7 @@
  */
 import * as t from '@babel/types';
 import { addJSDocComment, buildSelectionArgsCall, callExpr, constDecl, createFunctionParam, createImportDeclaration, createSTypeParam, createTypeReExport, destructureParamsWithSelection, exportDeclareFunction, exportFunction, generateHookFileCode, getClientCallUnwrap, inferSelectResultType, objectProp, omitType, returnUseMutation, selectionConfigType, shorthandProp, spreadObj, sRef, typeRef, typeLiteralWithProps, useMutationOptionsType, useMutationResultType, voidStatement, } from './hooks-ast';
-import { getBulkCreateMutationFileName, getBulkCreateMutationHookName, getBulkDeleteMutationFileName, getBulkDeleteMutationHookName, getBulkUpdateMutationFileName, getBulkUpdateMutationHookName, getBulkUpsertMutationFileName, getBulkUpsertMutationHookName, getCreateMutationFileName, getCreateMutationHookName, getCreateMutationName, getDeleteMutationFileName, getDeleteMutationHookName, getDeleteMutationName, getPrimaryKeyInfo, getTableNames, getUpdateMutationFileName, getUpdateMutationHookName, getUpdateMutationName, hasValidPrimaryKey, lcFirst, } from './utils';
+import { getBulkCreateMutationFileName, getBulkCreateMutationHookName, getBulkDeleteMutationFileName, getBulkDeleteMutationHookName, getBulkUpdateMutationFileName, getBulkUpdateMutationHookName, getBulkUpsertMutationFileName, getBulkUpsertMutationHookName, getCreateMutationFileName, getCreateMutationHookName, getCreateMutationName, getDeleteInputTypeName, getDeleteMutationFileName, getDeleteMutationHookName, getDeleteMutationName, getExtraInputKeys, getPrimaryKeyInfo, getTableNames, getUpdateInputTypeName, getUpdateMutationFileName, getUpdateMutationHookName, getUpdateMutationName, hasValidPrimaryKey, lcFirst, } from './utils';
 function buildMutationResultType(mutationName, singularName, relationTypeName, selectType) {
     return typeLiteralWithProps([
         {
@@ -133,7 +133,7 @@ export function generateCreateMutationHook(table, options = {}) {
     };
 }
 export function generateUpdateMutationHook(table, options = {}) {
-    const { reactQueryEnabled = true, useCentralizedKeys = true } = options;
+    const { reactQueryEnabled = true, useCentralizedKeys = true, typeRegistry } = options;
     if (!reactQueryEnabled)
         return null;
     if (table.query?.update === null)
@@ -152,6 +152,8 @@ export function generateUpdateMutationHook(table, options = {}) {
     const pkField = pkFields[0];
     const patchFieldName = table.query?.patchFieldName ?? lcFirst(typeName) + 'Patch';
     const pkTsType = pkField.tsType === 'string' ? t.tsStringKeyword() : t.tsNumberKeyword();
+    const updateInputTypeName = getUpdateInputTypeName(table);
+    const updateExtraKeys = getExtraInputKeys(updateInputTypeName, new Set(pkFields.map((pk) => pk.name)), patchFieldName, typeRegistry);
     const statements = [];
     // Imports
     statements.push(createImportDeclaration('@tanstack/react-query', [
@@ -170,9 +172,12 @@ export function generateUpdateMutationHook(table, options = {}) {
     statements.push(createImportDeclaration('../../orm/select-types', ['InferSelectResult', 'HookStrictSelect'], true));
     // Re-exports
     statements.push(createTypeReExport([selectTypeName, relationTypeName, patchTypeName], '../../orm/input-types'));
-    // Variable type: { pkField: type; patchFieldName: PatchType }
+    // Variable type: { pkField: type; extraKeys...; patchFieldName: PatchType }
     const updateVarType = t.tsTypeLiteral([
         t.tsPropertySignature(t.identifier(pkField.name), t.tsTypeAnnotation(pkTsType)),
+        ...updateExtraKeys.map((ek) => t.tsPropertySignature(t.identifier(ek.name), t.tsTypeAnnotation(ek.tsType === 'number' ? t.tsNumberKeyword() :
+            ek.tsType === 'boolean' ? t.tsBooleanKeyword() :
+                t.tsStringKeyword()))),
         t.tsPropertySignature(t.identifier(patchFieldName), t.tsTypeAnnotation(typeRef(patchTypeName))),
     ]);
     const resultType = (sel) => buildMutationResultType(mutationName, singularName, relationTypeName, sel);
@@ -219,11 +224,15 @@ export function generateUpdateMutationHook(table, options = {}) {
     //   getClient().singular.update({ where: { pkField }, data: patchFieldName, select: ... }).unwrap()
     const destructParam = t.objectPattern([
         shorthandProp(pkField.name),
+        ...updateExtraKeys.map((ek) => shorthandProp(ek.name)),
         shorthandProp(patchFieldName),
     ]);
     destructParam.typeAnnotation = t.tsTypeAnnotation(updateVarType);
     const mutationFnExpr = t.arrowFunctionExpression([destructParam], getClientCallUnwrap(singularName, 'update', t.objectExpression([
-        objectProp('where', t.objectExpression([shorthandProp(pkField.name)])),
+        objectProp('where', t.objectExpression([
+            shorthandProp(pkField.name),
+            ...updateExtraKeys.map((ek) => shorthandProp(ek.name)),
+        ])),
         objectProp('data', t.identifier(patchFieldName)),
         objectProp('select', t.memberExpression(t.identifier('args'), t.identifier('select'))),
     ])));
@@ -260,7 +269,7 @@ export function generateUpdateMutationHook(table, options = {}) {
     };
 }
 export function generateDeleteMutationHook(table, options = {}) {
-    const { reactQueryEnabled = true, useCentralizedKeys = true } = options;
+    const { reactQueryEnabled = true, useCentralizedKeys = true, typeRegistry } = options;
     if (!reactQueryEnabled)
         return null;
     if (table.query?.delete === null)
@@ -277,6 +286,8 @@ export function generateDeleteMutationHook(table, options = {}) {
     const pkFields = getPrimaryKeyInfo(table);
     const pkField = pkFields[0];
     const pkTsType = pkField.tsType === 'string' ? t.tsStringKeyword() : t.tsNumberKeyword();
+    const deleteInputTypeName = getDeleteInputTypeName(table);
+    const deleteExtraKeys = getExtraInputKeys(deleteInputTypeName, new Set(pkFields.map((pk) => pk.name)), null, typeRegistry);
     const statements = [];
     // Imports
     statements.push(createImportDeclaration('@tanstack/react-query', [
@@ -295,9 +306,12 @@ export function generateDeleteMutationHook(table, options = {}) {
     statements.push(createImportDeclaration('../../orm/select-types', ['InferSelectResult', 'HookStrictSelect'], true));
     // Re-exports
     statements.push(createTypeReExport([selectTypeName, relationTypeName], '../../orm/input-types'));
-    // Variable type: { pkField: type }
+    // Variable type: { pkField: type; extraKeys... }
     const deleteVarType = t.tsTypeLiteral([
         t.tsPropertySignature(t.identifier(pkField.name), t.tsTypeAnnotation(pkTsType)),
+        ...deleteExtraKeys.map((ek) => t.tsPropertySignature(t.identifier(ek.name), t.tsTypeAnnotation(ek.tsType === 'number' ? t.tsNumberKeyword() :
+            ek.tsType === 'boolean' ? t.tsBooleanKeyword() :
+                t.tsStringKeyword()))),
     ]);
     const resultType = (sel) => buildMutationResultType(mutationName, singularName, relationTypeName, sel);
     // Overload 1: with fields
@@ -341,10 +355,16 @@ export function generateDeleteMutationHook(table, options = {}) {
         : undefined;
     // mutationFn: ({ pkField }: { pkField: type }) =>
     //   getClient().singular.delete({ where: { pkField }, select: ... }).unwrap()
-    const destructParam = t.objectPattern([shorthandProp(pkField.name)]);
+    const destructParam = t.objectPattern([
+        shorthandProp(pkField.name),
+        ...deleteExtraKeys.map((ek) => shorthandProp(ek.name)),
+    ]);
     destructParam.typeAnnotation = t.tsTypeAnnotation(deleteVarType);
     const mutationFnExpr = t.arrowFunctionExpression([destructParam], getClientCallUnwrap(singularName, 'delete', t.objectExpression([
-        objectProp('where', t.objectExpression([shorthandProp(pkField.name)])),
+        objectProp('where', t.objectExpression([
+            shorthandProp(pkField.name),
+            ...deleteExtraKeys.map((ek) => shorthandProp(ek.name)),
+        ])),
         objectProp('select', t.memberExpression(t.identifier('args'), t.identifier('select'))),
     ])));
     // onSuccess: remove detail, invalidate lists

package/esm/core/codegen/orm/index.js

@@ -31,7 +31,7 @@ export function generateOrm(options) {
         content: selectTypesFile.content,
     });
     // 2. Generate model files
-    const modelFiles = generateAllModelFiles(tables, useSharedTypes);
+    const modelFiles = generateAllModelFiles(tables, useSharedTypes, typeRegistry);
     for (const modelFile of modelFiles) {
         files.push({
             path: `models/${modelFile.fileName}`,

package/esm/core/codegen/orm/model-generator.d.ts

@@ -1,9 +1,9 @@
-import type { Table } from '../../../types/schema';
+import type { Table, TypeRegistry } from '../../../types/schema';
 export interface GeneratedModelFile {
     fileName: string;
     content: string;
     modelName: string;
     tableName: string;
 }
-export declare function generateModelFile(table: Table, _useSharedTypes: boolean, options?: Record<string, never>, allTables?: Table[]): GeneratedModelFile;
-export declare function generateAllModelFiles(tables: Table[], useSharedTypes: boolean): GeneratedModelFile[];
+export declare function generateModelFile(table: Table, _useSharedTypes: boolean, options?: Record<string, never>, allTables?: Table[], typeRegistry?: TypeRegistry): GeneratedModelFile;
+export declare function generateAllModelFiles(tables: Table[], useSharedTypes: boolean, typeRegistry?: TypeRegistry): GeneratedModelFile[];

package/esm/core/codegen/orm/model-generator.js

@@ -7,7 +7,7 @@
 import * as t from '@babel/types';
 import { singularize } from 'inflekt';
 import { generateCode } from '../babel-ast';
-import { getCreateInputTypeName, getCreateMutationName, getDeleteInputTypeName, getDeleteMutationName, getFilterTypeName, getGeneratedFileHeader, getOrderByTypeName, getPrimaryKeyInfo, getTableNames, hasValidPrimaryKey, lcFirst, ucFirst, } from '../utils';
+import { getCreateInputTypeName, getCreateMutationName, getDeleteInputTypeName, getDeleteMutationName, getExtraInputKeys, getFilterTypeName, getGeneratedFileHeader, getOrderByTypeName, getPrimaryKeyInfo, getTableNames, getUpdateInputTypeName, hasValidPrimaryKey, lcFirst, ucFirst, } from '../utils';
 function createImportDeclaration(moduleSpecifier, namedImports, typeOnly = false) {
     const specifiers = namedImports.map((name) => t.importSpecifier(t.identifier(name), t.identifier(name)));
     const decl = t.importDeclaration(specifiers, t.stringLiteral(moduleSpecifier));
@@ -66,7 +66,8 @@ function strictSelectGuard(selectTypeName) {
         t.tsTypeReference(t.identifier(selectTypeName)),
     ]));
 }
-export function generateModelFile(table, _useSharedTypes, options, allTables) {
+// ExtraInputKey type and getExtraInputKeys are imported from ../utils
+export function generateModelFile(table, _useSharedTypes, options, allTables, typeRegistry) {
     const { typeName, singularName, pluralName } = getTableNames(table);
     const modelName = `${typeName}Model`;
     const baseFileName = lcFirst(typeName);
@@ -77,7 +78,7 @@ export function generateModelFile(table, _useSharedTypes, options, allTables) {
     const whereTypeName = getFilterTypeName(table);
     const orderByTypeName = getOrderByTypeName(table);
     const createInputTypeName = `Create${typeName}Input`;
-    const updateInputTypeName = `Update${typeName}Input`;
+    const updateInputTypeName = getUpdateInputTypeName(table);
     const patchTypeName = `${typeName}Patch`;
     const pkFields = getPrimaryKeyInfo(table);
     const pkField = pkFields[0];
@@ -364,12 +365,19 @@ export function generateModelFile(table, _useSharedTypes, options, allTables) {
     }
     // ── update ─────────────────────────────────────────────────────────────
     if (updateMutationName) {
+        const patchFieldName = table.query?.patchFieldName ?? lcFirst(typeName) + 'Patch';
+        const updateExtraKeys = getExtraInputKeys(updateInputTypeName, new Set(pkFields.map((pk) => pk.name)), patchFieldName, typeRegistry);
         const whereLiteral = () => t.tsTypeLiteral([
             (() => {
                 const prop = t.tsPropertySignature(t.identifier(pkField.name), t.tsTypeAnnotation(pkTsType()));
                 prop.optional = false;
                 return prop;
             })(),
+            ...updateExtraKeys.map((ek) => {
+                const prop = t.tsPropertySignature(t.identifier(ek.name), t.tsTypeAnnotation(tsTypeFromPrimitive(ek.tsType)));
+                prop.optional = false;
+                return prop;
+            }),
         ]);
         const argsType = (sel) => t.tsTypeReference(t.identifier('UpdateArgs'), t.tsTypeParameterInstantiation([
             sel,
@@ -393,7 +401,10 @@ export function generateModelFile(table, _useSharedTypes, options, allTables) {
             strictSelectGuard(selectTypeName),
         ]));
         const selectExpr = t.memberExpression(t.identifier('args'), t.identifier('select'));
-        const patchFieldName = table.query?.patchFieldName ?? lcFirst(typeName) + 'Patch';
+        // Build extraKeys object for partitioned table fields
+        const extraKeysArg = updateExtraKeys.length > 0
+            ? t.objectExpression(updateExtraKeys.map((ek) => t.objectProperty(t.identifier(ek.name), t.memberExpression(t.memberExpression(t.identifier('args'), t.identifier('where')), t.identifier(ek.name)))))
+            : t.identifier('undefined');
         const bodyArgs = [
             t.stringLiteral(typeName),
             t.stringLiteral(updateMutationName),
@@ -405,17 +416,26 @@ export function generateModelFile(table, _useSharedTypes, options, allTables) {
             t.stringLiteral(pkField.name),
             t.stringLiteral(patchFieldName),
             t.identifier('connectionFieldsMap'),
+            extraKeysArg,
         ];
         classBody.push(createClassMethod('update', createTypeParam(selectTypeName), [implParam], retType(sRef()), buildMethodBody('buildUpdateByPkDocument', bodyArgs, 'mutation', typeName, updateMutationName)));
     }
     // ── delete ─────────────────────────────────────────────────────────────
     if (deleteMutationName) {
-        // Build where type with ALL PK fields (supports composite PKs)
-        const whereLiteral = () => t.tsTypeLiteral(pkFields.map((pk) => {
-            const prop = t.tsPropertySignature(t.identifier(pk.name), t.tsTypeAnnotation(tsTypeFromPrimitive(pk.tsType ?? 'string')));
-            prop.optional = false;
-            return prop;
-        }));
+        const deleteExtraKeys = getExtraInputKeys(deleteInputTypeName, new Set(pkFields.map((pk) => pk.name)), null, typeRegistry);
+        // Build where type with ALL PK fields + extra keys (supports composite PKs + partition keys)
+        const whereLiteral = () => t.tsTypeLiteral([
+            ...pkFields.map((pk) => {
+                const prop = t.tsPropertySignature(t.identifier(pk.name), t.tsTypeAnnotation(tsTypeFromPrimitive(pk.tsType ?? 'string')));
+                prop.optional = false;
+                return prop;
+            }),
+            ...deleteExtraKeys.map((ek) => {
+                const prop = t.tsPropertySignature(t.identifier(ek.name), t.tsTypeAnnotation(tsTypeFromPrimitive(ek.tsType)));
+                prop.optional = false;
+                return prop;
+            }),
+        ]);
         const argsType = (sel) => t.tsTypeReference(t.identifier('DeleteArgs'), t.tsTypeParameterInstantiation([whereLiteral(), sel]));
         const retType = (sel) => t.tsTypeAnnotation(t.tsTypeReference(t.identifier('QueryBuilder'), t.tsTypeParameterInstantiation([
             t.tsTypeLiteral([
@@ -434,8 +454,11 @@ export function generateModelFile(table, _useSharedTypes, options, allTables) {
             strictSelectGuard(selectTypeName),
         ]));
         const selectExpr = t.memberExpression(t.identifier('args'), t.identifier('select'));
-        // Build keys object: { field1: args.where.field1, field2: args.where.field2, ... }
-        const keysObj = t.objectExpression(pkFields.map((pk) => t.objectProperty(t.identifier(pk.name), t.memberExpression(t.memberExpression(t.identifier('args'), t.identifier('where')), t.identifier(pk.name)))));
+        // Build keys object: { field1: args.where.field1, ... } including extra keys
+        const keysObj = t.objectExpression([
+            ...pkFields.map((pk) => t.objectProperty(t.identifier(pk.name), t.memberExpression(t.memberExpression(t.identifier('args'), t.identifier('where')), t.identifier(pk.name)))),
+            ...deleteExtraKeys.map((ek) => t.objectProperty(t.identifier(ek.name), t.memberExpression(t.memberExpression(t.identifier('args'), t.identifier('where')), t.identifier(ek.name)))),
+        ]);
         const bodyArgs = [
             t.stringLiteral(typeName),
             t.stringLiteral(deleteMutationName),
@@ -711,6 +734,6 @@ export function generateModelFile(table, _useSharedTypes, options, allTables) {
         tableName: table.name,
     };
 }
-export function generateAllModelFiles(tables, useSharedTypes) {
-    return tables.map((table) => generateModelFile(table, useSharedTypes, undefined, tables));
+export function generateAllModelFiles(tables, useSharedTypes, typeRegistry) {
+    return tables.map((table) => generateModelFile(table, useSharedTypes, undefined, tables, typeRegistry));
 }

package/esm/core/codegen/utils.d.ts

@@ -146,6 +146,17 @@ export declare function getUpdateInputTypeName(table: Table): string;
  * like DeletePostTagByPostIdAndTagIdInput), falls back to Delete${Entity}Input.
  */
 export declare function getDeleteInputTypeName(table: Table): string;
+export interface ExtraInputKey {
+    name: string;
+    gqlType: string;
+    tsType: string;
+}
+/**
+ * Discover extra required fields on a mutation input type beyond the standard
+ * ones (clientMutationId, PK fields, patch field). PostGraphile adds these for
+ * partitioned tables (e.g. databaseId as the partition key).
+ */
+export declare function getExtraInputKeys(inputTypeName: string, pkFieldNames: Set<string>, patchFieldName: string | null, typeRegistry?: TypeRegistry): ExtraInputKey[];
 /**
  * Convert GraphQL type to TypeScript type
  */

package/esm/core/codegen/utils.js

@@ -240,6 +240,38 @@ export function getDeleteInputTypeName(table) {
     const mutationName = table.query?.delete;
     return mutationName ? ucFirst(mutationName) + 'Input' : `Delete${table.name}Input`;
 }
+/**
+ * Discover extra required fields on a mutation input type beyond the standard
+ * ones (clientMutationId, PK fields, patch field). PostGraphile adds these for
+ * partitioned tables (e.g. databaseId as the partition key).
+ */
+export function getExtraInputKeys(inputTypeName, pkFieldNames, patchFieldName, typeRegistry) {
+    if (!typeRegistry)
+        return [];
+    const inputType = typeRegistry.get(inputTypeName);
+    if (!inputType || inputType.kind !== 'INPUT_OBJECT' || !inputType.inputFields)
+        return [];
+    const skip = new Set(['clientMutationId', ...(patchFieldName ? [patchFieldName] : [])]);
+    for (const pk of pkFieldNames)
+        skip.add(pk);
+    const extras = [];
+    for (const field of inputType.inputFields) {
+        if (skip.has(field.name))
+            continue;
+        if (field.type.kind !== 'NON_NULL')
+            continue;
+        const innerName = field.type.ofType?.name;
+        if (!innerName)
+            continue;
+        let tsType = 'string';
+        if (innerName === 'Int' || innerName === 'Float' || innerName === 'BigFloat')
+            tsType = 'number';
+        else if (innerName === 'Boolean')
+            tsType = 'boolean';
+        extras.push({ name: field.name, gqlType: innerName, tsType });
+    }
+    return extras;
+}
 // ============================================================================
 // Type mapping: GraphQL → TypeScript
 // ============================================================================

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@constructive-io/graphql-codegen",
-  "version": "4.47.9",
+  "version": "4.47.11",
   "description": "GraphQL SDK generator for Constructive databases with React Query hooks",
   "keywords": [
     "graphql",
@@ -100,5 +100,5 @@
     "tsx": "^4.21.0",
     "typescript": "^5.9.3"
   },
-  "gitHead": "16c2ae1c30850db1d71c021346c9d34361361d8e"
+  "gitHead": "1dc015ea6eb2c968db440d87ca65d5e57dd4f8c3"
 }