@wireio/stake 1.0.1 → 2.0.0

package/src/networks/solana/utils.ts

@@ -4,53 +4,26 @@ import {
     Connection,
     Keypair,
     PublicKey,
-    SystemProgram,
-    StakeProgram,
 } from '@solana/web3.js';
 import {
     TOKEN_2022_PROGRAM_ID,
-    ASSOCIATED_TOKEN_PROGRAM_ID,
     getAssociatedTokenAddress,
 } from '@solana/spl-token';
 
 import {
-    LIQSOL_CORE,
-    LIQSOL_TOKEN,
-    PAY_RATE_SCALE_FACTOR,
     DEFAULT_AVERAGE_PAY_RATE,
     EPHEMERAL_RENT_EXEMPTION,
-    LAMPORTS_PER_SOL,
-    lamportsToSol,
     solToLamports,
-    deriveDepositAuthorityPda,
-    deriveLiqsolMintPda,
-    deriveLiqsolMintAuthorityPda,
-    deriveVaultPda,
-    deriveReservePoolPda,
-    deriveStakeControllerStatePda,
-    derivePayoutStatePda,
-    deriveBucketAuthorityPda,
-    deriveDistributionStatePda,
-    deriveUserRecordPda,
-    derivePayRateHistoryPda,
-    deriveStakeControllerVaultPda,
-    deriveEphemeralStakeAddress,
     DEFAULT_PAY_RATE_LOOKBACK,
-    deriveOutpostGlobalStatePda,
-    deriveOutpostPoolAuthorityPda,
-    deriveOutpostAccountPda,
-    deriveTrancheStatePda,
-    deriveUserPretokenRecordPda,
     CHAINLINK_FEED,
     CHAINLINK_PROGRAM,
-    deriveExtraAccountMetaListPda,
 } from './constants';
 
-import liqsolCoreIDL from '../../assets/solana/idl/liqsol_core.json';
-import { LiqsolCore } from '../../assets/solana/types/liqsol_core';
+import { LiqsolCore } from '../../assets/solana/devnet/types/liqsol_core';
 import { GlobalState, TrancheState } from './types';
 import { TrancheLadderItem, TrancheSnapshot } from '../../types';
 import { ChainID } from '@wireio/core';
+import { SolanaProgramService } from './program';
 
 // -----------------------------------------------------------------------------
 // Tranche Support
@@ -268,265 +241,6 @@ export function buildSolanaTrancheSnapshot(options: {
     };
 }
 
