86 lines
2.4 KiB
TypeScript
86 lines
2.4 KiB
TypeScript
import { PublicKey, Connection } from '@solana/web3.js'
|
|
import { TokenInstructions } from '@project-serum/serum'
|
|
import { toUiDecimals } from '@blockworks-foundation/mango-v4'
|
|
import { STAKEABLE_TOKENS_DATA } from './constants'
|
|
|
|
export class TokenAccount {
|
|
publicKey!: PublicKey
|
|
mint!: PublicKey
|
|
owner!: PublicKey
|
|
amount!: number
|
|
decimals!: number
|
|
uiAmount: number
|
|
|
|
constructor(
|
|
publicKey: PublicKey,
|
|
decoded: {
|
|
mint: PublicKey
|
|
owner: PublicKey
|
|
amount: number
|
|
decimals: number
|
|
uiAmount: number
|
|
},
|
|
) {
|
|
this.publicKey = publicKey
|
|
this.uiAmount = 0
|
|
Object.assign(this, decoded)
|
|
}
|
|
}
|
|
|
|
export function exists<T>(item: T | null | undefined): item is T {
|
|
return !!item
|
|
}
|
|
|
|
export async function getTokenAccountsByOwnerWithWrappedSol(
|
|
connection: Connection,
|
|
owner: PublicKey,
|
|
): Promise<TokenAccount[]> {
|
|
const solReq = connection.getAccountInfo(owner)
|
|
const tokenReq = connection.getParsedTokenAccountsByOwner(owner, {
|
|
programId: TokenInstructions.TOKEN_PROGRAM_ID,
|
|
})
|
|
|
|
// fetch data
|
|
const [solResp, tokenResp] = await Promise.all([solReq, tokenReq])
|
|
|
|
// parse token accounts
|
|
const tokenAccounts = tokenResp.value.map((t) => {
|
|
return {
|
|
publicKey: t.pubkey,
|
|
mint: t.account.data.parsed.info.mint,
|
|
owner: t.account.data.parsed.info.owner,
|
|
amount: t.account.data.parsed.info.tokenAmount.amount,
|
|
uiAmount: t.account.data.parsed.info.tokenAmount.uiAmount,
|
|
decimals: t.account.data.parsed.info.tokenAmount.decimals,
|
|
}
|
|
})
|
|
// create fake wrapped sol account to reflect sol balances in user's wallet
|
|
const lamports = solResp?.lamports || 0
|
|
const solAccount = new TokenAccount(owner, {
|
|
mint: TokenInstructions.WRAPPED_SOL_MINT,
|
|
owner,
|
|
amount: lamports,
|
|
uiAmount: toUiDecimals(lamports, 9),
|
|
decimals: 9,
|
|
})
|
|
|
|
// prepend SOL account to beginning of list
|
|
return [solAccount].concat(tokenAccounts)
|
|
}
|
|
|
|
export const formatTokenSymbol = (symbol: string) => {
|
|
if (symbol.toLowerCase().includes('portal')) {
|
|
const truncSymbol = symbol.split(' ')[0].toUpperCase()
|
|
return truncSymbol === 'WBTC' ? 'wBTC' : truncSymbol
|
|
}
|
|
return symbol === 'MSOL' ? 'mSOL' : symbol
|
|
}
|
|
|
|
export const getStakableTokensDataForMint = (mintPk: string) => {
|
|
return STAKEABLE_TOKENS_DATA.find((x) => x.mint_address === mintPk)!
|
|
}
|
|
|
|
export const getStakableTokensDataForTokenName = (tokenName: string) => {
|
|
return STAKEABLE_TOKENS_DATA.find((x) => x.name === tokenName)!
|
|
}
|