@cheny56/node-client 1.0.9 → 1.0.11

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cheny56/node-client",
-  "version": "1.0.9",
+  "version": "1.0.11",
   "description": "Client library for Quorum blockchain with Post-Quantum Cryptography (PQC) and Zero-Knowledge Proof (ZK) support",
   "main": "./src/index.js",
   "types": "src/index.d.ts",

package/src/contract.js

@@ -160,6 +160,10 @@ class ERC20Token extends Contract {
      */
     async transferPQC(wallet, provider, to, amount, txOptions = {}) {
         const data = this.encodeFunctionData('transfer', [to, amount]);
+        // Ensure wallet is initialized
+        if (!wallet._initialized) {
+            await wallet._ensureKeyPair();
+        }
         const nonce = await provider.getTransactionCount(wallet.address, 'pending');
         
         const tx = new PQCLegacyTransaction({
@@ -172,7 +176,7 @@ class ERC20Token extends Contract {
             data: data,
         });
 
-        tx.sign(wallet);
+        await tx.sign(wallet);
         const serialized = tx.getHex();
         return await provider.sendRawTransaction(serialized);
     }
@@ -188,7 +192,9 @@ class ERC20Token extends Contract {
      */
     async transferHybrid(wallet, provider, to, amount, txOptions = {}) {
         const data = this.encodeFunctionData('transfer', [to, amount]);
-        const nonce = await provider.getTransactionCount(wallet.address, 'pending');
+        // Ensure wallet is initialized
+        const address = await wallet.getAddress();
+        const nonce = await provider.getTransactionCount(address, 'pending');
         
         const tx = new HybridLegacyTransaction({
             chainId: txOptions.chainId || 1337,
@@ -200,7 +206,7 @@ class ERC20Token extends Contract {
             data: data,
         });
 
-        tx.sign(wallet.ecdsaWallet, wallet.pqcWallet);
+        await tx.sign(wallet.ecdsaWallet, wallet.pqcWallet);
         const serialized = tx.getHex();
         return await provider.sendRawTransaction(serialized);
     }

package/src/transaction.js

@@ -118,30 +118,28 @@ class PQCLegacyTransaction extends BaseTransaction {
     /**
      * Sign with PQC wallet
      * @param {PQCWallet} wallet - PQC wallet
-     * @returns {PQCLegacyTransaction} This transaction
+     * @returns {Promise<PQCLegacyTransaction>} This transaction
      */
-    sign(wallet) {
+    async sign(wallet) {
+        // Ensure wallet is initialized
+        if (!wallet._initialized) {
+            await wallet._ensureKeyPair();
+        }
+        
         const hash = ethers.getBytes(this.getSigningHash());
         this.pqcPubKey = wallet.publicKey;
-        // dilithium-crystals API: sign(secretKey, message) or signWithContext(secretKey, message, ctx)
-        if (dilithium.sign) {
-            this.pqcSig = dilithium.sign(wallet.secretKey, hash);
-        } else if (dilithium.signWithContext) {
-            const ctx = new TextEncoder().encode('ML-DSA-65');
-            this.pqcSig = dilithium.signWithContext(wallet.secretKey, hash, ctx);
-        } else {
-            throw new Error('dilithium-crystals does not expose sign method');
-        }
+        // Use wallet's sign method which handles the async properly
+        this.pqcSig = await wallet.sign(hash);
         this.v = Number(this.chainId) * 2 + 35;
         this.r = '0x0';
         this.s = '0x0';
         return this;
     }
 
-    verify() {
+    async verify() {
         if (!this.pqcSig || !this.pqcPubKey) return false;
         const hash = ethers.getBytes(this.getSigningHash());
-        // dilithium-crystals API: verify(publicKey, message, signature) or verifyWithContext
+        // Use dilithium verify directly
         if (dilithium.verify) {
             return dilithium.verify(this.pqcPubKey, hash, this.pqcSig);
         } else if (dilithium.verifyWithContext) {
@@ -235,9 +233,14 @@ class HybridLegacyTransaction extends BaseTransaction {
      * Sign with both ECDSA and PQC wallets
      * @param {ECDSAWallet} ecdsaWallet - ECDSA wallet
      * @param {PQCWallet} pqcWallet - PQC wallet
-     * @returns {HybridLegacyTransaction} This transaction
+     * @returns {Promise<HybridLegacyTransaction>} This transaction
      */
-    sign(ecdsaWallet, pqcWallet) {
+    async sign(ecdsaWallet, pqcWallet) {
+        // Ensure PQC wallet is initialized
+        if (!pqcWallet._initialized) {
+            await pqcWallet._ensureKeyPair();
+        }
+        
         const hash = this.getSigningHash();
         const hashBytes = ethers.getBytes(hash);
 
@@ -248,16 +251,9 @@ class HybridLegacyTransaction extends BaseTransaction {
         this.r = ecdsaSig.r;
         this.s = ecdsaSig.s;
 
-        // PQC signature
+        // PQC signature - use wallet's sign method
         this.pqcPubKey = pqcWallet.publicKey;
-        if (dilithium.sign) {
-            this.pqcSig = dilithium.sign(pqcWallet.secretKey, hashBytes);
-        } else if (dilithium.signWithContext) {
-            const ctx = new TextEncoder().encode('ML-DSA-65');
-            this.pqcSig = dilithium.signWithContext(pqcWallet.secretKey, hashBytes, ctx);
-        } else {
-            throw new Error('dilithium-crystals does not expose sign method');
-        }
+        this.pqcSig = await pqcWallet.sign(hashBytes);
 
         return this;
     }
@@ -391,21 +387,19 @@ class PQCAccessListTransaction extends BaseTransaction {
         return ethers.keccak256(ethers.hexlify(prefixed));
     }
 
-    sign(wallet) {
+    async sign(wallet) {
+        // Ensure wallet is initialized
+        if (!wallet._initialized) {
+            await wallet._ensureKeyPair();
+        }
+        
         const hash = ethers.getBytes(this.getSigningHash());
         this.pqcPubKey = wallet.publicKey;
-        if (dilithium.sign) {
-            this.pqcSig = dilithium.sign(wallet.secretKey, hash);
-        } else if (dilithium.signWithContext) {
-            const ctx = new TextEncoder().encode('ML-DSA-65');
-            this.pqcSig = dilithium.signWithContext(wallet.secretKey, hash, ctx);
-        } else {
-            throw new Error('dilithium-crystals does not expose sign method');
-        }
+        this.pqcSig = await wallet.sign(hash);
         return this;
     }
 
-    verify() {
+    async verify() {
         if (!this.pqcSig || !this.pqcPubKey) return false;
         const hash = ethers.getBytes(this.getSigningHash());
         if (dilithium.verify) {
@@ -550,21 +544,19 @@ class PQCDynamicFeeTransaction extends BaseTransaction {
         return ethers.keccak256(ethers.hexlify(prefixed));
     }
 
-    sign(wallet) {
+    async sign(wallet) {
+        // Ensure wallet is initialized
+        if (!wallet._initialized) {
+            await wallet._ensureKeyPair();
+        }
+        
         const hash = ethers.getBytes(this.getSigningHash());
         this.pqcPubKey = wallet.publicKey;
-        if (dilithium.sign) {
-            this.pqcSig = dilithium.sign(wallet.secretKey, hash);
-        } else if (dilithium.signWithContext) {
-            const ctx = new TextEncoder().encode('ML-DSA-65');
-            this.pqcSig = dilithium.signWithContext(wallet.secretKey, hash, ctx);
-        } else {
-            throw new Error('dilithium-crystals does not expose sign method');
-        }
+        this.pqcSig = await wallet.sign(hash);
         return this;
     }
 
-    verify() {
+    async verify() {
         if (!this.pqcSig || !this.pqcPubKey) return false;
         const hash = ethers.getBytes(this.getSigningHash());
         if (dilithium.verify) {

package/src/wallet.js

@@ -66,14 +66,36 @@ class PQCWallet {
         if (secretKey && publicKey) {
             this.secretKey = secretKey instanceof Uint8Array ? secretKey : new Uint8Array(secretKey);
             this.publicKey = publicKey instanceof Uint8Array ? publicKey : new Uint8Array(publicKey);
+            this.address = derivePQCAddress(this.publicKey);
+            this._initialized = true;
         } else {
             // Generate new key pair using dilithium-crystals
-            // dilithium-crystals typically uses keygen() or generateKeyPair()
-            const keyPair = dilithium.keygen ? dilithium.keygen() : dilithium.generateKeyPair();
-            this.secretKey = keyPair.secretKey || keyPair.privateKey;
-            this.publicKey = keyPair.publicKey;
+            // dilithium-crystals uses keyPair() which is async and returns { publicKey, secretKey }
+            this.secretKey = null;
+            this.publicKey = null;
+            this.address = null;
+            this._initialized = false;
         }
+    }
+
+    /**
+     * Initialize key pair if not already done (async)
+     * @private
+     */
+    async _ensureKeyPair() {
+        if (this._initialized) {
+            return;
+        }
+        
+        if (!dilithium.keyPair) {
+            throw new Error('dilithium-crystals does not expose keyPair method. Please check the package API.');
+        }
+        
+        const keyPair = await dilithium.keyPair();
+        this.secretKey = keyPair.secretKey;
+        this.publicKey = keyPair.publicKey;
         this.address = derivePQCAddress(this.publicKey);
+        this._initialized = true;
     }
 
     /**
@@ -96,9 +118,14 @@ class PQCWallet {
     /**
      * Sign a message hash
      * @param {Uint8Array} hash - Message hash (32 bytes)
-     * @returns {Uint8Array} Signature
+     * @returns {Promise<Uint8Array>} Signature
      */
-    sign(hash) {
+    async sign(hash) {
+        // Ensure key pair is initialized
+        if (!this._initialized) {
+            await this._ensureKeyPair();
+        }
+        
         // dilithium-crystals API: sign(secretKey, message)
         // Some implementations may require context, but we'll try without first
         if (dilithium.sign) {
@@ -115,9 +142,14 @@ class PQCWallet {
      * Verify a signature
      * @param {Uint8Array} hash - Message hash
      * @param {Uint8Array} signature - Signature to verify
-     * @returns {boolean} True if signature is valid
+     * @returns {Promise<boolean>} True if signature is valid
      */
-    verify(hash, signature) {
+    async verify(hash, signature) {
+        // Ensure key pair is initialized
+        if (!this._initialized) {
+            await this._ensureKeyPair();
+        }
+        
         // dilithium-crystals API: verify(publicKey, message, signature)
         if (dilithium.verify) {
             return dilithium.verify(this.publicKey, hash, signature);
@@ -131,17 +163,23 @@ class PQCWallet {
 
     /**
      * Get public key as hex string
-     * @returns {string} Public key (hex)
+     * @returns {Promise<string>} Public key (hex)
      */
-    getPublicKeyHex() {
+    async getPublicKeyHex() {
+        if (!this._initialized) {
+            await this._ensureKeyPair();
+        }
         return ethers.hexlify(this.publicKey);
     }
 
     /**
      * Get secret key as hex string (use with caution!)
-     * @returns {string} Secret key (hex)
+     * @returns {Promise<string>} Secret key (hex)
      */
-    getSecretKeyHex() {
+    async getSecretKeyHex() {
+        if (!this._initialized) {
+            await this._ensureKeyPair();
+        }
         return ethers.hexlify(this.secretKey);
     }
 }
@@ -156,10 +194,30 @@ class HybridWallet {
         }
         this.ecdsaWallet = ecdsaWallet instanceof ECDSAWallet ? ecdsaWallet : new ECDSAWallet(ecdsaWallet);
         this.pqcWallet = pqcWallet instanceof PQCWallet ? pqcWallet : new PQCWallet();
+        this.address = null; // Will be computed when PQC wallet is initialized
+        this._initialized = false;
+    }
+
+    /**
+     * Get the hybrid address (computed from ECDSA + PQC public keys)
+     * @returns {Promise<string>} Hybrid address
+     */
+    async getAddress() {
+        if (this._initialized && this.address) {
+            return this.address;
+        }
+        
+        // Ensure PQC wallet has keys
+        if (!this.pqcWallet._initialized) {
+            await this.pqcWallet._ensureKeyPair();
+        }
+        
         this.address = deriveHybridAddress(
             ethers.getBytes(this.ecdsaWallet.getPublicKey()),
             this.pqcWallet.publicKey
         );
+        this._initialized = true;
+        return this.address;
     }
 
     /**
@@ -172,7 +230,13 @@ class HybridWallet {
     static fromKeys(ecdsaPrivateKey, pqcSecretKey, pqcPublicKey) {
         const ecdsaWallet = new ECDSAWallet(ecdsaPrivateKey);
         const pqcWallet = new PQCWallet(pqcSecretKey, pqcPublicKey);
-        return new HybridWallet(ecdsaWallet, pqcWallet);
+        const wallet = new HybridWallet(ecdsaWallet, pqcWallet);
+        wallet.address = deriveHybridAddress(
+            ethers.getBytes(ecdsaWallet.getPublicKey()),
+            pqcWallet.publicKey
+        );
+        wallet._initialized = true;
+        return wallet;
     }
 
     /**