@exodus/solana-api 1.3.3 → 1.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exodus/solana-api",
-  "version": "1.3.3",
+  "version": "1.4.0",
   "description": "Exodus internal Solana asset API wrapper",
   "main": "src/index.js",
   "files": [
@@ -14,7 +14,7 @@
   },
   "dependencies": {
     "@exodus/asset-json-rpc": "^1.0.0",
-    "@exodus/solana-lib": "^1.2.29",
+    "@exodus/solana-lib": "^1.3.0",
     "lodash": "^4.17.11",
     "url-join": "4.0.0",
     "wretch": "^1.5.2"
@@ -22,5 +22,5 @@
   "devDependencies": {
     "node-fetch": "~1.6.3"
   },
-  "gitHead": "69134d5324df0b3977a84b39e1098876a5ebffe7"
+  "gitHead": "158aae6b4aeb871b1566b2500456a5f475ef9d5f"
 }

package/src/index.js

@@ -3,10 +3,14 @@ import createApi from '@exodus/asset-json-rpc'
 import {
   getMetadataAccount,
   deserializeMetaplexMetadata,
+  getTransactionSimulationParams,
+  filterAccountsByOwner,
   tokens,
   SYSTEM_PROGRAM_ID,
   STAKE_PROGRAM_ID,
   TOKEN_PROGRAM_ID,
+  SOL_DECIMAL,
+  computeBalance,
 } from '@exodus/solana-lib'
 import assert from 'assert'
 import lodash from 'lodash'
@@ -409,6 +413,18 @@ class Api {
     }
   }
 
+  // Returns account balance of a SPL Token account.
+  async getTokenBalance(tokenAddress: string) {
+    const {
+      value: { amount },
+    } = await this.api.post({
+      method: 'getTokenAccountBalance',
+      params: [tokenAddress],
+    })
+
+    return amount
+  }
+
   async getAccountInfo(address: string) {
     const { value } = await this.api.post({
       method: 'getAccountInfo',
@@ -560,6 +576,91 @@ class Api {
     console.log(`tx ${JSON.stringify(result)} sent!`)
     return result || null
   }
+
+  simulateTransaction = async (transaction, options) => {
+    const {
+      value: { accounts },
+    } = await this.api.post({
+      method: 'simulateTransaction',
+      params: [transaction.serialize({ verifySignatures: false }).toString('base64'), options],
+    })
+
+    return accounts
+  }
+
+  resolveSimulationSideEffects = async (solAccounts, tokenAccounts) => {
+    const willReceive = []
+    const willSend = []
+
+    const resolveSols = solAccounts.map(async (account) => {
+      const currentAmount = await this.getBalance(account.address)
+      const balance = computeBalance(account.amount, currentAmount)
+      return {
+        name: 'SOL',
+        symbol: 'SOL',
+        balance,
+        decimal: SOL_DECIMAL,
+        type: 'SOL',
+      }
+    })
+    const _getTokenBalance = async (address) => {
+      try {
+        return await this.getTokenBalance(address)
+      } catch (error) {
+        if (error.message && error.message.includes('could not find account')) {
+          return '0'
+        }
+        throw error
+      }
+    }
+
+    const resolveTokens = tokenAccounts.map(async (account) => {
+      const [_tokenMetaPlex, currentAmount, decimal] = await Promise.all([
+        this.getMetaplexMetadata(account.mint),
+        _getTokenBalance(account.address),
+        this.getDecimals(account.mint),
+      ])
+
+      const balance = computeBalance(account.amount, currentAmount)
+      const tokenMetaPlex = _tokenMetaPlex || { name: null, symbol: null }
+      return {
+        name: tokenMetaPlex.name,
+        symbol: tokenMetaPlex.symbol,
+        balance,
+        decimal,
+        type: 'TOKEN',
+      }
+    })
+
+    const accounts = await Promise.all([...resolveSols, ...resolveTokens])
+    accounts.forEach((account) => {
+      if (account.balance > 0) {
+        willReceive.push(account)
+      } else {
+        willSend.push(account)
+      }
+    })
+
+    return {
+      willReceive,
+      willSend,
+    }
+  }
+
+  /**
+   * Simulate transaction and return side effects
+   */
+  simulateAndRetrieveSideEffects = async (transaction, publicKey: string) => {
+    const { config, accountAddresses } = getTransactionSimulationParams(transaction)
+    const futureAccountsState = await this.simulateTransaction(transaction, config)
+    const { solAccounts, tokenAccounts } = filterAccountsByOwner(
+      futureAccountsState,
+      accountAddresses,
+      publicKey
+    )
+
+    return this.resolveSimulationSideEffects(solAccounts, tokenAccounts)
+  }
 }
 
 function isTokenSupported(mint: string) {