mango-ui-v3/utils/tokens.ts

import { ACCOUNT_LAYOUT } from '@blockworks-foundation/mango-client'
import { Connection, PublicKey } from '@solana/web3.js'
import * as bs58 from 'bs58'
import { TokenInstructions } from '@project-serum/serum'
import { WRAPPED_SOL_MINT } from '@project-serum/serum/lib/token-instructions'

export const TOKEN_PROGRAM_ID = new PublicKey(
  'TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA'
)

export type ProgramAccount<T> = {
  publicKey: PublicKey
  account: T
}

export function parseTokenAccountData(data: Buffer): {
  mint: PublicKey
  owner: PublicKey
  amount: number
} {
  const { mint, owner, amount } = ACCOUNT_LAYOUT.decode(data)
  return {
    mint: new PublicKey(mint),
    owner: new PublicKey(owner),
    amount,
  }
}

// export async function getWalletTokenInfo(
//   connection: Connection,
//   ownerPublicKey: PublicKey
// ) {
//   const splAccounts = await getOwnedTokenAccounts(connection, ownerPublicKey)
//   const account = await connection.getAccountInfo(ownerPublicKey)
//   if (!account) return splAccounts
//   return [
//     {
//       publicKey: ownerPublicKey,
//       account: {
//         mint: WRAPPED_SOL_MINT,
//         owner: ownerPublicKey,
//         amount: account.lamports,
//       },
//     },
//   ].concat(splAccounts)
// }

// export async function getOwnedTokenAccounts(
//   connection: Connection,
//   publicKey: PublicKey
// ): Promise<any[]> {
//   const filters = getOwnedAccountsFilters(publicKey)
//   // @ts-ignore
//   const resp = await connection._rpcRequest('getProgramAccounts', [
//     TokenInstructions.TOKEN_PROGRAM_ID.toBase58(),
//     {
//       commitment: connection.commitment,
//       filters,
//     },
//   ])
//   if (resp.error) {
//     throw new Error(
//       'failed to get token accounts owned by ' +
//         publicKey.toBase58() +
//         ': ' +
//         resp.error.message
//     )
//   }
//   return resp.result.map(({ pubkey, account: { data } }) => {
//     data = bs58.decode(data)
//     return {
//       publicKey: new PublicKey(pubkey),
//       account: parseTokenAccountData(data),
//     }
//   })
// }

// export function getOwnedAccountsFilters(publicKey: PublicKey) {
//   return [
//     {
//       memcmp: {
//         offset: ACCOUNT_LAYOUT.offsetOf('owner'),
//         bytes: publicKey.toBase58(),
//       },
//     },
//     {
//       dataSize: ACCOUNT_LAYOUT.span,
//     },
//   ]
// }