adds bitcoin xpub support

src/__test__/e2e/xpub.test.ts

@@ -0,0 +1,24 @@
+/* eslint-disable quotes */
+import { question } from 'readline-sync';
+import { fetchAddressesByDerivationPath, pair } from '../../api';
+import { setupClient } from '../utils/setup';
+import { LatticeGetAddressesFlag } from '../../protocol';
+
+describe('XPUB', () => {
+  test('pair', async () => {
+    const isPaired = await setupClient();
+    if (!isPaired) {
+      const secret = question('Please enter the pairing secret: ');
+      console.log('secret', secret);
+      await pair(secret.toUpperCase());
+    }
+  });
+
+    test('fetch bitcoin xpub', async () => {
+    const xpub = await fetchAddressesByDerivationPath("44'/0'/0'", {
+      flag: LatticeGetAddressesFlag.secp256k1Xpub,
+    });
+    expect(xpub).toHaveLength(1);
+    expect(xpub[0]).toMatch(/^xpub/); // XPUB should start with "xpub"
+    });
+});

src/api/addresses.ts

@@ -23,6 +23,7 @@ import {
 type FetchAddressesParams = {
   n?: number;
   startPathIndex?: number;
+  flag?: number;
 };
 
 export const fetchAddresses = async (overrides?: GetAddressesRequestParams) => {
@@ -195,11 +196,11 @@ export const fetchBip44ChangeAddresses = async ({
 
 export async function fetchAddressesByDerivationPath(
   path: string,
-  { n = 1, startPathIndex = 0 }: FetchAddressesParams = {},
+  { n = 1, startPathIndex = 0, flag }: FetchAddressesParams = {},
 ): Promise<string[]> {
   const components = path.split('/').filter(Boolean);
   const parsedPath = parseDerivationPathComponents(components);
-  const flag = getFlagFromPath(parsedPath);
+  const _flag = getFlagFromPath(parsedPath);
   const wildcardIndex = components.findIndex((part) =>
     part.toLowerCase().includes('x'),
   );
@@ -208,7 +209,7 @@ export async function fetchAddressesByDerivationPath(
     return queue((client) =>
       client.getAddresses({
         startPath: parsedPath,
-        flag,
+        flag: flag || _flag,
         n,
       }),
     );
@@ -223,7 +224,7 @@ export async function fetchAddressesByDerivationPath(
     const result = await queue((client) =>
       client.getAddresses({
         startPath: currentPath,
-        flag,
+        flag: flag || _flag,
         n: 1,
       }),
     );

src/constants.ts

@@ -18,6 +18,7 @@ export const EXTERNAL = {
     SECP256K1_PUB: LatticeGetAddressesFlag.secp256k1Pubkey,
     ED25519_PUB: LatticeGetAddressesFlag.ed25519Pubkey,
     BLS12_381_G1_PUB: LatticeGetAddressesFlag.bls12_381Pubkey,
+    SECP256K1_XPUB: LatticeGetAddressesFlag.secp256k1Xpub,
   },
   // Options for building general signing requests
   SIGNING: {

src/functions/getAddresses.ts

@@ -102,7 +102,8 @@ export const encodeGetAddressesRequest = ({
     (flag === LatticeGetAddressesFlag.ed25519Pubkey ||
       flag === LatticeGetAddressesFlag.secp256k1Pubkey ||
       flag === LatticeGetAddressesFlag.bls12_381Pubkey);
-  if (!isPubkeyOnly && !isValidAssetPath(startPath, fwConstants)) {
+  const isXpub = flag === LatticeGetAddressesFlag.secp256k1Xpub;
+  if (!isPubkeyOnly && !isXpub && !isValidAssetPath(startPath, fwConstants)) {
     throw new Error(
       'Derivation path or flag is not supported. Try updating Lattice firmware.',
     );
@@ -191,7 +192,7 @@ export const decodeGetAddressesResponse = (
       }
       off += 65;
     } else {
-      // Otherwise we are dealing with address strings
+      // Otherwise we are dealing with address strings or XPUB strings
       const addrBytes = data.slice(off, off + ProtocolConstants.addrStrLen);
       off += ProtocolConstants.addrStrLen;
       // Return the UTF-8 representation

src/protocol/latticeConstants.ts

@@ -48,6 +48,7 @@ export enum LatticeGetAddressesFlag {
   secp256k1Pubkey = 3,
   ed25519Pubkey = 4,
   bls12_381Pubkey = 5,
+  secp256k1Xpub = 6, // For Bitcoin XPUB
 }
 
 export enum LatticeSignSchema {