@cofhe/abi 0.8.0 → 1.0.0-alpha.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/index.cjs +49 -25
- package/dist/index.cjs.map +1 -1
- package/dist/index.js +49 -25
- package/dist/index.js.map +1 -1
- package/package.json +2 -2
- package/src/encryptedInputs.ts +35 -15
- package/src/encryptedReturnTypes.ts +20 -11
- package/src/utils.ts +9 -0
package/dist/index.cjs
CHANGED
|
@@ -40,6 +40,9 @@ function extractArrayParameterType(type) {
|
|
|
40
40
|
const size = match[2];
|
|
41
41
|
return [head, size === "" ? "" : size];
|
|
42
42
|
}
|
|
43
|
+
function tupleComponentsAreNamed(components) {
|
|
44
|
+
return components.length > 0 && components.every((component) => component.name != null && component.name !== "");
|
|
45
|
+
}
|
|
43
46
|
function isAbiFunction(item) {
|
|
44
47
|
if (typeof item !== "object" || item === null) return false;
|
|
45
48
|
return "type" in item && item.type === "function" && "name" in item && "outputs" in item;
|
|
@@ -130,9 +133,18 @@ function extractEncryptableValues(abi, functionName, args) {
|
|
|
130
133
|
if ("components" in param && Array.isArray(param.components)) {
|
|
131
134
|
const components = param.components;
|
|
132
135
|
const processTuple = (tupleValue) => {
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
+
if (tupleComponentsAreNamed(components)) {
|
|
137
|
+
const valueObj = tupleValue;
|
|
138
|
+
components.forEach((component) => {
|
|
139
|
+
processParameter(component, valueObj[component.name]);
|
|
140
|
+
});
|
|
141
|
+
return;
|
|
142
|
+
}
|
|
143
|
+
if (!Array.isArray(tupleValue)) {
|
|
144
|
+
throw new Error("Expected an array value for tuple with unnamed components");
|
|
145
|
+
}
|
|
146
|
+
components.forEach((component, index) => {
|
|
147
|
+
processParameter(component, tupleValue[index]);
|
|
136
148
|
});
|
|
137
149
|
};
|
|
138
150
|
if (typeSize != null) {
|
|
@@ -228,18 +240,24 @@ function insertEncryptedValues(abi, functionName, args, encryptedResult) {
|
|
|
228
240
|
if ("components" in param && Array.isArray(param.components)) {
|
|
229
241
|
const components = param.components;
|
|
230
242
|
const processTuple = (tupleValue) => {
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
|
|
238
|
-
|
|
243
|
+
if (tupleComponentsAreNamed(components)) {
|
|
244
|
+
const valueObj = tupleValue;
|
|
245
|
+
const result2 = {};
|
|
246
|
+
components.forEach((component) => {
|
|
247
|
+
const componentName = component.name;
|
|
248
|
+
if (componentName) {
|
|
249
|
+
const componentValue = valueObj[componentName];
|
|
250
|
+
if (componentValue !== void 0) {
|
|
251
|
+
result2[componentName] = processParameter(component, componentValue);
|
|
252
|
+
}
|
|
239
253
|
}
|
|
240
|
-
}
|
|
241
|
-
|
|
242
|
-
|
|
254
|
+
});
|
|
255
|
+
return result2;
|
|
256
|
+
}
|
|
257
|
+
if (!Array.isArray(tupleValue)) {
|
|
258
|
+
throw new Error("Expected an array value for tuple with unnamed components");
|
|
259
|
+
}
|
|
260
|
+
return components.map((component, index) => processParameter(component, tupleValue[index]));
|
|
243
261
|
};
|
|
244
262
|
if (typeSize != null) {
|
|
245
263
|
if (!Array.isArray(value)) {
|
|
@@ -360,18 +378,24 @@ function transformEncryptedReturnTypes(abi, functionName, data) {
|
|
|
360
378
|
if ("components" in param && Array.isArray(param.components)) {
|
|
361
379
|
const components = param.components;
|
|
362
380
|
const processTuple = (tupleValue) => {
|
|
363
|
-
|
|
364
|
-
|
|
365
|
-
|
|
366
|
-
|
|
367
|
-
|
|
368
|
-
|
|
369
|
-
|
|
370
|
-
|
|
381
|
+
if (tupleComponentsAreNamed(components)) {
|
|
382
|
+
const valueObj = tupleValue;
|
|
383
|
+
const result = {};
|
|
384
|
+
components.forEach((component) => {
|
|
385
|
+
const componentName = component.name;
|
|
386
|
+
if (componentName) {
|
|
387
|
+
const componentValue = valueObj[componentName];
|
|
388
|
+
if (componentValue !== void 0) {
|
|
389
|
+
result[componentName] = processParameter(component, componentValue);
|
|
390
|
+
}
|
|
371
391
|
}
|
|
372
|
-
}
|
|
373
|
-
|
|
374
|
-
|
|
392
|
+
});
|
|
393
|
+
return result;
|
|
394
|
+
}
|
|
395
|
+
if (!Array.isArray(tupleValue)) {
|
|
396
|
+
throw new Error("Expected an array value for tuple with unnamed components");
|
|
397
|
+
}
|
|
398
|
+
return components.map((component, index) => processParameter(component, tupleValue[index]));
|
|
375
399
|
};
|
|
376
400
|
if (typeSize != null) {
|
|
377
401
|
if (!Array.isArray(value)) {
|
package/dist/index.cjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/index.ts","../src/encryptedInputs.ts","../src/utils.ts","../src/encryptedReturnTypes.ts"],"sourcesContent":["export * from './encryptedInputs';\nexport * from './encryptedReturnTypes';\nexport * from './fhenixMap';\nexport type { Abi } from 'abitype';\nexport type { ContractReturnType } from './utils';\nexport { transformEncryptedReturnTypes } from './encryptedReturnTypes';\n","import {\n Encryptable,\n type AnyExternalHash,\n type EncryptableAddress,\n type EncryptableBool,\n type EncryptableItem,\n type EncryptableUint128,\n type EncryptableUint16,\n type EncryptableUint32,\n type EncryptableUint64,\n type EncryptableUint8,\n type ExternalAddressHash,\n type ExternalBoolHash,\n type ExternalUint128Hash,\n type ExternalUint16Hash,\n type ExternalUint32Hash,\n type ExternalUint64Hash,\n type ExternalUint8Hash,\n type LiteralToPrimitive,\n type Primitive,\n} from '@cofhe/sdk';\nimport type { Abi, AbiFunction, AbiParameter, ExtractAbiFunction } from 'abitype';\nimport type { CofheAbiParametersToPrimitiveTypes } from './fhenixMap';\nimport {\n extractArrayParameterType,\n getAbiFunction,\n type MaybeExtractArrayParameterType,\n type MaybePartialBy,\n type ReadonlyWiden,\n} from './utils';\n\nexport type CofheInputArgs<abi extends Abi | readonly unknown[] = Abi, functionName extends string = string> = GetArgs<\n abi,\n functionName\n>['args'];\n\ntype ExternalHashToInputMap<E extends AnyExternalHash> = E extends ExternalBoolHash\n ? EncryptableBool['data']\n : E extends ExternalUint8Hash\n ? EncryptableUint8['data']\n : E extends ExternalUint16Hash\n ? EncryptableUint16['data']\n : E extends ExternalUint32Hash\n ? EncryptableUint32['data']\n : E extends ExternalUint64Hash\n ? EncryptableUint64['data']\n : E extends ExternalUint128Hash\n ? EncryptableUint128['data']\n : E extends ExternalAddressHash\n ? EncryptableAddress['data']\n : never;\n\n// Note: AnyExternalHash is checked before Primitive - the branded hash types are structurally\n// `0x${string}` (a subtype of `string`, which is itself a Primitive), so Primitive must not be\n// checked first or every hash would be (incorrectly) widened via LiteralToPrimitive instead.\ntype EncryptedInputsToInputs<T> = T extends AnyExternalHash\n ? ExternalHashToInputMap<T>\n : T extends Primitive\n ? LiteralToPrimitive<T>\n : {\n [K in keyof T]: EncryptedInputsToInputs<T[K]>;\n };\n\n/**\n * Drops the batch-signature argument from an args tuple.\n *\n * The signature slot is the first non-`external*` parameter that follows the contiguous run of\n * `external*` parameters - `FHE.asEuintXX(hash, proof)` is a pair, so the proof sits with the\n * hashes it authenticates rather than at the end of the parameter list. Walks the ABI parameters\n * and the args tuple in lockstep and elides the arg at that position.\n */\ntype DropSignatureSlotTuple<\n params extends readonly unknown[],\n args extends readonly unknown[],\n afterExternalRun extends boolean = false,\n> = params extends readonly [infer P, ...infer PRest extends readonly unknown[]]\n ? args extends readonly [infer A, ...infer ARest extends readonly unknown[]]\n ? P extends AbiParameter\n ? ParamHasExternalInput<P> extends true\n ? readonly [A, ...DropSignatureSlotTuple<PRest, ARest, true>]\n : afterExternalRun extends true\n ? DropSignatureSlotTuple<PRest, ARest, false>\n : readonly [A, ...DropSignatureSlotTuple<PRest, ARest, false>]\n : readonly [A, ...DropSignatureSlotTuple<PRest, ARest, afterExternalRun>]\n : readonly []\n : args;\n\n/** Non-tuple args (e.g. `undefined`) pass through unchanged, as the previous `DropLast` did. */\ntype DropSignatureSlot<params extends readonly unknown[], args> = args extends readonly unknown[]\n ? DropSignatureSlotTuple<params, args>\n : args;\n\nconst EXTERNAL_INPUT_INTERNAL_TYPES = [\n 'externalEbool',\n 'externalEuint8',\n 'externalEuint16',\n 'externalEuint32',\n 'externalEuint64',\n 'externalEuint128',\n 'externalEaddress',\n] as const;\ntype ExternalInputInternalType = (typeof EXTERNAL_INPUT_INTERNAL_TYPES)[number];\n\n/** Type-level: does this ABI parameter carry (or, if a tuple, recursively contain) an `external*` input? */\ntype ParamHasExternalInput<param extends AbiParameter> =\n MaybeExtractArrayParameterType<param['internalType']> extends [infer head extends string, any]\n ? head extends ExternalInputInternalType\n ? true\n : // A struct array (`tuple[]` / `tuple[N]`) lands here too, with `head` the struct's own\n // internalType. Falling through to the components keeps `Order[]` visible as carrying\n // encrypted inputs, the way the runtime `paramHasExternalInput` already sees it.\n ParamComponentsHaveExternalInput<param>\n : param['internalType'] extends ExternalInputInternalType\n ? true\n : ParamComponentsHaveExternalInput<param>;\n\n/** Recurses into a parameter's `components`, for tuples at any array depth. */\ntype ParamComponentsHaveExternalInput<param extends AbiParameter> = param extends {\n components: infer components extends readonly AbiParameter[];\n}\n ? ParamsHaveExternalInput<components>\n : false;\n\ntype ParamsHaveExternalInput<params extends readonly AbiParameter[]> = params extends readonly [\n infer Head extends AbiParameter,\n ...infer Rest extends readonly AbiParameter[],\n]\n ? ParamHasExternalInput<Head> extends true\n ? true\n : ParamsHaveExternalInput<Rest>\n : false;\n\n/**\n * Pre-transform args for a function ABI: the raw plain values a caller supplies before encryption.\n *\n * If the function has one or more `external*` (encrypted) inputs, the plain `bytes` parameter\n * immediately following them is the shared batch signature slot - the caller never supplies it\n * directly (the SDK injects it after encryption), so it's dropped from this type entirely.\n */\nexport type CofheInputArgsPreTransform<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n abiFunction extends AbiFunction = abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction,\n> = EncryptedInputsToInputs<\n ParamsHaveExternalInput<abiFunction['inputs']> extends true\n ? DropSignatureSlot<abiFunction['inputs'], CofheInputArgs<abi, functionName>>\n : CofheInputArgs<abi, functionName>\n>;\n\n/// GetArgs from abitype (not exported from abitype)\ntype GetArgs<\n abi extends Abi | readonly unknown[] = Abi, // `readonly unknown[]` allows for non-const asserted types\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly [],\n ///\n abiFunction extends AbiFunction = abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction,\n primitiveTypes = CofheAbiParametersToPrimitiveTypes<abiFunction['inputs']>,\n args_ =\n | primitiveTypes // show all values\n | (abi extends Abi\n ? args extends primitiveTypes // infer value (if valid)\n ? primitiveTypes extends args // make sure `args` exactly matches `primitiveTypes` (e.g. avoid `args: readonly [{ foo: string; bar: number; }] | readonly [{ foo: string; }]`)\n ? // make inferred value of `args` match `primitiveTypes` (e.g. avoid union `args: readonly [123n] | readonly [bigint]`)\n ReadonlyWiden<args>\n : never\n : never\n : never)\n | (Abi extends abi ? readonly unknown[] : never), // fallback if `abi` is declared as `Abi`\n> = MaybePartialBy<{ args: args_ }, readonly [] extends primitiveTypes ? 'args' : Abi extends abi ? 'args' : string>;\n\nfunction internalTypeIsExternalInput(internalType: string): internalType is ExternalInputInternalType {\n return EXTERNAL_INPUT_INTERNAL_TYPES.includes(internalType as any);\n}\n\nfunction paramHasExternalInput(param: AbiParameter): boolean {\n const [internalTypeHead] = extractArrayParameterType(param.internalType);\n if (internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) return true;\n if ('components' in param && Array.isArray(param.components)) {\n return param.components.some((component) => paramHasExternalInput(component));\n }\n return false;\n}\n\nfunction transformSingleExternalToEncryptable(internalType: ExternalInputInternalType, data: unknown): EncryptableItem {\n switch (internalType) {\n case 'externalEbool':\n return Encryptable.bool(data as boolean);\n case 'externalEuint8':\n return Encryptable.uint8(data as string | bigint);\n case 'externalEuint16':\n return Encryptable.uint16(data as string | bigint);\n case 'externalEuint32':\n return Encryptable.uint32(data as string | bigint);\n case 'externalEuint64':\n return Encryptable.uint64(data as string | bigint);\n case 'externalEuint128':\n return Encryptable.uint128(data as string | bigint);\n case 'externalEaddress':\n return Encryptable.address(data as string | bigint);\n default:\n throw new Error(`Unknown external input type: ${internalType}`);\n }\n}\n\nfunction transformArrayOfExternalsToEncryptables(\n internalType: ExternalInputInternalType,\n size: string | undefined,\n data: unknown\n): EncryptableItem[] {\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n\n if (size != null && size !== '' && parseInt(size) !== data.length) {\n throw new Error(`Array size mismatch: ${size} !== ${data.length}`);\n }\n\n return data.map((item) => transformSingleExternalToEncryptable(internalType, item));\n}\n\n/**\n * Extracts encryptable values from function arguments based on the ABI.\n * Transforms raw data values into EncryptableItem objects that can be passed to the encrypt function.\n *\n * @param abi - The ABI containing the function definition\n * @param functionName - Name of the function\n * @param args - Function arguments in the format of CofheInputArgsPreTransform (raw data values;\n * if the function has `external*` inputs, the trailing batch-signature slot is NOT included)\n * @returns Array of EncryptableItem objects ready for encryption, in ABI parameter order\n */\nexport function extractEncryptableValues<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n args: CofheInputArgsPreTransform<TAbi, TFunctionName>\n): EncryptableItem[] {\n const abiFunction = getAbiFunction(abi, functionName);\n const inputs = abiFunction?.inputs;\n if (abiFunction == null || inputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n\n if (!Array.isArray(args)) {\n throw new Error('Arguments must be an array');\n }\n\n const hasExternalInputs = inputs.some((input) => paramHasExternalInput(input));\n const signatureSlotIndex = findSignatureSlotIndex(inputs, hasExternalInputs, functionName);\n\n // Collect encrypted values as EncryptableItem objects in order (flat array)\n const encryptableItems: EncryptableItem[] = [];\n\n function processParameter(param: AbiParameter, value: unknown): void {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead, internalTypeSize] = extractArrayParameterType(param.internalType);\n\n // Bare external input (e.g. `externalEuint32`, type: 'bytes32')\n if (typeSize == null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n encryptableItems.push(transformSingleExternalToEncryptable(internalTypeHead, value));\n return;\n }\n\n // Array of external inputs (e.g. `externalEuint32[]`/`externalEuint32[2]`, type: 'bytes32[]'/'bytes32[2]')\n if (typeSize != null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n const encryptables = transformArrayOfExternalsToEncryptables(internalTypeHead, typeSize, value);\n encryptableItems.push(...encryptables);\n return;\n }\n\n // Tuple recursive case (struct that may itself contain external inputs)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): void => {\n const valueObj = tupleValue as Record<string, unknown>;\n components.forEach((component) => {\n processParameter(component, valueObj[component.name]);\n });\n };\n\n // `tuple[]` / `tuple[N]`: walk each element as its own tuple. Treating the array\n // itself as one tuple would look up named components on the array object, find\n // none, and hand `undefined` to the encryptable transform.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n value.forEach(processTuple);\n return;\n }\n\n processTuple(value);\n }\n return;\n }\n\n // Not an external input, and cannot contain one\n return;\n }\n\n // Process all inputs, skipping the batch-signature slot (nothing to extract there). `args` is the\n // pre-transform tuple, which omits that slot, so its indices lag by one past the signature.\n let argIndex = 0;\n inputs.forEach((input, index) => {\n if (index === signatureSlotIndex) return;\n const arg = args[argIndex];\n if (arg == null) {\n throw new Error(`Argument ${argIndex} is undefined`);\n }\n argIndex++;\n processParameter(input, arg);\n });\n\n return encryptableItems;\n}\n\n/**\n * Index of the shared batch-signature parameter: the plain `bytes` immediately after the\n * contiguous run of `external*` parameters. Returns -1 when the function has no encrypted inputs.\n *\n * The proof pairs with the hashes it authenticates (`FHE.asEuintXX(hash, proof)`), so it does not\n * have to be the last parameter - anything may follow it.\n */\nfunction findSignatureSlotIndex(\n inputs: readonly AbiParameter[],\n hasExternalInputs: boolean,\n functionName: string\n): number {\n if (!hasExternalInputs) return -1;\n\n const externalIndices = inputs.reduce<number[]>((acc, input, index) => {\n if (paramHasExternalInput(input)) acc.push(index);\n return acc;\n }, []);\n\n const first = externalIndices[0]!;\n const last = externalIndices[externalIndices.length - 1]!;\n if (last - first + 1 !== externalIndices.length) {\n throw new Error(\n `Function ${functionName} has encrypted (external*) inputs at non-adjacent positions ` +\n `(${externalIndices.join(', ')}). They share one batch signature, so they must be contiguous ` +\n `with the 'bytes' signature parameter immediately after them`\n );\n }\n\n const slotIndex = last + 1;\n const slot = inputs[slotIndex];\n if (slot?.type !== 'bytes') {\n throw new Error(\n `Function ${functionName} has encrypted (external*) inputs, so the parameter immediately after ` +\n `them must be a plain 'bytes' parameter to receive the shared batch signature - found ` +\n `'${slot?.type ?? 'nothing'}' at position ${slotIndex} instead`\n );\n }\n\n return slotIndex;\n}\n\n/**\n * Re-inserts encrypted values back into function arguments based on the ABI.\n * Takes the batch-verified result (per-input hashes, in order, followed by the single shared\n * signature) and the original args structure, and produces the final call args: each `external*`\n * position gets its corresponding hash, and the trailing `bytes` parameter gets the signature.\n *\n * This function mirrors the extraction logic in extractEncryptableValues, ensuring values\n * are inserted in the exact same order and locations where they were extracted.\n *\n * @param abi - The ABI containing the function definition\n * @param functionName - Name of the function\n * @param args - Original function arguments in the format of CofheInputArgsPreTransform\n * @param encryptedResult - `[...hashes, signature]`, as returned by `EncryptInputsBuilder.execute()`\n * @returns Function arguments with encrypted values inserted (format of CofheInputArgs)\n */\nexport function insertEncryptedValues<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n args: CofheInputArgsPreTransform<TAbi, TFunctionName>,\n encryptedResult: readonly `0x${string}`[]\n): CofheInputArgs<TAbi, TFunctionName> {\n const abiFunction = getAbiFunction(abi, functionName);\n const inputs = abiFunction?.inputs;\n if (abiFunction == null || inputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n\n if (!Array.isArray(args)) {\n throw new Error('Arguments must be an array');\n }\n\n const hasExternalInputs = inputs.some((input) => paramHasExternalInput(input));\n const signatureSlotIndex = findSignatureSlotIndex(inputs, hasExternalInputs, functionName);\n\n const hashes = hasExternalInputs ? encryptedResult.slice(0, -1) : [];\n const signature = hasExternalInputs ? encryptedResult[encryptedResult.length - 1] : undefined;\n\n // Track position in the hashes array\n let hashIndex = 0;\n\n function processParameter(param: AbiParameter, value: unknown): unknown {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead] = extractArrayParameterType(param.internalType);\n\n // Bare external input - substitute its hash\n if (typeSize == null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n if (hashIndex >= hashes.length) {\n throw new Error(`Not enough encrypted hashes: expected at least ${hashIndex + 1}, got ${hashes.length}`);\n }\n const hash = hashes[hashIndex];\n hashIndex++;\n return hash;\n }\n\n // Array of external inputs - substitute the corresponding slice of hashes\n if (typeSize != null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n const arrayLength = Array.isArray(value) ? value.length : 0;\n if (arrayLength === 0) {\n return [];\n }\n\n if (hashIndex + arrayLength > hashes.length) {\n throw new Error(\n `Not enough encrypted hashes: expected at least ${hashIndex + arrayLength}, got ${hashes.length}`\n );\n }\n\n const hashArray = hashes.slice(hashIndex, hashIndex + arrayLength);\n hashIndex += arrayLength;\n return hashArray;\n }\n\n // Tuple recursive case (struct that may itself contain external inputs)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): unknown => {\n const valueObj = tupleValue as Record<string, unknown>;\n const result: Record<string, unknown> = {};\n\n components.forEach((component) => {\n const componentName = component.name;\n if (componentName) {\n const componentValue = valueObj[componentName];\n if (componentValue !== undefined) {\n result[componentName] = processParameter(component, componentValue);\n }\n }\n });\n\n return result;\n };\n\n // `tuple[]` / `tuple[N]`: rebuild each element as its own tuple, in the same order\n // extractEncryptableValues walked them, so the hashes line up. Treating the array as\n // one tuple would consume no hashes and collapse the argument to `{}`.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n return value.map(processTuple);\n }\n\n return processTuple(value);\n }\n return value;\n }\n\n // Not an external input, return original value\n return value;\n }\n\n // Rebuild the full arg list; the signature slot is injected directly. `args` omits that slot, so\n // its indices lag by one past the signature position.\n let argIndex = 0;\n const result = inputs.map((input, index) => {\n if (index === signatureSlotIndex) {\n return signature;\n }\n const arg = args[argIndex];\n if (arg == null) {\n throw new Error(`Argument ${argIndex} is undefined`);\n }\n argIndex++;\n return processParameter(input, arg);\n });\n\n // Verify we used all encrypted hashes\n if (hashIndex !== hashes.length) {\n throw new Error(`Mismatch in encrypted hashes count: used ${hashIndex}, but provided ${hashes.length}`);\n }\n\n return result as CofheInputArgs<TAbi, TFunctionName>;\n}\n","import type {\n Abi,\n AbiConstructor,\n AbiError,\n AbiEvent,\n AbiFallback,\n AbiFunction,\n AbiParameter,\n AbiParametersToPrimitiveTypes,\n AbiReceive,\n AbiType,\n Address,\n ExtractAbiFunction,\n ResolvedRegister,\n SolidityArray,\n SolidityFixedArrayRange,\n SolidityTuple,\n} from 'abitype';\n\nexport type ContractReturnType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n ///\n abiFunction extends AbiFunction = (\n abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction\n ) extends infer abiFunction_ extends AbiFunction\n ? IsUnion<abiFunction_> extends true // narrow overloads by `args` by converting to tuple and filtering out overloads that don't match\n ? UnionToTuple<abiFunction_> extends infer abiFunctions extends readonly AbiFunction[]\n ? {\n [K in keyof abiFunctions]: (\n readonly unknown[] | undefined extends args // for functions that don't have inputs, `args` can be `undefined` so fallback to `readonly []`\n ? readonly []\n : args\n ) extends AbiParametersToPrimitiveTypes<abiFunctions[K]['inputs'], 'inputs'>\n ? abiFunctions[K]\n : never;\n }[number] // convert back to union (removes `never` tuple entries: `['foo', never, 'bar'][number]` => `'foo' | 'bar'`)\n : never\n : abiFunction_\n : never,\n outputs extends readonly AbiParameter[] = abiFunction['outputs'],\n primitiveTypes extends readonly unknown[] = AbiParametersToPrimitiveTypes<outputs, 'outputs', true>,\n> = [abiFunction] extends [never]\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` declared as `Abi`)\n : readonly unknown[] extends primitiveTypes\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` not const-asserted)\n : primitiveTypes extends readonly [] // unwrap `primitiveTypes`\n ? // biome-ignore lint/suspicious/noConfusingVoidType: <explanation>\n void // no outputs\n : primitiveTypes extends readonly [infer primitiveType]\n ? primitiveType // single output\n : primitiveTypes;\n\nexport type Error<messages extends string | string[]> = messages extends string\n ? [\n // Surrounding with array to prevent `messages` from being widened to `string`\n `Error: ${messages}`,\n ]\n : {\n [key in keyof messages]: messages[key] extends infer message extends string ? `Error: ${message}` : never;\n };\n\ntype PartialBy<TType, TKeys extends keyof TType> = ExactPartial<Pick<TType, TKeys>> & Omit<TType, TKeys>;\ntype ExactPartial<T> = { [K in keyof T]?: T[K] | undefined };\n\nexport type MaybePartialBy<TType, TKeys extends string> = TKeys extends keyof TType ? PartialBy<TType, TKeys> : TType;\n\nexport type ReadonlyWiden<TType> =\n | (TType extends Function ? TType : never)\n | (TType extends ResolvedRegister['bigIntType'] ? bigint : never)\n | (TType extends boolean ? boolean : never)\n | (TType extends ResolvedRegister['intType'] ? number : never)\n | (TType extends string\n ? TType extends Address\n ? Address\n : TType extends ResolvedRegister['bytesType']['inputs']\n ? ResolvedRegister['bytesType']\n : string\n : never)\n | (TType extends readonly [] ? readonly [] : never)\n | (TType extends Record<string, unknown> ? { [K in keyof TType]: ReadonlyWiden<TType[K]> } : never)\n | (TType extends { length: number }\n ? {\n [K in keyof TType]: ReadonlyWiden<TType[K]>;\n } extends infer Val extends unknown[]\n ? readonly [...Val]\n : never\n : never);\n\nexport type AbiBasicType = Exclude<AbiType, SolidityTuple | SolidityArray>;\n\n/**\n * First, infer `Head` against a known size type (either fixed-length array value or `\"\"`).\n *\n * | Input | Head |\n * | --------------- | ------------ |\n * | `string[]` | `string` |\n * | `string[][][3]` | `string[][]` |\n */\nexport type MaybeExtractArrayParameterType<type> = type extends `${infer head}[${'' | `${SolidityFixedArrayRange}`}]`\n ? // * Then, infer in the opposite direction, using the known `head` to infer the exact `size` value.\n // *\n // * | Input | Size |\n // * | ------------ | ---- |\n // * | `${head}[]` | `\"\"` |\n // * | `${head}[3]` | `3` |\n // */\n type extends `${head}[${infer size}]`\n ? [head, size]\n : undefined\n : undefined;\n\nexport type IsUnion<T, C = T> = T extends C ? ([C] extends [T] ? false : true) : never;\nexport type UnionToTuple<U, Last = LastInUnion<U>> = [U] extends [never]\n ? []\n : [...UnionToTuple<Exclude<U, Last>>, Last];\ntype LastInUnion<U> =\n UnionToIntersection<U extends unknown ? (x: U) => 0 : never> extends (x: infer L) => 0 ? L : never;\ntype UnionToIntersection<U> = (U extends unknown ? (arg: U) => 0 : never) extends (arg: infer I) => 0 ? I : never;\n\n/**\n * Create tuple of {@link type} type with {@link size} size\n *\n * @param Type - Type of tuple\n * @param Size - Size of tuple\n * @returns Tuple of {@link type} type with {@link size} size\n *\n * @example\n * type Result = Tuple<string, 2>\n * // ^? type Result = [string, string]\n */\n// https://github.com/Microsoft/TypeScript/issues/26223#issuecomment-674500430\nexport type Tuple<type, size extends number> = size extends size\n ? number extends size\n ? type[]\n : _TupleOf<type, size, []>\n : never;\ntype _TupleOf<length, size extends number, acc extends readonly unknown[]> = acc['length'] extends size\n ? acc\n : _TupleOf<length, size, readonly [length, ...acc]>;\n\n/**\n * Merges two object types into new type\n *\n * @param object1 - Object to merge into\n * @param object2 - Object to merge and override keys from {@link object1}\n * @returns New object type with keys from {@link object1} and {@link object2}. If a key exists in both {@link object1} and {@link object2}, the key from {@link object2} will be used.\n *\n * @example\n * type Result = Merge<{ foo: string }, { foo: number; bar: string }>\n * // ^? type Result = { foo: number; bar: string }\n */\nexport type Merge<object1, object2> = Omit<object1, keyof object2> & object2;\n\n// FUNCTIONS\n\n/**\n * Extracts array parameter type information from a type string.\n * Always returns a tuple, with undefined for non-array types.\n *\n * @param type - Type string like \"tuple\", \"tuple[]\", or \"tuple[2]\"\n * @returns [head, size] where:\n * - head is the base type\n * - size is undefined for non-arrays, \"\" for dynamic arrays, or the number as a string for fixed arrays\n *\n * @example\n * extractArrayParameterType(\"tuple\") // [\"tuple\", undefined]\n * extractArrayParameterType(\"tuple[]\") // [\"tuple\", \"\"]\n * extractArrayParameterType(\"tuple[2]\") // [\"tuple\", \"2\"]\n */\nexport function extractArrayParameterType<T extends string | undefined>(type: T): [T, string | undefined] {\n if (type == null) return [type, undefined];\n\n const match = type.match(/^(.+)\\[(\\d*)\\]$/);\n\n if (!match) {\n // Not an array type, return [type, undefined]\n return [type, undefined];\n }\n\n const head = match[1];\n const size = match[2]; // Empty string for dynamic arrays, or digits for fixed arrays\n\n // Return empty string for dynamic arrays, or the size string for fixed arrays\n return [head as T, size === '' ? '' : size];\n}\n\ntype AbiItem = AbiConstructor | AbiError | AbiEvent | AbiFallback | AbiFunction | AbiReceive;\n\nexport function isAbiFunction(item: AbiItem | unknown): item is AbiFunction {\n if (typeof item !== 'object' || item === null) return false;\n return 'type' in item && item.type === 'function' && 'name' in item && 'outputs' in item;\n}\n\nexport function getAbiFunction<TAbi extends Abi | readonly unknown[], TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName\n): AbiFunction | undefined {\n return abi.find((item) => {\n const isFunction = isAbiFunction(item);\n if (!isFunction) return false;\n return item.name === functionName;\n }) as AbiFunction | undefined;\n}\n","import { type Primitive, type LiteralToPrimitive, FheTypes } from '@cofhe/sdk';\nimport type { Abi, AbiFunction, ExtractAbiFunction, AbiParameter } from 'abitype';\nimport type {\n EncryptedReturnType,\n CofheAbiParametersToPrimitiveTypes,\n EBool,\n EAddress,\n EUint128,\n EUint16,\n EUint32,\n EUint64,\n EUint8,\n} from './fhenixMap';\nimport {\n getAbiFunction,\n type IsUnion,\n type UnionToTuple,\n type ContractReturnType,\n extractArrayParameterType,\n} from './utils';\n\ntype FirstCofheReturnValue<T> = [T] extends [void]\n ? never\n : T extends readonly [infer Head, ...(readonly unknown[])]\n ? Head\n : T;\n\ntype FheTypeFromCofheReturnValue<T> = T extends { utype: infer U }\n ? U\n : T extends readonly (infer Element)[]\n ? FheTypeFromCofheReturnValue<Element>\n : never;\n\n/**\n * Extracts the `FheTypes` of the first returned value of a CoFHE call.\n *\n * Examples:\n * - `returns (ebool)` -> `FheTypes.Bool`\n * - `returns (euint8, euint64)` -> `FheTypes.Uint8`\n * - `returns (euint16[])` -> `FheTypes.Uint16`\n */\nexport type CofheFirstReturnFheType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n> = FheTypeFromCofheReturnValue<FirstCofheReturnValue<CofheReturnType<abi, functionName, args>>>;\n\nexport type CofheReturnType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n ///\n abiFunction extends AbiFunction = (\n abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction\n ) extends infer abiFunction_ extends AbiFunction\n ? IsUnion<abiFunction_> extends true // narrow overloads by `args` by converting to tuple and filtering out overloads that don't match\n ? UnionToTuple<abiFunction_> extends infer abiFunctions extends readonly AbiFunction[]\n ? {\n [K in keyof abiFunctions]: (\n readonly unknown[] | undefined extends args // for functions that don't have inputs, `args` can be `undefined` so fallback to `readonly []`\n ? readonly []\n : args\n ) extends CofheAbiParametersToPrimitiveTypes<abiFunctions[K]['inputs'], 'inputs'>\n ? abiFunctions[K]\n : never;\n }[number] // convert back to union (removes `never` tuple entries: `['foo', never, 'bar'][number]` => `'foo' | 'bar'`)\n : never\n : abiFunction_\n : never,\n outputs extends readonly AbiParameter[] = abiFunction['outputs'],\n primitiveTypes extends readonly unknown[] = CofheAbiParametersToPrimitiveTypes<outputs, 'outputs'>,\n> = [abiFunction] extends [never]\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` declared as `Abi`)\n : readonly unknown[] extends primitiveTypes\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` not const-asserted)\n : primitiveTypes extends readonly [] // unwrap `primitiveTypes`\n ? // biome-ignore lint/suspicious/noConfusingVoidType: <explanation>\n void // no outputs\n : primitiveTypes extends readonly [infer primitiveType]\n ? primitiveType // single output\n : primitiveTypes;\n\ntype EncryptedReturnTypeToReturnTypeMap<E extends EncryptedReturnType> = E extends EBool\n ? boolean\n : E extends EUint8\n ? bigint\n : E extends EUint16\n ? bigint\n : E extends EUint32\n ? bigint\n : E extends EUint64\n ? bigint\n : E extends EUint128\n ? bigint\n : E extends EAddress\n ? string\n : never;\n\ntype EncryptedReturnTypesToReturnTypes<T> = T extends Primitive\n ? LiteralToPrimitive<T>\n : T extends EncryptedReturnType\n ? EncryptedReturnTypeToReturnTypeMap<T>\n : {\n [K in keyof T]: EncryptedReturnTypesToReturnTypes<T[K]>;\n };\n\nexport type CofheReturnTypePostTransform<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n> = EncryptedReturnTypesToReturnTypes<CofheReturnType<abi, functionName, args>>;\n\nconst ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES = [\n 'ebool',\n 'euint8',\n 'euint16',\n 'euint32',\n 'euint64',\n 'euint128',\n 'eaddress',\n] as const;\ntype EncryptedReturnTypeInternalType = (typeof ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES)[number];\n\ntype EncryptedReturnTypeInternalTypeToReturnTypeMap<T extends EncryptedReturnTypeInternalType> = T extends 'ebool'\n ? EBool\n : T extends 'euint8'\n ? EUint8\n : T extends 'euint16'\n ? EUint16\n : T extends 'euint32'\n ? EUint32\n : T extends 'euint64'\n ? EUint64\n : T extends 'euint128'\n ? EUint128\n : T extends 'eaddress'\n ? EAddress\n : never;\n\nfunction transformSingleEncryptedReturnTypeToReturnType<T extends EncryptedReturnTypeInternalType>(\n internalType: T,\n data: string\n): EncryptedReturnTypeInternalTypeToReturnTypeMap<T> {\n switch (internalType) {\n case 'ebool':\n return {\n ctHash: data,\n utype: FheTypes.Bool,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint8':\n return {\n ctHash: data,\n utype: FheTypes.Uint8,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint16':\n return {\n ctHash: data,\n utype: FheTypes.Uint16,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint32':\n return {\n ctHash: data,\n utype: FheTypes.Uint32,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint64':\n return {\n ctHash: data,\n utype: FheTypes.Uint64,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint128':\n return {\n ctHash: data,\n utype: FheTypes.Uint128,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'eaddress':\n return {\n ctHash: data,\n utype: FheTypes.Uint160,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n default:\n throw new Error(`Unknown encrypted return type: ${internalType}`);\n }\n}\n\nfunction transformArrayOfEncryptedReturnTypesToReturnTypes<T extends EncryptedReturnTypeInternalType>(\n internalType: T,\n size: string | undefined,\n data: string[]\n): EncryptedReturnTypeInternalTypeToReturnTypeMap<T>[] {\n // Ensure data is an array\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n\n // Ensure length of fixed length tuple matches the length of the data array\n if (size != null && size !== '' && parseInt(size) !== data.length) {\n throw new Error(`Array size mismatch: ${size} !== ${data.length}`);\n }\n\n // Transform each item in the data array into an return type\n return data.map((item) => transformSingleEncryptedReturnTypeToReturnType(internalType, item));\n}\n\nfunction internalTypeIsEncryptedReturnType(internalType: string): internalType is EncryptedReturnTypeInternalType {\n return ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES.includes(internalType as any);\n}\n\nexport function transformEncryptedReturnTypes<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n data: ContractReturnType<TAbi, TFunctionName>\n): CofheReturnType<TAbi, TFunctionName> {\n const abiFunction = getAbiFunction(abi, functionName);\n const outputs = abiFunction?.outputs;\n if (abiFunction == null || outputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n const outputsLength = outputs.length;\n\n function processParameter(param: AbiParameter, value: unknown): unknown {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead, internalTypeSize] = extractArrayParameterType(param.internalType);\n\n // Is encrypted return type\n if (internalTypeHead != null && internalTypeIsEncryptedReturnType(internalTypeHead)) {\n // Is single encrypted return type\n if (internalTypeSize == null) {\n return transformSingleEncryptedReturnTypeToReturnType(internalTypeHead, value as string);\n }\n\n // Is array of encrypted return types\n if (internalTypeSize != null) {\n return transformArrayOfEncryptedReturnTypesToReturnTypes(internalTypeHead, internalTypeSize, value as string[]);\n }\n }\n\n // Tuple but not an encrypted return type (recursive case)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): unknown => {\n const valueObj = tupleValue as Record<string, unknown>;\n const result: Record<string, unknown> = {};\n\n components.forEach((component) => {\n const componentName = component.name;\n if (componentName) {\n const componentValue = valueObj[componentName];\n if (componentValue !== undefined) {\n result[componentName] = processParameter(component, componentValue);\n }\n }\n });\n\n return result;\n };\n\n // `tuple[]` / `tuple[N]`: transform each element as its own tuple. Treating the\n // array itself as one tuple would look up named components on the array object,\n // find none, and collapse the whole result to `{}`.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n return value.map(processTuple);\n }\n\n return processTuple(value);\n }\n\n return value;\n }\n\n // Not an encrypted return type, return original value\n return value;\n }\n\n // Process results (array case)\n if (outputsLength > 1) {\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n if (outputsLength !== data.length) {\n throw new Error(`Mismatch in outputs length: ${outputsLength} !== ${data.length}`);\n }\n return data.map((item, index) => {\n return processParameter(outputs[index], item);\n }) as CofheReturnType<TAbi, TFunctionName>;\n }\n\n // Process retuls (single item case)\n return processParameter(outputs[0], data) as CofheReturnType<TAbi, TFunctionName>;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,iBAoBO;;;ACuJA,SAAS,0BAAwD,MAAkC;AACxG,MAAI,QAAQ,KAAM,QAAO,CAAC,MAAM,MAAS;AAEzC,QAAM,QAAQ,KAAK,MAAM,iBAAiB;AAE1C,MAAI,CAAC,OAAO;AAEV,WAAO,CAAC,MAAM,MAAS;AAAA,EACzB;AAEA,QAAM,OAAO,MAAM,CAAC;AACpB,QAAM,OAAO,MAAM,CAAC;AAGpB,SAAO,CAAC,MAAW,SAAS,KAAK,KAAK,IAAI;AAC5C;AAIO,SAAS,cAAc,MAA8C;AAC1E,MAAI,OAAO,SAAS,YAAY,SAAS,KAAM,QAAO;AACtD,SAAO,UAAU,QAAQ,KAAK,SAAS,cAAc,UAAU,QAAQ,aAAa;AACtF;AAEO,SAAS,eACd,KACA,cACyB;AACzB,SAAO,IAAI,KAAK,CAAC,SAAS;AACxB,UAAM,aAAa,cAAc,IAAI;AACrC,QAAI,CAAC,WAAY,QAAO;AACxB,WAAO,KAAK,SAAS;AAAA,EACvB,CAAC;AACH;;;ADhHA,IAAM,gCAAgC;AAAA,EACpC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAsEA,SAAS,4BAA4B,cAAiE;AACpG,SAAO,8BAA8B,SAAS,YAAmB;AACnE;AAEA,SAAS,sBAAsB,OAA8B;AAC3D,QAAM,CAAC,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AACvE,MAAI,oBAAoB,QAAQ,4BAA4B,gBAAgB,EAAG,QAAO;AACtF,MAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,WAAO,MAAM,WAAW,KAAK,CAAC,cAAc,sBAAsB,SAAS,CAAC;AAAA,EAC9E;AACA,SAAO;AACT;AAEA,SAAS,qCAAqC,cAAyC,MAAgC;AACrH,UAAQ,cAAc;AAAA,IACpB,KAAK;AACH,aAAO,uBAAY,KAAK,IAAe;AAAA,IACzC,KAAK;AACH,aAAO,uBAAY,MAAM,IAAuB;AAAA,IAClD,KAAK;AACH,aAAO,uBAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,uBAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,uBAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,uBAAY,QAAQ,IAAuB;AAAA,IACpD,KAAK;AACH,aAAO,uBAAY,QAAQ,IAAuB;AAAA,IACpD;AACE,YAAM,IAAI,MAAM,gCAAgC,YAAY,EAAE;AAAA,EAClE;AACF;AAEA,SAAS,wCACP,cACA,MACA,MACmB;AACnB,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,uBAAuB;AAAA,EACzC;AAEA,MAAI,QAAQ,QAAQ,SAAS,MAAM,SAAS,IAAI,MAAM,KAAK,QAAQ;AACjE,UAAM,IAAI,MAAM,wBAAwB,IAAI,QAAQ,KAAK,MAAM,EAAE;AAAA,EACnE;AAEA,SAAO,KAAK,IAAI,CAAC,SAAS,qCAAqC,cAAc,IAAI,CAAC;AACpF;AAYO,SAAS,yBACd,KACA,cACA,MACmB;AACnB,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,SAAS,aAAa;AAC5B,MAAI,eAAe,QAAQ,UAAU,MAAM;AACzC,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AAEA,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,4BAA4B;AAAA,EAC9C;AAEA,QAAM,oBAAoB,OAAO,KAAK,CAAC,UAAU,sBAAsB,KAAK,CAAC;AAC7E,QAAM,qBAAqB,uBAAuB,QAAQ,mBAAmB,YAAY;AAGzF,QAAM,mBAAsC,CAAC;AAE7C,WAAS,iBAAiB,OAAqB,OAAsB;AACnE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,kBAAkB,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGzF,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,uBAAiB,KAAK,qCAAqC,kBAAkB,KAAK,CAAC;AACnF;AAAA,IACF;AAGA,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,YAAM,eAAe,wCAAwC,kBAAkB,UAAU,KAAK;AAC9F,uBAAiB,KAAK,GAAG,YAAY;AACrC;AAAA,IACF;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAA8B;AAClD,gBAAM,WAAW;AACjB,qBAAW,QAAQ,CAAC,cAAc;AAChC,6BAAiB,WAAW,SAAS,UAAU,IAAI,CAAC;AAAA,UACtD,CAAC;AAAA,QACH;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,gBAAM,QAAQ,YAAY;AAC1B;AAAA,QACF;AAEA,qBAAa,KAAK;AAAA,MACpB;AACA;AAAA,IACF;AAGA;AAAA,EACF;AAIA,MAAI,WAAW;AACf,SAAO,QAAQ,CAAC,OAAO,UAAU;AAC/B,QAAI,UAAU,mBAAoB;AAClC,UAAM,MAAM,KAAK,QAAQ;AACzB,QAAI,OAAO,MAAM;AACf,YAAM,IAAI,MAAM,YAAY,QAAQ,eAAe;AAAA,IACrD;AACA;AACA,qBAAiB,OAAO,GAAG;AAAA,EAC7B,CAAC;AAED,SAAO;AACT;AASA,SAAS,uBACP,QACA,mBACA,cACQ;AACR,MAAI,CAAC,kBAAmB,QAAO;AAE/B,QAAM,kBAAkB,OAAO,OAAiB,CAAC,KAAK,OAAO,UAAU;AACrE,QAAI,sBAAsB,KAAK,EAAG,KAAI,KAAK,KAAK;AAChD,WAAO;AAAA,EACT,GAAG,CAAC,CAAC;AAEL,QAAM,QAAQ,gBAAgB,CAAC;AAC/B,QAAM,OAAO,gBAAgB,gBAAgB,SAAS,CAAC;AACvD,MAAI,OAAO,QAAQ,MAAM,gBAAgB,QAAQ;AAC/C,UAAM,IAAI;AAAA,MACR,YAAY,YAAY,gEAClB,gBAAgB,KAAK,IAAI,CAAC;AAAA,IAElC;AAAA,EACF;AAEA,QAAM,YAAY,OAAO;AACzB,QAAM,OAAO,OAAO,SAAS;AAC7B,MAAI,MAAM,SAAS,SAAS;AAC1B,UAAM,IAAI;AAAA,MACR,YAAY,YAAY,+JAElB,MAAM,QAAQ,SAAS,iBAAiB,SAAS;AAAA,IACzD;AAAA,EACF;AAEA,SAAO;AACT;AAiBO,SAAS,sBACd,KACA,cACA,MACA,iBACqC;AACrC,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,SAAS,aAAa;AAC5B,MAAI,eAAe,QAAQ,UAAU,MAAM;AACzC,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AAEA,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,4BAA4B;AAAA,EAC9C;AAEA,QAAM,oBAAoB,OAAO,KAAK,CAAC,UAAU,sBAAsB,KAAK,CAAC;AAC7E,QAAM,qBAAqB,uBAAuB,QAAQ,mBAAmB,YAAY;AAEzF,QAAM,SAAS,oBAAoB,gBAAgB,MAAM,GAAG,EAAE,IAAI,CAAC;AACnE,QAAM,YAAY,oBAAoB,gBAAgB,gBAAgB,SAAS,CAAC,IAAI;AAGpF,MAAI,YAAY;AAEhB,WAAS,iBAAiB,OAAqB,OAAyB;AACtE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGvE,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,UAAI,aAAa,OAAO,QAAQ;AAC9B,cAAM,IAAI,MAAM,kDAAkD,YAAY,CAAC,SAAS,OAAO,MAAM,EAAE;AAAA,MACzG;AACA,YAAM,OAAO,OAAO,SAAS;AAC7B;AACA,aAAO;AAAA,IACT;AAGA,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,YAAM,cAAc,MAAM,QAAQ,KAAK,IAAI,MAAM,SAAS;AAC1D,UAAI,gBAAgB,GAAG;AACrB,eAAO,CAAC;AAAA,MACV;AAEA,UAAI,YAAY,cAAc,OAAO,QAAQ;AAC3C,cAAM,IAAI;AAAA,UACR,kDAAkD,YAAY,WAAW,SAAS,OAAO,MAAM;AAAA,QACjG;AAAA,MACF;AAEA,YAAM,YAAY,OAAO,MAAM,WAAW,YAAY,WAAW;AACjE,mBAAa;AACb,aAAO;AAAA,IACT;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAAiC;AACrD,gBAAM,WAAW;AACjB,gBAAMA,UAAkC,CAAC;AAEzC,qBAAW,QAAQ,CAAC,cAAc;AAChC,kBAAM,gBAAgB,UAAU;AAChC,gBAAI,eAAe;AACjB,oBAAM,iBAAiB,SAAS,aAAa;AAC7C,kBAAI,mBAAmB,QAAW;AAChC,gBAAAA,QAAO,aAAa,IAAI,iBAAiB,WAAW,cAAc;AAAA,cACpE;AAAA,YACF;AAAA,UACF,CAAC;AAED,iBAAOA;AAAA,QACT;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,iBAAO,MAAM,IAAI,YAAY;AAAA,QAC/B;AAEA,eAAO,aAAa,KAAK;AAAA,MAC3B;AACA,aAAO;AAAA,IACT;AAGA,WAAO;AAAA,EACT;AAIA,MAAI,WAAW;AACf,QAAM,SAAS,OAAO,IAAI,CAAC,OAAO,UAAU;AAC1C,QAAI,UAAU,oBAAoB;AAChC,aAAO;AAAA,IACT;AACA,UAAM,MAAM,KAAK,QAAQ;AACzB,QAAI,OAAO,MAAM;AACf,YAAM,IAAI,MAAM,YAAY,QAAQ,eAAe;AAAA,IACrD;AACA;AACA,WAAO,iBAAiB,OAAO,GAAG;AAAA,EACpC,CAAC;AAGD,MAAI,cAAc,OAAO,QAAQ;AAC/B,UAAM,IAAI,MAAM,4CAA4C,SAAS,kBAAkB,OAAO,MAAM,EAAE;AAAA,EACxG;AAEA,SAAO;AACT;;;AElfA,IAAAC,cAAkE;AAgHlE,IAAM,uCAAuC;AAAA,EAC3C;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAmBA,SAAS,+CACP,cACA,MACmD;AACnD,UAAQ,cAAc;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF;AACE,YAAM,IAAI,MAAM,kCAAkC,YAAY,EAAE;AAAA,EACpE;AACF;AAEA,SAAS,kDACP,cACA,MACA,MACqD;AAErD,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,uBAAuB;AAAA,EACzC;AAGA,MAAI,QAAQ,QAAQ,SAAS,MAAM,SAAS,IAAI,MAAM,KAAK,QAAQ;AACjE,UAAM,IAAI,MAAM,wBAAwB,IAAI,QAAQ,KAAK,MAAM,EAAE;AAAA,EACnE;AAGA,SAAO,KAAK,IAAI,CAAC,SAAS,+CAA+C,cAAc,IAAI,CAAC;AAC9F;AAEA,SAAS,kCAAkC,cAAuE;AAChH,SAAO,qCAAqC,SAAS,YAAmB;AAC1E;AAEO,SAAS,8BACd,KACA,cACA,MACsC;AACtC,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,UAAU,aAAa;AAC7B,MAAI,eAAe,QAAQ,WAAW,MAAM;AAC1C,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AACA,QAAM,gBAAgB,QAAQ;AAE9B,WAAS,iBAAiB,OAAqB,OAAyB;AACtE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,kBAAkB,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGzF,QAAI,oBAAoB,QAAQ,kCAAkC,gBAAgB,GAAG;AAEnF,UAAI,oBAAoB,MAAM;AAC5B,eAAO,+CAA+C,kBAAkB,KAAe;AAAA,MACzF;AAGA,UAAI,oBAAoB,MAAM;AAC5B,eAAO,kDAAkD,kBAAkB,kBAAkB,KAAiB;AAAA,MAChH;AAAA,IACF;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAAiC;AACrD,gBAAM,WAAW;AACjB,gBAAM,SAAkC,CAAC;AAEzC,qBAAW,QAAQ,CAAC,cAAc;AAChC,kBAAM,gBAAgB,UAAU;AAChC,gBAAI,eAAe;AACjB,oBAAM,iBAAiB,SAAS,aAAa;AAC7C,kBAAI,mBAAmB,QAAW;AAChC,uBAAO,aAAa,IAAI,iBAAiB,WAAW,cAAc;AAAA,cACpE;AAAA,YACF;AAAA,UACF,CAAC;AAED,iBAAO;AAAA,QACT;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,iBAAO,MAAM,IAAI,YAAY;AAAA,QAC/B;AAEA,eAAO,aAAa,KAAK;AAAA,MAC3B;AAEA,aAAO;AAAA,IACT;AAGA,WAAO;AAAA,EACT;AAGA,MAAI,gBAAgB,GAAG;AACrB,QAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,YAAM,IAAI,MAAM,uBAAuB;AAAA,IACzC;AACA,QAAI,kBAAkB,KAAK,QAAQ;AACjC,YAAM,IAAI,MAAM,+BAA+B,aAAa,QAAQ,KAAK,MAAM,EAAE;AAAA,IACnF;AACA,WAAO,KAAK,IAAI,CAAC,MAAM,UAAU;AAC/B,aAAO,iBAAiB,QAAQ,KAAK,GAAG,IAAI;AAAA,IAC9C,CAAC;AAAA,EACH;AAGA,SAAO,iBAAiB,QAAQ,CAAC,GAAG,IAAI;AAC1C;","names":["result","import_sdk"]}
|
|
1
|
+
{"version":3,"sources":["../src/index.ts","../src/encryptedInputs.ts","../src/utils.ts","../src/encryptedReturnTypes.ts"],"sourcesContent":["export * from './encryptedInputs';\nexport * from './encryptedReturnTypes';\nexport * from './fhenixMap';\nexport type { Abi } from 'abitype';\nexport type { ContractReturnType } from './utils';\nexport { transformEncryptedReturnTypes } from './encryptedReturnTypes';\n","import {\n Encryptable,\n type AnyExternalHash,\n type EncryptableAddress,\n type EncryptableBool,\n type EncryptableItem,\n type EncryptableUint128,\n type EncryptableUint16,\n type EncryptableUint32,\n type EncryptableUint64,\n type EncryptableUint8,\n type ExternalAddressHash,\n type ExternalBoolHash,\n type ExternalUint128Hash,\n type ExternalUint16Hash,\n type ExternalUint32Hash,\n type ExternalUint64Hash,\n type ExternalUint8Hash,\n type LiteralToPrimitive,\n type Primitive,\n} from '@cofhe/sdk';\nimport type { Abi, AbiFunction, AbiParameter, ExtractAbiFunction } from 'abitype';\nimport type { CofheAbiParametersToPrimitiveTypes } from './fhenixMap';\nimport {\n extractArrayParameterType,\n getAbiFunction,\n type MaybeExtractArrayParameterType,\n type MaybePartialBy,\n type ReadonlyWiden,\n tupleComponentsAreNamed,\n} from './utils';\n\nexport type CofheInputArgs<abi extends Abi | readonly unknown[] = Abi, functionName extends string = string> = GetArgs<\n abi,\n functionName\n>['args'];\n\ntype ExternalHashToInputMap<E extends AnyExternalHash> = E extends ExternalBoolHash\n ? EncryptableBool['data']\n : E extends ExternalUint8Hash\n ? EncryptableUint8['data']\n : E extends ExternalUint16Hash\n ? EncryptableUint16['data']\n : E extends ExternalUint32Hash\n ? EncryptableUint32['data']\n : E extends ExternalUint64Hash\n ? EncryptableUint64['data']\n : E extends ExternalUint128Hash\n ? EncryptableUint128['data']\n : E extends ExternalAddressHash\n ? EncryptableAddress['data']\n : never;\n\n// Note: AnyExternalHash is checked before Primitive - the branded hash types are structurally\n// `0x${string}` (a subtype of `string`, which is itself a Primitive), so Primitive must not be\n// checked first or every hash would be (incorrectly) widened via LiteralToPrimitive instead.\ntype EncryptedInputsToInputs<T> = T extends AnyExternalHash\n ? ExternalHashToInputMap<T>\n : T extends Primitive\n ? LiteralToPrimitive<T>\n : {\n [K in keyof T]: EncryptedInputsToInputs<T[K]>;\n };\n\n/**\n * Drops the batch-signature argument from an args tuple.\n *\n * The signature slot is the first non-`external*` parameter that follows the contiguous run of\n * `external*` parameters - `FHE.asEuintXX(hash, proof)` is a pair, so the proof sits with the\n * hashes it authenticates rather than at the end of the parameter list. Walks the ABI parameters\n * and the args tuple in lockstep and elides the arg at that position.\n */\ntype DropSignatureSlotTuple<\n params extends readonly unknown[],\n args extends readonly unknown[],\n afterExternalRun extends boolean = false,\n> = params extends readonly [infer P, ...infer PRest extends readonly unknown[]]\n ? args extends readonly [infer A, ...infer ARest extends readonly unknown[]]\n ? P extends AbiParameter\n ? ParamHasExternalInput<P> extends true\n ? readonly [A, ...DropSignatureSlotTuple<PRest, ARest, true>]\n : afterExternalRun extends true\n ? DropSignatureSlotTuple<PRest, ARest, false>\n : readonly [A, ...DropSignatureSlotTuple<PRest, ARest, false>]\n : readonly [A, ...DropSignatureSlotTuple<PRest, ARest, afterExternalRun>]\n : readonly []\n : args;\n\n/** Non-tuple args (e.g. `undefined`) pass through unchanged, as the previous `DropLast` did. */\ntype DropSignatureSlot<params extends readonly unknown[], args> = args extends readonly unknown[]\n ? DropSignatureSlotTuple<params, args>\n : args;\n\nconst EXTERNAL_INPUT_INTERNAL_TYPES = [\n 'externalEbool',\n 'externalEuint8',\n 'externalEuint16',\n 'externalEuint32',\n 'externalEuint64',\n 'externalEuint128',\n 'externalEaddress',\n] as const;\ntype ExternalInputInternalType = (typeof EXTERNAL_INPUT_INTERNAL_TYPES)[number];\n\n/** Type-level: does this ABI parameter carry (or, if a tuple, recursively contain) an `external*` input? */\ntype ParamHasExternalInput<param extends AbiParameter> =\n MaybeExtractArrayParameterType<param['internalType']> extends [infer head extends string, any]\n ? head extends ExternalInputInternalType\n ? true\n : // A struct array (`tuple[]` / `tuple[N]`) lands here too, with `head` the struct's own\n // internalType. Falling through to the components keeps `Order[]` visible as carrying\n // encrypted inputs, the way the runtime `paramHasExternalInput` already sees it.\n ParamComponentsHaveExternalInput<param>\n : param['internalType'] extends ExternalInputInternalType\n ? true\n : ParamComponentsHaveExternalInput<param>;\n\n/** Recurses into a parameter's `components`, for tuples at any array depth. */\ntype ParamComponentsHaveExternalInput<param extends AbiParameter> = param extends {\n components: infer components extends readonly AbiParameter[];\n}\n ? ParamsHaveExternalInput<components>\n : false;\n\ntype ParamsHaveExternalInput<params extends readonly AbiParameter[]> = params extends readonly [\n infer Head extends AbiParameter,\n ...infer Rest extends readonly AbiParameter[],\n]\n ? ParamHasExternalInput<Head> extends true\n ? true\n : ParamsHaveExternalInput<Rest>\n : false;\n\n/**\n * Pre-transform args for a function ABI: the raw plain values a caller supplies before encryption.\n *\n * If the function has one or more `external*` (encrypted) inputs, the plain `bytes` parameter\n * immediately following them is the shared batch signature slot - the caller never supplies it\n * directly (the SDK injects it after encryption), so it's dropped from this type entirely.\n */\nexport type CofheInputArgsPreTransform<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n abiFunction extends AbiFunction = abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction,\n> = EncryptedInputsToInputs<\n ParamsHaveExternalInput<abiFunction['inputs']> extends true\n ? DropSignatureSlot<abiFunction['inputs'], CofheInputArgs<abi, functionName>>\n : CofheInputArgs<abi, functionName>\n>;\n\n/// GetArgs from abitype (not exported from abitype)\ntype GetArgs<\n abi extends Abi | readonly unknown[] = Abi, // `readonly unknown[]` allows for non-const asserted types\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly [],\n ///\n abiFunction extends AbiFunction = abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction,\n primitiveTypes = CofheAbiParametersToPrimitiveTypes<abiFunction['inputs']>,\n args_ =\n | primitiveTypes // show all values\n | (abi extends Abi\n ? args extends primitiveTypes // infer value (if valid)\n ? primitiveTypes extends args // make sure `args` exactly matches `primitiveTypes` (e.g. avoid `args: readonly [{ foo: string; bar: number; }] | readonly [{ foo: string; }]`)\n ? // make inferred value of `args` match `primitiveTypes` (e.g. avoid union `args: readonly [123n] | readonly [bigint]`)\n ReadonlyWiden<args>\n : never\n : never\n : never)\n | (Abi extends abi ? readonly unknown[] : never), // fallback if `abi` is declared as `Abi`\n> = MaybePartialBy<{ args: args_ }, readonly [] extends primitiveTypes ? 'args' : Abi extends abi ? 'args' : string>;\n\nfunction internalTypeIsExternalInput(internalType: string): internalType is ExternalInputInternalType {\n return EXTERNAL_INPUT_INTERNAL_TYPES.includes(internalType as any);\n}\n\nfunction paramHasExternalInput(param: AbiParameter): boolean {\n const [internalTypeHead] = extractArrayParameterType(param.internalType);\n if (internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) return true;\n if ('components' in param && Array.isArray(param.components)) {\n return param.components.some((component) => paramHasExternalInput(component));\n }\n return false;\n}\n\nfunction transformSingleExternalToEncryptable(internalType: ExternalInputInternalType, data: unknown): EncryptableItem {\n switch (internalType) {\n case 'externalEbool':\n return Encryptable.bool(data as boolean);\n case 'externalEuint8':\n return Encryptable.uint8(data as string | bigint);\n case 'externalEuint16':\n return Encryptable.uint16(data as string | bigint);\n case 'externalEuint32':\n return Encryptable.uint32(data as string | bigint);\n case 'externalEuint64':\n return Encryptable.uint64(data as string | bigint);\n case 'externalEuint128':\n return Encryptable.uint128(data as string | bigint);\n case 'externalEaddress':\n return Encryptable.address(data as string | bigint);\n default:\n throw new Error(`Unknown external input type: ${internalType}`);\n }\n}\n\nfunction transformArrayOfExternalsToEncryptables(\n internalType: ExternalInputInternalType,\n size: string | undefined,\n data: unknown\n): EncryptableItem[] {\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n\n if (size != null && size !== '' && parseInt(size) !== data.length) {\n throw new Error(`Array size mismatch: ${size} !== ${data.length}`);\n }\n\n return data.map((item) => transformSingleExternalToEncryptable(internalType, item));\n}\n\n/**\n * Extracts encryptable values from function arguments based on the ABI.\n * Transforms raw data values into EncryptableItem objects that can be passed to the encrypt function.\n *\n * @param abi - The ABI containing the function definition\n * @param functionName - Name of the function\n * @param args - Function arguments in the format of CofheInputArgsPreTransform (raw data values;\n * if the function has `external*` inputs, the trailing batch-signature slot is NOT included)\n * @returns Array of EncryptableItem objects ready for encryption, in ABI parameter order\n */\nexport function extractEncryptableValues<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n args: CofheInputArgsPreTransform<TAbi, TFunctionName>\n): EncryptableItem[] {\n const abiFunction = getAbiFunction(abi, functionName);\n const inputs = abiFunction?.inputs;\n if (abiFunction == null || inputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n\n if (!Array.isArray(args)) {\n throw new Error('Arguments must be an array');\n }\n\n const hasExternalInputs = inputs.some((input) => paramHasExternalInput(input));\n const signatureSlotIndex = findSignatureSlotIndex(inputs, hasExternalInputs, functionName);\n\n // Collect encrypted values as EncryptableItem objects in order (flat array)\n const encryptableItems: EncryptableItem[] = [];\n\n function processParameter(param: AbiParameter, value: unknown): void {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead, internalTypeSize] = extractArrayParameterType(param.internalType);\n\n // Bare external input (e.g. `externalEuint32`, type: 'bytes32')\n if (typeSize == null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n encryptableItems.push(transformSingleExternalToEncryptable(internalTypeHead, value));\n return;\n }\n\n // Array of external inputs (e.g. `externalEuint32[]`/`externalEuint32[2]`, type: 'bytes32[]'/'bytes32[2]')\n if (typeSize != null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n const encryptables = transformArrayOfExternalsToEncryptables(internalTypeHead, typeSize, value);\n encryptableItems.push(...encryptables);\n return;\n }\n\n // Tuple recursive case (struct that may itself contain external inputs)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): void => {\n if (tupleComponentsAreNamed(components)) {\n const valueObj = tupleValue as Record<string, unknown>;\n components.forEach((component) => {\n processParameter(component, valueObj[component.name]);\n });\n return;\n }\n\n if (!Array.isArray(tupleValue)) {\n throw new Error('Expected an array value for tuple with unnamed components');\n }\n\n components.forEach((component, index) => {\n processParameter(component, tupleValue[index]);\n });\n };\n\n // `tuple[]` / `tuple[N]`: walk each element as its own tuple. Treating the array\n // itself as one tuple would look up named components on the array object, find\n // none, and hand `undefined` to the encryptable transform.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n value.forEach(processTuple);\n return;\n }\n\n processTuple(value);\n }\n return;\n }\n\n // Not an external input, and cannot contain one\n return;\n }\n\n // Process all inputs, skipping the batch-signature slot (nothing to extract there). `args` is the\n // pre-transform tuple, which omits that slot, so its indices lag by one past the signature.\n let argIndex = 0;\n inputs.forEach((input, index) => {\n if (index === signatureSlotIndex) return;\n const arg = args[argIndex];\n if (arg == null) {\n throw new Error(`Argument ${argIndex} is undefined`);\n }\n argIndex++;\n processParameter(input, arg);\n });\n\n return encryptableItems;\n}\n\n/**\n * Index of the shared batch-signature parameter: the plain `bytes` immediately after the\n * contiguous run of `external*` parameters. Returns -1 when the function has no encrypted inputs.\n *\n * The proof pairs with the hashes it authenticates (`FHE.asEuintXX(hash, proof)`), so it does not\n * have to be the last parameter - anything may follow it.\n */\nfunction findSignatureSlotIndex(\n inputs: readonly AbiParameter[],\n hasExternalInputs: boolean,\n functionName: string\n): number {\n if (!hasExternalInputs) return -1;\n\n const externalIndices = inputs.reduce<number[]>((acc, input, index) => {\n if (paramHasExternalInput(input)) acc.push(index);\n return acc;\n }, []);\n\n const first = externalIndices[0]!;\n const last = externalIndices[externalIndices.length - 1]!;\n if (last - first + 1 !== externalIndices.length) {\n throw new Error(\n `Function ${functionName} has encrypted (external*) inputs at non-adjacent positions ` +\n `(${externalIndices.join(', ')}). They share one batch signature, so they must be contiguous ` +\n `with the 'bytes' signature parameter immediately after them`\n );\n }\n\n const slotIndex = last + 1;\n const slot = inputs[slotIndex];\n if (slot?.type !== 'bytes') {\n throw new Error(\n `Function ${functionName} has encrypted (external*) inputs, so the parameter immediately after ` +\n `them must be a plain 'bytes' parameter to receive the shared batch signature - found ` +\n `'${slot?.type ?? 'nothing'}' at position ${slotIndex} instead`\n );\n }\n\n return slotIndex;\n}\n\n/**\n * Re-inserts encrypted values back into function arguments based on the ABI.\n * Takes the batch-verified result (per-input hashes, in order, followed by the single shared\n * signature) and the original args structure, and produces the final call args: each `external*`\n * position gets its corresponding hash, and the trailing `bytes` parameter gets the signature.\n *\n * This function mirrors the extraction logic in extractEncryptableValues, ensuring values\n * are inserted in the exact same order and locations where they were extracted.\n *\n * @param abi - The ABI containing the function definition\n * @param functionName - Name of the function\n * @param args - Original function arguments in the format of CofheInputArgsPreTransform\n * @param encryptedResult - `[...hashes, signature]`, as returned by `EncryptInputsBuilder.execute()`\n * @returns Function arguments with encrypted values inserted (format of CofheInputArgs)\n */\nexport function insertEncryptedValues<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n args: CofheInputArgsPreTransform<TAbi, TFunctionName>,\n encryptedResult: readonly `0x${string}`[]\n): CofheInputArgs<TAbi, TFunctionName> {\n const abiFunction = getAbiFunction(abi, functionName);\n const inputs = abiFunction?.inputs;\n if (abiFunction == null || inputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n\n if (!Array.isArray(args)) {\n throw new Error('Arguments must be an array');\n }\n\n const hasExternalInputs = inputs.some((input) => paramHasExternalInput(input));\n const signatureSlotIndex = findSignatureSlotIndex(inputs, hasExternalInputs, functionName);\n\n const hashes = hasExternalInputs ? encryptedResult.slice(0, -1) : [];\n const signature = hasExternalInputs ? encryptedResult[encryptedResult.length - 1] : undefined;\n\n // Track position in the hashes array\n let hashIndex = 0;\n\n function processParameter(param: AbiParameter, value: unknown): unknown {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead] = extractArrayParameterType(param.internalType);\n\n // Bare external input - substitute its hash\n if (typeSize == null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n if (hashIndex >= hashes.length) {\n throw new Error(`Not enough encrypted hashes: expected at least ${hashIndex + 1}, got ${hashes.length}`);\n }\n const hash = hashes[hashIndex];\n hashIndex++;\n return hash;\n }\n\n // Array of external inputs - substitute the corresponding slice of hashes\n if (typeSize != null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n const arrayLength = Array.isArray(value) ? value.length : 0;\n if (arrayLength === 0) {\n return [];\n }\n\n if (hashIndex + arrayLength > hashes.length) {\n throw new Error(\n `Not enough encrypted hashes: expected at least ${hashIndex + arrayLength}, got ${hashes.length}`\n );\n }\n\n const hashArray = hashes.slice(hashIndex, hashIndex + arrayLength);\n hashIndex += arrayLength;\n return hashArray;\n }\n\n // Tuple recursive case (struct that may itself contain external inputs)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): unknown => {\n if (tupleComponentsAreNamed(components)) {\n const valueObj = tupleValue as Record<string, unknown>;\n const result: Record<string, unknown> = {};\n\n components.forEach((component) => {\n const componentName = component.name;\n if (componentName) {\n const componentValue = valueObj[componentName];\n if (componentValue !== undefined) {\n result[componentName] = processParameter(component, componentValue);\n }\n }\n });\n\n return result;\n }\n\n if (!Array.isArray(tupleValue)) {\n throw new Error('Expected an array value for tuple with unnamed components');\n }\n\n return components.map((component, index) => processParameter(component, tupleValue[index]));\n };\n\n // `tuple[]` / `tuple[N]`: rebuild each element as its own tuple, in the same order\n // extractEncryptableValues walked them, so the hashes line up. Treating the array as\n // one tuple would consume no hashes and collapse the argument to `{}`.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n return value.map(processTuple);\n }\n\n return processTuple(value);\n }\n return value;\n }\n\n // Not an external input, return original value\n return value;\n }\n\n // Rebuild the full arg list; the signature slot is injected directly. `args` omits that slot, so\n // its indices lag by one past the signature position.\n let argIndex = 0;\n const result = inputs.map((input, index) => {\n if (index === signatureSlotIndex) {\n return signature;\n }\n const arg = args[argIndex];\n if (arg == null) {\n throw new Error(`Argument ${argIndex} is undefined`);\n }\n argIndex++;\n return processParameter(input, arg);\n });\n\n // Verify we used all encrypted hashes\n if (hashIndex !== hashes.length) {\n throw new Error(`Mismatch in encrypted hashes count: used ${hashIndex}, but provided ${hashes.length}`);\n }\n\n return result as CofheInputArgs<TAbi, TFunctionName>;\n}\n","import type {\n Abi,\n AbiConstructor,\n AbiError,\n AbiEvent,\n AbiFallback,\n AbiFunction,\n AbiParameter,\n AbiParametersToPrimitiveTypes,\n AbiReceive,\n AbiType,\n Address,\n ExtractAbiFunction,\n ResolvedRegister,\n SolidityArray,\n SolidityFixedArrayRange,\n SolidityTuple,\n} from 'abitype';\n\nexport type ContractReturnType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n ///\n abiFunction extends AbiFunction = (\n abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction\n ) extends infer abiFunction_ extends AbiFunction\n ? IsUnion<abiFunction_> extends true // narrow overloads by `args` by converting to tuple and filtering out overloads that don't match\n ? UnionToTuple<abiFunction_> extends infer abiFunctions extends readonly AbiFunction[]\n ? {\n [K in keyof abiFunctions]: (\n readonly unknown[] | undefined extends args // for functions that don't have inputs, `args` can be `undefined` so fallback to `readonly []`\n ? readonly []\n : args\n ) extends AbiParametersToPrimitiveTypes<abiFunctions[K]['inputs'], 'inputs'>\n ? abiFunctions[K]\n : never;\n }[number] // convert back to union (removes `never` tuple entries: `['foo', never, 'bar'][number]` => `'foo' | 'bar'`)\n : never\n : abiFunction_\n : never,\n outputs extends readonly AbiParameter[] = abiFunction['outputs'],\n primitiveTypes extends readonly unknown[] = AbiParametersToPrimitiveTypes<outputs, 'outputs', true>,\n> = [abiFunction] extends [never]\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` declared as `Abi`)\n : readonly unknown[] extends primitiveTypes\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` not const-asserted)\n : primitiveTypes extends readonly [] // unwrap `primitiveTypes`\n ? // biome-ignore lint/suspicious/noConfusingVoidType: <explanation>\n void // no outputs\n : primitiveTypes extends readonly [infer primitiveType]\n ? primitiveType // single output\n : primitiveTypes;\n\nexport type Error<messages extends string | string[]> = messages extends string\n ? [\n // Surrounding with array to prevent `messages` from being widened to `string`\n `Error: ${messages}`,\n ]\n : {\n [key in keyof messages]: messages[key] extends infer message extends string ? `Error: ${message}` : never;\n };\n\ntype PartialBy<TType, TKeys extends keyof TType> = ExactPartial<Pick<TType, TKeys>> & Omit<TType, TKeys>;\ntype ExactPartial<T> = { [K in keyof T]?: T[K] | undefined };\n\nexport type MaybePartialBy<TType, TKeys extends string> = TKeys extends keyof TType ? PartialBy<TType, TKeys> : TType;\n\nexport type ReadonlyWiden<TType> =\n | (TType extends Function ? TType : never)\n | (TType extends ResolvedRegister['bigIntType'] ? bigint : never)\n | (TType extends boolean ? boolean : never)\n | (TType extends ResolvedRegister['intType'] ? number : never)\n | (TType extends string\n ? TType extends Address\n ? Address\n : TType extends ResolvedRegister['bytesType']['inputs']\n ? ResolvedRegister['bytesType']\n : string\n : never)\n | (TType extends readonly [] ? readonly [] : never)\n | (TType extends Record<string, unknown> ? { [K in keyof TType]: ReadonlyWiden<TType[K]> } : never)\n | (TType extends { length: number }\n ? {\n [K in keyof TType]: ReadonlyWiden<TType[K]>;\n } extends infer Val extends unknown[]\n ? readonly [...Val]\n : never\n : never);\n\nexport type AbiBasicType = Exclude<AbiType, SolidityTuple | SolidityArray>;\n\n/**\n * First, infer `Head` against a known size type (either fixed-length array value or `\"\"`).\n *\n * | Input | Head |\n * | --------------- | ------------ |\n * | `string[]` | `string` |\n * | `string[][][3]` | `string[][]` |\n */\nexport type MaybeExtractArrayParameterType<type> = type extends `${infer head}[${'' | `${SolidityFixedArrayRange}`}]`\n ? // * Then, infer in the opposite direction, using the known `head` to infer the exact `size` value.\n // *\n // * | Input | Size |\n // * | ------------ | ---- |\n // * | `${head}[]` | `\"\"` |\n // * | `${head}[3]` | `3` |\n // */\n type extends `${head}[${infer size}]`\n ? [head, size]\n : undefined\n : undefined;\n\nexport type IsUnion<T, C = T> = T extends C ? ([C] extends [T] ? false : true) : never;\nexport type UnionToTuple<U, Last = LastInUnion<U>> = [U] extends [never]\n ? []\n : [...UnionToTuple<Exclude<U, Last>>, Last];\ntype LastInUnion<U> =\n UnionToIntersection<U extends unknown ? (x: U) => 0 : never> extends (x: infer L) => 0 ? L : never;\ntype UnionToIntersection<U> = (U extends unknown ? (arg: U) => 0 : never) extends (arg: infer I) => 0 ? I : never;\n\n/**\n * Create tuple of {@link type} type with {@link size} size\n *\n * @param Type - Type of tuple\n * @param Size - Size of tuple\n * @returns Tuple of {@link type} type with {@link size} size\n *\n * @example\n * type Result = Tuple<string, 2>\n * // ^? type Result = [string, string]\n */\n// https://github.com/Microsoft/TypeScript/issues/26223#issuecomment-674500430\nexport type Tuple<type, size extends number> = size extends size\n ? number extends size\n ? type[]\n : _TupleOf<type, size, []>\n : never;\ntype _TupleOf<length, size extends number, acc extends readonly unknown[]> = acc['length'] extends size\n ? acc\n : _TupleOf<length, size, readonly [length, ...acc]>;\n\n/**\n * Merges two object types into new type\n *\n * @param object1 - Object to merge into\n * @param object2 - Object to merge and override keys from {@link object1}\n * @returns New object type with keys from {@link object1} and {@link object2}. If a key exists in both {@link object1} and {@link object2}, the key from {@link object2} will be used.\n *\n * @example\n * type Result = Merge<{ foo: string }, { foo: number; bar: string }>\n * // ^? type Result = { foo: number; bar: string }\n */\nexport type Merge<object1, object2> = Omit<object1, keyof object2> & object2;\n\n// FUNCTIONS\n\n/**\n * Extracts array parameter type information from a type string.\n * Always returns a tuple, with undefined for non-array types.\n *\n * @param type - Type string like \"tuple\", \"tuple[]\", or \"tuple[2]\"\n * @returns [head, size] where:\n * - head is the base type\n * - size is undefined for non-arrays, \"\" for dynamic arrays, or the number as a string for fixed arrays\n *\n * @example\n * extractArrayParameterType(\"tuple\") // [\"tuple\", undefined]\n * extractArrayParameterType(\"tuple[]\") // [\"tuple\", \"\"]\n * extractArrayParameterType(\"tuple[2]\") // [\"tuple\", \"2\"]\n */\nexport function extractArrayParameterType<T extends string | undefined>(type: T): [T, string | undefined] {\n if (type == null) return [type, undefined];\n\n const match = type.match(/^(.+)\\[(\\d*)\\]$/);\n\n if (!match) {\n // Not an array type, return [type, undefined]\n return [type, undefined];\n }\n\n const head = match[1];\n const size = match[2]; // Empty string for dynamic arrays, or digits for fixed arrays\n\n // Return empty string for dynamic arrays, or the size string for fixed arrays\n return [head as T, size === '' ? '' : size];\n}\n\n/**\n * Matches the tuple-shape rule used by `CofheAbiComponentsToPrimitiveType`:\n * tuples are objects only when every component has a name. Otherwise they are\n * represented positionally as arrays.\n */\nexport function tupleComponentsAreNamed(components: readonly AbiParameter[]): boolean {\n return components.length > 0 && components.every((component) => component.name != null && component.name !== '');\n}\n\ntype AbiItem = AbiConstructor | AbiError | AbiEvent | AbiFallback | AbiFunction | AbiReceive;\n\nexport function isAbiFunction(item: AbiItem | unknown): item is AbiFunction {\n if (typeof item !== 'object' || item === null) return false;\n return 'type' in item && item.type === 'function' && 'name' in item && 'outputs' in item;\n}\n\nexport function getAbiFunction<TAbi extends Abi | readonly unknown[], TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName\n): AbiFunction | undefined {\n return abi.find((item) => {\n const isFunction = isAbiFunction(item);\n if (!isFunction) return false;\n return item.name === functionName;\n }) as AbiFunction | undefined;\n}\n","import { type Primitive, type LiteralToPrimitive, FheTypes } from '@cofhe/sdk';\nimport type { Abi, AbiFunction, ExtractAbiFunction, AbiParameter } from 'abitype';\nimport type {\n EncryptedReturnType,\n CofheAbiParametersToPrimitiveTypes,\n EBool,\n EAddress,\n EUint128,\n EUint16,\n EUint32,\n EUint64,\n EUint8,\n} from './fhenixMap';\nimport {\n getAbiFunction,\n type IsUnion,\n type UnionToTuple,\n type ContractReturnType,\n extractArrayParameterType,\n tupleComponentsAreNamed,\n} from './utils';\n\ntype FirstCofheReturnValue<T> = [T] extends [void]\n ? never\n : T extends readonly [infer Head, ...(readonly unknown[])]\n ? Head\n : T;\n\ntype FheTypeFromCofheReturnValue<T> = T extends { utype: infer U }\n ? U\n : T extends readonly (infer Element)[]\n ? FheTypeFromCofheReturnValue<Element>\n : never;\n\n/**\n * Extracts the `FheTypes` of the first returned value of a CoFHE call.\n *\n * Examples:\n * - `returns (ebool)` -> `FheTypes.Bool`\n * - `returns (euint8, euint64)` -> `FheTypes.Uint8`\n * - `returns (euint16[])` -> `FheTypes.Uint16`\n */\nexport type CofheFirstReturnFheType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n> = FheTypeFromCofheReturnValue<FirstCofheReturnValue<CofheReturnType<abi, functionName, args>>>;\n\nexport type CofheReturnType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n ///\n abiFunction extends AbiFunction = (\n abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction\n ) extends infer abiFunction_ extends AbiFunction\n ? IsUnion<abiFunction_> extends true // narrow overloads by `args` by converting to tuple and filtering out overloads that don't match\n ? UnionToTuple<abiFunction_> extends infer abiFunctions extends readonly AbiFunction[]\n ? {\n [K in keyof abiFunctions]: (\n readonly unknown[] | undefined extends args // for functions that don't have inputs, `args` can be `undefined` so fallback to `readonly []`\n ? readonly []\n : args\n ) extends CofheAbiParametersToPrimitiveTypes<abiFunctions[K]['inputs'], 'inputs'>\n ? abiFunctions[K]\n : never;\n }[number] // convert back to union (removes `never` tuple entries: `['foo', never, 'bar'][number]` => `'foo' | 'bar'`)\n : never\n : abiFunction_\n : never,\n outputs extends readonly AbiParameter[] = abiFunction['outputs'],\n primitiveTypes extends readonly unknown[] = CofheAbiParametersToPrimitiveTypes<outputs, 'outputs'>,\n> = [abiFunction] extends [never]\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` declared as `Abi`)\n : readonly unknown[] extends primitiveTypes\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` not const-asserted)\n : primitiveTypes extends readonly [] // unwrap `primitiveTypes`\n ? // biome-ignore lint/suspicious/noConfusingVoidType: <explanation>\n void // no outputs\n : primitiveTypes extends readonly [infer primitiveType]\n ? primitiveType // single output\n : primitiveTypes;\n\ntype EncryptedReturnTypeToReturnTypeMap<E extends EncryptedReturnType> = E extends EBool\n ? boolean\n : E extends EUint8\n ? bigint\n : E extends EUint16\n ? bigint\n : E extends EUint32\n ? bigint\n : E extends EUint64\n ? bigint\n : E extends EUint128\n ? bigint\n : E extends EAddress\n ? string\n : never;\n\ntype EncryptedReturnTypesToReturnTypes<T> = T extends Primitive\n ? LiteralToPrimitive<T>\n : T extends EncryptedReturnType\n ? EncryptedReturnTypeToReturnTypeMap<T>\n : {\n [K in keyof T]: EncryptedReturnTypesToReturnTypes<T[K]>;\n };\n\nexport type CofheReturnTypePostTransform<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n> = EncryptedReturnTypesToReturnTypes<CofheReturnType<abi, functionName, args>>;\n\nconst ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES = [\n 'ebool',\n 'euint8',\n 'euint16',\n 'euint32',\n 'euint64',\n 'euint128',\n 'eaddress',\n] as const;\ntype EncryptedReturnTypeInternalType = (typeof ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES)[number];\n\ntype EncryptedReturnTypeInternalTypeToReturnTypeMap<T extends EncryptedReturnTypeInternalType> = T extends 'ebool'\n ? EBool\n : T extends 'euint8'\n ? EUint8\n : T extends 'euint16'\n ? EUint16\n : T extends 'euint32'\n ? EUint32\n : T extends 'euint64'\n ? EUint64\n : T extends 'euint128'\n ? EUint128\n : T extends 'eaddress'\n ? EAddress\n : never;\n\nfunction transformSingleEncryptedReturnTypeToReturnType<T extends EncryptedReturnTypeInternalType>(\n internalType: T,\n data: string\n): EncryptedReturnTypeInternalTypeToReturnTypeMap<T> {\n switch (internalType) {\n case 'ebool':\n return {\n ctHash: data,\n utype: FheTypes.Bool,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint8':\n return {\n ctHash: data,\n utype: FheTypes.Uint8,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint16':\n return {\n ctHash: data,\n utype: FheTypes.Uint16,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint32':\n return {\n ctHash: data,\n utype: FheTypes.Uint32,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint64':\n return {\n ctHash: data,\n utype: FheTypes.Uint64,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint128':\n return {\n ctHash: data,\n utype: FheTypes.Uint128,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'eaddress':\n return {\n ctHash: data,\n utype: FheTypes.Uint160,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n default:\n throw new Error(`Unknown encrypted return type: ${internalType}`);\n }\n}\n\nfunction transformArrayOfEncryptedReturnTypesToReturnTypes<T extends EncryptedReturnTypeInternalType>(\n internalType: T,\n size: string | undefined,\n data: string[]\n): EncryptedReturnTypeInternalTypeToReturnTypeMap<T>[] {\n // Ensure data is an array\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n\n // Ensure length of fixed length tuple matches the length of the data array\n if (size != null && size !== '' && parseInt(size) !== data.length) {\n throw new Error(`Array size mismatch: ${size} !== ${data.length}`);\n }\n\n // Transform each item in the data array into an return type\n return data.map((item) => transformSingleEncryptedReturnTypeToReturnType(internalType, item));\n}\n\nfunction internalTypeIsEncryptedReturnType(internalType: string): internalType is EncryptedReturnTypeInternalType {\n return ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES.includes(internalType as any);\n}\n\nexport function transformEncryptedReturnTypes<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n data: ContractReturnType<TAbi, TFunctionName>\n): CofheReturnType<TAbi, TFunctionName> {\n const abiFunction = getAbiFunction(abi, functionName);\n const outputs = abiFunction?.outputs;\n if (abiFunction == null || outputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n const outputsLength = outputs.length;\n\n function processParameter(param: AbiParameter, value: unknown): unknown {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead, internalTypeSize] = extractArrayParameterType(param.internalType);\n\n // Is encrypted return type\n if (internalTypeHead != null && internalTypeIsEncryptedReturnType(internalTypeHead)) {\n // Is single encrypted return type\n if (internalTypeSize == null) {\n return transformSingleEncryptedReturnTypeToReturnType(internalTypeHead, value as string);\n }\n\n // Is array of encrypted return types\n if (internalTypeSize != null) {\n return transformArrayOfEncryptedReturnTypesToReturnTypes(internalTypeHead, internalTypeSize, value as string[]);\n }\n }\n\n // Tuple but not an encrypted return type (recursive case)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): unknown => {\n if (tupleComponentsAreNamed(components)) {\n const valueObj = tupleValue as Record<string, unknown>;\n const result: Record<string, unknown> = {};\n\n components.forEach((component) => {\n const componentName = component.name;\n if (componentName) {\n const componentValue = valueObj[componentName];\n if (componentValue !== undefined) {\n result[componentName] = processParameter(component, componentValue);\n }\n }\n });\n\n return result;\n }\n\n if (!Array.isArray(tupleValue)) {\n throw new Error('Expected an array value for tuple with unnamed components');\n }\n\n return components.map((component, index) => processParameter(component, tupleValue[index]));\n };\n\n // `tuple[]` / `tuple[N]`: transform each element as its own tuple. Treating the\n // array itself as one tuple would look up named components on the array object,\n // find none, and collapse the whole result to `{}`.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n return value.map(processTuple);\n }\n\n return processTuple(value);\n }\n\n return value;\n }\n\n // Not an encrypted return type, return original value\n return value;\n }\n\n // Process results (array case)\n if (outputsLength > 1) {\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n if (outputsLength !== data.length) {\n throw new Error(`Mismatch in outputs length: ${outputsLength} !== ${data.length}`);\n }\n return data.map((item, index) => {\n return processParameter(outputs[index], item);\n }) as CofheReturnType<TAbi, TFunctionName>;\n }\n\n // Process retuls (single item case)\n return processParameter(outputs[0], data) as CofheReturnType<TAbi, TFunctionName>;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,iBAoBO;;;ACuJA,SAAS,0BAAwD,MAAkC;AACxG,MAAI,QAAQ,KAAM,QAAO,CAAC,MAAM,MAAS;AAEzC,QAAM,QAAQ,KAAK,MAAM,iBAAiB;AAE1C,MAAI,CAAC,OAAO;AAEV,WAAO,CAAC,MAAM,MAAS;AAAA,EACzB;AAEA,QAAM,OAAO,MAAM,CAAC;AACpB,QAAM,OAAO,MAAM,CAAC;AAGpB,SAAO,CAAC,MAAW,SAAS,KAAK,KAAK,IAAI;AAC5C;AAOO,SAAS,wBAAwB,YAA8C;AACpF,SAAO,WAAW,SAAS,KAAK,WAAW,MAAM,CAAC,cAAc,UAAU,QAAQ,QAAQ,UAAU,SAAS,EAAE;AACjH;AAIO,SAAS,cAAc,MAA8C;AAC1E,MAAI,OAAO,SAAS,YAAY,SAAS,KAAM,QAAO;AACtD,SAAO,UAAU,QAAQ,KAAK,SAAS,cAAc,UAAU,QAAQ,aAAa;AACtF;AAEO,SAAS,eACd,KACA,cACyB;AACzB,SAAO,IAAI,KAAK,CAAC,SAAS;AACxB,UAAM,aAAa,cAAc,IAAI;AACrC,QAAI,CAAC,WAAY,QAAO;AACxB,WAAO,KAAK,SAAS;AAAA,EACvB,CAAC;AACH;;;ADxHA,IAAM,gCAAgC;AAAA,EACpC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAsEA,SAAS,4BAA4B,cAAiE;AACpG,SAAO,8BAA8B,SAAS,YAAmB;AACnE;AAEA,SAAS,sBAAsB,OAA8B;AAC3D,QAAM,CAAC,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AACvE,MAAI,oBAAoB,QAAQ,4BAA4B,gBAAgB,EAAG,QAAO;AACtF,MAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,WAAO,MAAM,WAAW,KAAK,CAAC,cAAc,sBAAsB,SAAS,CAAC;AAAA,EAC9E;AACA,SAAO;AACT;AAEA,SAAS,qCAAqC,cAAyC,MAAgC;AACrH,UAAQ,cAAc;AAAA,IACpB,KAAK;AACH,aAAO,uBAAY,KAAK,IAAe;AAAA,IACzC,KAAK;AACH,aAAO,uBAAY,MAAM,IAAuB;AAAA,IAClD,KAAK;AACH,aAAO,uBAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,uBAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,uBAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,uBAAY,QAAQ,IAAuB;AAAA,IACpD,KAAK;AACH,aAAO,uBAAY,QAAQ,IAAuB;AAAA,IACpD;AACE,YAAM,IAAI,MAAM,gCAAgC,YAAY,EAAE;AAAA,EAClE;AACF;AAEA,SAAS,wCACP,cACA,MACA,MACmB;AACnB,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,uBAAuB;AAAA,EACzC;AAEA,MAAI,QAAQ,QAAQ,SAAS,MAAM,SAAS,IAAI,MAAM,KAAK,QAAQ;AACjE,UAAM,IAAI,MAAM,wBAAwB,IAAI,QAAQ,KAAK,MAAM,EAAE;AAAA,EACnE;AAEA,SAAO,KAAK,IAAI,CAAC,SAAS,qCAAqC,cAAc,IAAI,CAAC;AACpF;AAYO,SAAS,yBACd,KACA,cACA,MACmB;AACnB,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,SAAS,aAAa;AAC5B,MAAI,eAAe,QAAQ,UAAU,MAAM;AACzC,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AAEA,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,4BAA4B;AAAA,EAC9C;AAEA,QAAM,oBAAoB,OAAO,KAAK,CAAC,UAAU,sBAAsB,KAAK,CAAC;AAC7E,QAAM,qBAAqB,uBAAuB,QAAQ,mBAAmB,YAAY;AAGzF,QAAM,mBAAsC,CAAC;AAE7C,WAAS,iBAAiB,OAAqB,OAAsB;AACnE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,kBAAkB,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGzF,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,uBAAiB,KAAK,qCAAqC,kBAAkB,KAAK,CAAC;AACnF;AAAA,IACF;AAGA,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,YAAM,eAAe,wCAAwC,kBAAkB,UAAU,KAAK;AAC9F,uBAAiB,KAAK,GAAG,YAAY;AACrC;AAAA,IACF;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAA8B;AAClD,cAAI,wBAAwB,UAAU,GAAG;AACvC,kBAAM,WAAW;AACjB,uBAAW,QAAQ,CAAC,cAAc;AAChC,+BAAiB,WAAW,SAAS,UAAU,IAAI,CAAC;AAAA,YACtD,CAAC;AACD;AAAA,UACF;AAEA,cAAI,CAAC,MAAM,QAAQ,UAAU,GAAG;AAC9B,kBAAM,IAAI,MAAM,2DAA2D;AAAA,UAC7E;AAEA,qBAAW,QAAQ,CAAC,WAAW,UAAU;AACvC,6BAAiB,WAAW,WAAW,KAAK,CAAC;AAAA,UAC/C,CAAC;AAAA,QACH;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,gBAAM,QAAQ,YAAY;AAC1B;AAAA,QACF;AAEA,qBAAa,KAAK;AAAA,MACpB;AACA;AAAA,IACF;AAGA;AAAA,EACF;AAIA,MAAI,WAAW;AACf,SAAO,QAAQ,CAAC,OAAO,UAAU;AAC/B,QAAI,UAAU,mBAAoB;AAClC,UAAM,MAAM,KAAK,QAAQ;AACzB,QAAI,OAAO,MAAM;AACf,YAAM,IAAI,MAAM,YAAY,QAAQ,eAAe;AAAA,IACrD;AACA;AACA,qBAAiB,OAAO,GAAG;AAAA,EAC7B,CAAC;AAED,SAAO;AACT;AASA,SAAS,uBACP,QACA,mBACA,cACQ;AACR,MAAI,CAAC,kBAAmB,QAAO;AAE/B,QAAM,kBAAkB,OAAO,OAAiB,CAAC,KAAK,OAAO,UAAU;AACrE,QAAI,sBAAsB,KAAK,EAAG,KAAI,KAAK,KAAK;AAChD,WAAO;AAAA,EACT,GAAG,CAAC,CAAC;AAEL,QAAM,QAAQ,gBAAgB,CAAC;AAC/B,QAAM,OAAO,gBAAgB,gBAAgB,SAAS,CAAC;AACvD,MAAI,OAAO,QAAQ,MAAM,gBAAgB,QAAQ;AAC/C,UAAM,IAAI;AAAA,MACR,YAAY,YAAY,gEAClB,gBAAgB,KAAK,IAAI,CAAC;AAAA,IAElC;AAAA,EACF;AAEA,QAAM,YAAY,OAAO;AACzB,QAAM,OAAO,OAAO,SAAS;AAC7B,MAAI,MAAM,SAAS,SAAS;AAC1B,UAAM,IAAI;AAAA,MACR,YAAY,YAAY,+JAElB,MAAM,QAAQ,SAAS,iBAAiB,SAAS;AAAA,IACzD;AAAA,EACF;AAEA,SAAO;AACT;AAiBO,SAAS,sBACd,KACA,cACA,MACA,iBACqC;AACrC,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,SAAS,aAAa;AAC5B,MAAI,eAAe,QAAQ,UAAU,MAAM;AACzC,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AAEA,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,4BAA4B;AAAA,EAC9C;AAEA,QAAM,oBAAoB,OAAO,KAAK,CAAC,UAAU,sBAAsB,KAAK,CAAC;AAC7E,QAAM,qBAAqB,uBAAuB,QAAQ,mBAAmB,YAAY;AAEzF,QAAM,SAAS,oBAAoB,gBAAgB,MAAM,GAAG,EAAE,IAAI,CAAC;AACnE,QAAM,YAAY,oBAAoB,gBAAgB,gBAAgB,SAAS,CAAC,IAAI;AAGpF,MAAI,YAAY;AAEhB,WAAS,iBAAiB,OAAqB,OAAyB;AACtE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGvE,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,UAAI,aAAa,OAAO,QAAQ;AAC9B,cAAM,IAAI,MAAM,kDAAkD,YAAY,CAAC,SAAS,OAAO,MAAM,EAAE;AAAA,MACzG;AACA,YAAM,OAAO,OAAO,SAAS;AAC7B;AACA,aAAO;AAAA,IACT;AAGA,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,YAAM,cAAc,MAAM,QAAQ,KAAK,IAAI,MAAM,SAAS;AAC1D,UAAI,gBAAgB,GAAG;AACrB,eAAO,CAAC;AAAA,MACV;AAEA,UAAI,YAAY,cAAc,OAAO,QAAQ;AAC3C,cAAM,IAAI;AAAA,UACR,kDAAkD,YAAY,WAAW,SAAS,OAAO,MAAM;AAAA,QACjG;AAAA,MACF;AAEA,YAAM,YAAY,OAAO,MAAM,WAAW,YAAY,WAAW;AACjE,mBAAa;AACb,aAAO;AAAA,IACT;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAAiC;AACrD,cAAI,wBAAwB,UAAU,GAAG;AACvC,kBAAM,WAAW;AACjB,kBAAMA,UAAkC,CAAC;AAEzC,uBAAW,QAAQ,CAAC,cAAc;AAChC,oBAAM,gBAAgB,UAAU;AAChC,kBAAI,eAAe;AACjB,sBAAM,iBAAiB,SAAS,aAAa;AAC7C,oBAAI,mBAAmB,QAAW;AAChC,kBAAAA,QAAO,aAAa,IAAI,iBAAiB,WAAW,cAAc;AAAA,gBACpE;AAAA,cACF;AAAA,YACF,CAAC;AAED,mBAAOA;AAAA,UACT;AAEA,cAAI,CAAC,MAAM,QAAQ,UAAU,GAAG;AAC9B,kBAAM,IAAI,MAAM,2DAA2D;AAAA,UAC7E;AAEA,iBAAO,WAAW,IAAI,CAAC,WAAW,UAAU,iBAAiB,WAAW,WAAW,KAAK,CAAC,CAAC;AAAA,QAC5F;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,iBAAO,MAAM,IAAI,YAAY;AAAA,QAC/B;AAEA,eAAO,aAAa,KAAK;AAAA,MAC3B;AACA,aAAO;AAAA,IACT;AAGA,WAAO;AAAA,EACT;AAIA,MAAI,WAAW;AACf,QAAM,SAAS,OAAO,IAAI,CAAC,OAAO,UAAU;AAC1C,QAAI,UAAU,oBAAoB;AAChC,aAAO;AAAA,IACT;AACA,UAAM,MAAM,KAAK,QAAQ;AACzB,QAAI,OAAO,MAAM;AACf,YAAM,IAAI,MAAM,YAAY,QAAQ,eAAe;AAAA,IACrD;AACA;AACA,WAAO,iBAAiB,OAAO,GAAG;AAAA,EACpC,CAAC;AAGD,MAAI,cAAc,OAAO,QAAQ;AAC/B,UAAM,IAAI,MAAM,4CAA4C,SAAS,kBAAkB,OAAO,MAAM,EAAE;AAAA,EACxG;AAEA,SAAO;AACT;;;AEtgBA,IAAAC,cAAkE;AAiHlE,IAAM,uCAAuC;AAAA,EAC3C;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAmBA,SAAS,+CACP,cACA,MACmD;AACnD,UAAQ,cAAc;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,qBAAS;AAAA,MAClB;AAAA,IACF;AACE,YAAM,IAAI,MAAM,kCAAkC,YAAY,EAAE;AAAA,EACpE;AACF;AAEA,SAAS,kDACP,cACA,MACA,MACqD;AAErD,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,uBAAuB;AAAA,EACzC;AAGA,MAAI,QAAQ,QAAQ,SAAS,MAAM,SAAS,IAAI,MAAM,KAAK,QAAQ;AACjE,UAAM,IAAI,MAAM,wBAAwB,IAAI,QAAQ,KAAK,MAAM,EAAE;AAAA,EACnE;AAGA,SAAO,KAAK,IAAI,CAAC,SAAS,+CAA+C,cAAc,IAAI,CAAC;AAC9F;AAEA,SAAS,kCAAkC,cAAuE;AAChH,SAAO,qCAAqC,SAAS,YAAmB;AAC1E;AAEO,SAAS,8BACd,KACA,cACA,MACsC;AACtC,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,UAAU,aAAa;AAC7B,MAAI,eAAe,QAAQ,WAAW,MAAM;AAC1C,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AACA,QAAM,gBAAgB,QAAQ;AAE9B,WAAS,iBAAiB,OAAqB,OAAyB;AACtE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,kBAAkB,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGzF,QAAI,oBAAoB,QAAQ,kCAAkC,gBAAgB,GAAG;AAEnF,UAAI,oBAAoB,MAAM;AAC5B,eAAO,+CAA+C,kBAAkB,KAAe;AAAA,MACzF;AAGA,UAAI,oBAAoB,MAAM;AAC5B,eAAO,kDAAkD,kBAAkB,kBAAkB,KAAiB;AAAA,MAChH;AAAA,IACF;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAAiC;AACrD,cAAI,wBAAwB,UAAU,GAAG;AACvC,kBAAM,WAAW;AACjB,kBAAM,SAAkC,CAAC;AAEzC,uBAAW,QAAQ,CAAC,cAAc;AAChC,oBAAM,gBAAgB,UAAU;AAChC,kBAAI,eAAe;AACjB,sBAAM,iBAAiB,SAAS,aAAa;AAC7C,oBAAI,mBAAmB,QAAW;AAChC,yBAAO,aAAa,IAAI,iBAAiB,WAAW,cAAc;AAAA,gBACpE;AAAA,cACF;AAAA,YACF,CAAC;AAED,mBAAO;AAAA,UACT;AAEA,cAAI,CAAC,MAAM,QAAQ,UAAU,GAAG;AAC9B,kBAAM,IAAI,MAAM,2DAA2D;AAAA,UAC7E;AAEA,iBAAO,WAAW,IAAI,CAAC,WAAW,UAAU,iBAAiB,WAAW,WAAW,KAAK,CAAC,CAAC;AAAA,QAC5F;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,iBAAO,MAAM,IAAI,YAAY;AAAA,QAC/B;AAEA,eAAO,aAAa,KAAK;AAAA,MAC3B;AAEA,aAAO;AAAA,IACT;AAGA,WAAO;AAAA,EACT;AAGA,MAAI,gBAAgB,GAAG;AACrB,QAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,YAAM,IAAI,MAAM,uBAAuB;AAAA,IACzC;AACA,QAAI,kBAAkB,KAAK,QAAQ;AACjC,YAAM,IAAI,MAAM,+BAA+B,aAAa,QAAQ,KAAK,MAAM,EAAE;AAAA,IACnF;AACA,WAAO,KAAK,IAAI,CAAC,MAAM,UAAU;AAC/B,aAAO,iBAAiB,QAAQ,KAAK,GAAG,IAAI;AAAA,IAC9C,CAAC;AAAA,EACH;AAGA,SAAO,iBAAiB,QAAQ,CAAC,GAAG,IAAI;AAC1C;","names":["result","import_sdk"]}
|
package/dist/index.js
CHANGED
|
@@ -14,6 +14,9 @@ function extractArrayParameterType(type) {
|
|
|
14
14
|
const size = match[2];
|
|
15
15
|
return [head, size === "" ? "" : size];
|
|
16
16
|
}
|
|
17
|
+
function tupleComponentsAreNamed(components) {
|
|
18
|
+
return components.length > 0 && components.every((component) => component.name != null && component.name !== "");
|
|
19
|
+
}
|
|
17
20
|
function isAbiFunction(item) {
|
|
18
21
|
if (typeof item !== "object" || item === null) return false;
|
|
19
22
|
return "type" in item && item.type === "function" && "name" in item && "outputs" in item;
|
|
@@ -104,9 +107,18 @@ function extractEncryptableValues(abi, functionName, args) {
|
|
|
104
107
|
if ("components" in param && Array.isArray(param.components)) {
|
|
105
108
|
const components = param.components;
|
|
106
109
|
const processTuple = (tupleValue) => {
|
|
107
|
-
|
|
108
|
-
|
|
109
|
-
|
|
110
|
+
if (tupleComponentsAreNamed(components)) {
|
|
111
|
+
const valueObj = tupleValue;
|
|
112
|
+
components.forEach((component) => {
|
|
113
|
+
processParameter(component, valueObj[component.name]);
|
|
114
|
+
});
|
|
115
|
+
return;
|
|
116
|
+
}
|
|
117
|
+
if (!Array.isArray(tupleValue)) {
|
|
118
|
+
throw new Error("Expected an array value for tuple with unnamed components");
|
|
119
|
+
}
|
|
120
|
+
components.forEach((component, index) => {
|
|
121
|
+
processParameter(component, tupleValue[index]);
|
|
110
122
|
});
|
|
111
123
|
};
|
|
112
124
|
if (typeSize != null) {
|
|
@@ -202,18 +214,24 @@ function insertEncryptedValues(abi, functionName, args, encryptedResult) {
|
|
|
202
214
|
if ("components" in param && Array.isArray(param.components)) {
|
|
203
215
|
const components = param.components;
|
|
204
216
|
const processTuple = (tupleValue) => {
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
217
|
+
if (tupleComponentsAreNamed(components)) {
|
|
218
|
+
const valueObj = tupleValue;
|
|
219
|
+
const result2 = {};
|
|
220
|
+
components.forEach((component) => {
|
|
221
|
+
const componentName = component.name;
|
|
222
|
+
if (componentName) {
|
|
223
|
+
const componentValue = valueObj[componentName];
|
|
224
|
+
if (componentValue !== void 0) {
|
|
225
|
+
result2[componentName] = processParameter(component, componentValue);
|
|
226
|
+
}
|
|
213
227
|
}
|
|
214
|
-
}
|
|
215
|
-
|
|
216
|
-
|
|
228
|
+
});
|
|
229
|
+
return result2;
|
|
230
|
+
}
|
|
231
|
+
if (!Array.isArray(tupleValue)) {
|
|
232
|
+
throw new Error("Expected an array value for tuple with unnamed components");
|
|
233
|
+
}
|
|
234
|
+
return components.map((component, index) => processParameter(component, tupleValue[index]));
|
|
217
235
|
};
|
|
218
236
|
if (typeSize != null) {
|
|
219
237
|
if (!Array.isArray(value)) {
|
|
@@ -334,18 +352,24 @@ function transformEncryptedReturnTypes(abi, functionName, data) {
|
|
|
334
352
|
if ("components" in param && Array.isArray(param.components)) {
|
|
335
353
|
const components = param.components;
|
|
336
354
|
const processTuple = (tupleValue) => {
|
|
337
|
-
|
|
338
|
-
|
|
339
|
-
|
|
340
|
-
|
|
341
|
-
|
|
342
|
-
|
|
343
|
-
|
|
344
|
-
|
|
355
|
+
if (tupleComponentsAreNamed(components)) {
|
|
356
|
+
const valueObj = tupleValue;
|
|
357
|
+
const result = {};
|
|
358
|
+
components.forEach((component) => {
|
|
359
|
+
const componentName = component.name;
|
|
360
|
+
if (componentName) {
|
|
361
|
+
const componentValue = valueObj[componentName];
|
|
362
|
+
if (componentValue !== void 0) {
|
|
363
|
+
result[componentName] = processParameter(component, componentValue);
|
|
364
|
+
}
|
|
345
365
|
}
|
|
346
|
-
}
|
|
347
|
-
|
|
348
|
-
|
|
366
|
+
});
|
|
367
|
+
return result;
|
|
368
|
+
}
|
|
369
|
+
if (!Array.isArray(tupleValue)) {
|
|
370
|
+
throw new Error("Expected an array value for tuple with unnamed components");
|
|
371
|
+
}
|
|
372
|
+
return components.map((component, index) => processParameter(component, tupleValue[index]));
|
|
349
373
|
};
|
|
350
374
|
if (typeSize != null) {
|
|
351
375
|
if (!Array.isArray(value)) {
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/encryptedInputs.ts","../src/utils.ts","../src/encryptedReturnTypes.ts"],"sourcesContent":["import {\n Encryptable,\n type AnyExternalHash,\n type EncryptableAddress,\n type EncryptableBool,\n type EncryptableItem,\n type EncryptableUint128,\n type EncryptableUint16,\n type EncryptableUint32,\n type EncryptableUint64,\n type EncryptableUint8,\n type ExternalAddressHash,\n type ExternalBoolHash,\n type ExternalUint128Hash,\n type ExternalUint16Hash,\n type ExternalUint32Hash,\n type ExternalUint64Hash,\n type ExternalUint8Hash,\n type LiteralToPrimitive,\n type Primitive,\n} from '@cofhe/sdk';\nimport type { Abi, AbiFunction, AbiParameter, ExtractAbiFunction } from 'abitype';\nimport type { CofheAbiParametersToPrimitiveTypes } from './fhenixMap';\nimport {\n extractArrayParameterType,\n getAbiFunction,\n type MaybeExtractArrayParameterType,\n type MaybePartialBy,\n type ReadonlyWiden,\n} from './utils';\n\nexport type CofheInputArgs<abi extends Abi | readonly unknown[] = Abi, functionName extends string = string> = GetArgs<\n abi,\n functionName\n>['args'];\n\ntype ExternalHashToInputMap<E extends AnyExternalHash> = E extends ExternalBoolHash\n ? EncryptableBool['data']\n : E extends ExternalUint8Hash\n ? EncryptableUint8['data']\n : E extends ExternalUint16Hash\n ? EncryptableUint16['data']\n : E extends ExternalUint32Hash\n ? EncryptableUint32['data']\n : E extends ExternalUint64Hash\n ? EncryptableUint64['data']\n : E extends ExternalUint128Hash\n ? EncryptableUint128['data']\n : E extends ExternalAddressHash\n ? EncryptableAddress['data']\n : never;\n\n// Note: AnyExternalHash is checked before Primitive - the branded hash types are structurally\n// `0x${string}` (a subtype of `string`, which is itself a Primitive), so Primitive must not be\n// checked first or every hash would be (incorrectly) widened via LiteralToPrimitive instead.\ntype EncryptedInputsToInputs<T> = T extends AnyExternalHash\n ? ExternalHashToInputMap<T>\n : T extends Primitive\n ? LiteralToPrimitive<T>\n : {\n [K in keyof T]: EncryptedInputsToInputs<T[K]>;\n };\n\n/**\n * Drops the batch-signature argument from an args tuple.\n *\n * The signature slot is the first non-`external*` parameter that follows the contiguous run of\n * `external*` parameters - `FHE.asEuintXX(hash, proof)` is a pair, so the proof sits with the\n * hashes it authenticates rather than at the end of the parameter list. Walks the ABI parameters\n * and the args tuple in lockstep and elides the arg at that position.\n */\ntype DropSignatureSlotTuple<\n params extends readonly unknown[],\n args extends readonly unknown[],\n afterExternalRun extends boolean = false,\n> = params extends readonly [infer P, ...infer PRest extends readonly unknown[]]\n ? args extends readonly [infer A, ...infer ARest extends readonly unknown[]]\n ? P extends AbiParameter\n ? ParamHasExternalInput<P> extends true\n ? readonly [A, ...DropSignatureSlotTuple<PRest, ARest, true>]\n : afterExternalRun extends true\n ? DropSignatureSlotTuple<PRest, ARest, false>\n : readonly [A, ...DropSignatureSlotTuple<PRest, ARest, false>]\n : readonly [A, ...DropSignatureSlotTuple<PRest, ARest, afterExternalRun>]\n : readonly []\n : args;\n\n/** Non-tuple args (e.g. `undefined`) pass through unchanged, as the previous `DropLast` did. */\ntype DropSignatureSlot<params extends readonly unknown[], args> = args extends readonly unknown[]\n ? DropSignatureSlotTuple<params, args>\n : args;\n\nconst EXTERNAL_INPUT_INTERNAL_TYPES = [\n 'externalEbool',\n 'externalEuint8',\n 'externalEuint16',\n 'externalEuint32',\n 'externalEuint64',\n 'externalEuint128',\n 'externalEaddress',\n] as const;\ntype ExternalInputInternalType = (typeof EXTERNAL_INPUT_INTERNAL_TYPES)[number];\n\n/** Type-level: does this ABI parameter carry (or, if a tuple, recursively contain) an `external*` input? */\ntype ParamHasExternalInput<param extends AbiParameter> =\n MaybeExtractArrayParameterType<param['internalType']> extends [infer head extends string, any]\n ? head extends ExternalInputInternalType\n ? true\n : // A struct array (`tuple[]` / `tuple[N]`) lands here too, with `head` the struct's own\n // internalType. Falling through to the components keeps `Order[]` visible as carrying\n // encrypted inputs, the way the runtime `paramHasExternalInput` already sees it.\n ParamComponentsHaveExternalInput<param>\n : param['internalType'] extends ExternalInputInternalType\n ? true\n : ParamComponentsHaveExternalInput<param>;\n\n/** Recurses into a parameter's `components`, for tuples at any array depth. */\ntype ParamComponentsHaveExternalInput<param extends AbiParameter> = param extends {\n components: infer components extends readonly AbiParameter[];\n}\n ? ParamsHaveExternalInput<components>\n : false;\n\ntype ParamsHaveExternalInput<params extends readonly AbiParameter[]> = params extends readonly [\n infer Head extends AbiParameter,\n ...infer Rest extends readonly AbiParameter[],\n]\n ? ParamHasExternalInput<Head> extends true\n ? true\n : ParamsHaveExternalInput<Rest>\n : false;\n\n/**\n * Pre-transform args for a function ABI: the raw plain values a caller supplies before encryption.\n *\n * If the function has one or more `external*` (encrypted) inputs, the plain `bytes` parameter\n * immediately following them is the shared batch signature slot - the caller never supplies it\n * directly (the SDK injects it after encryption), so it's dropped from this type entirely.\n */\nexport type CofheInputArgsPreTransform<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n abiFunction extends AbiFunction = abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction,\n> = EncryptedInputsToInputs<\n ParamsHaveExternalInput<abiFunction['inputs']> extends true\n ? DropSignatureSlot<abiFunction['inputs'], CofheInputArgs<abi, functionName>>\n : CofheInputArgs<abi, functionName>\n>;\n\n/// GetArgs from abitype (not exported from abitype)\ntype GetArgs<\n abi extends Abi | readonly unknown[] = Abi, // `readonly unknown[]` allows for non-const asserted types\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly [],\n ///\n abiFunction extends AbiFunction = abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction,\n primitiveTypes = CofheAbiParametersToPrimitiveTypes<abiFunction['inputs']>,\n args_ =\n | primitiveTypes // show all values\n | (abi extends Abi\n ? args extends primitiveTypes // infer value (if valid)\n ? primitiveTypes extends args // make sure `args` exactly matches `primitiveTypes` (e.g. avoid `args: readonly [{ foo: string; bar: number; }] | readonly [{ foo: string; }]`)\n ? // make inferred value of `args` match `primitiveTypes` (e.g. avoid union `args: readonly [123n] | readonly [bigint]`)\n ReadonlyWiden<args>\n : never\n : never\n : never)\n | (Abi extends abi ? readonly unknown[] : never), // fallback if `abi` is declared as `Abi`\n> = MaybePartialBy<{ args: args_ }, readonly [] extends primitiveTypes ? 'args' : Abi extends abi ? 'args' : string>;\n\nfunction internalTypeIsExternalInput(internalType: string): internalType is ExternalInputInternalType {\n return EXTERNAL_INPUT_INTERNAL_TYPES.includes(internalType as any);\n}\n\nfunction paramHasExternalInput(param: AbiParameter): boolean {\n const [internalTypeHead] = extractArrayParameterType(param.internalType);\n if (internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) return true;\n if ('components' in param && Array.isArray(param.components)) {\n return param.components.some((component) => paramHasExternalInput(component));\n }\n return false;\n}\n\nfunction transformSingleExternalToEncryptable(internalType: ExternalInputInternalType, data: unknown): EncryptableItem {\n switch (internalType) {\n case 'externalEbool':\n return Encryptable.bool(data as boolean);\n case 'externalEuint8':\n return Encryptable.uint8(data as string | bigint);\n case 'externalEuint16':\n return Encryptable.uint16(data as string | bigint);\n case 'externalEuint32':\n return Encryptable.uint32(data as string | bigint);\n case 'externalEuint64':\n return Encryptable.uint64(data as string | bigint);\n case 'externalEuint128':\n return Encryptable.uint128(data as string | bigint);\n case 'externalEaddress':\n return Encryptable.address(data as string | bigint);\n default:\n throw new Error(`Unknown external input type: ${internalType}`);\n }\n}\n\nfunction transformArrayOfExternalsToEncryptables(\n internalType: ExternalInputInternalType,\n size: string | undefined,\n data: unknown\n): EncryptableItem[] {\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n\n if (size != null && size !== '' && parseInt(size) !== data.length) {\n throw new Error(`Array size mismatch: ${size} !== ${data.length}`);\n }\n\n return data.map((item) => transformSingleExternalToEncryptable(internalType, item));\n}\n\n/**\n * Extracts encryptable values from function arguments based on the ABI.\n * Transforms raw data values into EncryptableItem objects that can be passed to the encrypt function.\n *\n * @param abi - The ABI containing the function definition\n * @param functionName - Name of the function\n * @param args - Function arguments in the format of CofheInputArgsPreTransform (raw data values;\n * if the function has `external*` inputs, the trailing batch-signature slot is NOT included)\n * @returns Array of EncryptableItem objects ready for encryption, in ABI parameter order\n */\nexport function extractEncryptableValues<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n args: CofheInputArgsPreTransform<TAbi, TFunctionName>\n): EncryptableItem[] {\n const abiFunction = getAbiFunction(abi, functionName);\n const inputs = abiFunction?.inputs;\n if (abiFunction == null || inputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n\n if (!Array.isArray(args)) {\n throw new Error('Arguments must be an array');\n }\n\n const hasExternalInputs = inputs.some((input) => paramHasExternalInput(input));\n const signatureSlotIndex = findSignatureSlotIndex(inputs, hasExternalInputs, functionName);\n\n // Collect encrypted values as EncryptableItem objects in order (flat array)\n const encryptableItems: EncryptableItem[] = [];\n\n function processParameter(param: AbiParameter, value: unknown): void {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead, internalTypeSize] = extractArrayParameterType(param.internalType);\n\n // Bare external input (e.g. `externalEuint32`, type: 'bytes32')\n if (typeSize == null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n encryptableItems.push(transformSingleExternalToEncryptable(internalTypeHead, value));\n return;\n }\n\n // Array of external inputs (e.g. `externalEuint32[]`/`externalEuint32[2]`, type: 'bytes32[]'/'bytes32[2]')\n if (typeSize != null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n const encryptables = transformArrayOfExternalsToEncryptables(internalTypeHead, typeSize, value);\n encryptableItems.push(...encryptables);\n return;\n }\n\n // Tuple recursive case (struct that may itself contain external inputs)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): void => {\n const valueObj = tupleValue as Record<string, unknown>;\n components.forEach((component) => {\n processParameter(component, valueObj[component.name]);\n });\n };\n\n // `tuple[]` / `tuple[N]`: walk each element as its own tuple. Treating the array\n // itself as one tuple would look up named components on the array object, find\n // none, and hand `undefined` to the encryptable transform.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n value.forEach(processTuple);\n return;\n }\n\n processTuple(value);\n }\n return;\n }\n\n // Not an external input, and cannot contain one\n return;\n }\n\n // Process all inputs, skipping the batch-signature slot (nothing to extract there). `args` is the\n // pre-transform tuple, which omits that slot, so its indices lag by one past the signature.\n let argIndex = 0;\n inputs.forEach((input, index) => {\n if (index === signatureSlotIndex) return;\n const arg = args[argIndex];\n if (arg == null) {\n throw new Error(`Argument ${argIndex} is undefined`);\n }\n argIndex++;\n processParameter(input, arg);\n });\n\n return encryptableItems;\n}\n\n/**\n * Index of the shared batch-signature parameter: the plain `bytes` immediately after the\n * contiguous run of `external*` parameters. Returns -1 when the function has no encrypted inputs.\n *\n * The proof pairs with the hashes it authenticates (`FHE.asEuintXX(hash, proof)`), so it does not\n * have to be the last parameter - anything may follow it.\n */\nfunction findSignatureSlotIndex(\n inputs: readonly AbiParameter[],\n hasExternalInputs: boolean,\n functionName: string\n): number {\n if (!hasExternalInputs) return -1;\n\n const externalIndices = inputs.reduce<number[]>((acc, input, index) => {\n if (paramHasExternalInput(input)) acc.push(index);\n return acc;\n }, []);\n\n const first = externalIndices[0]!;\n const last = externalIndices[externalIndices.length - 1]!;\n if (last - first + 1 !== externalIndices.length) {\n throw new Error(\n `Function ${functionName} has encrypted (external*) inputs at non-adjacent positions ` +\n `(${externalIndices.join(', ')}). They share one batch signature, so they must be contiguous ` +\n `with the 'bytes' signature parameter immediately after them`\n );\n }\n\n const slotIndex = last + 1;\n const slot = inputs[slotIndex];\n if (slot?.type !== 'bytes') {\n throw new Error(\n `Function ${functionName} has encrypted (external*) inputs, so the parameter immediately after ` +\n `them must be a plain 'bytes' parameter to receive the shared batch signature - found ` +\n `'${slot?.type ?? 'nothing'}' at position ${slotIndex} instead`\n );\n }\n\n return slotIndex;\n}\n\n/**\n * Re-inserts encrypted values back into function arguments based on the ABI.\n * Takes the batch-verified result (per-input hashes, in order, followed by the single shared\n * signature) and the original args structure, and produces the final call args: each `external*`\n * position gets its corresponding hash, and the trailing `bytes` parameter gets the signature.\n *\n * This function mirrors the extraction logic in extractEncryptableValues, ensuring values\n * are inserted in the exact same order and locations where they were extracted.\n *\n * @param abi - The ABI containing the function definition\n * @param functionName - Name of the function\n * @param args - Original function arguments in the format of CofheInputArgsPreTransform\n * @param encryptedResult - `[...hashes, signature]`, as returned by `EncryptInputsBuilder.execute()`\n * @returns Function arguments with encrypted values inserted (format of CofheInputArgs)\n */\nexport function insertEncryptedValues<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n args: CofheInputArgsPreTransform<TAbi, TFunctionName>,\n encryptedResult: readonly `0x${string}`[]\n): CofheInputArgs<TAbi, TFunctionName> {\n const abiFunction = getAbiFunction(abi, functionName);\n const inputs = abiFunction?.inputs;\n if (abiFunction == null || inputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n\n if (!Array.isArray(args)) {\n throw new Error('Arguments must be an array');\n }\n\n const hasExternalInputs = inputs.some((input) => paramHasExternalInput(input));\n const signatureSlotIndex = findSignatureSlotIndex(inputs, hasExternalInputs, functionName);\n\n const hashes = hasExternalInputs ? encryptedResult.slice(0, -1) : [];\n const signature = hasExternalInputs ? encryptedResult[encryptedResult.length - 1] : undefined;\n\n // Track position in the hashes array\n let hashIndex = 0;\n\n function processParameter(param: AbiParameter, value: unknown): unknown {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead] = extractArrayParameterType(param.internalType);\n\n // Bare external input - substitute its hash\n if (typeSize == null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n if (hashIndex >= hashes.length) {\n throw new Error(`Not enough encrypted hashes: expected at least ${hashIndex + 1}, got ${hashes.length}`);\n }\n const hash = hashes[hashIndex];\n hashIndex++;\n return hash;\n }\n\n // Array of external inputs - substitute the corresponding slice of hashes\n if (typeSize != null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n const arrayLength = Array.isArray(value) ? value.length : 0;\n if (arrayLength === 0) {\n return [];\n }\n\n if (hashIndex + arrayLength > hashes.length) {\n throw new Error(\n `Not enough encrypted hashes: expected at least ${hashIndex + arrayLength}, got ${hashes.length}`\n );\n }\n\n const hashArray = hashes.slice(hashIndex, hashIndex + arrayLength);\n hashIndex += arrayLength;\n return hashArray;\n }\n\n // Tuple recursive case (struct that may itself contain external inputs)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): unknown => {\n const valueObj = tupleValue as Record<string, unknown>;\n const result: Record<string, unknown> = {};\n\n components.forEach((component) => {\n const componentName = component.name;\n if (componentName) {\n const componentValue = valueObj[componentName];\n if (componentValue !== undefined) {\n result[componentName] = processParameter(component, componentValue);\n }\n }\n });\n\n return result;\n };\n\n // `tuple[]` / `tuple[N]`: rebuild each element as its own tuple, in the same order\n // extractEncryptableValues walked them, so the hashes line up. Treating the array as\n // one tuple would consume no hashes and collapse the argument to `{}`.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n return value.map(processTuple);\n }\n\n return processTuple(value);\n }\n return value;\n }\n\n // Not an external input, return original value\n return value;\n }\n\n // Rebuild the full arg list; the signature slot is injected directly. `args` omits that slot, so\n // its indices lag by one past the signature position.\n let argIndex = 0;\n const result = inputs.map((input, index) => {\n if (index === signatureSlotIndex) {\n return signature;\n }\n const arg = args[argIndex];\n if (arg == null) {\n throw new Error(`Argument ${argIndex} is undefined`);\n }\n argIndex++;\n return processParameter(input, arg);\n });\n\n // Verify we used all encrypted hashes\n if (hashIndex !== hashes.length) {\n throw new Error(`Mismatch in encrypted hashes count: used ${hashIndex}, but provided ${hashes.length}`);\n }\n\n return result as CofheInputArgs<TAbi, TFunctionName>;\n}\n","import type {\n Abi,\n AbiConstructor,\n AbiError,\n AbiEvent,\n AbiFallback,\n AbiFunction,\n AbiParameter,\n AbiParametersToPrimitiveTypes,\n AbiReceive,\n AbiType,\n Address,\n ExtractAbiFunction,\n ResolvedRegister,\n SolidityArray,\n SolidityFixedArrayRange,\n SolidityTuple,\n} from 'abitype';\n\nexport type ContractReturnType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n ///\n abiFunction extends AbiFunction = (\n abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction\n ) extends infer abiFunction_ extends AbiFunction\n ? IsUnion<abiFunction_> extends true // narrow overloads by `args` by converting to tuple and filtering out overloads that don't match\n ? UnionToTuple<abiFunction_> extends infer abiFunctions extends readonly AbiFunction[]\n ? {\n [K in keyof abiFunctions]: (\n readonly unknown[] | undefined extends args // for functions that don't have inputs, `args` can be `undefined` so fallback to `readonly []`\n ? readonly []\n : args\n ) extends AbiParametersToPrimitiveTypes<abiFunctions[K]['inputs'], 'inputs'>\n ? abiFunctions[K]\n : never;\n }[number] // convert back to union (removes `never` tuple entries: `['foo', never, 'bar'][number]` => `'foo' | 'bar'`)\n : never\n : abiFunction_\n : never,\n outputs extends readonly AbiParameter[] = abiFunction['outputs'],\n primitiveTypes extends readonly unknown[] = AbiParametersToPrimitiveTypes<outputs, 'outputs', true>,\n> = [abiFunction] extends [never]\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` declared as `Abi`)\n : readonly unknown[] extends primitiveTypes\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` not const-asserted)\n : primitiveTypes extends readonly [] // unwrap `primitiveTypes`\n ? // biome-ignore lint/suspicious/noConfusingVoidType: <explanation>\n void // no outputs\n : primitiveTypes extends readonly [infer primitiveType]\n ? primitiveType // single output\n : primitiveTypes;\n\nexport type Error<messages extends string | string[]> = messages extends string\n ? [\n // Surrounding with array to prevent `messages` from being widened to `string`\n `Error: ${messages}`,\n ]\n : {\n [key in keyof messages]: messages[key] extends infer message extends string ? `Error: ${message}` : never;\n };\n\ntype PartialBy<TType, TKeys extends keyof TType> = ExactPartial<Pick<TType, TKeys>> & Omit<TType, TKeys>;\ntype ExactPartial<T> = { [K in keyof T]?: T[K] | undefined };\n\nexport type MaybePartialBy<TType, TKeys extends string> = TKeys extends keyof TType ? PartialBy<TType, TKeys> : TType;\n\nexport type ReadonlyWiden<TType> =\n | (TType extends Function ? TType : never)\n | (TType extends ResolvedRegister['bigIntType'] ? bigint : never)\n | (TType extends boolean ? boolean : never)\n | (TType extends ResolvedRegister['intType'] ? number : never)\n | (TType extends string\n ? TType extends Address\n ? Address\n : TType extends ResolvedRegister['bytesType']['inputs']\n ? ResolvedRegister['bytesType']\n : string\n : never)\n | (TType extends readonly [] ? readonly [] : never)\n | (TType extends Record<string, unknown> ? { [K in keyof TType]: ReadonlyWiden<TType[K]> } : never)\n | (TType extends { length: number }\n ? {\n [K in keyof TType]: ReadonlyWiden<TType[K]>;\n } extends infer Val extends unknown[]\n ? readonly [...Val]\n : never\n : never);\n\nexport type AbiBasicType = Exclude<AbiType, SolidityTuple | SolidityArray>;\n\n/**\n * First, infer `Head` against a known size type (either fixed-length array value or `\"\"`).\n *\n * | Input | Head |\n * | --------------- | ------------ |\n * | `string[]` | `string` |\n * | `string[][][3]` | `string[][]` |\n */\nexport type MaybeExtractArrayParameterType<type> = type extends `${infer head}[${'' | `${SolidityFixedArrayRange}`}]`\n ? // * Then, infer in the opposite direction, using the known `head` to infer the exact `size` value.\n // *\n // * | Input | Size |\n // * | ------------ | ---- |\n // * | `${head}[]` | `\"\"` |\n // * | `${head}[3]` | `3` |\n // */\n type extends `${head}[${infer size}]`\n ? [head, size]\n : undefined\n : undefined;\n\nexport type IsUnion<T, C = T> = T extends C ? ([C] extends [T] ? false : true) : never;\nexport type UnionToTuple<U, Last = LastInUnion<U>> = [U] extends [never]\n ? []\n : [...UnionToTuple<Exclude<U, Last>>, Last];\ntype LastInUnion<U> =\n UnionToIntersection<U extends unknown ? (x: U) => 0 : never> extends (x: infer L) => 0 ? L : never;\ntype UnionToIntersection<U> = (U extends unknown ? (arg: U) => 0 : never) extends (arg: infer I) => 0 ? I : never;\n\n/**\n * Create tuple of {@link type} type with {@link size} size\n *\n * @param Type - Type of tuple\n * @param Size - Size of tuple\n * @returns Tuple of {@link type} type with {@link size} size\n *\n * @example\n * type Result = Tuple<string, 2>\n * // ^? type Result = [string, string]\n */\n// https://github.com/Microsoft/TypeScript/issues/26223#issuecomment-674500430\nexport type Tuple<type, size extends number> = size extends size\n ? number extends size\n ? type[]\n : _TupleOf<type, size, []>\n : never;\ntype _TupleOf<length, size extends number, acc extends readonly unknown[]> = acc['length'] extends size\n ? acc\n : _TupleOf<length, size, readonly [length, ...acc]>;\n\n/**\n * Merges two object types into new type\n *\n * @param object1 - Object to merge into\n * @param object2 - Object to merge and override keys from {@link object1}\n * @returns New object type with keys from {@link object1} and {@link object2}. If a key exists in both {@link object1} and {@link object2}, the key from {@link object2} will be used.\n *\n * @example\n * type Result = Merge<{ foo: string }, { foo: number; bar: string }>\n * // ^? type Result = { foo: number; bar: string }\n */\nexport type Merge<object1, object2> = Omit<object1, keyof object2> & object2;\n\n// FUNCTIONS\n\n/**\n * Extracts array parameter type information from a type string.\n * Always returns a tuple, with undefined for non-array types.\n *\n * @param type - Type string like \"tuple\", \"tuple[]\", or \"tuple[2]\"\n * @returns [head, size] where:\n * - head is the base type\n * - size is undefined for non-arrays, \"\" for dynamic arrays, or the number as a string for fixed arrays\n *\n * @example\n * extractArrayParameterType(\"tuple\") // [\"tuple\", undefined]\n * extractArrayParameterType(\"tuple[]\") // [\"tuple\", \"\"]\n * extractArrayParameterType(\"tuple[2]\") // [\"tuple\", \"2\"]\n */\nexport function extractArrayParameterType<T extends string | undefined>(type: T): [T, string | undefined] {\n if (type == null) return [type, undefined];\n\n const match = type.match(/^(.+)\\[(\\d*)\\]$/);\n\n if (!match) {\n // Not an array type, return [type, undefined]\n return [type, undefined];\n }\n\n const head = match[1];\n const size = match[2]; // Empty string for dynamic arrays, or digits for fixed arrays\n\n // Return empty string for dynamic arrays, or the size string for fixed arrays\n return [head as T, size === '' ? '' : size];\n}\n\ntype AbiItem = AbiConstructor | AbiError | AbiEvent | AbiFallback | AbiFunction | AbiReceive;\n\nexport function isAbiFunction(item: AbiItem | unknown): item is AbiFunction {\n if (typeof item !== 'object' || item === null) return false;\n return 'type' in item && item.type === 'function' && 'name' in item && 'outputs' in item;\n}\n\nexport function getAbiFunction<TAbi extends Abi | readonly unknown[], TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName\n): AbiFunction | undefined {\n return abi.find((item) => {\n const isFunction = isAbiFunction(item);\n if (!isFunction) return false;\n return item.name === functionName;\n }) as AbiFunction | undefined;\n}\n","import { type Primitive, type LiteralToPrimitive, FheTypes } from '@cofhe/sdk';\nimport type { Abi, AbiFunction, ExtractAbiFunction, AbiParameter } from 'abitype';\nimport type {\n EncryptedReturnType,\n CofheAbiParametersToPrimitiveTypes,\n EBool,\n EAddress,\n EUint128,\n EUint16,\n EUint32,\n EUint64,\n EUint8,\n} from './fhenixMap';\nimport {\n getAbiFunction,\n type IsUnion,\n type UnionToTuple,\n type ContractReturnType,\n extractArrayParameterType,\n} from './utils';\n\ntype FirstCofheReturnValue<T> = [T] extends [void]\n ? never\n : T extends readonly [infer Head, ...(readonly unknown[])]\n ? Head\n : T;\n\ntype FheTypeFromCofheReturnValue<T> = T extends { utype: infer U }\n ? U\n : T extends readonly (infer Element)[]\n ? FheTypeFromCofheReturnValue<Element>\n : never;\n\n/**\n * Extracts the `FheTypes` of the first returned value of a CoFHE call.\n *\n * Examples:\n * - `returns (ebool)` -> `FheTypes.Bool`\n * - `returns (euint8, euint64)` -> `FheTypes.Uint8`\n * - `returns (euint16[])` -> `FheTypes.Uint16`\n */\nexport type CofheFirstReturnFheType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n> = FheTypeFromCofheReturnValue<FirstCofheReturnValue<CofheReturnType<abi, functionName, args>>>;\n\nexport type CofheReturnType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n ///\n abiFunction extends AbiFunction = (\n abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction\n ) extends infer abiFunction_ extends AbiFunction\n ? IsUnion<abiFunction_> extends true // narrow overloads by `args` by converting to tuple and filtering out overloads that don't match\n ? UnionToTuple<abiFunction_> extends infer abiFunctions extends readonly AbiFunction[]\n ? {\n [K in keyof abiFunctions]: (\n readonly unknown[] | undefined extends args // for functions that don't have inputs, `args` can be `undefined` so fallback to `readonly []`\n ? readonly []\n : args\n ) extends CofheAbiParametersToPrimitiveTypes<abiFunctions[K]['inputs'], 'inputs'>\n ? abiFunctions[K]\n : never;\n }[number] // convert back to union (removes `never` tuple entries: `['foo', never, 'bar'][number]` => `'foo' | 'bar'`)\n : never\n : abiFunction_\n : never,\n outputs extends readonly AbiParameter[] = abiFunction['outputs'],\n primitiveTypes extends readonly unknown[] = CofheAbiParametersToPrimitiveTypes<outputs, 'outputs'>,\n> = [abiFunction] extends [never]\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` declared as `Abi`)\n : readonly unknown[] extends primitiveTypes\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` not const-asserted)\n : primitiveTypes extends readonly [] // unwrap `primitiveTypes`\n ? // biome-ignore lint/suspicious/noConfusingVoidType: <explanation>\n void // no outputs\n : primitiveTypes extends readonly [infer primitiveType]\n ? primitiveType // single output\n : primitiveTypes;\n\ntype EncryptedReturnTypeToReturnTypeMap<E extends EncryptedReturnType> = E extends EBool\n ? boolean\n : E extends EUint8\n ? bigint\n : E extends EUint16\n ? bigint\n : E extends EUint32\n ? bigint\n : E extends EUint64\n ? bigint\n : E extends EUint128\n ? bigint\n : E extends EAddress\n ? string\n : never;\n\ntype EncryptedReturnTypesToReturnTypes<T> = T extends Primitive\n ? LiteralToPrimitive<T>\n : T extends EncryptedReturnType\n ? EncryptedReturnTypeToReturnTypeMap<T>\n : {\n [K in keyof T]: EncryptedReturnTypesToReturnTypes<T[K]>;\n };\n\nexport type CofheReturnTypePostTransform<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n> = EncryptedReturnTypesToReturnTypes<CofheReturnType<abi, functionName, args>>;\n\nconst ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES = [\n 'ebool',\n 'euint8',\n 'euint16',\n 'euint32',\n 'euint64',\n 'euint128',\n 'eaddress',\n] as const;\ntype EncryptedReturnTypeInternalType = (typeof ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES)[number];\n\ntype EncryptedReturnTypeInternalTypeToReturnTypeMap<T extends EncryptedReturnTypeInternalType> = T extends 'ebool'\n ? EBool\n : T extends 'euint8'\n ? EUint8\n : T extends 'euint16'\n ? EUint16\n : T extends 'euint32'\n ? EUint32\n : T extends 'euint64'\n ? EUint64\n : T extends 'euint128'\n ? EUint128\n : T extends 'eaddress'\n ? EAddress\n : never;\n\nfunction transformSingleEncryptedReturnTypeToReturnType<T extends EncryptedReturnTypeInternalType>(\n internalType: T,\n data: string\n): EncryptedReturnTypeInternalTypeToReturnTypeMap<T> {\n switch (internalType) {\n case 'ebool':\n return {\n ctHash: data,\n utype: FheTypes.Bool,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint8':\n return {\n ctHash: data,\n utype: FheTypes.Uint8,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint16':\n return {\n ctHash: data,\n utype: FheTypes.Uint16,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint32':\n return {\n ctHash: data,\n utype: FheTypes.Uint32,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint64':\n return {\n ctHash: data,\n utype: FheTypes.Uint64,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint128':\n return {\n ctHash: data,\n utype: FheTypes.Uint128,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'eaddress':\n return {\n ctHash: data,\n utype: FheTypes.Uint160,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n default:\n throw new Error(`Unknown encrypted return type: ${internalType}`);\n }\n}\n\nfunction transformArrayOfEncryptedReturnTypesToReturnTypes<T extends EncryptedReturnTypeInternalType>(\n internalType: T,\n size: string | undefined,\n data: string[]\n): EncryptedReturnTypeInternalTypeToReturnTypeMap<T>[] {\n // Ensure data is an array\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n\n // Ensure length of fixed length tuple matches the length of the data array\n if (size != null && size !== '' && parseInt(size) !== data.length) {\n throw new Error(`Array size mismatch: ${size} !== ${data.length}`);\n }\n\n // Transform each item in the data array into an return type\n return data.map((item) => transformSingleEncryptedReturnTypeToReturnType(internalType, item));\n}\n\nfunction internalTypeIsEncryptedReturnType(internalType: string): internalType is EncryptedReturnTypeInternalType {\n return ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES.includes(internalType as any);\n}\n\nexport function transformEncryptedReturnTypes<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n data: ContractReturnType<TAbi, TFunctionName>\n): CofheReturnType<TAbi, TFunctionName> {\n const abiFunction = getAbiFunction(abi, functionName);\n const outputs = abiFunction?.outputs;\n if (abiFunction == null || outputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n const outputsLength = outputs.length;\n\n function processParameter(param: AbiParameter, value: unknown): unknown {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead, internalTypeSize] = extractArrayParameterType(param.internalType);\n\n // Is encrypted return type\n if (internalTypeHead != null && internalTypeIsEncryptedReturnType(internalTypeHead)) {\n // Is single encrypted return type\n if (internalTypeSize == null) {\n return transformSingleEncryptedReturnTypeToReturnType(internalTypeHead, value as string);\n }\n\n // Is array of encrypted return types\n if (internalTypeSize != null) {\n return transformArrayOfEncryptedReturnTypesToReturnTypes(internalTypeHead, internalTypeSize, value as string[]);\n }\n }\n\n // Tuple but not an encrypted return type (recursive case)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): unknown => {\n const valueObj = tupleValue as Record<string, unknown>;\n const result: Record<string, unknown> = {};\n\n components.forEach((component) => {\n const componentName = component.name;\n if (componentName) {\n const componentValue = valueObj[componentName];\n if (componentValue !== undefined) {\n result[componentName] = processParameter(component, componentValue);\n }\n }\n });\n\n return result;\n };\n\n // `tuple[]` / `tuple[N]`: transform each element as its own tuple. Treating the\n // array itself as one tuple would look up named components on the array object,\n // find none, and collapse the whole result to `{}`.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n return value.map(processTuple);\n }\n\n return processTuple(value);\n }\n\n return value;\n }\n\n // Not an encrypted return type, return original value\n return value;\n }\n\n // Process results (array case)\n if (outputsLength > 1) {\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n if (outputsLength !== data.length) {\n throw new Error(`Mismatch in outputs length: ${outputsLength} !== ${data.length}`);\n }\n return data.map((item, index) => {\n return processParameter(outputs[index], item);\n }) as CofheReturnType<TAbi, TFunctionName>;\n }\n\n // Process retuls (single item case)\n return processParameter(outputs[0], data) as CofheReturnType<TAbi, TFunctionName>;\n}\n"],"mappings":";AAAA;AAAA,EACE;AAAA,OAmBK;;;ACuJA,SAAS,0BAAwD,MAAkC;AACxG,MAAI,QAAQ,KAAM,QAAO,CAAC,MAAM,MAAS;AAEzC,QAAM,QAAQ,KAAK,MAAM,iBAAiB;AAE1C,MAAI,CAAC,OAAO;AAEV,WAAO,CAAC,MAAM,MAAS;AAAA,EACzB;AAEA,QAAM,OAAO,MAAM,CAAC;AACpB,QAAM,OAAO,MAAM,CAAC;AAGpB,SAAO,CAAC,MAAW,SAAS,KAAK,KAAK,IAAI;AAC5C;AAIO,SAAS,cAAc,MAA8C;AAC1E,MAAI,OAAO,SAAS,YAAY,SAAS,KAAM,QAAO;AACtD,SAAO,UAAU,QAAQ,KAAK,SAAS,cAAc,UAAU,QAAQ,aAAa;AACtF;AAEO,SAAS,eACd,KACA,cACyB;AACzB,SAAO,IAAI,KAAK,CAAC,SAAS;AACxB,UAAM,aAAa,cAAc,IAAI;AACrC,QAAI,CAAC,WAAY,QAAO;AACxB,WAAO,KAAK,SAAS;AAAA,EACvB,CAAC;AACH;;;ADhHA,IAAM,gCAAgC;AAAA,EACpC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAsEA,SAAS,4BAA4B,cAAiE;AACpG,SAAO,8BAA8B,SAAS,YAAmB;AACnE;AAEA,SAAS,sBAAsB,OAA8B;AAC3D,QAAM,CAAC,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AACvE,MAAI,oBAAoB,QAAQ,4BAA4B,gBAAgB,EAAG,QAAO;AACtF,MAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,WAAO,MAAM,WAAW,KAAK,CAAC,cAAc,sBAAsB,SAAS,CAAC;AAAA,EAC9E;AACA,SAAO;AACT;AAEA,SAAS,qCAAqC,cAAyC,MAAgC;AACrH,UAAQ,cAAc;AAAA,IACpB,KAAK;AACH,aAAO,YAAY,KAAK,IAAe;AAAA,IACzC,KAAK;AACH,aAAO,YAAY,MAAM,IAAuB;AAAA,IAClD,KAAK;AACH,aAAO,YAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,YAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,YAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,YAAY,QAAQ,IAAuB;AAAA,IACpD,KAAK;AACH,aAAO,YAAY,QAAQ,IAAuB;AAAA,IACpD;AACE,YAAM,IAAI,MAAM,gCAAgC,YAAY,EAAE;AAAA,EAClE;AACF;AAEA,SAAS,wCACP,cACA,MACA,MACmB;AACnB,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,uBAAuB;AAAA,EACzC;AAEA,MAAI,QAAQ,QAAQ,SAAS,MAAM,SAAS,IAAI,MAAM,KAAK,QAAQ;AACjE,UAAM,IAAI,MAAM,wBAAwB,IAAI,QAAQ,KAAK,MAAM,EAAE;AAAA,EACnE;AAEA,SAAO,KAAK,IAAI,CAAC,SAAS,qCAAqC,cAAc,IAAI,CAAC;AACpF;AAYO,SAAS,yBACd,KACA,cACA,MACmB;AACnB,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,SAAS,aAAa;AAC5B,MAAI,eAAe,QAAQ,UAAU,MAAM;AACzC,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AAEA,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,4BAA4B;AAAA,EAC9C;AAEA,QAAM,oBAAoB,OAAO,KAAK,CAAC,UAAU,sBAAsB,KAAK,CAAC;AAC7E,QAAM,qBAAqB,uBAAuB,QAAQ,mBAAmB,YAAY;AAGzF,QAAM,mBAAsC,CAAC;AAE7C,WAAS,iBAAiB,OAAqB,OAAsB;AACnE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,kBAAkB,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGzF,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,uBAAiB,KAAK,qCAAqC,kBAAkB,KAAK,CAAC;AACnF;AAAA,IACF;AAGA,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,YAAM,eAAe,wCAAwC,kBAAkB,UAAU,KAAK;AAC9F,uBAAiB,KAAK,GAAG,YAAY;AACrC;AAAA,IACF;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAA8B;AAClD,gBAAM,WAAW;AACjB,qBAAW,QAAQ,CAAC,cAAc;AAChC,6BAAiB,WAAW,SAAS,UAAU,IAAI,CAAC;AAAA,UACtD,CAAC;AAAA,QACH;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,gBAAM,QAAQ,YAAY;AAC1B;AAAA,QACF;AAEA,qBAAa,KAAK;AAAA,MACpB;AACA;AAAA,IACF;AAGA;AAAA,EACF;AAIA,MAAI,WAAW;AACf,SAAO,QAAQ,CAAC,OAAO,UAAU;AAC/B,QAAI,UAAU,mBAAoB;AAClC,UAAM,MAAM,KAAK,QAAQ;AACzB,QAAI,OAAO,MAAM;AACf,YAAM,IAAI,MAAM,YAAY,QAAQ,eAAe;AAAA,IACrD;AACA;AACA,qBAAiB,OAAO,GAAG;AAAA,EAC7B,CAAC;AAED,SAAO;AACT;AASA,SAAS,uBACP,QACA,mBACA,cACQ;AACR,MAAI,CAAC,kBAAmB,QAAO;AAE/B,QAAM,kBAAkB,OAAO,OAAiB,CAAC,KAAK,OAAO,UAAU;AACrE,QAAI,sBAAsB,KAAK,EAAG,KAAI,KAAK,KAAK;AAChD,WAAO;AAAA,EACT,GAAG,CAAC,CAAC;AAEL,QAAM,QAAQ,gBAAgB,CAAC;AAC/B,QAAM,OAAO,gBAAgB,gBAAgB,SAAS,CAAC;AACvD,MAAI,OAAO,QAAQ,MAAM,gBAAgB,QAAQ;AAC/C,UAAM,IAAI;AAAA,MACR,YAAY,YAAY,gEAClB,gBAAgB,KAAK,IAAI,CAAC;AAAA,IAElC;AAAA,EACF;AAEA,QAAM,YAAY,OAAO;AACzB,QAAM,OAAO,OAAO,SAAS;AAC7B,MAAI,MAAM,SAAS,SAAS;AAC1B,UAAM,IAAI;AAAA,MACR,YAAY,YAAY,+JAElB,MAAM,QAAQ,SAAS,iBAAiB,SAAS;AAAA,IACzD;AAAA,EACF;AAEA,SAAO;AACT;AAiBO,SAAS,sBACd,KACA,cACA,MACA,iBACqC;AACrC,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,SAAS,aAAa;AAC5B,MAAI,eAAe,QAAQ,UAAU,MAAM;AACzC,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AAEA,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,4BAA4B;AAAA,EAC9C;AAEA,QAAM,oBAAoB,OAAO,KAAK,CAAC,UAAU,sBAAsB,KAAK,CAAC;AAC7E,QAAM,qBAAqB,uBAAuB,QAAQ,mBAAmB,YAAY;AAEzF,QAAM,SAAS,oBAAoB,gBAAgB,MAAM,GAAG,EAAE,IAAI,CAAC;AACnE,QAAM,YAAY,oBAAoB,gBAAgB,gBAAgB,SAAS,CAAC,IAAI;AAGpF,MAAI,YAAY;AAEhB,WAAS,iBAAiB,OAAqB,OAAyB;AACtE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGvE,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,UAAI,aAAa,OAAO,QAAQ;AAC9B,cAAM,IAAI,MAAM,kDAAkD,YAAY,CAAC,SAAS,OAAO,MAAM,EAAE;AAAA,MACzG;AACA,YAAM,OAAO,OAAO,SAAS;AAC7B;AACA,aAAO;AAAA,IACT;AAGA,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,YAAM,cAAc,MAAM,QAAQ,KAAK,IAAI,MAAM,SAAS;AAC1D,UAAI,gBAAgB,GAAG;AACrB,eAAO,CAAC;AAAA,MACV;AAEA,UAAI,YAAY,cAAc,OAAO,QAAQ;AAC3C,cAAM,IAAI;AAAA,UACR,kDAAkD,YAAY,WAAW,SAAS,OAAO,MAAM;AAAA,QACjG;AAAA,MACF;AAEA,YAAM,YAAY,OAAO,MAAM,WAAW,YAAY,WAAW;AACjE,mBAAa;AACb,aAAO;AAAA,IACT;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAAiC;AACrD,gBAAM,WAAW;AACjB,gBAAMA,UAAkC,CAAC;AAEzC,qBAAW,QAAQ,CAAC,cAAc;AAChC,kBAAM,gBAAgB,UAAU;AAChC,gBAAI,eAAe;AACjB,oBAAM,iBAAiB,SAAS,aAAa;AAC7C,kBAAI,mBAAmB,QAAW;AAChC,gBAAAA,QAAO,aAAa,IAAI,iBAAiB,WAAW,cAAc;AAAA,cACpE;AAAA,YACF;AAAA,UACF,CAAC;AAED,iBAAOA;AAAA,QACT;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,iBAAO,MAAM,IAAI,YAAY;AAAA,QAC/B;AAEA,eAAO,aAAa,KAAK;AAAA,MAC3B;AACA,aAAO;AAAA,IACT;AAGA,WAAO;AAAA,EACT;AAIA,MAAI,WAAW;AACf,QAAM,SAAS,OAAO,IAAI,CAAC,OAAO,UAAU;AAC1C,QAAI,UAAU,oBAAoB;AAChC,aAAO;AAAA,IACT;AACA,UAAM,MAAM,KAAK,QAAQ;AACzB,QAAI,OAAO,MAAM;AACf,YAAM,IAAI,MAAM,YAAY,QAAQ,eAAe;AAAA,IACrD;AACA;AACA,WAAO,iBAAiB,OAAO,GAAG;AAAA,EACpC,CAAC;AAGD,MAAI,cAAc,OAAO,QAAQ;AAC/B,UAAM,IAAI,MAAM,4CAA4C,SAAS,kBAAkB,OAAO,MAAM,EAAE;AAAA,EACxG;AAEA,SAAO;AACT;;;AElfA,SAAkD,gBAAgB;AAgHlE,IAAM,uCAAuC;AAAA,EAC3C;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAmBA,SAAS,+CACP,cACA,MACmD;AACnD,UAAQ,cAAc;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF;AACE,YAAM,IAAI,MAAM,kCAAkC,YAAY,EAAE;AAAA,EACpE;AACF;AAEA,SAAS,kDACP,cACA,MACA,MACqD;AAErD,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,uBAAuB;AAAA,EACzC;AAGA,MAAI,QAAQ,QAAQ,SAAS,MAAM,SAAS,IAAI,MAAM,KAAK,QAAQ;AACjE,UAAM,IAAI,MAAM,wBAAwB,IAAI,QAAQ,KAAK,MAAM,EAAE;AAAA,EACnE;AAGA,SAAO,KAAK,IAAI,CAAC,SAAS,+CAA+C,cAAc,IAAI,CAAC;AAC9F;AAEA,SAAS,kCAAkC,cAAuE;AAChH,SAAO,qCAAqC,SAAS,YAAmB;AAC1E;AAEO,SAAS,8BACd,KACA,cACA,MACsC;AACtC,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,UAAU,aAAa;AAC7B,MAAI,eAAe,QAAQ,WAAW,MAAM;AAC1C,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AACA,QAAM,gBAAgB,QAAQ;AAE9B,WAAS,iBAAiB,OAAqB,OAAyB;AACtE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,kBAAkB,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGzF,QAAI,oBAAoB,QAAQ,kCAAkC,gBAAgB,GAAG;AAEnF,UAAI,oBAAoB,MAAM;AAC5B,eAAO,+CAA+C,kBAAkB,KAAe;AAAA,MACzF;AAGA,UAAI,oBAAoB,MAAM;AAC5B,eAAO,kDAAkD,kBAAkB,kBAAkB,KAAiB;AAAA,MAChH;AAAA,IACF;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAAiC;AACrD,gBAAM,WAAW;AACjB,gBAAM,SAAkC,CAAC;AAEzC,qBAAW,QAAQ,CAAC,cAAc;AAChC,kBAAM,gBAAgB,UAAU;AAChC,gBAAI,eAAe;AACjB,oBAAM,iBAAiB,SAAS,aAAa;AAC7C,kBAAI,mBAAmB,QAAW;AAChC,uBAAO,aAAa,IAAI,iBAAiB,WAAW,cAAc;AAAA,cACpE;AAAA,YACF;AAAA,UACF,CAAC;AAED,iBAAO;AAAA,QACT;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,iBAAO,MAAM,IAAI,YAAY;AAAA,QAC/B;AAEA,eAAO,aAAa,KAAK;AAAA,MAC3B;AAEA,aAAO;AAAA,IACT;AAGA,WAAO;AAAA,EACT;AAGA,MAAI,gBAAgB,GAAG;AACrB,QAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,YAAM,IAAI,MAAM,uBAAuB;AAAA,IACzC;AACA,QAAI,kBAAkB,KAAK,QAAQ;AACjC,YAAM,IAAI,MAAM,+BAA+B,aAAa,QAAQ,KAAK,MAAM,EAAE;AAAA,IACnF;AACA,WAAO,KAAK,IAAI,CAAC,MAAM,UAAU;AAC/B,aAAO,iBAAiB,QAAQ,KAAK,GAAG,IAAI;AAAA,IAC9C,CAAC;AAAA,EACH;AAGA,SAAO,iBAAiB,QAAQ,CAAC,GAAG,IAAI;AAC1C;","names":["result"]}
|
|
1
|
+
{"version":3,"sources":["../src/encryptedInputs.ts","../src/utils.ts","../src/encryptedReturnTypes.ts"],"sourcesContent":["import {\n Encryptable,\n type AnyExternalHash,\n type EncryptableAddress,\n type EncryptableBool,\n type EncryptableItem,\n type EncryptableUint128,\n type EncryptableUint16,\n type EncryptableUint32,\n type EncryptableUint64,\n type EncryptableUint8,\n type ExternalAddressHash,\n type ExternalBoolHash,\n type ExternalUint128Hash,\n type ExternalUint16Hash,\n type ExternalUint32Hash,\n type ExternalUint64Hash,\n type ExternalUint8Hash,\n type LiteralToPrimitive,\n type Primitive,\n} from '@cofhe/sdk';\nimport type { Abi, AbiFunction, AbiParameter, ExtractAbiFunction } from 'abitype';\nimport type { CofheAbiParametersToPrimitiveTypes } from './fhenixMap';\nimport {\n extractArrayParameterType,\n getAbiFunction,\n type MaybeExtractArrayParameterType,\n type MaybePartialBy,\n type ReadonlyWiden,\n tupleComponentsAreNamed,\n} from './utils';\n\nexport type CofheInputArgs<abi extends Abi | readonly unknown[] = Abi, functionName extends string = string> = GetArgs<\n abi,\n functionName\n>['args'];\n\ntype ExternalHashToInputMap<E extends AnyExternalHash> = E extends ExternalBoolHash\n ? EncryptableBool['data']\n : E extends ExternalUint8Hash\n ? EncryptableUint8['data']\n : E extends ExternalUint16Hash\n ? EncryptableUint16['data']\n : E extends ExternalUint32Hash\n ? EncryptableUint32['data']\n : E extends ExternalUint64Hash\n ? EncryptableUint64['data']\n : E extends ExternalUint128Hash\n ? EncryptableUint128['data']\n : E extends ExternalAddressHash\n ? EncryptableAddress['data']\n : never;\n\n// Note: AnyExternalHash is checked before Primitive - the branded hash types are structurally\n// `0x${string}` (a subtype of `string`, which is itself a Primitive), so Primitive must not be\n// checked first or every hash would be (incorrectly) widened via LiteralToPrimitive instead.\ntype EncryptedInputsToInputs<T> = T extends AnyExternalHash\n ? ExternalHashToInputMap<T>\n : T extends Primitive\n ? LiteralToPrimitive<T>\n : {\n [K in keyof T]: EncryptedInputsToInputs<T[K]>;\n };\n\n/**\n * Drops the batch-signature argument from an args tuple.\n *\n * The signature slot is the first non-`external*` parameter that follows the contiguous run of\n * `external*` parameters - `FHE.asEuintXX(hash, proof)` is a pair, so the proof sits with the\n * hashes it authenticates rather than at the end of the parameter list. Walks the ABI parameters\n * and the args tuple in lockstep and elides the arg at that position.\n */\ntype DropSignatureSlotTuple<\n params extends readonly unknown[],\n args extends readonly unknown[],\n afterExternalRun extends boolean = false,\n> = params extends readonly [infer P, ...infer PRest extends readonly unknown[]]\n ? args extends readonly [infer A, ...infer ARest extends readonly unknown[]]\n ? P extends AbiParameter\n ? ParamHasExternalInput<P> extends true\n ? readonly [A, ...DropSignatureSlotTuple<PRest, ARest, true>]\n : afterExternalRun extends true\n ? DropSignatureSlotTuple<PRest, ARest, false>\n : readonly [A, ...DropSignatureSlotTuple<PRest, ARest, false>]\n : readonly [A, ...DropSignatureSlotTuple<PRest, ARest, afterExternalRun>]\n : readonly []\n : args;\n\n/** Non-tuple args (e.g. `undefined`) pass through unchanged, as the previous `DropLast` did. */\ntype DropSignatureSlot<params extends readonly unknown[], args> = args extends readonly unknown[]\n ? DropSignatureSlotTuple<params, args>\n : args;\n\nconst EXTERNAL_INPUT_INTERNAL_TYPES = [\n 'externalEbool',\n 'externalEuint8',\n 'externalEuint16',\n 'externalEuint32',\n 'externalEuint64',\n 'externalEuint128',\n 'externalEaddress',\n] as const;\ntype ExternalInputInternalType = (typeof EXTERNAL_INPUT_INTERNAL_TYPES)[number];\n\n/** Type-level: does this ABI parameter carry (or, if a tuple, recursively contain) an `external*` input? */\ntype ParamHasExternalInput<param extends AbiParameter> =\n MaybeExtractArrayParameterType<param['internalType']> extends [infer head extends string, any]\n ? head extends ExternalInputInternalType\n ? true\n : // A struct array (`tuple[]` / `tuple[N]`) lands here too, with `head` the struct's own\n // internalType. Falling through to the components keeps `Order[]` visible as carrying\n // encrypted inputs, the way the runtime `paramHasExternalInput` already sees it.\n ParamComponentsHaveExternalInput<param>\n : param['internalType'] extends ExternalInputInternalType\n ? true\n : ParamComponentsHaveExternalInput<param>;\n\n/** Recurses into a parameter's `components`, for tuples at any array depth. */\ntype ParamComponentsHaveExternalInput<param extends AbiParameter> = param extends {\n components: infer components extends readonly AbiParameter[];\n}\n ? ParamsHaveExternalInput<components>\n : false;\n\ntype ParamsHaveExternalInput<params extends readonly AbiParameter[]> = params extends readonly [\n infer Head extends AbiParameter,\n ...infer Rest extends readonly AbiParameter[],\n]\n ? ParamHasExternalInput<Head> extends true\n ? true\n : ParamsHaveExternalInput<Rest>\n : false;\n\n/**\n * Pre-transform args for a function ABI: the raw plain values a caller supplies before encryption.\n *\n * If the function has one or more `external*` (encrypted) inputs, the plain `bytes` parameter\n * immediately following them is the shared batch signature slot - the caller never supplies it\n * directly (the SDK injects it after encryption), so it's dropped from this type entirely.\n */\nexport type CofheInputArgsPreTransform<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n abiFunction extends AbiFunction = abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction,\n> = EncryptedInputsToInputs<\n ParamsHaveExternalInput<abiFunction['inputs']> extends true\n ? DropSignatureSlot<abiFunction['inputs'], CofheInputArgs<abi, functionName>>\n : CofheInputArgs<abi, functionName>\n>;\n\n/// GetArgs from abitype (not exported from abitype)\ntype GetArgs<\n abi extends Abi | readonly unknown[] = Abi, // `readonly unknown[]` allows for non-const asserted types\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly [],\n ///\n abiFunction extends AbiFunction = abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction,\n primitiveTypes = CofheAbiParametersToPrimitiveTypes<abiFunction['inputs']>,\n args_ =\n | primitiveTypes // show all values\n | (abi extends Abi\n ? args extends primitiveTypes // infer value (if valid)\n ? primitiveTypes extends args // make sure `args` exactly matches `primitiveTypes` (e.g. avoid `args: readonly [{ foo: string; bar: number; }] | readonly [{ foo: string; }]`)\n ? // make inferred value of `args` match `primitiveTypes` (e.g. avoid union `args: readonly [123n] | readonly [bigint]`)\n ReadonlyWiden<args>\n : never\n : never\n : never)\n | (Abi extends abi ? readonly unknown[] : never), // fallback if `abi` is declared as `Abi`\n> = MaybePartialBy<{ args: args_ }, readonly [] extends primitiveTypes ? 'args' : Abi extends abi ? 'args' : string>;\n\nfunction internalTypeIsExternalInput(internalType: string): internalType is ExternalInputInternalType {\n return EXTERNAL_INPUT_INTERNAL_TYPES.includes(internalType as any);\n}\n\nfunction paramHasExternalInput(param: AbiParameter): boolean {\n const [internalTypeHead] = extractArrayParameterType(param.internalType);\n if (internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) return true;\n if ('components' in param && Array.isArray(param.components)) {\n return param.components.some((component) => paramHasExternalInput(component));\n }\n return false;\n}\n\nfunction transformSingleExternalToEncryptable(internalType: ExternalInputInternalType, data: unknown): EncryptableItem {\n switch (internalType) {\n case 'externalEbool':\n return Encryptable.bool(data as boolean);\n case 'externalEuint8':\n return Encryptable.uint8(data as string | bigint);\n case 'externalEuint16':\n return Encryptable.uint16(data as string | bigint);\n case 'externalEuint32':\n return Encryptable.uint32(data as string | bigint);\n case 'externalEuint64':\n return Encryptable.uint64(data as string | bigint);\n case 'externalEuint128':\n return Encryptable.uint128(data as string | bigint);\n case 'externalEaddress':\n return Encryptable.address(data as string | bigint);\n default:\n throw new Error(`Unknown external input type: ${internalType}`);\n }\n}\n\nfunction transformArrayOfExternalsToEncryptables(\n internalType: ExternalInputInternalType,\n size: string | undefined,\n data: unknown\n): EncryptableItem[] {\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n\n if (size != null && size !== '' && parseInt(size) !== data.length) {\n throw new Error(`Array size mismatch: ${size} !== ${data.length}`);\n }\n\n return data.map((item) => transformSingleExternalToEncryptable(internalType, item));\n}\n\n/**\n * Extracts encryptable values from function arguments based on the ABI.\n * Transforms raw data values into EncryptableItem objects that can be passed to the encrypt function.\n *\n * @param abi - The ABI containing the function definition\n * @param functionName - Name of the function\n * @param args - Function arguments in the format of CofheInputArgsPreTransform (raw data values;\n * if the function has `external*` inputs, the trailing batch-signature slot is NOT included)\n * @returns Array of EncryptableItem objects ready for encryption, in ABI parameter order\n */\nexport function extractEncryptableValues<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n args: CofheInputArgsPreTransform<TAbi, TFunctionName>\n): EncryptableItem[] {\n const abiFunction = getAbiFunction(abi, functionName);\n const inputs = abiFunction?.inputs;\n if (abiFunction == null || inputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n\n if (!Array.isArray(args)) {\n throw new Error('Arguments must be an array');\n }\n\n const hasExternalInputs = inputs.some((input) => paramHasExternalInput(input));\n const signatureSlotIndex = findSignatureSlotIndex(inputs, hasExternalInputs, functionName);\n\n // Collect encrypted values as EncryptableItem objects in order (flat array)\n const encryptableItems: EncryptableItem[] = [];\n\n function processParameter(param: AbiParameter, value: unknown): void {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead, internalTypeSize] = extractArrayParameterType(param.internalType);\n\n // Bare external input (e.g. `externalEuint32`, type: 'bytes32')\n if (typeSize == null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n encryptableItems.push(transformSingleExternalToEncryptable(internalTypeHead, value));\n return;\n }\n\n // Array of external inputs (e.g. `externalEuint32[]`/`externalEuint32[2]`, type: 'bytes32[]'/'bytes32[2]')\n if (typeSize != null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n const encryptables = transformArrayOfExternalsToEncryptables(internalTypeHead, typeSize, value);\n encryptableItems.push(...encryptables);\n return;\n }\n\n // Tuple recursive case (struct that may itself contain external inputs)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): void => {\n if (tupleComponentsAreNamed(components)) {\n const valueObj = tupleValue as Record<string, unknown>;\n components.forEach((component) => {\n processParameter(component, valueObj[component.name]);\n });\n return;\n }\n\n if (!Array.isArray(tupleValue)) {\n throw new Error('Expected an array value for tuple with unnamed components');\n }\n\n components.forEach((component, index) => {\n processParameter(component, tupleValue[index]);\n });\n };\n\n // `tuple[]` / `tuple[N]`: walk each element as its own tuple. Treating the array\n // itself as one tuple would look up named components on the array object, find\n // none, and hand `undefined` to the encryptable transform.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n value.forEach(processTuple);\n return;\n }\n\n processTuple(value);\n }\n return;\n }\n\n // Not an external input, and cannot contain one\n return;\n }\n\n // Process all inputs, skipping the batch-signature slot (nothing to extract there). `args` is the\n // pre-transform tuple, which omits that slot, so its indices lag by one past the signature.\n let argIndex = 0;\n inputs.forEach((input, index) => {\n if (index === signatureSlotIndex) return;\n const arg = args[argIndex];\n if (arg == null) {\n throw new Error(`Argument ${argIndex} is undefined`);\n }\n argIndex++;\n processParameter(input, arg);\n });\n\n return encryptableItems;\n}\n\n/**\n * Index of the shared batch-signature parameter: the plain `bytes` immediately after the\n * contiguous run of `external*` parameters. Returns -1 when the function has no encrypted inputs.\n *\n * The proof pairs with the hashes it authenticates (`FHE.asEuintXX(hash, proof)`), so it does not\n * have to be the last parameter - anything may follow it.\n */\nfunction findSignatureSlotIndex(\n inputs: readonly AbiParameter[],\n hasExternalInputs: boolean,\n functionName: string\n): number {\n if (!hasExternalInputs) return -1;\n\n const externalIndices = inputs.reduce<number[]>((acc, input, index) => {\n if (paramHasExternalInput(input)) acc.push(index);\n return acc;\n }, []);\n\n const first = externalIndices[0]!;\n const last = externalIndices[externalIndices.length - 1]!;\n if (last - first + 1 !== externalIndices.length) {\n throw new Error(\n `Function ${functionName} has encrypted (external*) inputs at non-adjacent positions ` +\n `(${externalIndices.join(', ')}). They share one batch signature, so they must be contiguous ` +\n `with the 'bytes' signature parameter immediately after them`\n );\n }\n\n const slotIndex = last + 1;\n const slot = inputs[slotIndex];\n if (slot?.type !== 'bytes') {\n throw new Error(\n `Function ${functionName} has encrypted (external*) inputs, so the parameter immediately after ` +\n `them must be a plain 'bytes' parameter to receive the shared batch signature - found ` +\n `'${slot?.type ?? 'nothing'}' at position ${slotIndex} instead`\n );\n }\n\n return slotIndex;\n}\n\n/**\n * Re-inserts encrypted values back into function arguments based on the ABI.\n * Takes the batch-verified result (per-input hashes, in order, followed by the single shared\n * signature) and the original args structure, and produces the final call args: each `external*`\n * position gets its corresponding hash, and the trailing `bytes` parameter gets the signature.\n *\n * This function mirrors the extraction logic in extractEncryptableValues, ensuring values\n * are inserted in the exact same order and locations where they were extracted.\n *\n * @param abi - The ABI containing the function definition\n * @param functionName - Name of the function\n * @param args - Original function arguments in the format of CofheInputArgsPreTransform\n * @param encryptedResult - `[...hashes, signature]`, as returned by `EncryptInputsBuilder.execute()`\n * @returns Function arguments with encrypted values inserted (format of CofheInputArgs)\n */\nexport function insertEncryptedValues<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n args: CofheInputArgsPreTransform<TAbi, TFunctionName>,\n encryptedResult: readonly `0x${string}`[]\n): CofheInputArgs<TAbi, TFunctionName> {\n const abiFunction = getAbiFunction(abi, functionName);\n const inputs = abiFunction?.inputs;\n if (abiFunction == null || inputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n\n if (!Array.isArray(args)) {\n throw new Error('Arguments must be an array');\n }\n\n const hasExternalInputs = inputs.some((input) => paramHasExternalInput(input));\n const signatureSlotIndex = findSignatureSlotIndex(inputs, hasExternalInputs, functionName);\n\n const hashes = hasExternalInputs ? encryptedResult.slice(0, -1) : [];\n const signature = hasExternalInputs ? encryptedResult[encryptedResult.length - 1] : undefined;\n\n // Track position in the hashes array\n let hashIndex = 0;\n\n function processParameter(param: AbiParameter, value: unknown): unknown {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead] = extractArrayParameterType(param.internalType);\n\n // Bare external input - substitute its hash\n if (typeSize == null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n if (hashIndex >= hashes.length) {\n throw new Error(`Not enough encrypted hashes: expected at least ${hashIndex + 1}, got ${hashes.length}`);\n }\n const hash = hashes[hashIndex];\n hashIndex++;\n return hash;\n }\n\n // Array of external inputs - substitute the corresponding slice of hashes\n if (typeSize != null && internalTypeHead != null && internalTypeIsExternalInput(internalTypeHead)) {\n const arrayLength = Array.isArray(value) ? value.length : 0;\n if (arrayLength === 0) {\n return [];\n }\n\n if (hashIndex + arrayLength > hashes.length) {\n throw new Error(\n `Not enough encrypted hashes: expected at least ${hashIndex + arrayLength}, got ${hashes.length}`\n );\n }\n\n const hashArray = hashes.slice(hashIndex, hashIndex + arrayLength);\n hashIndex += arrayLength;\n return hashArray;\n }\n\n // Tuple recursive case (struct that may itself contain external inputs)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): unknown => {\n if (tupleComponentsAreNamed(components)) {\n const valueObj = tupleValue as Record<string, unknown>;\n const result: Record<string, unknown> = {};\n\n components.forEach((component) => {\n const componentName = component.name;\n if (componentName) {\n const componentValue = valueObj[componentName];\n if (componentValue !== undefined) {\n result[componentName] = processParameter(component, componentValue);\n }\n }\n });\n\n return result;\n }\n\n if (!Array.isArray(tupleValue)) {\n throw new Error('Expected an array value for tuple with unnamed components');\n }\n\n return components.map((component, index) => processParameter(component, tupleValue[index]));\n };\n\n // `tuple[]` / `tuple[N]`: rebuild each element as its own tuple, in the same order\n // extractEncryptableValues walked them, so the hashes line up. Treating the array as\n // one tuple would consume no hashes and collapse the argument to `{}`.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n return value.map(processTuple);\n }\n\n return processTuple(value);\n }\n return value;\n }\n\n // Not an external input, return original value\n return value;\n }\n\n // Rebuild the full arg list; the signature slot is injected directly. `args` omits that slot, so\n // its indices lag by one past the signature position.\n let argIndex = 0;\n const result = inputs.map((input, index) => {\n if (index === signatureSlotIndex) {\n return signature;\n }\n const arg = args[argIndex];\n if (arg == null) {\n throw new Error(`Argument ${argIndex} is undefined`);\n }\n argIndex++;\n return processParameter(input, arg);\n });\n\n // Verify we used all encrypted hashes\n if (hashIndex !== hashes.length) {\n throw new Error(`Mismatch in encrypted hashes count: used ${hashIndex}, but provided ${hashes.length}`);\n }\n\n return result as CofheInputArgs<TAbi, TFunctionName>;\n}\n","import type {\n Abi,\n AbiConstructor,\n AbiError,\n AbiEvent,\n AbiFallback,\n AbiFunction,\n AbiParameter,\n AbiParametersToPrimitiveTypes,\n AbiReceive,\n AbiType,\n Address,\n ExtractAbiFunction,\n ResolvedRegister,\n SolidityArray,\n SolidityFixedArrayRange,\n SolidityTuple,\n} from 'abitype';\n\nexport type ContractReturnType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n ///\n abiFunction extends AbiFunction = (\n abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction\n ) extends infer abiFunction_ extends AbiFunction\n ? IsUnion<abiFunction_> extends true // narrow overloads by `args` by converting to tuple and filtering out overloads that don't match\n ? UnionToTuple<abiFunction_> extends infer abiFunctions extends readonly AbiFunction[]\n ? {\n [K in keyof abiFunctions]: (\n readonly unknown[] | undefined extends args // for functions that don't have inputs, `args` can be `undefined` so fallback to `readonly []`\n ? readonly []\n : args\n ) extends AbiParametersToPrimitiveTypes<abiFunctions[K]['inputs'], 'inputs'>\n ? abiFunctions[K]\n : never;\n }[number] // convert back to union (removes `never` tuple entries: `['foo', never, 'bar'][number]` => `'foo' | 'bar'`)\n : never\n : abiFunction_\n : never,\n outputs extends readonly AbiParameter[] = abiFunction['outputs'],\n primitiveTypes extends readonly unknown[] = AbiParametersToPrimitiveTypes<outputs, 'outputs', true>,\n> = [abiFunction] extends [never]\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` declared as `Abi`)\n : readonly unknown[] extends primitiveTypes\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` not const-asserted)\n : primitiveTypes extends readonly [] // unwrap `primitiveTypes`\n ? // biome-ignore lint/suspicious/noConfusingVoidType: <explanation>\n void // no outputs\n : primitiveTypes extends readonly [infer primitiveType]\n ? primitiveType // single output\n : primitiveTypes;\n\nexport type Error<messages extends string | string[]> = messages extends string\n ? [\n // Surrounding with array to prevent `messages` from being widened to `string`\n `Error: ${messages}`,\n ]\n : {\n [key in keyof messages]: messages[key] extends infer message extends string ? `Error: ${message}` : never;\n };\n\ntype PartialBy<TType, TKeys extends keyof TType> = ExactPartial<Pick<TType, TKeys>> & Omit<TType, TKeys>;\ntype ExactPartial<T> = { [K in keyof T]?: T[K] | undefined };\n\nexport type MaybePartialBy<TType, TKeys extends string> = TKeys extends keyof TType ? PartialBy<TType, TKeys> : TType;\n\nexport type ReadonlyWiden<TType> =\n | (TType extends Function ? TType : never)\n | (TType extends ResolvedRegister['bigIntType'] ? bigint : never)\n | (TType extends boolean ? boolean : never)\n | (TType extends ResolvedRegister['intType'] ? number : never)\n | (TType extends string\n ? TType extends Address\n ? Address\n : TType extends ResolvedRegister['bytesType']['inputs']\n ? ResolvedRegister['bytesType']\n : string\n : never)\n | (TType extends readonly [] ? readonly [] : never)\n | (TType extends Record<string, unknown> ? { [K in keyof TType]: ReadonlyWiden<TType[K]> } : never)\n | (TType extends { length: number }\n ? {\n [K in keyof TType]: ReadonlyWiden<TType[K]>;\n } extends infer Val extends unknown[]\n ? readonly [...Val]\n : never\n : never);\n\nexport type AbiBasicType = Exclude<AbiType, SolidityTuple | SolidityArray>;\n\n/**\n * First, infer `Head` against a known size type (either fixed-length array value or `\"\"`).\n *\n * | Input | Head |\n * | --------------- | ------------ |\n * | `string[]` | `string` |\n * | `string[][][3]` | `string[][]` |\n */\nexport type MaybeExtractArrayParameterType<type> = type extends `${infer head}[${'' | `${SolidityFixedArrayRange}`}]`\n ? // * Then, infer in the opposite direction, using the known `head` to infer the exact `size` value.\n // *\n // * | Input | Size |\n // * | ------------ | ---- |\n // * | `${head}[]` | `\"\"` |\n // * | `${head}[3]` | `3` |\n // */\n type extends `${head}[${infer size}]`\n ? [head, size]\n : undefined\n : undefined;\n\nexport type IsUnion<T, C = T> = T extends C ? ([C] extends [T] ? false : true) : never;\nexport type UnionToTuple<U, Last = LastInUnion<U>> = [U] extends [never]\n ? []\n : [...UnionToTuple<Exclude<U, Last>>, Last];\ntype LastInUnion<U> =\n UnionToIntersection<U extends unknown ? (x: U) => 0 : never> extends (x: infer L) => 0 ? L : never;\ntype UnionToIntersection<U> = (U extends unknown ? (arg: U) => 0 : never) extends (arg: infer I) => 0 ? I : never;\n\n/**\n * Create tuple of {@link type} type with {@link size} size\n *\n * @param Type - Type of tuple\n * @param Size - Size of tuple\n * @returns Tuple of {@link type} type with {@link size} size\n *\n * @example\n * type Result = Tuple<string, 2>\n * // ^? type Result = [string, string]\n */\n// https://github.com/Microsoft/TypeScript/issues/26223#issuecomment-674500430\nexport type Tuple<type, size extends number> = size extends size\n ? number extends size\n ? type[]\n : _TupleOf<type, size, []>\n : never;\ntype _TupleOf<length, size extends number, acc extends readonly unknown[]> = acc['length'] extends size\n ? acc\n : _TupleOf<length, size, readonly [length, ...acc]>;\n\n/**\n * Merges two object types into new type\n *\n * @param object1 - Object to merge into\n * @param object2 - Object to merge and override keys from {@link object1}\n * @returns New object type with keys from {@link object1} and {@link object2}. If a key exists in both {@link object1} and {@link object2}, the key from {@link object2} will be used.\n *\n * @example\n * type Result = Merge<{ foo: string }, { foo: number; bar: string }>\n * // ^? type Result = { foo: number; bar: string }\n */\nexport type Merge<object1, object2> = Omit<object1, keyof object2> & object2;\n\n// FUNCTIONS\n\n/**\n * Extracts array parameter type information from a type string.\n * Always returns a tuple, with undefined for non-array types.\n *\n * @param type - Type string like \"tuple\", \"tuple[]\", or \"tuple[2]\"\n * @returns [head, size] where:\n * - head is the base type\n * - size is undefined for non-arrays, \"\" for dynamic arrays, or the number as a string for fixed arrays\n *\n * @example\n * extractArrayParameterType(\"tuple\") // [\"tuple\", undefined]\n * extractArrayParameterType(\"tuple[]\") // [\"tuple\", \"\"]\n * extractArrayParameterType(\"tuple[2]\") // [\"tuple\", \"2\"]\n */\nexport function extractArrayParameterType<T extends string | undefined>(type: T): [T, string | undefined] {\n if (type == null) return [type, undefined];\n\n const match = type.match(/^(.+)\\[(\\d*)\\]$/);\n\n if (!match) {\n // Not an array type, return [type, undefined]\n return [type, undefined];\n }\n\n const head = match[1];\n const size = match[2]; // Empty string for dynamic arrays, or digits for fixed arrays\n\n // Return empty string for dynamic arrays, or the size string for fixed arrays\n return [head as T, size === '' ? '' : size];\n}\n\n/**\n * Matches the tuple-shape rule used by `CofheAbiComponentsToPrimitiveType`:\n * tuples are objects only when every component has a name. Otherwise they are\n * represented positionally as arrays.\n */\nexport function tupleComponentsAreNamed(components: readonly AbiParameter[]): boolean {\n return components.length > 0 && components.every((component) => component.name != null && component.name !== '');\n}\n\ntype AbiItem = AbiConstructor | AbiError | AbiEvent | AbiFallback | AbiFunction | AbiReceive;\n\nexport function isAbiFunction(item: AbiItem | unknown): item is AbiFunction {\n if (typeof item !== 'object' || item === null) return false;\n return 'type' in item && item.type === 'function' && 'name' in item && 'outputs' in item;\n}\n\nexport function getAbiFunction<TAbi extends Abi | readonly unknown[], TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName\n): AbiFunction | undefined {\n return abi.find((item) => {\n const isFunction = isAbiFunction(item);\n if (!isFunction) return false;\n return item.name === functionName;\n }) as AbiFunction | undefined;\n}\n","import { type Primitive, type LiteralToPrimitive, FheTypes } from '@cofhe/sdk';\nimport type { Abi, AbiFunction, ExtractAbiFunction, AbiParameter } from 'abitype';\nimport type {\n EncryptedReturnType,\n CofheAbiParametersToPrimitiveTypes,\n EBool,\n EAddress,\n EUint128,\n EUint16,\n EUint32,\n EUint64,\n EUint8,\n} from './fhenixMap';\nimport {\n getAbiFunction,\n type IsUnion,\n type UnionToTuple,\n type ContractReturnType,\n extractArrayParameterType,\n tupleComponentsAreNamed,\n} from './utils';\n\ntype FirstCofheReturnValue<T> = [T] extends [void]\n ? never\n : T extends readonly [infer Head, ...(readonly unknown[])]\n ? Head\n : T;\n\ntype FheTypeFromCofheReturnValue<T> = T extends { utype: infer U }\n ? U\n : T extends readonly (infer Element)[]\n ? FheTypeFromCofheReturnValue<Element>\n : never;\n\n/**\n * Extracts the `FheTypes` of the first returned value of a CoFHE call.\n *\n * Examples:\n * - `returns (ebool)` -> `FheTypes.Bool`\n * - `returns (euint8, euint64)` -> `FheTypes.Uint8`\n * - `returns (euint16[])` -> `FheTypes.Uint16`\n */\nexport type CofheFirstReturnFheType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n> = FheTypeFromCofheReturnValue<FirstCofheReturnValue<CofheReturnType<abi, functionName, args>>>;\n\nexport type CofheReturnType<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n ///\n abiFunction extends AbiFunction = (\n abi extends Abi ? ExtractAbiFunction<abi, functionName> : AbiFunction\n ) extends infer abiFunction_ extends AbiFunction\n ? IsUnion<abiFunction_> extends true // narrow overloads by `args` by converting to tuple and filtering out overloads that don't match\n ? UnionToTuple<abiFunction_> extends infer abiFunctions extends readonly AbiFunction[]\n ? {\n [K in keyof abiFunctions]: (\n readonly unknown[] | undefined extends args // for functions that don't have inputs, `args` can be `undefined` so fallback to `readonly []`\n ? readonly []\n : args\n ) extends CofheAbiParametersToPrimitiveTypes<abiFunctions[K]['inputs'], 'inputs'>\n ? abiFunctions[K]\n : never;\n }[number] // convert back to union (removes `never` tuple entries: `['foo', never, 'bar'][number]` => `'foo' | 'bar'`)\n : never\n : abiFunction_\n : never,\n outputs extends readonly AbiParameter[] = abiFunction['outputs'],\n primitiveTypes extends readonly unknown[] = CofheAbiParametersToPrimitiveTypes<outputs, 'outputs'>,\n> = [abiFunction] extends [never]\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` declared as `Abi`)\n : readonly unknown[] extends primitiveTypes\n ? unknown // `abiFunction` was not inferrable (e.g. `abi` not const-asserted)\n : primitiveTypes extends readonly [] // unwrap `primitiveTypes`\n ? // biome-ignore lint/suspicious/noConfusingVoidType: <explanation>\n void // no outputs\n : primitiveTypes extends readonly [infer primitiveType]\n ? primitiveType // single output\n : primitiveTypes;\n\ntype EncryptedReturnTypeToReturnTypeMap<E extends EncryptedReturnType> = E extends EBool\n ? boolean\n : E extends EUint8\n ? bigint\n : E extends EUint16\n ? bigint\n : E extends EUint32\n ? bigint\n : E extends EUint64\n ? bigint\n : E extends EUint128\n ? bigint\n : E extends EAddress\n ? string\n : never;\n\ntype EncryptedReturnTypesToReturnTypes<T> = T extends Primitive\n ? LiteralToPrimitive<T>\n : T extends EncryptedReturnType\n ? EncryptedReturnTypeToReturnTypeMap<T>\n : {\n [K in keyof T]: EncryptedReturnTypesToReturnTypes<T[K]>;\n };\n\nexport type CofheReturnTypePostTransform<\n abi extends Abi | readonly unknown[] = Abi,\n functionName extends string = string,\n args extends readonly unknown[] | undefined = readonly unknown[] | undefined,\n> = EncryptedReturnTypesToReturnTypes<CofheReturnType<abi, functionName, args>>;\n\nconst ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES = [\n 'ebool',\n 'euint8',\n 'euint16',\n 'euint32',\n 'euint64',\n 'euint128',\n 'eaddress',\n] as const;\ntype EncryptedReturnTypeInternalType = (typeof ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES)[number];\n\ntype EncryptedReturnTypeInternalTypeToReturnTypeMap<T extends EncryptedReturnTypeInternalType> = T extends 'ebool'\n ? EBool\n : T extends 'euint8'\n ? EUint8\n : T extends 'euint16'\n ? EUint16\n : T extends 'euint32'\n ? EUint32\n : T extends 'euint64'\n ? EUint64\n : T extends 'euint128'\n ? EUint128\n : T extends 'eaddress'\n ? EAddress\n : never;\n\nfunction transformSingleEncryptedReturnTypeToReturnType<T extends EncryptedReturnTypeInternalType>(\n internalType: T,\n data: string\n): EncryptedReturnTypeInternalTypeToReturnTypeMap<T> {\n switch (internalType) {\n case 'ebool':\n return {\n ctHash: data,\n utype: FheTypes.Bool,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint8':\n return {\n ctHash: data,\n utype: FheTypes.Uint8,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint16':\n return {\n ctHash: data,\n utype: FheTypes.Uint16,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint32':\n return {\n ctHash: data,\n utype: FheTypes.Uint32,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint64':\n return {\n ctHash: data,\n utype: FheTypes.Uint64,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'euint128':\n return {\n ctHash: data,\n utype: FheTypes.Uint128,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n case 'eaddress':\n return {\n ctHash: data,\n utype: FheTypes.Uint160,\n } as EncryptedReturnTypeInternalTypeToReturnTypeMap<T>;\n default:\n throw new Error(`Unknown encrypted return type: ${internalType}`);\n }\n}\n\nfunction transformArrayOfEncryptedReturnTypesToReturnTypes<T extends EncryptedReturnTypeInternalType>(\n internalType: T,\n size: string | undefined,\n data: string[]\n): EncryptedReturnTypeInternalTypeToReturnTypeMap<T>[] {\n // Ensure data is an array\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n\n // Ensure length of fixed length tuple matches the length of the data array\n if (size != null && size !== '' && parseInt(size) !== data.length) {\n throw new Error(`Array size mismatch: ${size} !== ${data.length}`);\n }\n\n // Transform each item in the data array into an return type\n return data.map((item) => transformSingleEncryptedReturnTypeToReturnType(internalType, item));\n}\n\nfunction internalTypeIsEncryptedReturnType(internalType: string): internalType is EncryptedReturnTypeInternalType {\n return ENCRYPTED_RETURN_TYPE_INTERNAL_TYPES.includes(internalType as any);\n}\n\nexport function transformEncryptedReturnTypes<TAbi extends Abi, TFunctionName extends string>(\n abi: TAbi,\n functionName: TFunctionName,\n data: ContractReturnType<TAbi, TFunctionName>\n): CofheReturnType<TAbi, TFunctionName> {\n const abiFunction = getAbiFunction(abi, functionName);\n const outputs = abiFunction?.outputs;\n if (abiFunction == null || outputs == null) {\n throw new Error(`Function ${functionName} not found in ABI`);\n }\n const outputsLength = outputs.length;\n\n function processParameter(param: AbiParameter, value: unknown): unknown {\n const [typeHead, typeSize] = extractArrayParameterType(param.type);\n const [internalTypeHead, internalTypeSize] = extractArrayParameterType(param.internalType);\n\n // Is encrypted return type\n if (internalTypeHead != null && internalTypeIsEncryptedReturnType(internalTypeHead)) {\n // Is single encrypted return type\n if (internalTypeSize == null) {\n return transformSingleEncryptedReturnTypeToReturnType(internalTypeHead, value as string);\n }\n\n // Is array of encrypted return types\n if (internalTypeSize != null) {\n return transformArrayOfEncryptedReturnTypesToReturnTypes(internalTypeHead, internalTypeSize, value as string[]);\n }\n }\n\n // Tuple but not an encrypted return type (recursive case)\n if (typeHead === 'tuple') {\n if ('components' in param && Array.isArray(param.components)) {\n const components = param.components;\n\n const processTuple = (tupleValue: unknown): unknown => {\n if (tupleComponentsAreNamed(components)) {\n const valueObj = tupleValue as Record<string, unknown>;\n const result: Record<string, unknown> = {};\n\n components.forEach((component) => {\n const componentName = component.name;\n if (componentName) {\n const componentValue = valueObj[componentName];\n if (componentValue !== undefined) {\n result[componentName] = processParameter(component, componentValue);\n }\n }\n });\n\n return result;\n }\n\n if (!Array.isArray(tupleValue)) {\n throw new Error('Expected an array value for tuple with unnamed components');\n }\n\n return components.map((component, index) => processParameter(component, tupleValue[index]));\n };\n\n // `tuple[]` / `tuple[N]`: transform each element as its own tuple. Treating the\n // array itself as one tuple would look up named components on the array object,\n // find none, and collapse the whole result to `{}`.\n if (typeSize != null) {\n if (!Array.isArray(value)) {\n throw new Error(`Expected an array value for ${param.type}`);\n }\n if (typeSize !== '' && parseInt(typeSize) !== value.length) {\n throw new Error(`Array size mismatch: ${typeSize} !== ${value.length}`);\n }\n return value.map(processTuple);\n }\n\n return processTuple(value);\n }\n\n return value;\n }\n\n // Not an encrypted return type, return original value\n return value;\n }\n\n // Process results (array case)\n if (outputsLength > 1) {\n if (!Array.isArray(data)) {\n throw new Error('Data must be an array');\n }\n if (outputsLength !== data.length) {\n throw new Error(`Mismatch in outputs length: ${outputsLength} !== ${data.length}`);\n }\n return data.map((item, index) => {\n return processParameter(outputs[index], item);\n }) as CofheReturnType<TAbi, TFunctionName>;\n }\n\n // Process retuls (single item case)\n return processParameter(outputs[0], data) as CofheReturnType<TAbi, TFunctionName>;\n}\n"],"mappings":";AAAA;AAAA,EACE;AAAA,OAmBK;;;ACuJA,SAAS,0BAAwD,MAAkC;AACxG,MAAI,QAAQ,KAAM,QAAO,CAAC,MAAM,MAAS;AAEzC,QAAM,QAAQ,KAAK,MAAM,iBAAiB;AAE1C,MAAI,CAAC,OAAO;AAEV,WAAO,CAAC,MAAM,MAAS;AAAA,EACzB;AAEA,QAAM,OAAO,MAAM,CAAC;AACpB,QAAM,OAAO,MAAM,CAAC;AAGpB,SAAO,CAAC,MAAW,SAAS,KAAK,KAAK,IAAI;AAC5C;AAOO,SAAS,wBAAwB,YAA8C;AACpF,SAAO,WAAW,SAAS,KAAK,WAAW,MAAM,CAAC,cAAc,UAAU,QAAQ,QAAQ,UAAU,SAAS,EAAE;AACjH;AAIO,SAAS,cAAc,MAA8C;AAC1E,MAAI,OAAO,SAAS,YAAY,SAAS,KAAM,QAAO;AACtD,SAAO,UAAU,QAAQ,KAAK,SAAS,cAAc,UAAU,QAAQ,aAAa;AACtF;AAEO,SAAS,eACd,KACA,cACyB;AACzB,SAAO,IAAI,KAAK,CAAC,SAAS;AACxB,UAAM,aAAa,cAAc,IAAI;AACrC,QAAI,CAAC,WAAY,QAAO;AACxB,WAAO,KAAK,SAAS;AAAA,EACvB,CAAC;AACH;;;ADxHA,IAAM,gCAAgC;AAAA,EACpC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAsEA,SAAS,4BAA4B,cAAiE;AACpG,SAAO,8BAA8B,SAAS,YAAmB;AACnE;AAEA,SAAS,sBAAsB,OAA8B;AAC3D,QAAM,CAAC,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AACvE,MAAI,oBAAoB,QAAQ,4BAA4B,gBAAgB,EAAG,QAAO;AACtF,MAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,WAAO,MAAM,WAAW,KAAK,CAAC,cAAc,sBAAsB,SAAS,CAAC;AAAA,EAC9E;AACA,SAAO;AACT;AAEA,SAAS,qCAAqC,cAAyC,MAAgC;AACrH,UAAQ,cAAc;AAAA,IACpB,KAAK;AACH,aAAO,YAAY,KAAK,IAAe;AAAA,IACzC,KAAK;AACH,aAAO,YAAY,MAAM,IAAuB;AAAA,IAClD,KAAK;AACH,aAAO,YAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,YAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,YAAY,OAAO,IAAuB;AAAA,IACnD,KAAK;AACH,aAAO,YAAY,QAAQ,IAAuB;AAAA,IACpD,KAAK;AACH,aAAO,YAAY,QAAQ,IAAuB;AAAA,IACpD;AACE,YAAM,IAAI,MAAM,gCAAgC,YAAY,EAAE;AAAA,EAClE;AACF;AAEA,SAAS,wCACP,cACA,MACA,MACmB;AACnB,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,uBAAuB;AAAA,EACzC;AAEA,MAAI,QAAQ,QAAQ,SAAS,MAAM,SAAS,IAAI,MAAM,KAAK,QAAQ;AACjE,UAAM,IAAI,MAAM,wBAAwB,IAAI,QAAQ,KAAK,MAAM,EAAE;AAAA,EACnE;AAEA,SAAO,KAAK,IAAI,CAAC,SAAS,qCAAqC,cAAc,IAAI,CAAC;AACpF;AAYO,SAAS,yBACd,KACA,cACA,MACmB;AACnB,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,SAAS,aAAa;AAC5B,MAAI,eAAe,QAAQ,UAAU,MAAM;AACzC,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AAEA,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,4BAA4B;AAAA,EAC9C;AAEA,QAAM,oBAAoB,OAAO,KAAK,CAAC,UAAU,sBAAsB,KAAK,CAAC;AAC7E,QAAM,qBAAqB,uBAAuB,QAAQ,mBAAmB,YAAY;AAGzF,QAAM,mBAAsC,CAAC;AAE7C,WAAS,iBAAiB,OAAqB,OAAsB;AACnE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,kBAAkB,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGzF,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,uBAAiB,KAAK,qCAAqC,kBAAkB,KAAK,CAAC;AACnF;AAAA,IACF;AAGA,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,YAAM,eAAe,wCAAwC,kBAAkB,UAAU,KAAK;AAC9F,uBAAiB,KAAK,GAAG,YAAY;AACrC;AAAA,IACF;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAA8B;AAClD,cAAI,wBAAwB,UAAU,GAAG;AACvC,kBAAM,WAAW;AACjB,uBAAW,QAAQ,CAAC,cAAc;AAChC,+BAAiB,WAAW,SAAS,UAAU,IAAI,CAAC;AAAA,YACtD,CAAC;AACD;AAAA,UACF;AAEA,cAAI,CAAC,MAAM,QAAQ,UAAU,GAAG;AAC9B,kBAAM,IAAI,MAAM,2DAA2D;AAAA,UAC7E;AAEA,qBAAW,QAAQ,CAAC,WAAW,UAAU;AACvC,6BAAiB,WAAW,WAAW,KAAK,CAAC;AAAA,UAC/C,CAAC;AAAA,QACH;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,gBAAM,QAAQ,YAAY;AAC1B;AAAA,QACF;AAEA,qBAAa,KAAK;AAAA,MACpB;AACA;AAAA,IACF;AAGA;AAAA,EACF;AAIA,MAAI,WAAW;AACf,SAAO,QAAQ,CAAC,OAAO,UAAU;AAC/B,QAAI,UAAU,mBAAoB;AAClC,UAAM,MAAM,KAAK,QAAQ;AACzB,QAAI,OAAO,MAAM;AACf,YAAM,IAAI,MAAM,YAAY,QAAQ,eAAe;AAAA,IACrD;AACA;AACA,qBAAiB,OAAO,GAAG;AAAA,EAC7B,CAAC;AAED,SAAO;AACT;AASA,SAAS,uBACP,QACA,mBACA,cACQ;AACR,MAAI,CAAC,kBAAmB,QAAO;AAE/B,QAAM,kBAAkB,OAAO,OAAiB,CAAC,KAAK,OAAO,UAAU;AACrE,QAAI,sBAAsB,KAAK,EAAG,KAAI,KAAK,KAAK;AAChD,WAAO;AAAA,EACT,GAAG,CAAC,CAAC;AAEL,QAAM,QAAQ,gBAAgB,CAAC;AAC/B,QAAM,OAAO,gBAAgB,gBAAgB,SAAS,CAAC;AACvD,MAAI,OAAO,QAAQ,MAAM,gBAAgB,QAAQ;AAC/C,UAAM,IAAI;AAAA,MACR,YAAY,YAAY,gEAClB,gBAAgB,KAAK,IAAI,CAAC;AAAA,IAElC;AAAA,EACF;AAEA,QAAM,YAAY,OAAO;AACzB,QAAM,OAAO,OAAO,SAAS;AAC7B,MAAI,MAAM,SAAS,SAAS;AAC1B,UAAM,IAAI;AAAA,MACR,YAAY,YAAY,+JAElB,MAAM,QAAQ,SAAS,iBAAiB,SAAS;AAAA,IACzD;AAAA,EACF;AAEA,SAAO;AACT;AAiBO,SAAS,sBACd,KACA,cACA,MACA,iBACqC;AACrC,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,SAAS,aAAa;AAC5B,MAAI,eAAe,QAAQ,UAAU,MAAM;AACzC,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AAEA,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,4BAA4B;AAAA,EAC9C;AAEA,QAAM,oBAAoB,OAAO,KAAK,CAAC,UAAU,sBAAsB,KAAK,CAAC;AAC7E,QAAM,qBAAqB,uBAAuB,QAAQ,mBAAmB,YAAY;AAEzF,QAAM,SAAS,oBAAoB,gBAAgB,MAAM,GAAG,EAAE,IAAI,CAAC;AACnE,QAAM,YAAY,oBAAoB,gBAAgB,gBAAgB,SAAS,CAAC,IAAI;AAGpF,MAAI,YAAY;AAEhB,WAAS,iBAAiB,OAAqB,OAAyB;AACtE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGvE,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,UAAI,aAAa,OAAO,QAAQ;AAC9B,cAAM,IAAI,MAAM,kDAAkD,YAAY,CAAC,SAAS,OAAO,MAAM,EAAE;AAAA,MACzG;AACA,YAAM,OAAO,OAAO,SAAS;AAC7B;AACA,aAAO;AAAA,IACT;AAGA,QAAI,YAAY,QAAQ,oBAAoB,QAAQ,4BAA4B,gBAAgB,GAAG;AACjG,YAAM,cAAc,MAAM,QAAQ,KAAK,IAAI,MAAM,SAAS;AAC1D,UAAI,gBAAgB,GAAG;AACrB,eAAO,CAAC;AAAA,MACV;AAEA,UAAI,YAAY,cAAc,OAAO,QAAQ;AAC3C,cAAM,IAAI;AAAA,UACR,kDAAkD,YAAY,WAAW,SAAS,OAAO,MAAM;AAAA,QACjG;AAAA,MACF;AAEA,YAAM,YAAY,OAAO,MAAM,WAAW,YAAY,WAAW;AACjE,mBAAa;AACb,aAAO;AAAA,IACT;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAAiC;AACrD,cAAI,wBAAwB,UAAU,GAAG;AACvC,kBAAM,WAAW;AACjB,kBAAMA,UAAkC,CAAC;AAEzC,uBAAW,QAAQ,CAAC,cAAc;AAChC,oBAAM,gBAAgB,UAAU;AAChC,kBAAI,eAAe;AACjB,sBAAM,iBAAiB,SAAS,aAAa;AAC7C,oBAAI,mBAAmB,QAAW;AAChC,kBAAAA,QAAO,aAAa,IAAI,iBAAiB,WAAW,cAAc;AAAA,gBACpE;AAAA,cACF;AAAA,YACF,CAAC;AAED,mBAAOA;AAAA,UACT;AAEA,cAAI,CAAC,MAAM,QAAQ,UAAU,GAAG;AAC9B,kBAAM,IAAI,MAAM,2DAA2D;AAAA,UAC7E;AAEA,iBAAO,WAAW,IAAI,CAAC,WAAW,UAAU,iBAAiB,WAAW,WAAW,KAAK,CAAC,CAAC;AAAA,QAC5F;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,iBAAO,MAAM,IAAI,YAAY;AAAA,QAC/B;AAEA,eAAO,aAAa,KAAK;AAAA,MAC3B;AACA,aAAO;AAAA,IACT;AAGA,WAAO;AAAA,EACT;AAIA,MAAI,WAAW;AACf,QAAM,SAAS,OAAO,IAAI,CAAC,OAAO,UAAU;AAC1C,QAAI,UAAU,oBAAoB;AAChC,aAAO;AAAA,IACT;AACA,UAAM,MAAM,KAAK,QAAQ;AACzB,QAAI,OAAO,MAAM;AACf,YAAM,IAAI,MAAM,YAAY,QAAQ,eAAe;AAAA,IACrD;AACA;AACA,WAAO,iBAAiB,OAAO,GAAG;AAAA,EACpC,CAAC;AAGD,MAAI,cAAc,OAAO,QAAQ;AAC/B,UAAM,IAAI,MAAM,4CAA4C,SAAS,kBAAkB,OAAO,MAAM,EAAE;AAAA,EACxG;AAEA,SAAO;AACT;;;AEtgBA,SAAkD,gBAAgB;AAiHlE,IAAM,uCAAuC;AAAA,EAC3C;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAmBA,SAAS,+CACP,cACA,MACmD;AACnD,UAAQ,cAAc;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO,SAAS;AAAA,MAClB;AAAA,IACF;AACE,YAAM,IAAI,MAAM,kCAAkC,YAAY,EAAE;AAAA,EACpE;AACF;AAEA,SAAS,kDACP,cACA,MACA,MACqD;AAErD,MAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,uBAAuB;AAAA,EACzC;AAGA,MAAI,QAAQ,QAAQ,SAAS,MAAM,SAAS,IAAI,MAAM,KAAK,QAAQ;AACjE,UAAM,IAAI,MAAM,wBAAwB,IAAI,QAAQ,KAAK,MAAM,EAAE;AAAA,EACnE;AAGA,SAAO,KAAK,IAAI,CAAC,SAAS,+CAA+C,cAAc,IAAI,CAAC;AAC9F;AAEA,SAAS,kCAAkC,cAAuE;AAChH,SAAO,qCAAqC,SAAS,YAAmB;AAC1E;AAEO,SAAS,8BACd,KACA,cACA,MACsC;AACtC,QAAM,cAAc,eAAe,KAAK,YAAY;AACpD,QAAM,UAAU,aAAa;AAC7B,MAAI,eAAe,QAAQ,WAAW,MAAM;AAC1C,UAAM,IAAI,MAAM,YAAY,YAAY,mBAAmB;AAAA,EAC7D;AACA,QAAM,gBAAgB,QAAQ;AAE9B,WAAS,iBAAiB,OAAqB,OAAyB;AACtE,UAAM,CAAC,UAAU,QAAQ,IAAI,0BAA0B,MAAM,IAAI;AACjE,UAAM,CAAC,kBAAkB,gBAAgB,IAAI,0BAA0B,MAAM,YAAY;AAGzF,QAAI,oBAAoB,QAAQ,kCAAkC,gBAAgB,GAAG;AAEnF,UAAI,oBAAoB,MAAM;AAC5B,eAAO,+CAA+C,kBAAkB,KAAe;AAAA,MACzF;AAGA,UAAI,oBAAoB,MAAM;AAC5B,eAAO,kDAAkD,kBAAkB,kBAAkB,KAAiB;AAAA,MAChH;AAAA,IACF;AAGA,QAAI,aAAa,SAAS;AACxB,UAAI,gBAAgB,SAAS,MAAM,QAAQ,MAAM,UAAU,GAAG;AAC5D,cAAM,aAAa,MAAM;AAEzB,cAAM,eAAe,CAAC,eAAiC;AACrD,cAAI,wBAAwB,UAAU,GAAG;AACvC,kBAAM,WAAW;AACjB,kBAAM,SAAkC,CAAC;AAEzC,uBAAW,QAAQ,CAAC,cAAc;AAChC,oBAAM,gBAAgB,UAAU;AAChC,kBAAI,eAAe;AACjB,sBAAM,iBAAiB,SAAS,aAAa;AAC7C,oBAAI,mBAAmB,QAAW;AAChC,yBAAO,aAAa,IAAI,iBAAiB,WAAW,cAAc;AAAA,gBACpE;AAAA,cACF;AAAA,YACF,CAAC;AAED,mBAAO;AAAA,UACT;AAEA,cAAI,CAAC,MAAM,QAAQ,UAAU,GAAG;AAC9B,kBAAM,IAAI,MAAM,2DAA2D;AAAA,UAC7E;AAEA,iBAAO,WAAW,IAAI,CAAC,WAAW,UAAU,iBAAiB,WAAW,WAAW,KAAK,CAAC,CAAC;AAAA,QAC5F;AAKA,YAAI,YAAY,MAAM;AACpB,cAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,kBAAM,IAAI,MAAM,+BAA+B,MAAM,IAAI,EAAE;AAAA,UAC7D;AACA,cAAI,aAAa,MAAM,SAAS,QAAQ,MAAM,MAAM,QAAQ;AAC1D,kBAAM,IAAI,MAAM,wBAAwB,QAAQ,QAAQ,MAAM,MAAM,EAAE;AAAA,UACxE;AACA,iBAAO,MAAM,IAAI,YAAY;AAAA,QAC/B;AAEA,eAAO,aAAa,KAAK;AAAA,MAC3B;AAEA,aAAO;AAAA,IACT;AAGA,WAAO;AAAA,EACT;AAGA,MAAI,gBAAgB,GAAG;AACrB,QAAI,CAAC,MAAM,QAAQ,IAAI,GAAG;AACxB,YAAM,IAAI,MAAM,uBAAuB;AAAA,IACzC;AACA,QAAI,kBAAkB,KAAK,QAAQ;AACjC,YAAM,IAAI,MAAM,+BAA+B,aAAa,QAAQ,KAAK,MAAM,EAAE;AAAA,IACnF;AACA,WAAO,KAAK,IAAI,CAAC,MAAM,UAAU;AAC/B,aAAO,iBAAiB,QAAQ,KAAK,GAAG,IAAI;AAAA,IAC9C,CAAC;AAAA,EACH;AAGA,SAAO,iBAAiB,QAAQ,CAAC,GAAG,IAAI;AAC1C;","names":["result"]}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@cofhe/abi",
|
|
3
|
-
"version": "0.
|
|
3
|
+
"version": "1.0.0-alpha.0",
|
|
4
4
|
"type": "module",
|
|
5
5
|
"description": "ABI utilities for Fhenix FHE contracts",
|
|
6
6
|
"types": "./dist/index.d.ts",
|
|
@@ -27,7 +27,7 @@
|
|
|
27
27
|
],
|
|
28
28
|
"dependencies": {
|
|
29
29
|
"abitype": "1.2.3",
|
|
30
|
-
"@cofhe/sdk": "0.
|
|
30
|
+
"@cofhe/sdk": "1.0.0-alpha.0"
|
|
31
31
|
},
|
|
32
32
|
"devDependencies": {
|
|
33
33
|
"@types/node": "20.19.15",
|
package/src/encryptedInputs.ts
CHANGED
|
@@ -27,6 +27,7 @@ import {
|
|
|
27
27
|
type MaybeExtractArrayParameterType,
|
|
28
28
|
type MaybePartialBy,
|
|
29
29
|
type ReadonlyWiden,
|
|
30
|
+
tupleComponentsAreNamed,
|
|
30
31
|
} from './utils';
|
|
31
32
|
|
|
32
33
|
export type CofheInputArgs<abi extends Abi | readonly unknown[] = Abi, functionName extends string = string> = GetArgs<
|
|
@@ -272,9 +273,20 @@ export function extractEncryptableValues<TAbi extends Abi, TFunctionName extends
|
|
|
272
273
|
const components = param.components;
|
|
273
274
|
|
|
274
275
|
const processTuple = (tupleValue: unknown): void => {
|
|
275
|
-
|
|
276
|
-
|
|
277
|
-
|
|
276
|
+
if (tupleComponentsAreNamed(components)) {
|
|
277
|
+
const valueObj = tupleValue as Record<string, unknown>;
|
|
278
|
+
components.forEach((component) => {
|
|
279
|
+
processParameter(component, valueObj[component.name]);
|
|
280
|
+
});
|
|
281
|
+
return;
|
|
282
|
+
}
|
|
283
|
+
|
|
284
|
+
if (!Array.isArray(tupleValue)) {
|
|
285
|
+
throw new Error('Expected an array value for tuple with unnamed components');
|
|
286
|
+
}
|
|
287
|
+
|
|
288
|
+
components.forEach((component, index) => {
|
|
289
|
+
processParameter(component, tupleValue[index]);
|
|
278
290
|
});
|
|
279
291
|
};
|
|
280
292
|
|
|
@@ -437,20 +449,28 @@ export function insertEncryptedValues<TAbi extends Abi, TFunctionName extends st
|
|
|
437
449
|
const components = param.components;
|
|
438
450
|
|
|
439
451
|
const processTuple = (tupleValue: unknown): unknown => {
|
|
440
|
-
|
|
441
|
-
|
|
442
|
-
|
|
443
|
-
|
|
444
|
-
|
|
445
|
-
|
|
446
|
-
|
|
447
|
-
|
|
448
|
-
|
|
452
|
+
if (tupleComponentsAreNamed(components)) {
|
|
453
|
+
const valueObj = tupleValue as Record<string, unknown>;
|
|
454
|
+
const result: Record<string, unknown> = {};
|
|
455
|
+
|
|
456
|
+
components.forEach((component) => {
|
|
457
|
+
const componentName = component.name;
|
|
458
|
+
if (componentName) {
|
|
459
|
+
const componentValue = valueObj[componentName];
|
|
460
|
+
if (componentValue !== undefined) {
|
|
461
|
+
result[componentName] = processParameter(component, componentValue);
|
|
462
|
+
}
|
|
449
463
|
}
|
|
450
|
-
}
|
|
451
|
-
|
|
464
|
+
});
|
|
465
|
+
|
|
466
|
+
return result;
|
|
467
|
+
}
|
|
468
|
+
|
|
469
|
+
if (!Array.isArray(tupleValue)) {
|
|
470
|
+
throw new Error('Expected an array value for tuple with unnamed components');
|
|
471
|
+
}
|
|
452
472
|
|
|
453
|
-
return
|
|
473
|
+
return components.map((component, index) => processParameter(component, tupleValue[index]));
|
|
454
474
|
};
|
|
455
475
|
|
|
456
476
|
// `tuple[]` / `tuple[N]`: rebuild each element as its own tuple, in the same order
|
|
@@ -17,6 +17,7 @@ import {
|
|
|
17
17
|
type UnionToTuple,
|
|
18
18
|
type ContractReturnType,
|
|
19
19
|
extractArrayParameterType,
|
|
20
|
+
tupleComponentsAreNamed,
|
|
20
21
|
} from './utils';
|
|
21
22
|
|
|
22
23
|
type FirstCofheReturnValue<T> = [T] extends [void]
|
|
@@ -240,20 +241,28 @@ export function transformEncryptedReturnTypes<TAbi extends Abi, TFunctionName ex
|
|
|
240
241
|
const components = param.components;
|
|
241
242
|
|
|
242
243
|
const processTuple = (tupleValue: unknown): unknown => {
|
|
243
|
-
|
|
244
|
-
|
|
244
|
+
if (tupleComponentsAreNamed(components)) {
|
|
245
|
+
const valueObj = tupleValue as Record<string, unknown>;
|
|
246
|
+
const result: Record<string, unknown> = {};
|
|
245
247
|
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
248
|
+
components.forEach((component) => {
|
|
249
|
+
const componentName = component.name;
|
|
250
|
+
if (componentName) {
|
|
251
|
+
const componentValue = valueObj[componentName];
|
|
252
|
+
if (componentValue !== undefined) {
|
|
253
|
+
result[componentName] = processParameter(component, componentValue);
|
|
254
|
+
}
|
|
252
255
|
}
|
|
253
|
-
}
|
|
254
|
-
});
|
|
256
|
+
});
|
|
255
257
|
|
|
256
|
-
|
|
258
|
+
return result;
|
|
259
|
+
}
|
|
260
|
+
|
|
261
|
+
if (!Array.isArray(tupleValue)) {
|
|
262
|
+
throw new Error('Expected an array value for tuple with unnamed components');
|
|
263
|
+
}
|
|
264
|
+
|
|
265
|
+
return components.map((component, index) => processParameter(component, tupleValue[index]));
|
|
257
266
|
};
|
|
258
267
|
|
|
259
268
|
// `tuple[]` / `tuple[N]`: transform each element as its own tuple. Treating the
|
package/src/utils.ts
CHANGED
|
@@ -186,6 +186,15 @@ export function extractArrayParameterType<T extends string | undefined>(type: T)
|
|
|
186
186
|
return [head as T, size === '' ? '' : size];
|
|
187
187
|
}
|
|
188
188
|
|
|
189
|
+
/**
|
|
190
|
+
* Matches the tuple-shape rule used by `CofheAbiComponentsToPrimitiveType`:
|
|
191
|
+
* tuples are objects only when every component has a name. Otherwise they are
|
|
192
|
+
* represented positionally as arrays.
|
|
193
|
+
*/
|
|
194
|
+
export function tupleComponentsAreNamed(components: readonly AbiParameter[]): boolean {
|
|
195
|
+
return components.length > 0 && components.every((component) => component.name != null && component.name !== '');
|
|
196
|
+
}
|
|
197
|
+
|
|
189
198
|
type AbiItem = AbiConstructor | AbiError | AbiEvent | AbiFallback | AbiFunction | AbiReceive;
|
|
190
199
|
|
|
191
200
|
export function isAbiFunction(item: AbiItem | unknown): item is AbiFunction {
|