Merge pull request #872 from nervosnetwork/compute-cycles

feat: compute cycles
nervosnetwork · Aug 19, 2019 · 3d673f8 · 3d673f8
2 parents 6f3b695 + 01e9165
commit 3d673f8
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 4 deletions.
diff --git a/packages/neuron-wallet/src/controllers/wallets/index.ts b/packages/neuron-wallet/src/controllers/wallets/index.ts
@@ -372,6 +372,26 @@ export default class WalletsController {
     }
   }
 
+  @CatchControllerError
+  public static async computeCycles(params: { id: string; walletID: string; capacities: string }) {
+    if (!params) {
+      throw new IsRequired('Parameters')
+    }
+    try {
+      const walletsService = WalletsService.getInstance()
+      const cycles = await walletsService.computeCycles(params.walletID, params.capacities)
+      return {
+        status: ResponseCode.Success,
+        result: cycles,
+      }
+    } catch (err) {
+      return {
+        status: ResponseCode.Fail,
+        msg: `Error: "${err.message}"`,
+      }
+    }
+  }
+
   @CatchControllerError
   public static async updateAddressDescription({
     walletID,

diff --git a/packages/neuron-wallet/src/services/wallets.ts b/packages/neuron-wallet/src/services/wallets.ts
@@ -16,6 +16,7 @@ import { WalletListSubject, CurrentWalletSubject } from 'models/subjects/wallets
 import dataUpdateSubject from 'models/subjects/data-update'
 import CommandSubject from 'models/subjects/command'
 import WindowManager from 'models/window-manager'
+import CellsService from 'services/cells'
 
 import NodeService from './node'
 import FileService from './file'
@@ -27,6 +28,7 @@ const fileService = FileService.getInstance()
 
 const MODULE_NAME = 'wallets'
 const DEBOUNCE_TIME = 200
+const SECP_CYCLES = BigInt('1440000')
 
 export interface Wallet {
   id: string
@@ -315,7 +317,7 @@ export default class WalletService {
       throw new IsRequired('Password')
     }
 
-    const addressInfos = await this.getAddressInfos()
+    const addressInfos = await this.getAddressInfos(walletID)
 
     const addresses: string[] = addressInfos.map(info => info.address)
 
@@ -371,10 +373,31 @@ export default class WalletService {
     return txHash
   }
 
+  public computeCycles = async (walletID: string = '', capacities: string): Promise<string> => {
+    const wallet = await this.get(walletID)
+    if (!wallet) {
+      throw new WalletNotFound(walletID)
+    }
+
+    const addressInfos = await this.getAddressInfos(walletID)
+
+    const addresses: string[] = addressInfos.map(info => info.address)
+
+    const lockHashes: string[] = await LockUtils.addressesToAllLockHashes(addresses)
+
+    const { inputs } = await CellsService.gatherInputs(capacities, lockHashes, '0')
+    const cycles = SECP_CYCLES * BigInt(inputs.length)
+
+    return cycles.toString()
+  }
+
   // path is a BIP44 full path such as "m/44'/309'/0'/0/0"
-  public getAddressInfos = async (): Promise<AddressInterface[]> => {
-    const walletId = this.getCurrent()!.id
-    const addrs = await AddressService.allAddressesByWalletId(walletId)
+  public getAddressInfos = async (walletID: string): Promise<AddressInterface[]> => {
+    const wallet = await this.get(walletID)
+    if (!wallet) {
+      throw new WalletNotFound(walletID)
+    }
+    const addrs = await AddressService.allAddressesByWalletId(walletID)
     return addrs
   }