chore: add dao script support in indexer sync

nervosnetwork · Nov 7, 2019 · 2756a15 · 2756a15
1 parent 77a505d
commit 2756a15
Showing 1 changed file with 22 additions and 3 deletions.
diff --git a/packages/neuron-wallet/src/services/indexer/queue.ts b/packages/neuron-wallet/src/services/indexer/queue.ts
@@ -15,6 +15,7 @@ import HexUtils from 'utils/hex'
 import { TxUniqueFlagCache } from './tx-unique-flag'
 import { TransactionCache } from './transaction-cache'
 import TransactionEntity from 'database/chain/entities/transaction'
+import DaoUtils from 'models/dao-utils'
 
 export interface LockHashInfo {
   lockHash: string
@@ -90,13 +91,19 @@ export default class IndexerQueue {
             await this.indexLockHashes(lockHashInfos)
             this.indexed = true
           }
+          const daoScriptInfo = await DaoUtils.daoScript(this.url)
+          const daoScriptHash = LockUtils.computeScriptHash({
+            codeHash: daoScriptInfo.codeHash,
+            args: "0x",
+            hashType: daoScriptInfo.hashType,
+          })
           const lockHashes: string[] = lockHashInfos.map(info => info.lockHash)
           const minBlockNumber = await this.getCurrentBlockNumber(lockHashes)
           for (const lockHash of lockHashes) {
-            await this.pipeline(lockHash, TxPointType.CreatedBy, currentBlockNumber)
+            await this.pipeline(lockHash, TxPointType.CreatedBy, currentBlockNumber, daoScriptHash)
           }
           for (const lockHash of lockHashes) {
-            await this.pipeline(lockHash, TxPointType.ConsumedBy, currentBlockNumber)
+            await this.pipeline(lockHash, TxPointType.ConsumedBy, currentBlockNumber, daoScriptHash)
           }
           if (minBlockNumber) {
             await this.blockNumberService.updateCurrent(minBlockNumber)
@@ -171,7 +178,7 @@ export default class IndexerQueue {
   }
 
   // type: 'createdBy' | 'consumedBy'
-  public pipeline = async (lockHash: string, type: TxPointType, startBlockNumber: bigint) => {
+  public pipeline = async (lockHash: string, type: TxPointType, startBlockNumber: bigint, daoScriptHash: string) => {
     let page = 0
     let stopped = false
     while (!stopped) {
@@ -220,6 +227,18 @@ export default class IndexerQueue {
                 input.lockHash = LockUtils.lockScriptToHash(input.lock)
                 input.capacity = previousOutput.capacity
               }
+
+              const outputs = transaction.outputs!
+              for (let i = 0; i < outputs.length; ++i) {
+                const output = outputs[i]
+                const typeScript = output.type
+                if (typeScript) {
+                  output.typeHash = LockUtils.computeScriptHash(typeScript)
+                  if (output.typeHash === daoScriptHash) {
+                    output.daoData = transaction.outputsData![i]
+                  }
+                }
+              }
             }
             const { blockHash } = transactionWithStatus.txStatus
             if (blockHash) {