mirror of
https://github.com/zama-ai/fhevm-solidity.git
synced 2026-01-10 13:07:56 -05:00
367290f93d
* feat: use latest core contracts, updated lib API, renamed to fhevm and fhe chore: add dummy addresses for CI test: add tests for makePubliclyDecryptable method * chore: update package-lock.json for linux * chore: update package-lock.json * fix: solidity comments (#746) * chore: update oracle version * chore: update fhevm core-contracts version * feat: update fhevm sdk and mocked userDecrypt * chore: fix coverage
60 lines
2.6 KiB
TypeScript
60 lines
2.6 KiB
TypeScript
import { mkdirSync, writeFileSync } from 'fs';
|
|
import path from 'path';
|
|
|
|
import { validateFHETypes, validateOperators } from './common';
|
|
import { generateOverloads } from './generateOverloads';
|
|
import { ALL_OPERATORS } from './operators';
|
|
import { generateSolidityFHELib, generateSolidityFheType, generateSolidityImplLib } from './templates';
|
|
import {
|
|
generateSolidityOverloadTestFiles,
|
|
generateSolidityUnitTestContracts,
|
|
generateTypeScriptTestCode,
|
|
splitOverloadsToShards,
|
|
} from './testgen';
|
|
import { ALL_FHE_TYPES } from './types';
|
|
|
|
/**
|
|
* Generates all necessary files including Solidity contracts and TypeScript test files.
|
|
*
|
|
* This function performs the following steps:
|
|
* 1. Generates FHE types from a JSON file.
|
|
* 2. Validates and processes the list of operators.
|
|
* 3. Generates Solidity source code for FHE and implementation contracts.
|
|
* 4. Splits the generated overloads into multiple shards to avoid exceeding Solidity's contract size limit.
|
|
* 5. Writes the generated Solidity contracts and test files to the appropriate directories.
|
|
* 6. Generates TypeScript test code for the split overloads and writes them to the test directory.
|
|
*
|
|
*/
|
|
function generateAllFiles() {
|
|
const numberOfTestSplits = 12;
|
|
|
|
// Validate the FHE types
|
|
validateFHETypes(ALL_FHE_TYPES);
|
|
// Validate the operators
|
|
validateOperators(ALL_OPERATORS);
|
|
|
|
/// Generate core Solidity contract files.
|
|
writeFileSync('lib/FheType.sol', generateSolidityFheType(ALL_FHE_TYPES));
|
|
writeFileSync('lib/Impl.sol', generateSolidityImplLib(ALL_OPERATORS));
|
|
writeFileSync('lib/FHE.sol', generateSolidityFHELib(ALL_OPERATORS, ALL_FHE_TYPES));
|
|
|
|
// TODO: For now, the testgen only supports automatically generated tests for euintXX.
|
|
/// Generate overloads, split them into shards, and generate Solidity contracts to be used for TypeScript unit test files.
|
|
writeFileSync(
|
|
`${path.resolve(__dirname)}/overloads.json`,
|
|
JSON.stringify(generateOverloads(ALL_FHE_TYPES), (_key, value) =>
|
|
typeof value === 'bigint' ? value.toString() : value,
|
|
),
|
|
);
|
|
const overloadShards = splitOverloadsToShards(generateSolidityOverloadTestFiles(ALL_OPERATORS, ALL_FHE_TYPES));
|
|
mkdirSync('contracts/tests', { recursive: true });
|
|
overloadShards.forEach((os) => {
|
|
writeFileSync(`examples/tests/FHEVMTestSuite${os.shardNumber}.sol`, generateSolidityUnitTestContracts(os));
|
|
});
|
|
|
|
const tsSplits: string[] = generateTypeScriptTestCode(overloadShards, numberOfTestSplits);
|
|
tsSplits.forEach((split, splitIdx) => writeFileSync(`test/fhevmOperations/fhevmOperations${splitIdx + 1}.ts`, split));
|
|
}
|
|
|
|
generateAllFiles();
|