-
-// -----------------------------------------------------------------------------
-// Read-only liqsol_core Program helper
-// -----------------------------------------------------------------------------
-
-let _liqsolCoreProgram: Program<LiqsolCore> | null = null;
-
-export function getLiqsolCoreProgram(
-    connection: Connection,
-): Program<LiqsolCore> {
-    if (_liqsolCoreProgram && _liqsolCoreProgram.provider.connection === connection) {
-        return _liqsolCoreProgram;
-    }
-
-    // Dummy wallet – we're only doing read-only account fetches here.
-    const tmpKeypair = Keypair.generate();
-    const wallet: any = {
-        publicKey: tmpKeypair.publicKey,
-        signAllTransactions: async () => [],
-        signTransaction: async () => tmpKeypair,
-    };
-
-    const provider = new AnchorProvider(connection, wallet, {
-        commitment: 'confirmed',
-    });
-
-    const program = new Program(
-        liqsolCoreIDL,
-        provider,
-    ) as Program<LiqsolCore>;
-
-    _liqsolCoreProgram = program;
-    return program;
-}
-
-
-// -----------------------------------------------------------------------------
-// Balance + state getters (UI-friendly)
-// -----------------------------------------------------------------------------
-
-export async function getUserLiqSolBalance(
-    connection: Connection,
-    user: PublicKey,
-): Promise<number> {
-    const liqsolMint = deriveLiqsolMintPda();
-    const ata = await getAssociatedTokenAddress(
-        liqsolMint,
-        user,
-        false,
-        TOKEN_2022_PROGRAM_ID,
-    );
-
-    try {
-        const bal = await connection.getTokenAccountBalance(ata);
-        return Number(bal.value.amount); // raw lamports, not decimals-adjusted
-    } catch {
-        return 0;
-    }
-}
-
-export async function getBucketLiqSolBalance(
-    connection: Connection,
-): Promise<number> {
-    const liqsolMint = deriveLiqsolMintPda();
-    const bucketAuthority = deriveBucketAuthorityPda();
-    const ata = await getAssociatedTokenAddress(
-        liqsolMint,
-        bucketAuthority,
-        true,
-        TOKEN_2022_PROGRAM_ID,
-    );
-
-    try {
-        const bal = await connection.getTokenAccountBalance(ata);
-        return Number(bal.value.amount);
-    } catch {
-        return 0;
-    }
-}
-
-export async function getReservePoolBalance(
-    connection: Connection,
-): Promise<number> {
-    const reservePool = deriveReservePoolPda();
-    try {
-        const lamports = await connection.getBalance(reservePool);
-        return lamports;
-    } catch {
-        return 0;
-    }
-}
-
-/**
- * Raw account info for the stake_controller state (Anchor decode is left
- * to the caller so the SDK can stay IDL-agnostic at this layer).
- */
-export async function getStakeControllerStateRaw(
-    connection: Connection,
-): Promise<Uint8Array | null> {
-    const pda = deriveStakeControllerStatePda();
-    const info = await connection.getAccountInfo(pda);
-    return info?.data ?? null;
-}
-
-/**
- * Raw account info for the payout-state account.
- */
-export async function getPayoutStateRaw(
-    connection: Connection,
-): Promise<Uint8Array | null> {
-    const pda = derivePayoutStatePda();
-    const info = await connection.getAccountInfo(pda);
-    return info?.data ?? null;
-}
-
-/**
- * Raw account info for a user's distribution user_record.
- *
- * NOTE: user_record is keyed by the TOKEN ACCOUNT (ATA), not the wallet.
- * Here we derive the user's liqSOL ATA and then its user_record PDA.
- */
-export async function getUserRecordRaw(
-    connection: Connection,
-    user: PublicKey,
-): Promise<Uint8Array | null> {
-    const liqsolMint = deriveLiqsolMintPda();
-    const userAta = await getAssociatedTokenAddress(
-        liqsolMint,
-        user,
-        false,
-        TOKEN_2022_PROGRAM_ID,
-    );
-    const pda = deriveUserRecordPda(userAta);
-    const info = await connection.getAccountInfo(pda);
-    return info?.data ?? null;
-}
-
-
-// -----------------------------------------------------------------------------
-// Pay-rate & fee utilities (on-chain compatible)
-// -----------------------------------------------------------------------------
-
-export async function getAveragePayRate(
-    connection: Connection,
-    lookback: number = DEFAULT_PAY_RATE_LOOKBACK,
-): Promise<bigint> {
-    const program = getLiqsolCoreProgram(connection);
-    const payRateHistoryPda = derivePayRateHistoryPda();
-
-    try {
-        const anyProgram = program;
-        const payRateHistoryAccount = await anyProgram.account.payRateHistory.fetch(
-            payRateHistoryPda,
-        );
-
-        const entries = payRateHistoryAccount.entries ?? [];
-        const totalEntriesAdded = Number(
-            payRateHistoryAccount.totalEntriesAdded ?? 0,
-        );
-        const currentIndex: number = payRateHistoryAccount.currentIndex ?? 0;
-        const maxEntries: number =
-            payRateHistoryAccount.maxEntries ?? entries.length;
-
-        if (!entries.length) {
-            return DEFAULT_AVERAGE_PAY_RATE;
-        }
-
-        const entriesToFetch = Math.min(lookback, maxEntries, entries.length);
-
-        let idx: number;
-        if (totalEntriesAdded === 0) {
-            idx = 0;
-        } else if (currentIndex === 0) {
-            idx = maxEntries - 1;
-        } else {
-            idx = currentIndex - 1;
-        }
-
-        let sum = BigInt(0);
-        let validCount = BigInt(0);
-
-        for (let i = 0; i < entriesToFetch; i++) {
-            const entry = entries[idx];
-            if (entry && typeof entry.scaledRate !== 'undefined') {
-                const rate = BigInt(entry.scaledRate.toString());
-                if (rate > BigInt(0)) {
-                    sum += rate;
-                    validCount += BigInt(1);
-                }
-            }
-
-            if (totalEntriesAdded === 0) {
-                idx = (idx + 1) % maxEntries;
-            } else {
-                idx = idx === 0 ? maxEntries - 1 : idx - 1;
-            }
-        }
-
-        if (validCount === BigInt(0)) {
-            return DEFAULT_AVERAGE_PAY_RATE;
-        }
-
-        return sum / validCount;
-    } catch (err) {
-        // Fallback to default when history missing/unavailable
-        return DEFAULT_AVERAGE_PAY_RATE;
-    }
-}
-
-/**
- * On-chain fee formula:
- *   fee = (average_pay_rate * 4 * deposit_amount_lamports) / 10^12
- */
-export function calculateExpectedFee(
-    depositAmountLamports: BN,
-    averagePayRate: BN,
-): BN {
-    return averagePayRate
-        .mul(new BN(4))
-        .mul(depositAmountLamports)
-        .div(new BN(1_000_000_000_000)); // 10^12
-}
-
-/**
- * Convenience helper to preview how a deposit splits between user + bucket
- * and how much goes into the reserve, assuming the simple model:
- *
- * - userLiqSol = amount - fee + EPHEMERAL_RENT_EXEMPTION
- * - bucketLiqSol = fee
- * - reserveLamports = amount + EPHEMERAL_RENT_EXEMPTION
- */
-export function previewDepositEffects(params: {
-    depositAmountLamports: BN;
-    averagePayRate: BN;
-    rentExemptionLamports?: number;
-}): {
-    feeLamports: BN;
-    userLiqSolLamports: BN;
-    bucketLiqSolLamports: BN;
-    reserveIncreaseLamports: BN;
-} {
-    const { depositAmountLamports, averagePayRate } = params;
-    const rent = new BN(
-        params.rentExemptionLamports ?? EPHEMERAL_RENT_EXEMPTION,
-    );
-
-    const fee = calculateExpectedFee(depositAmountLamports, averagePayRate);
-    const userLiqSol = depositAmountLamports.sub(fee).add(rent);
-    const bucketLiqSol = fee;
-    const reserveIncrease = depositAmountLamports.add(rent);
-
-    return {
-        feeLamports: fee,
-        userLiqSolLamports: userLiqSol,
-        bucketLiqSolLamports: bucketLiqSol,
-        reserveIncreaseLamports: reserveIncrease,
-    };
-}
-
 /**
  * ---------------------------------------------------------------------------
  * Outpost-specific helpers
@@ -569,17 +283,18 @@ export interface OutpostAccounts {
 export async function buildOutpostAccounts(
     connection: Connection,
     user: PublicKey,
+    pgs: SolanaProgramService,
 ): Promise<OutpostAccounts> {
-    const globalState = deriveOutpostGlobalStatePda();
-    const poolAuthority = deriveOutpostPoolAuthorityPda();
-    const liqsolMint = deriveLiqsolMintPda();
-    const distributionState = deriveDistributionStatePda();
-    const bucketAuthority = deriveBucketAuthorityPda();
-    const trancheState = deriveTrancheStatePda();
+    const globalState = pgs.deriveOutpostGlobalStatePda();
+    const poolAuthority = pgs.deriveOutpostPoolAuthorityPda();
+    const liqsolMint = pgs.deriveLiqsolMintPda();
+    const distributionState = pgs.deriveDistributionStatePda();
+    const bucketAuthority = pgs.deriveBucketAuthorityPda();
+    const trancheState = pgs.deriveTrancheStatePda();
 
     // Outpost / pretoken PDAs
-    const outpostAccount = deriveOutpostAccountPda(user);
-    const userPretokenRecord = deriveUserPretokenRecordPda(user);
+    const outpostAccount = pgs.deriveOutpostAccountPda(user);
+    const userPretokenRecord = pgs.deriveUserPretokenRecordPda(user);
 
     // ATAs
     const liqsolPoolAta = await getAssociatedTokenAddress(
@@ -604,18 +319,18 @@ export async function buildOutpostAccounts(
     );
 
     // user_record PDAs keyed by TOKEN ACCOUNTS (matches capital-staking)
-    const bucketUserRecord = deriveUserRecordPda(bucketTokenAccount);
-    const liqsolPoolUserRecord = deriveUserRecordPda(liqsolPoolAta);
-    const userUserRecord = deriveUserRecordPda(userAta);
+    const bucketUserRecord = pgs.deriveUserRecordPda(bucketTokenAccount);
+    const liqsolPoolUserRecord = pgs.deriveUserRecordPda(liqsolPoolAta);
+    const userUserRecord = pgs.deriveUserRecordPda(userAta);
 
-    const extraAccountMetaList = deriveExtraAccountMetaListPda(liqsolMint);
+    const extraAccountMetaList = pgs.deriveExtraAccountMetaListPda(liqsolMint);
 
     // Chainlink program feeds
     let chainLinkFeed = CHAINLINK_FEED;
     let chainLinkProgram = CHAINLINK_PROGRAM;
 
     try {
-        const program = getLiqsolCoreProgram(connection);
+        const program = pgs.getProgram('liqsolCore');
         const ts: TrancheState =
             await program.account.trancheState.fetch(trancheState);
 

package/src/staker.ts

@@ -37,10 +37,12 @@ export class Staker {
         config.forEach((cfg) => {
             switch (cfg.network.chainId) {
                 case SolChainID.Mainnet:
+                case SolChainID.Devnet:
                     this.clients.set(cfg.network.chainId, new SolanaStakingClient(cfg));
                     break;
 
                 case EvmChainID.Ethereum:
+                case EvmChainID.Hoodi:
                     this.clients.set(cfg.network.chainId, new EthereumStakingClient(cfg));
                     break;
 
@@ -52,13 +54,20 @@ export class Staker {
 
         this.selectedChainID = selectedChainID;
     }
-    
+
     /**
      * Select a chain. Returns true if a client exists for it, false otherwise.
      * (We still record the selectedChainID so the UI can reflect the chosen chain.)
      */
     setChain(chainID: ChainID): boolean {
+        const hasClient = this.clients.has(chainID);
+
+        if (!hasClient) {
+            console.warn(`[Staker] setChain: no client configured for chain ${chainID}`);
+            return false;
+        }
+
         this.selectedChainID = chainID;
-        return this.clients.has(chainID);
+        return true;
     }
 }

