• Index
• Search by APIs
• Search by Words
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• amman-master
  • .prettierrc
  • LICENSE
  • sh
    • update-docs
  • amman-client
    • src
      • diagnostics
        • api.ts
        • token-balances.ts
        • token-registry.ts
        • address-labels.ts
        • address-label-mapper.ts
      • utils
        • keypair.ts
        • api.ts
        • keys.ts
        • browser.ts
        • address.ts
        • log.ts
      • transactions
        • api.ts
        • types.ts
        • transaction-checker.ts
        • confirmed-transaction-assertable-promise.ts
        • consts.ts
        • transaction-handler.ts
      • storage
        • sdk-types.ts
        • api.ts
        • mock-storage-driver.ts
        • consts.ts
      • api.ts
      • types.ts
      • asserts
        • api.ts
        • asserts.ts
      • consts.ts
      • amman-client.ts
      • relay
        • api.ts
        • timeout.ts
        • types.ts
        • client.ts
        • consts.ts
      • assets
        • persistence.ts
        • api.ts
        • account-data-serializer.ts
    • typedoc.json
    • README.md
    • package.json
    • tsconfig.json
  • README.md
  • package.json
  • amman
    • src
      • diagnostics
        • programs
          • program-names.ts
          • log-parser.ts
          • index.ts
          • program-err.ts
      • utils
        • config.ts
        • guards.ts
        • index.ts
        • path.ts
        • fs.ts
        • log.ts
        • http.ts
      • storage
        • index.ts
        • types.ts
        • mock-server.ts
      • cli
        • utils
          • solana-logs.ts
          • resolvers.ts
          • index.ts
          • connection.ts
        • commands
          • run.ts
          • airdrop.ts
          • logs.ts
          • label.ts
          • snapshot.ts
          • index.ts
          • account.ts
          • start.ts
        • amman.ts
      • validator
        • index.ts
        • solana-validator.ts
        • process-accounts.ts
        • types.ts
        • ensure-validator-up.ts
        • init-validator.ts
        • prepare-config.ts
        • process-snapshot.ts
      • types.ts
      • accounts
        • providers.ts
        • state.ts
        • types.ts
      • amman.ts
      • relay
        • server.ts
        • types.ts
      • assets
        • persistence.ts
        • local-accounts.ts
        • index.ts
        • types.ts
        • consts.ts
    • typedoc.json
    • README.md
    • package.json
    • tsconfig.json
  • tsconfig.json
  • .gitignore

import { isValidPublicKeyAddress } from '@metaplex-foundation/amman-client'
import { PublicKey } from '@solana/web3.js'
import { spawnSync } from 'child_process'
import path from 'path'
import { logError, logInfo } from '../utils'
import { canAccess, ensureDirSync } from '../utils/fs'
import { Account } from '../validator/types'

export async function getExecutableAddress(programId: string): Promise<string> {
  const programPubkey = new PublicKey(programId)
  const [executableAddress] = await PublicKey.findProgramAddress(
    [programPubkey.toBytes()],
    new PublicKey('BPFLoaderUpgradeab1e11111111111111111111111')
  )
  return executableAddress.toString()
}

export async function saveAccount(
  accountId: string,
  endpoint: string,
  accountsFolder: string,
  executable = false
) {
  const makeRemainingArgs = (id: string) => [
    '-u',
    endpoint,
    '-o',
    `${accountsFolder}/${id}.json`,
    '--output',
    'json',
  ]
  logInfo(`Saving account ${accountId} from cluster ${endpoint}`)
  spawnSync('solana', ['account', accountId, ...makeRemainingArgs(accountId)])
  if (executable) {
    logInfo(`Saving executable data for ${accountId} from cluster ${endpoint}`)
    const executableId = await getExecutableAddress(accountId)
    spawnSync('solana', [
      'account',
      executableId,
      ...makeRemainingArgs(executableId),
    ])
  }
}

export async function handleFetchAccounts(
  accountsCluster: string,
  accounts: Account[],
  accountsFolder: string,
  force = false
) {
  if (accounts.length > 0) {
    ensureDirSync(accountsFolder)
    for (const { accountId, cluster, executable } of accounts) {
      if (accountId == null || !isValidPublicKeyAddress(accountId)) {
        throw new Error(
          `Account ID ${accountId} in accounts array from validator config is invalid`
        )
      }
      if (
        force ||
        !(await canAccess(path.join(accountsFolder, `${accountId}.json`)))
      ) {
        const endpoint = cluster ?? accountsCluster
        try {
          await saveAccount(accountId, endpoint, accountsFolder, executable)
        } catch (err) {
          logError(`Failed to load ${accountId} from cluster ${endpoint}`)
          throw err
        }
      }
    }
  }
}