@arcium-hq/client 0.11.0 → 0.11.2

package/src/idl/arcium_staking.ts

@@ -296,7 +296,7 @@ export type ArciumStaking = {
         },
         {
           "name": "mint",
-          "address": "BoVUVSJAhob2cvM5z4JZTdS6VWRSznh3bbzQD1TFjvDP"
+          "address": "ARXwZkNAtzPfdcoqQiduJn8EPv9fKiDfGn2KyggyDrFs"
         },
         {
           "name": "poolAccount",
@@ -1136,7 +1136,7 @@ export type ArciumStaking = {
         },
         {
           "name": "mint",
-          "address": "BoVUVSJAhob2cvM5z4JZTdS6VWRSznh3bbzQD1TFjvDP"
+          "address": "ARXwZkNAtzPfdcoqQiduJn8EPv9fKiDfGn2KyggyDrFs"
         },
         {
           "name": "poolAccount",
@@ -1852,38 +1852,38 @@ export type ArciumStaking = {
               {
                 "kind": "const",
                 "value": [
-                  160,
-                  125,
-                  200,
-                  55,
-                  211,
-                  178,
-                  66,
+                  140,
+                  2,
                   27,
-                  149,
-                  22,
-                  219,
-                  191,
-                  28,
-                  218,
-                  171,
+                  73,
+                  195,
+                  17,
+                  0,
+                  246,
+                  183,
                   113,
-                  92,
-                  216,
-                  236,
-                  165,
-                  124,
-                  20,
-                  89,
-                  205,
-                  119,
-                  106,
-                  175,
-                  166,
-                  185,
-                  155,
-                  69,
-                  242
+                  43,
+                  241,
+                  63,
+                  58,
+                  114,
+                  3,
+                  95,
+                  72,
+                  247,
+                  95,
+                  41,
+                  87,
+                  67,
+                  226,
+                  103,
+                  238,
+                  224,
+                  193,
+                  5,
+                  200,
+                  127,
+                  194
                 ]
               }
             ],
@@ -2233,38 +2233,38 @@ export type ArciumStaking = {
               {
                 "kind": "const",
                 "value": [
-                  160,
-                  125,
-                  200,
-                  55,
-                  211,
-                  178,
-                  66,
+                  140,
+                  2,
                   27,
-                  149,
-                  22,
-                  219,
-                  191,
-                  28,
-                  218,
-                  171,
+                  73,
+                  195,
+                  17,
+                  0,
+                  246,
+                  183,
                   113,
-                  92,
-                  216,
-                  236,
-                  165,
-                  124,
-                  20,
-                  89,
-                  205,
-                  119,
-                  106,
-                  175,
-                  166,
-                  185,
-                  155,
-                  69,
-                  242
+                  43,
+                  241,
+                  63,
+                  58,
+                  114,
+                  3,
+                  95,
+                  72,
+                  247,
+                  95,
+                  41,
+                  87,
+                  67,
+                  226,
+                  103,
+                  238,
+                  224,
+                  193,
+                  5,
+                  200,
+                  127,
+                  194
                 ]
               }
             ],
@@ -2502,38 +2502,38 @@ export type ArciumStaking = {
               {
                 "kind": "const",
                 "value": [
-                  160,
-                  125,
-                  200,
-                  55,
-                  211,
-                  178,
-                  66,
+                  140,
+                  2,
                   27,
-                  149,
-                  22,
-                  219,
-                  191,
-                  28,
-                  218,
-                  171,
+                  73,
+                  195,
+                  17,
+                  0,
+                  246,
+                  183,
                   113,
-                  92,
-                  216,
-                  236,
-                  165,
-                  124,
-                  20,
-                  89,
-                  205,
-                  119,
-                  106,
-                  175,
-                  166,
-                  185,
-                  155,
-                  69,
-                  242
+                  43,
+                  241,
+                  63,
+                  58,
+                  114,
+                  3,
+                  95,
+                  72,
+                  247,
+                  95,
+                  41,
+                  87,
+                  67,
+                  226,
+                  103,
+                  238,
+                  224,
+                  193,
+                  5,
+                  200,
+                  127,
+                  194
                 ]
               }
             ],
@@ -2977,38 +2977,38 @@ export type ArciumStaking = {
               {
                 "kind": "const",
                 "value": [
-                  160,
-                  125,
-                  200,
-                  55,
-                  211,
-                  178,
-                  66,
+                  140,
+                  2,
                   27,
-                  149,
-                  22,
-                  219,
-                  191,
-                  28,
-                  218,
-                  171,
+                  73,
+                  195,
+                  17,
+                  0,
+                  246,
+                  183,
                   113,
-                  92,
-                  216,
-                  236,
-                  165,
-                  124,
-                  20,
-                  89,
-                  205,
-                  119,
-                  106,
-                  175,
-                  166,
-                  185,
-                  155,
-                  69,
-                  242
+                  43,
+                  241,
+                  63,
+                  58,
+                  114,
+                  3,
+                  95,
+                  72,
+                  247,
+                  95,
+                  41,
+                  87,
+                  67,
+                  226,
+                  103,
+                  238,
+                  224,
+                  193,
+                  5,
+                  200,
+                  127,
+                  194
                 ]
               }
             ],
@@ -3225,7 +3225,7 @@ export type ArciumStaking = {
         },
         {
           "name": "mint",
-          "address": "BoVUVSJAhob2cvM5z4JZTdS6VWRSznh3bbzQD1TFjvDP"
+          "address": "ARXwZkNAtzPfdcoqQiduJn8EPv9fKiDfGn2KyggyDrFs"
         },
         {
           "name": "poolAta",
@@ -3500,7 +3500,7 @@ export type ArciumStaking = {
         },
         {
           "name": "mint",
-          "address": "BoVUVSJAhob2cvM5z4JZTdS6VWRSznh3bbzQD1TFjvDP"
+          "address": "ARXwZkNAtzPfdcoqQiduJn8EPv9fKiDfGn2KyggyDrFs"
         },
         {
           "name": "poolAccount",
@@ -3868,7 +3868,7 @@ export type ArciumStaking = {
         },
         {
           "name": "mint",
-          "address": "BoVUVSJAhob2cvM5z4JZTdS6VWRSznh3bbzQD1TFjvDP"
+          "address": "ARXwZkNAtzPfdcoqQiduJn8EPv9fKiDfGn2KyggyDrFs"
         },
         {
           "name": "poolAccount",
@@ -4398,7 +4398,7 @@ export type ArciumStaking = {
         },
         {
           "name": "mint",
-          "address": "BoVUVSJAhob2cvM5z4JZTdS6VWRSznh3bbzQD1TFjvDP"
+          "address": "ARXwZkNAtzPfdcoqQiduJn8EPv9fKiDfGn2KyggyDrFs"
         },
         {
           "name": "poolAccount",
@@ -4588,6 +4588,26 @@ export type ArciumStaking = {
         {
           "name": "tokenProgram",
           "address": "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA"
+        },
+        {
+          "name": "arxNode",
+          "docs": [
+            "The `ArxNode` this stake is bound to. Required (and validated against the stored binding)",
+            "on withdrawals when the stake's `binding` is `Node(..)`, so we can read the node's",
+            "cluster and enforce that cluster's minimum primary stake. May be omitted for deposits,",
+            "unbound stakes, or recovery-peer bindings. The `Account` wrapper enforces it's owned by",
+            "the arcium program."
+          ],
+          "optional": true
+        },
+        {
+          "name": "mempool",
+          "docs": [
+            "The mempool account of the cluster the bound node is seated in (or proposed to). Required",
+            "alongside `arx_node` when that node belongs to a cluster. Only its discriminator (the",
+            "mempool size) is read."
+          ],
+          "optional": true
         }
       ],
       "args": [
@@ -5477,6 +5497,26 @@ export type ArciumStaking = {
       "code": 6034,
       "name": "delegatedRemovalNotMatured",
       "msg": "Delegated stake removal has not matured on the primary yet (finalize the primary first)"
+    },
+    {
+      "code": 6035,
+      "name": "insufficientPrimaryStake",
+      "msg": "Resulting primary stake is below the minimum for the bound node's cluster mempool size"
+    },
+    {
+      "code": 6036,
+      "name": "missingNodeAccount",
+      "msg": "Stake is bound to a node; the bound ArxNode account must be provided"
+    },
+    {
+      "code": 6037,
+      "name": "missingMempoolAccount",
+      "msg": "Bound node is seated in a cluster; the cluster's mempool account must be provided"
+    },
+    {
+      "code": 6038,
+      "name": "invalidMempool",
+      "msg": "Provided mempool account does not match the bound node's cluster"
     }
   ],
   "types": [

package/src/idl/index.ts

@@ -1,20 +1,33 @@
 import { Arcium } from './arcium.js';
 import * as ARCIUM_IDL_MODULE from './arcium.json' with { type: 'json' };
+import { ArciumStaking } from './arcium_staking.js';
+import * as ARCIUM_STAKING_IDL_MODULE from './arcium_staking.json' with { type: 'json' };
 
 // Handle both ESM (ts-node wraps as {default: ...}) and bundled (direct) imports
 const ARCIUM_IDL = ((ARCIUM_IDL_MODULE as { default?: Arcium }).default ?? ARCIUM_IDL_MODULE) as unknown as Arcium;
+const ARCIUM_STAKING_IDL = ((ARCIUM_STAKING_IDL_MODULE as { default?: ArciumStaking }).default ?? ARCIUM_STAKING_IDL_MODULE) as unknown as ArciumStaking;
 
 /**
  * The deployed address of the Arcium program, as specified in the IDL.
  */
 export const ARCIUM_ADDR = ARCIUM_IDL.address;
 
+/**
+ * The deployed address of the Arcium staking program, as specified in the IDL.
+ */
+export const ARCIUM_STAKING_ADDR = ARCIUM_STAKING_IDL.address;
+
 /**
  * The Anchor-generated IDL type for the Arcium program.
  */
 export type { Arcium as ArciumIdlType };
 
 /**
- * The Anchor-generated IDL JSON object for the Arcium program.
+ * The Anchor-generated IDL type for the Arcium staking program.
+ */
+export type { ArciumStaking as ArciumStakingIdlType };
+
+/**
+ * The Anchor-generated IDL JSON objects for the Arcium programs.
  */
-export { ARCIUM_IDL };
+export { ARCIUM_IDL, ARCIUM_STAKING_IDL };

package/src/onchain.ts

@@ -1,9 +1,10 @@
 import anchorPkg, { Program, AnchorProvider } from '@anchor-lang/core';
 import * as anchor from '@anchor-lang/core';
 import type { AccountInfo, ConfirmOptions, PublicKey } from '@solana/web3.js';
-import { ARCIUM_IDL, ArciumIdlType } from './idl/index.js';
+import { ARCIUM_IDL, ARCIUM_STAKING_IDL, ArciumIdlType } from './idl/index.js';
 import { sha256 } from './cryptography/cryptography.js';
 import { Arcium } from './idl/arcium.js';
+import { ArciumStaking } from './idl/arcium_staking.js';
 import {
     COMP_DEF_OFFSET_SIZE,
     DEFAULT_UPLOAD_TX_BATCH_SIZE,
@@ -18,6 +19,7 @@ import {
     getComputationAccAddress,
     getCompDefAccAddress,
     getFeePoolAccAddress,
+    getMxeRecoveryStakeStateAccAddress,
     getRawCircuitAccAddress,
     getLookupTableAddress,
 } from './pda.js';
@@ -607,6 +609,15 @@ export function getArciumProgram(provider: AnchorProvider): Program<Arcium> {
     return new Program(ARCIUM_IDL as unknown as Arcium, provider);
 }
 
+/**
+ * Return an Anchor program instance for the Arcium staking program.
+ * @param provider - Anchor provider to use.
+ * @returns Anchor program instance for the Arcium staking program.
+ */
+export function getArciumStakingProgram(provider: AnchorProvider): Program<ArciumStaking> {
+    return new Program(ARCIUM_STAKING_IDL as unknown as ArciumStaking, provider);
+}
+
 /**
  * Return the public key and offset for a computation definition account, given the circuit name and MXE program ID.
  * @param circuitName - Name of the circuit.
@@ -695,6 +706,21 @@ export async function initMxePart2(
             addressLookupTable: getLookupTableAddress(mxeProgramId, lutOffset),
         })
         .transaction();
+
+    // The Arcium program requires the staking program's initMxeRecoveryStakeState to immediately
+    // follow initMxePart2 in the same transaction (enforced by instruction introspection), so the
+    // MXE always ships with the recovery-stake state its fee-drain path depends on.
+    const recoveryStakeStateIx = await getArciumStakingProgram(provider)
+        .methods.initMxeRecoveryStakeState(mxeProgramId)
+        .accountsPartial({
+            payer: provider.publicKey,
+            mxeRecoveryStakeState: getMxeRecoveryStakeStateAccAddress(mxeProgramId),
+            instructionsSysvar: anchor.web3.SYSVAR_INSTRUCTIONS_PUBKEY,
+            systemProgram: anchor.web3.SystemProgram.programId,
+        })
+        .instruction();
+    tx.add(recoveryStakeStateIx);
+
     return signAndSendWithBlockhash(provider, tx, await provider.connection.getLatestBlockhash({ commitment: confirmOptions?.commitment || 'confirmed' }), confirmOptions);
 }
 

package/src/pda.ts

@@ -12,6 +12,7 @@ import {
     MEMPOOL_ACC_SEED,
     MXE_ACCOUNT_SEED,
     MXE_RECOVERY_ACC_SEED,
+    MXE_RECOVERY_STAKE_STATE_SEED,
     OPERATOR_ACC_SEED,
     OFFSET_BUFFER_SIZE,
     POOL_ACC_SEED,
@@ -19,7 +20,7 @@ import {
     RECOVERY_CLUSTER_ACC_SEED,
     RECOVERY_PEER_ACC_SEED,
 } from './constants.js';
-import { ARCIUM_ADDR } from './idl/index.js';
+import { ARCIUM_ADDR, ARCIUM_STAKING_ADDR } from './idl/index.js';
 
 const U64_SEED_SIZE = 8;
 const MAX_U8 = 0xff;
@@ -33,6 +34,14 @@ export function getArciumProgramId(): PublicKey {
     return new anchor.web3.PublicKey(ARCIUM_ADDR);
 }
 
+/**
+ * Return the public key of the deployed Arcium staking program on Solana.
+ * @returns Arcium staking program's public key.
+ */
+export function getArciumStakingProgramId(): PublicKey {
+    return new anchor.web3.PublicKey(ARCIUM_STAKING_ADDR);
+}
+
 /**
  * Derive the computation account address for a given cluster and computation offset.
  * @param clusterOffset - Offset of the cluster this computation will be executed by.
@@ -132,6 +141,19 @@ export function getMxeRecoveryAccAddress(originalMxeProgramId: PublicKey): Publi
     return findArciumPda([stringSeed(MXE_RECOVERY_ACC_SEED), pubkeySeed(originalMxeProgramId)]);
 }
 
+/**
+ * Derive the staking program's MxeRecoveryStakeState PDA for an MXE program.
+ * Derived under the staking program, not the Arcium program.
+ * @param mxeProgramId - Public key of the MXE program.
+ * @returns Derived MxeRecoveryStakeState account public key.
+ */
+export function getMxeRecoveryStakeStateAccAddress(mxeProgramId: PublicKey): PublicKey {
+    return findPda(getArciumStakingProgramId(), [
+        stringSeed(MXE_RECOVERY_STAKE_STATE_SEED),
+        pubkeySeed(mxeProgramId),
+    ]);
+}
+
 /**
  * Derive the recovery peer account address for a given recovery peer offset.
  * @param peerOffset - Recovery peer offset as a u32 in `[0, 2^32 - 1]`.

package/src/recoveryPeers.ts

@@ -3,7 +3,7 @@ import { MAX_RECOVERY_PEERS, MIN_RECOVERY_PEERS } from './constants.js';
 const MAX_U32 = 0xffffffff;
 
 /**
- * Mirrors the on-chain `init_mxe_part2` recovery-peer bounds before building the instruction.
+ * Mirrors the on-chain `init_mxe_part1` recovery-peer bounds before building the instruction.
  * @internal
  */
 export function assertValidRecoveryPeers(recoveryPeers: readonly number[]): void {