feat: store root account public key

anoma · Oct 31, 2023 · c267c06 · c267c06
1 parent 055524d
commit c267c06
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/apps/extension/src/background/keyring/crypto.ts b/apps/extension/src/background/keyring/crypto.ts
@@ -23,6 +23,7 @@ type CryptoArgs = {
   password: string;
   text: string;
   type: AccountType;
+  publicKey?: string;
 };
 
 export class Crypto {
@@ -38,6 +39,7 @@ export class Crypto {
       password,
       text,
       type,
+      publicKey,
     } = args;
 
     const { params, key, iv, salt } = this.encryptionParams(password);
@@ -54,6 +56,7 @@ export class Crypto {
       path,
       crypto,
       type,
+      publicKey
     };
   }
 

diff --git a/apps/extension/src/background/keyring/keyring.ts b/apps/extension/src/background/keyring/keyring.ts
@@ -235,7 +235,10 @@ export class KeyRing {
       const account = hdWallet.derive(bip44Path);
       const stringPointer = account.private().to_hex();
       const sk = readStringPointer(stringPointer, this._cryptoMemory);
-      const address = new Address(sk).implicit();
+      const address = new Address(sk);
+      const implicitAddress = address.implicit();
+      const publicKey = address.public();
+
       const { chainId } = this;
 
       // Generate unique ID for new parent account:
@@ -253,13 +256,14 @@ export class KeyRing {
       const mnemonicStore = crypto.encrypt({
         id,
         alias,
-        address,
-        owner: address,
+        address: implicitAddress,
+        owner: implicitAddress,
         chainId,
         password,
         path,
         text: phrase,
         type: AccountType.Mnemonic,
+        publicKey,
       });
 
       await this._keyStore.append(mnemonicStore);