@arcium-hq/client 0.11.1 → 0.11.2

package/build/types/pda.d.ts

@@ -5,6 +5,11 @@ import * as anchor from '@anchor-lang/core';
  * @returns Arcium program's public key.
  */
 export declare function getArciumProgramId(): PublicKey;
+/**
+ * Return the public key of the deployed Arcium staking program on Solana.
+ * @returns Arcium staking program's public key.
+ */
+export declare function getArciumStakingProgramId(): PublicKey;
 /**
  * Derive the computation account address for a given cluster and computation offset.
  * @param clusterOffset - Offset of the cluster this computation will be executed by.
@@ -71,6 +76,13 @@ export declare function getRecoveryClusterAccAddress(mxeProgramId: PublicKey): P
  * @returns Derived MXE recovery account public key.
  */
 export declare function getMxeRecoveryAccAddress(originalMxeProgramId: PublicKey): PublicKey;
+/**
+ * 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 declare function getMxeRecoveryStakeStateAccAddress(mxeProgramId: PublicKey): PublicKey;
 /**
  * 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/build/types/pda.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"pda.d.ts","sourceRoot":"","sources":["../../src/pda.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAA6B,MAAM,iBAAiB,CAAC;AACvE,OAAO,KAAK,MAAM,MAAM,mBAAmB,CAAC;AA0B5C;;;GAGG;AACH,wBAAgB,kBAAkB,IAAI,SAAS,CAE9C;AAED;;;;;GAKG;AACH,wBAAgB,wBAAwB,CAAC,aAAa,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,CAAC,EAAE,GAAG,SAAS,CAEvG;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,CAErE;AAED;;;;GAIG;AACH,wBAAgB,0BAA0B,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,CAE3E;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,IAAI,SAAS,CAEhD;AAED;;;GAGG;AACH,wBAAgB,kBAAkB,IAAI,SAAS,CAE9C;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,CAErE;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,UAAU,EAAE,MAAM,GAAG,SAAS,CAElE;AAED;;;;GAIG;AACH,wBAAgB,gBAAgB,CAAC,YAAY,EAAE,SAAS,GAAG,SAAS,CAEnE;AAED;;;;;GAKG;AACH,wBAAgB,oBAAoB,CAAC,YAAY,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,GAAG,SAAS,CAE9F;AAED;;;;GAIG;AACH,wBAAgB,4BAA4B,CAAC,YAAY,EAAE,SAAS,GAAG,SAAS,CAE/E;AAED;;;;GAIG;AACH,wBAAgB,wBAAwB,CAAC,oBAAoB,EAAE,SAAS,GAAG,SAAS,CAEnF;AAED;;;;GAIG;AACH,wBAAgB,yBAAyB,CAAC,UAAU,EAAE,MAAM,GAAG,SAAS,CAEvE;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CAAC,cAAc,EAAE,SAAS,GAAG,SAAS,CAE1E;AAED;;;;;GAKG;AACH,wBAAgB,yBAAyB,CAAC,YAAY,EAAE,SAAS,EAAE,iBAAiB,EAAE,MAAM,CAAC,EAAE,GAAG,SAAS,CAE1G;AAED;;;;;;;GAOG;AACH,wBAAgB,yBAAyB,CAAC,YAAY,EAAE,SAAS,GAAG,SAAS,CAE5E;AAED;;;;;GAKG;AACH,wBAAgB,uBAAuB,CAAC,aAAa,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,GAAG,SAAS,CAEpG;AAED;;;;;GAKG;AACH,wBAAgB,qBAAqB,CAAC,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,EAAE,GAAG,SAAS,CAG9F"}
+{"version":3,"file":"pda.d.ts","sourceRoot":"","sources":["../../src/pda.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAA6B,MAAM,iBAAiB,CAAC;AACvE,OAAO,KAAK,MAAM,MAAM,mBAAmB,CAAC;AA2B5C;;;GAGG;AACH,wBAAgB,kBAAkB,IAAI,SAAS,CAE9C;AAED;;;GAGG;AACH,wBAAgB,yBAAyB,IAAI,SAAS,CAErD;AAED;;;;;GAKG;AACH,wBAAgB,wBAAwB,CAAC,aAAa,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,CAAC,EAAE,GAAG,SAAS,CAEvG;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,CAErE;AAED;;;;GAIG;AACH,wBAAgB,0BAA0B,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,CAE3E;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,IAAI,SAAS,CAEhD;AAED;;;GAGG;AACH,wBAAgB,kBAAkB,IAAI,SAAS,CAE9C;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,CAErE;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,UAAU,EAAE,MAAM,GAAG,SAAS,CAElE;AAED;;;;GAIG;AACH,wBAAgB,gBAAgB,CAAC,YAAY,EAAE,SAAS,GAAG,SAAS,CAEnE;AAED;;;;;GAKG;AACH,wBAAgB,oBAAoB,CAAC,YAAY,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,GAAG,SAAS,CAE9F;AAED;;;;GAIG;AACH,wBAAgB,4BAA4B,CAAC,YAAY,EAAE,SAAS,GAAG,SAAS,CAE/E;AAED;;;;GAIG;AACH,wBAAgB,wBAAwB,CAAC,oBAAoB,EAAE,SAAS,GAAG,SAAS,CAEnF;AAED;;;;;GAKG;AACH,wBAAgB,kCAAkC,CAAC,YAAY,EAAE,SAAS,GAAG,SAAS,CAKrF;AAED;;;;GAIG;AACH,wBAAgB,yBAAyB,CAAC,UAAU,EAAE,MAAM,GAAG,SAAS,CAEvE;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CAAC,cAAc,EAAE,SAAS,GAAG,SAAS,CAE1E;AAED;;;;;GAKG;AACH,wBAAgB,yBAAyB,CAAC,YAAY,EAAE,SAAS,EAAE,iBAAiB,EAAE,MAAM,CAAC,EAAE,GAAG,SAAS,CAE1G;AAED;;;;;;;GAOG;AACH,wBAAgB,yBAAyB,CAAC,YAAY,EAAE,SAAS,GAAG,SAAS,CAE5E;AAED;;;;;GAKG;AACH,wBAAgB,uBAAuB,CAAC,aAAa,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,GAAG,SAAS,CAEpG;AAED;;;;;GAKG;AACH,wBAAgB,qBAAqB,CAAC,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,EAAE,GAAG,SAAS,CAG9F"}

package/build/types/recoveryPeers.d.ts

@@ -1,5 +1,5 @@
 /**
- * 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 declare function assertValidRecoveryPeers(recoveryPeers: readonly number[]): void;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arcium-hq/client",
-  "version": "0.11.1",
+  "version": "0.11.2",
   "description": "Client SDK for interacting with encrypted Solana programs",
   "author": "Arcium",
   "license": "GPL-3.0-only",

package/src/constants.ts

@@ -53,6 +53,11 @@ export const RECOVERY_CLUSTER_ACC_SEED = 'RecoveryClusterAccount';
  * @constant {string}
  */
 export const MXE_RECOVERY_ACC_SEED = 'MxeRecoveryAccount';
+/**
+ * Seed for the staking program's MxeRecoveryStakeState PDA.
+ * @constant {string}
+ */
+export const MXE_RECOVERY_STAKE_STATE_SEED = 'MxeRecoveryStakeState';
 /**
  * Seed for ComputationDefinitionRaw PDA.
  * @constant {string}

package/src/idl/arcium.json

@@ -2,7 +2,7 @@
   "address": "Arcj82pX7HxYKLR92qvgZUAd7vGS1k4hQvAFcPATFdEQ",
   "metadata": {
     "name": "arcium",
-    "version": "0.11.1",
+    "version": "0.11.2",
     "spec": "0.1.0",
     "description": "The Arcium program"
   },

package/src/idl/arcium.ts

@@ -8,7 +8,7 @@ export type Arcium = {
   "address": "Arcj82pX7HxYKLR92qvgZUAd7vGS1k4hQvAFcPATFdEQ",
   "metadata": {
     "name": "arcium",
-    "version": "0.11.1",
+    "version": "0.11.2",
     "spec": "0.1.0",
     "description": "The Arcium program"
   },

package/src/idl/arcium_staking.ts

@@ -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 {