From 75e66a5d4f27589297687b720b6d3a9f267c51f3 Mon Sep 17 00:00:00 2001
From: Charon-Fan <Charon.Dian@gmail.com>
Date: Fri, 26 Jul 2024 10:31:51 +0800
Subject: [PATCH] chore: add stakePub property to CardanoSignDataRequest

---
 packages/ur-registry-cardano/package.json              |  2 +-
 .../ur-registry-cardano/src/CardanoSignDataRequest.ts  | 10 ++++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/packages/ur-registry-cardano/package.json b/packages/ur-registry-cardano/package.json
index 8f78384..c224445 100644
--- a/packages/ur-registry-cardano/package.json
+++ b/packages/ur-registry-cardano/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@keystonehq/bc-ur-registry-cardano",
-  "version": "0.3.7",
+  "version": "0.3.8",
   "description": "bc-ur-registry extension for Cardano",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
diff --git a/packages/ur-registry-cardano/src/CardanoSignDataRequest.ts b/packages/ur-registry-cardano/src/CardanoSignDataRequest.ts
index 57e203f..fcf1a57 100644
--- a/packages/ur-registry-cardano/src/CardanoSignDataRequest.ts
+++ b/packages/ur-registry-cardano/src/CardanoSignDataRequest.ts
@@ -17,11 +17,13 @@ enum Keys {
   derivationPath,
   origin,
   signType,
+  stakePub,
 }
 
 type signRequestProps = {
   requestId?: Buffer;
   signData: Buffer;
+  stakePub: Buffer;
   derivationPath: CryptoKeypath;
   origin?: string;
 };
@@ -31,6 +33,7 @@ export class CardanoSignDataRequest extends RegistryItem {
   private signData: Buffer;
   private derivationPath: CryptoKeypath;
   private origin?: string;
+  private stakePub: Buffer;
 
   getRegistryType = () => ExtendedRegistryTypes.CARDANO_SIGN_DATA_REQUEST;
 
@@ -40,12 +43,14 @@ export class CardanoSignDataRequest extends RegistryItem {
     this.signData = args.signData;
     this.derivationPath = args.derivationPath;
     this.origin = args.origin;
+    this.stakePub = args.stakePub;
   }
 
   public getRequestId = () => this.requestId;
   public getSignData = () => this.signData;
   public getDerivationPath = () => this.derivationPath.getPath();
   public getOrigin = () => this.origin;
+  public getStakePub = () => this.stakePub;
 
   public toDataItem = () => {
     const map: DataItemMap = {};
@@ -65,6 +70,7 @@ export class CardanoSignDataRequest extends RegistryItem {
     const keyPath = this.derivationPath.toDataItem();
     keyPath.setTag(this.derivationPath.getRegistryType().getTag());
     map[Keys.derivationPath] = keyPath;
+    map[Keys.stakePub] = this.stakePub;
 
     return new DataItem(map);
   };
@@ -77,12 +83,14 @@ export class CardanoSignDataRequest extends RegistryItem {
       ? map[Keys.requestId].getData()
       : undefined;
     const origin = map[Keys.origin] ? map[Keys.origin] : undefined;
+    const stakePub = map[Keys.stakePub];
 
     return new CardanoSignDataRequest({
       requestId,
       signData,
       derivationPath,
       origin,
+      stakePub,
     });
   };
 
@@ -95,6 +103,7 @@ export class CardanoSignDataRequest extends RegistryItem {
     signData: string,
     hdPath: string,
     xfp: string,
+    stakePub: string,
     uuidString?: string,
     origin?: string
   ) {
@@ -118,6 +127,7 @@ export class CardanoSignDataRequest extends RegistryItem {
       signData: Buffer.from(signData, "hex"),
       derivationPath: hdpathObject,
       origin: origin || undefined,
+      stakePub: Buffer.from(stakePub, "hex"),
     });
   }
 }