package/src/types.ts

@@ -1,12 +1,29 @@
 import { BaseSignerWalletAdapter } from '@solana/wallet-adapter-base';
 import { PublicKey as SolPubKey } from '@solana/web3.js';
-import { ChainID, ExternalNetwork, PublicKey } from '@wireio/core';
+import { ChainID, EvmChainID, ExternalNetwork, PublicKey, SolChainID } from '@wireio/core';
 import { ethers } from 'ethers';
 
 export type StakerConfig = {
     network: ExternalNetwork;
     provider?: BaseSignerWalletAdapter | ethers.providers.Web3Provider;
     pubKey?: PublicKey;
+    extras?: {
+        squadsX?: SquadsXConfig;
+    }
+}
+
+export type SquadsXConfig = {
+    multisigPDA: string;        // REQUIRED, base58 multisig address string
+    vaultIndex?: number;        // default 0
+};
+
+export enum SupportedEvmChainID {
+    Ethereum = EvmChainID.Ethereum,
+    Hoodi = EvmChainID.Hoodi,
+}
+export enum SupportedSolChainID {
+    Mainnet = SolChainID.Mainnet,
+    Devnet = SolChainID.Devnet,
 }
 
 export interface IStakingClient {
@@ -33,11 +50,7 @@ export interface IStakingClient {
      *  - `TrancheSnapshot` when the chain supports pretoken/tranches
      *  - `null` if this chain has no WIRE/pretoken integration
      */
-    getTrancheSnapshot(options?: {
-        chainID?: ChainID;
-        windowBefore?: number;
-        windowAfter?: number;
-    }): Promise<TrancheSnapshot | null>;
+    getTrancheSnapshot(chainId: ChainID): Promise<TrancheSnapshot | null>;
 
     /**
      * Protocol fee charged for deposit from Native to LIQ