@ocap/resolver 1.18.150 → 1.18.152

package/lib/index.js

@@ -190,8 +190,6 @@ module.exports = class OCAPResolver {
       }
     }
 
-    this.buildMigrationChain();
-
     this.executor = createExecutor({
       filter,
       runAsLambda: typeof statedb.runAsLambda === 'function' ? statedb.runAsLambda.bind(statedb) : null,
@@ -331,12 +329,13 @@ module.exports = class OCAPResolver {
     return JSON.stringify(this.config);
   }
 
-  getAccountState({ address, traceMigration = true }, ctx) {
+  getAccountState({ address, traceMigration = true, expandContext = true }, ctx) {
     return this._getState({
       table: 'account',
       id: address,
       dataKey: 'data',
       extraParams: { traceMigration },
+      expandContext,
       onRead: async (state) => {
         if (state) {
           if (state.tokens) {
@@ -965,7 +964,9 @@ module.exports = class OCAPResolver {
       tx.receipts = getTxReceipts(tx, { config: this.config });
     }
 
-    if (get(tx, 'tx.itxJson.type_url') === 'fg:t:revoke_withdraw') {
+    const typeUrl = get(tx, 'tx.itxJson.type_url');
+
+    if (typeUrl === 'fg:t:revoke_withdraw' || typeUrl === 'fg:t:approve_withdraw') {
       const withdrawHash = get(tx, 'tx.itxJson.withdraw_tx_hash');
       // Avoid infinite loop by invalid data
       if (withdrawHash && withdrawHash !== tx.hash) {
@@ -975,10 +976,25 @@ module.exports = class OCAPResolver {
     }
 
     // https://github.com/blocklet/abt-wallet/issues/681
-    if (get(tx, 'tx.itxJson.type_url') === 'fg:t:deposit_token') {
+    if (typeUrl === 'fg:t:deposit_token' || typeUrl === 'fg:t:withdraw_token') {
       tx.receipts = getTxReceipts(tx, { config: this.config });
     }
 
+    if (typeUrl === 'fg:t:account_migrate') {
+      const fromState = await this.getAccountState({
+        address: tx.tx.from,
+        traceMigration: false,
+        expandContext: false,
+      });
+      if (fromState) {
+        tx.receipts = getTxReceipts(tx, { config: this.config, fromState });
+      }
+    }
+
+    if (typeUrl === 'fg:t:declare') {
+      tx.receipts = getTxReceipts(tx, { config: this.config, time: tx.time });
+    }
+
     tx.receipts = mergeTxReceipts(tx.receipts);
     tx.tokenSymbols = await this.getTxTokenSymbols(tx);
 
@@ -1153,6 +1169,49 @@ module.exports = class OCAPResolver {
     migrationChain.buildChains(migrationTxs);
     this.migrationChain = migrationChain;
   }
+
+  async getMigrationChain() {
+    if (!this.migrationChain) {
+      await this.buildMigrationChain();
+    }
+
+    return this.migrationChain;
+  }
+
+  /** fix accounts field of legacy approve_withdraw transactions in indexdb */
+  async fixApproveWithdrawAccounts() {
+    const approveWithdrawTxs = await this._getAllResults('transactions', (paging) =>
+      this.listTransactions({
+        paging,
+        typeFilter: { types: ['approve_withdraw'] },
+        validityFilter: { validity: ['VALID'] },
+      })
+    );
+
+    const updatedTxs = await Promise.all(
+      approveWithdrawTxs.map(async (tx) => {
+        const withdrawHash = tx.tx?.itxJson?.withdraw_tx_hash;
+
+        if (!withdrawHash) return;
+
+        const indexdbTx = await this.indexdb.tx.get(tx.hash);
+        const withdrawTx = await this.getTx({ hash: withdrawHash });
+        const from = withdrawTx?.tx?.from;
+
+        if (!from || !indexdbTx) return;
+
+        const accounts = uniq((indexdbTx.accounts || []).concat([from]).filter(Boolean));
+        debug('fixApproveWithdrawAccounts', { tx, from, accounts });
+
+        // update indexdb
+        const updatedTx = await this.indexdb.tx.update(tx.hash, { accounts });
+        // eslint-disable-next-line consistent-return
+        return updatedTx;
+      })
+    );
+
+    return updatedTxs.filter(Boolean);
+  }
 };
 
 module.exports.formatData = formatData;

package/lib/token-flow.js

@@ -4,6 +4,7 @@ const { BN, fromTokenToUnit } = require('@ocap/util');
 const { schemas, Joi } = require('@arcblock/validator');
 const { CustomError } = require('@ocap/util/lib/error');
 const uniq = require('lodash/uniq');
+const { FORGE_TOKEN_HOLDER } = require('@ocap/state/lib/states/tx');
 const debug = require('debug')(require('../package.json').name);
 
 const ZERO = new BN(0);
@@ -122,11 +123,11 @@ const getTransferFlow = (tx, tokenAddress) => {
 };
 
 const getVaultAccounts = (config) => {
-  return Object.values(config.vaults).flat();
+  return Object.values(config.vaults).flat().concat(FORGE_TOKEN_HOLDER);
 };
 
 const fixMigrateReceipts = async ({ accountAddress, tx }, resolver) => {
-  const { migrationChain } = resolver;
+  const migrationChain = await resolver.getMigrationChain();
   const address = migrationChain.findAddressAtTime(accountAddress, new Date(tx.time));
   const migrations = migrationChain.getMigrationHistory(accountAddress);
 

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.18.150",
+  "version": "1.18.152",
   "description": "GraphQL resolver built upon ocap statedb and GQL layer",
   "main": "lib/index.js",
   "files": [
@@ -22,18 +22,18 @@
     "jest": "^29.7.0"
   },
   "dependencies": {
-    "@arcblock/did": "1.18.150",
-    "@arcblock/did-util": "1.18.150",
-    "@arcblock/validator": "1.18.150",
-    "@ocap/config": "1.18.150",
-    "@ocap/indexdb": "1.18.150",
-    "@ocap/mcrypto": "1.18.150",
-    "@ocap/message": "1.18.150",
-    "@ocap/state": "1.18.150",
-    "@ocap/tx-protocols": "1.18.150",
-    "@ocap/util": "1.18.150",
+    "@arcblock/did": "1.18.152",
+    "@arcblock/did-util": "1.18.152",
+    "@arcblock/validator": "1.18.152",
+    "@ocap/config": "1.18.152",
+    "@ocap/indexdb": "1.18.152",
+    "@ocap/mcrypto": "1.18.152",
+    "@ocap/message": "1.18.152",
+    "@ocap/state": "1.18.152",
+    "@ocap/tx-protocols": "1.18.152",
+    "@ocap/util": "1.18.152",
     "debug": "^4.3.6",
     "lodash": "^4.17.21"
   },
-  "gitHead": "700ffd83b7fbb781d68ef761aa9b1b004a961d9f"
+  "gitHead": "4f31d06aa2a15b57a7d00fdec6241376f10ee873"
 }