@exodus/solana-lib 1.2.28 → 1.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exodus/solana-lib",
-  "version": "1.2.28",
+  "version": "1.3.0",
   "description": "Exodus internal Solana low-level library",
   "main": "src/index.js",
   "files": [
@@ -24,5 +24,6 @@
     "create-hash": "^1.1.3",
     "lodash": "^4.17.11",
     "tweetnacl": "^1.0.3"
-  }
+  },
+  "gitHead": "158aae6b4aeb871b1566b2500456a5f475ef9d5f"
 }

package/src/constants.js

@@ -14,3 +14,7 @@ export const MAGIC_EDEN_ESCROW_PROGRAM_ID = new PublicKey(
 )
 
 export const SEED = 'stake:0'
+
+export const LAMPORTS_PER_SOL = 1000000000
+
+export const SOL_DECIMAL = Math.log10(LAMPORTS_PER_SOL)

package/src/helpers/spl-token.js

@@ -43,6 +43,20 @@ export class U64 extends BN {
 
 const u64 = U64 // alias
 
+/**
+ * Layout for a public key
+ */
+const publicKey = (property = 'publicKey') => {
+  return BufferLayout.blob(32, property)
+}
+
+/**
+ * Layout for a 64bit unsigned value
+ */
+const uint64 = (property = 'uint64') => {
+  return BufferLayout.blob(8, property)
+}
+
 /**
  * An ERC20-like Token
  */
@@ -105,4 +119,20 @@ export class Token {
       data,
     })
   }
+
+  static decode(data: string) {
+    return BufferLayout.struct([
+      publicKey('mint'),
+      publicKey('owner'),
+      uint64('amount'),
+      BufferLayout.u32('delegateOption'),
+      publicKey('delegate'),
+      BufferLayout.u8('state'),
+      BufferLayout.u32('isNativeOption'),
+      uint64('isNative'),
+      uint64('delegatedAmount'),
+      BufferLayout.u32('closeAuthorityOption'),
+      publicKey('closeAuthority'),
+    ]).decode(data)
+  }
 }

package/src/index.js

@@ -4,6 +4,6 @@ export * from './constants'
 export * from './encode'
 export * from './keypair'
 export * from './tx'
-export { StakeInstruction, PublicKey } from './vendor'
+export { StakeInstruction, PublicKey, Transaction as SolanaWeb3Transaction, Message as SolanaWeb3Message } from './vendor'
 export { default as tokens } from './tokens'
 export { default as Transaction } from './transaction'

package/src/tx/index.js

@@ -2,3 +2,4 @@ export * from './sign-unsigned-tx'
 export * from './parse-unsigned-tx'
 export * from './create-unsigned-tx'
 export * from './create-and-sign-tx'
+export * from './simulate-and-sign-tx'

package/src/tx/simulate-and-sign-tx.js

@@ -0,0 +1,77 @@
+import { Token, U64 } from '../helpers/spl-token'
+import { PublicKey } from '../vendor/'
+
+export function computeBalance(futureAccountBalance: string, currentAccountBalance: string) {
+  return new U64(futureAccountBalance).sub(new U64(currentAccountBalance))
+}
+
+export function getTransactionSimulationParams(transaction) {
+  const config = {
+    encoding: 'base64',
+    commitment: 'confirmed',
+  }
+
+  const accountAddresses = transaction
+    .compileMessage()
+    .accountKeys.map((account) => account.toString())
+  config['accounts'] = {
+    encoding: 'base64',
+    addresses: accountAddresses,
+  }
+
+  return {
+    config,
+    accountAddresses,
+  }
+}
+
+const isSolAccount = (account) => account === '1111111111111111'
+
+export function filterAccountsByOwner(futureAccountsState, accountAddresses, publicKey) {
+  const solAccounts = []
+  const tokenAccounts = []
+
+  // Shouldn't happen
+  if (futureAccountsState.length !== accountAddresses.length) {
+    throw Error('Simulation returning wrong account length')
+  }
+
+  futureAccountsState.forEach((futureAccount, index) => {
+    try {
+      const accountAddress = accountAddresses[index]
+
+      // Check if it's SOL account (not token)
+      if (isSolAccount(futureAccount.owner.toString())) {
+        if (accountAddress.toString() === publicKey.toString()) {
+          solAccounts.push({
+            amount: futureAccount.lamports,
+            address: publicKey,
+          })
+        }
+        return
+      }
+
+      const data = Buffer.from(futureAccount.data[0], 'base64')
+      const token = Token.decode(data)
+      const owner = new PublicKey(token.owner).toString()
+      const amount = U64.fromBuffer(token.amount)
+
+      // Get tokens by owner using the public key
+      if (owner === publicKey.toString()) {
+        tokenAccounts.push({
+          amount,
+          mint: new PublicKey(token.mint).toString(),
+          owner: owner,
+          address: accountAddress.toString(),
+        })
+      }
+    } catch (error) {
+      console.warn(error)
+    }
+  })
+
+  return {
+    solAccounts,
+    tokenAccounts,
+  }
+}