feat(address): add 2 api for address. isAcp & minPaymentAmount

lay2dev · Oct 20, 2020 · f7ae547 · f7ae547
1 parent 55863d9
commit f7ae547
Show file tree

Hide file tree

Showing 5 changed files with 90 additions and 4 deletions.
diff --git a/src/builders/simple-sudt-builder.spec.ts b/src/builders/simple-sudt-builder.spec.ts
@@ -9,7 +9,7 @@ const test = anyTest as TestInterface<{ builder: SimpleSUDTBuilder }>;
 
 test.before(async (t) => {
   const address = new Address(
-    'ckt1qyqv4yga3pgw2h92hcnur7lepdfzmvg8wj7qn44vz8',
+    'ckt1qjr2r35c0f9vhcdgslx2fjwa9tylevr5qka7mfgmscd33wlhfykykn2zrv2y5rex4nnyfs2tqkde8zmayrls6d3kwa5',
     AddressType.ckb
   );
   const amount = new Amount('100');

diff --git a/src/constants.ts b/src/constants.ts
@@ -82,6 +82,18 @@ export const CHAIN_SPECS = {
         HashType.type
       ),
     },
+    acpLockList: [
+      new Script(
+        '0xbf43c3602455798c1a61a596e0d95278864c552fafe231c063b3fabf97a8febc',
+        '0x',
+        HashType.type
+      ),
+      new Script(
+        '0x0fb343953ee78c9986b091defb6252154e0bb51044fd2879fde5b27314506111',
+        '0x',
+        HashType.data
+      ),
+    ],
   },
 
   Aggron: {
@@ -155,6 +167,18 @@ export const CHAIN_SPECS = {
         HashType.type
       ),
     },
+    acpLockList: [
+      new Script(
+        '0x58c5f491aba6d61678b7cf7edf4910b1f5e00ec0cde2f42e0abb4fd9aff25a63',
+        '0x',
+        HashType.type
+      ),
+      new Script(
+        '0x86a1c6987a4acbe1a887cca4c9dd2ac9fcb07405bbeda51b861b18bbf7492c4b',
+        '0x',
+        HashType.type
+      ),
+    ],
   },
   // dev - lay2.ckb.dev
   Lay2: {
@@ -228,5 +252,12 @@ export const CHAIN_SPECS = {
         HashType.type
       ),
     },
+    acpLockList: [
+      new Script(
+        '0xc9eb3097397836e4d5b8fabed3c0cddd14fefe483caf238ca2e3095a111add0b',
+        '0x',
+        HashType.type
+      ),
+    ],
   },
 };
diff --git a/src/interfaces.ts b/src/interfaces.ts
@@ -33,4 +33,5 @@ export interface Config {
   multiSigLock: ConfigItem;
   pwLock: ConfigItem;
   sudtType: ConfigItem;
+  acpLockList: Script[];
 }
diff --git a/src/models/address.spec.ts b/src/models/address.spec.ts
@@ -57,3 +57,39 @@ test('to lock script', (t) => {
     hash_type: PWCore.config.pwLock.script.hashType,
   });
 });
+
+test('is acp address', (t) => {
+  t.true(
+    new Address(
+      'ckt1qjr2r35c0f9vhcdgslx2fjwa9tylevr5qka7mfgmscd33wlhfykykn2zrv2y5rex4nnyfs2tqkde8zmayrls6d3kwa5',
+      AddressType.ckb
+    ).isAcp()
+  );
+  t.false(
+    new Address(
+      'ckt1q35y83078t9h7nwzyvpe9qfuh8qjm08d2ktlegc22tcn4fgem6xnxwlwq4vae7nqgpkl6s59znsqmh9jkrtjhwct56efh7uep9y2xr04d4augtnmauya4s2cdvn0s6nxw9m6k7ndhf2l0un2g0tr7f88fegqns00nq',
+      AddressType.ckb
+    ).isAcp()
+  );
+});
+
+test('minimal pay amount', (t) => {
+  t.is(
+    new Address(
+      'ckt1q35y83078t9h7nwzyvpe9qfuh8qjm08d2ktlegc22tcn4fgem6xnxwlwq4vae7nqgpkl6s59znsqmh9jkrtjhwct56efh7uep9y2xr04d4augtnmauya4s2cdvn0s6nxw9m6k7ndhf2l0un2g0tr7f88fegqns00nq',
+      AddressType.ckb
+    )
+      .minPaymentAmount()
+      .toString(),
+    '105'
+  );
+  t.is(
+    new Address(
+      'ckt1qyqxpayn272n8km2k08hzldynj992egs0waqnr8zjs',
+      AddressType.ckb
+    )
+      .minPaymentAmount()
+      .toString(),
+    '61'
+  );
+});
diff --git a/src/models/address.ts b/src/models/address.ts
@@ -8,12 +8,14 @@ import {
   LumosConfigs,
   verifyCkbAddress,
   verifyEthAddress,
+  cellOccupiedBytes,
 } from '../utils';
 import {
   fullPayloadToAddress,
   AddressType as AType,
   AddressPrefix as APrefix,
 } from '@nervosnetwork/ckb-sdk-utils';
+import { Amount, AmountUnit } from './amount';
 
 export enum AddressPrefix {
   ckb,
@@ -69,9 +71,25 @@ export class Address {
     }
   }
 
-  // TODO need check code_hash is in the acplock list
-  isAcp() {
-    return true;
+  minPaymentAmount(): Amount {
+    if (this.isAcp()) {
+      return new Amount('1', AmountUnit.shannon);
+    }
+    const bytes = cellOccupiedBytes({
+      lock: this.toLockScript(),
+      type: null,
+      data: '0x',
+    });
+    return new Amount(bytes.toString());
+  }
+
+  isAcp(): boolean {
+    const script = this.toLockScript();
+    const { codeHash, hashType } = script;
+    const acpLock = PWCore.config.acpLockList.filter(
+      (x) => x.codeHash === codeHash && x.hashType === hashType
+    );
+    return acpLock && acpLock.length > 0;
   }
 
   toCKBAddress(): string {