Add network config code

Author: ericglau

packages/core/zama/src/confidentialFungible.test.ts

@@ -12,6 +12,7 @@ function testConfidentialFungible(title: string, opts: Partial<ConfidentialFungi
       name: 'MyToken',
       symbol: 'MTK',
       tokenURI: '',
+      networkConfig: 'zama-sepolia',
       ...opts,
     });
     t.snapshot(printContract(c));
@@ -30,6 +31,7 @@ function testAPIEquivalence(title: string, opts?: ConfidentialFungibleOptions) {
           name: 'MyToken',
           symbol: 'MTK',
           tokenURI: '',
+          networkConfig: 'zama-sepolia',
           ...opts,
         }),
       ),
@@ -57,6 +59,7 @@ function testPremint(scenario: string, premint: string, expectedError?: string)
           name: 'MyToken',
           symbol: 'MTK',
           tokenURI: '',
+          networkConfig: 'zama-sepolia',
           premint,
         }),
       );
@@ -66,6 +69,7 @@ function testPremint(scenario: string, premint: string, expectedError?: string)
         name: 'MyToken',
         symbol: 'MTK',
         tokenURI: '',
+        networkConfig: 'zama-sepolia',
         premint,
       });
       t.snapshot(printContract(c));
@@ -136,6 +140,7 @@ testAPIEquivalence('confidentialFungible API full upgradeable', {
   name: 'CustomToken',
   symbol: 'CTK',
   tokenURI: 'http://example.com',
+  networkConfig: 'zama-ethereum',
   premint: '2000',
   access: 'roles',
   wrappable: true,

packages/core/zama/src/confidentialFungible.ts

@@ -9,13 +9,14 @@ import { clockModeDefault, setClockMode } from './set-clock-mode';
 import { OptionsError } from './error';
 import { toUint256, UINT256_MAX } from './utils/convert-strings';
 
-export const crossChainBridgingOptions = [false, 'custom', 'superchain'] as const;
-export type CrossChainBridging = (typeof crossChainBridgingOptions)[number];
+export const networkConfigOptions = ['zama-sepolia', 'zama-ethereum'] as const;
+export type NetworkConfig = (typeof networkConfigOptions)[number];
 
 export interface ConfidentialFungibleOptions extends CommonOptions {
   name: string;
   symbol: string;
   tokenURI: string;
+  networkConfig: NetworkConfig;
   premint?: string;
   wrappable?: boolean;
   /**
@@ -29,6 +30,7 @@ export const defaults: Required<ConfidentialFungibleOptions> = {
   name: 'MyToken',
   symbol: 'MTK',
   tokenURI: '',
+  networkConfig: 'zama-sepolia',
   premint: '0',
   wrappable: false,
   votes: false,
@@ -59,6 +61,7 @@ export function buildConfidentialFungible(opts: ConfidentialFungibleOptions): Co
   const { info } = allOpts;
 
   addBase(c, allOpts.name, allOpts.symbol, allOpts.tokenURI);
+  addNetworkConfig(c, allOpts.networkConfig);
 
   if (allOpts.premint) {
     addPremint(c, allOpts.premint);
@@ -98,6 +101,20 @@ function addBase(c: ContractBuilder, name: string, symbol: string, tokenURI: str
   c.addOverride(ConfidentialFungibleToken, functions.decimals);
 }
 
+function addNetworkConfig(c: ContractBuilder, network: NetworkConfig) {
+  if (network === 'zama-sepolia') {
+    c.addParent({
+      name: 'SepoliaConfig',
+      path: '@fhevm/solidity/config/ZamaConfig.sol',
+    });
+  } else if (network === 'zama-ethereum') {
+    c.addParent({
+      name: 'EthereumConfig',
+      path: '@fhevm/solidity/config/ZamaConfig.sol',
+    });
+  }
+}
+
 export const premintPattern = /^(\d*)(?:\.(\d+))?(?:e(\d+))?$/;
 
 function scaleByPowerOfTen(base: bigint, exponent: number): bigint {

packages/core/zama/src/get-imports.test.ts

@@ -6,7 +6,7 @@ import { generateSources } from './generate/sources';
 import { buildGeneric } from './build-generic';
 
 test('confidential fungible basic', t => {
-  const c = buildConfidentialFungible({ name: 'MyToken', symbol: 'MTK', tokenURI: '' });
+  const c = buildConfidentialFungible({ name: 'MyToken', symbol: 'MTK', tokenURI: '', networkConfig: 'zama-sepolia' });
   const sources = getImports(c);
   const files = Object.keys(sources).sort();
 

packages/ui/src/zama/ConfidentialFungibleControls.svelte

@@ -47,14 +47,17 @@
   </label>
 
   <div class="labeled-input">
-    <span>Network</span>
+    <span class="flex justify-between pr-2">
+      Network Configuration
+      <HelpTooltip>Adds configuration to use FHEVM contracts provided by Zama on the specified network.</HelpTooltip>
+    </span>
     <div class="checkbox-group">
-      <label class:checked={opts.network === 'sepolia'}>
-        <input type="radio" bind:group={opts.network} value="sepolia" />
+      <label class:checked={opts.networkConfig === 'zama-sepolia'}>
+        <input type="radio" bind:group={opts.networkConfig} value="zama-sepolia" />
         Sepolia
       </label>
-      <label class:checked={opts.network === 'ethereum'}>
-        <input type="radio" bind:group={opts.network} value="ethereum" />
+      <label class:checked={opts.networkConfig === 'zama-ethereum'}>
+        <input type="radio" bind:group={opts.networkConfig} value="zama-ethereum" />
         Ethereum
       </label>
     </div>