go

Author: keyvankhademi

apps/staking/src/api.ts

@@ -435,7 +435,7 @@ export const reassignPublisherAccount = async (
   targetAccount: PublicKey,
   publisherKey: PublicKey,
 ): Promise<void> => {
-  return client.setPublisherStakeAccount(
+  return client.reassignPublisherStakeAccount(
     publisherKey,
     stakeAccount,
     targetAccount,
@@ -447,7 +447,7 @@ export const optPublisherOut = async (
   stakeAccount: PublicKey,
   publisherKey: PublicKey,
 ): Promise<void> => {
-  await client.setPublisherStakeAccount(publisherKey, stakeAccount, undefined);
+  await client.removePublisherStakeAccount(stakeAccount, publisherKey);
 };
 
 const MOCK_DELAY = 500;

governance/pyth_staking_sdk/src/pyth-staking-client.ts

@@ -644,7 +644,7 @@ export class PythStakingClient {
     };
   }
 
-  public async setPublisherStakeAccount(
+  async setPublisherStakeAccount(
     publisher: PublicKey,
     stakeAccountPositions: PublicKey,
     newStakeAccountPositions: PublicKey | undefined,
@@ -661,4 +661,27 @@ export class PythStakingClient {
     await sendTransaction([instruction], this.connection, this.wallet);
     return;
   }
+
+  public async reassignPublisherStakeAccount(
+    publisher: PublicKey,
+    stakeAccountPositions: PublicKey,
+    newStakeAccountPositions: PublicKey,
+  ) {
+    return this.setPublisherStakeAccount(
+      publisher,
+      stakeAccountPositions,
+      newStakeAccountPositions,
+    );
+  }
+
+  public async removePublisherStakeAccount(
+    publisher: PublicKey,
+    stakeAccountPositions: PublicKey,
+  ) {
+    return this.setPublisherStakeAccount(
+      publisher,
+      stakeAccountPositions,
+      undefined,
+    );
+  }
 }