@ckb-ccc/core 0.0.13-alpha.8 → 0.0.14-alpha.0

package/src/signer/ckb/signerCkbPublicKey.ts

@@ -1,7 +1,7 @@
 import { Address } from "../../address/index.js";
 import { bytesFrom } from "../../bytes/index.js";
-import { Transaction, TransactionLike } from "../../ckb/index.js";
-import { Client, KnownScript } from "../../client/index.js";
+import { Script, Transaction, TransactionLike } from "../../ckb/index.js";
+import { CellDepInfo, Client, KnownScript } from "../../client/index.js";
 import { hashCkb } from "../../hasher/index.js";
 import { Hex, HexLike, hexFrom } from "../../hex/index.js";
 import { Signer, SignerSignType, SignerType } from "../signer/index.js";
@@ -43,24 +43,109 @@ export class SignerCkbPublicKey extends Signer {
     return this.publicKey;
   }
 
+  async getAddressObjSecp256k1(): Promise<Address> {
+    return Address.fromKnownScript(
+      this.client,
+      KnownScript.Secp256k1Blake160,
+      bytesFrom(hashCkb(this.publicKey)).slice(0, 20),
+    );
+  }
+
+  async getRecommendedAddressObj(_preference?: unknown): Promise<Address> {
+    return this.getAddressObjSecp256k1();
+  }
+
   async getAddressObjs(): Promise<Address[]> {
-    return [
-      await Address.fromKnownScript(
+    const secp256k1 = await this.getAddressObjSecp256k1();
+
+    const addresses: Address[] = [];
+    let count = 0;
+    for await (const cell of this.client.findCells({
+      script: await Script.fromKnownScript(
         this.client,
-        KnownScript.Secp256k1Blake160,
-        bytesFrom(hashCkb(this.publicKey)).slice(0, 20),
+        KnownScript.AnyoneCanPay,
+        secp256k1.script.args,
       ),
-    ];
+      scriptType: "lock",
+      scriptSearchMode: "prefix",
+      withData: false,
+    })) {
+      if (count >= 10) {
+        break;
+      }
+      count += 1;
+
+      if (addresses.some(({ script }) => script.eq(cell.cellOutput.lock))) {
+        continue;
+      }
+
+      addresses.push(
+        Address.from({
+          prefix: this.client.addressPrefix,
+          script: cell.cellOutput.lock,
+        }),
+      );
+    }
+
+    return [secp256k1, ...addresses];
   }
 
-  async prepareTransaction(txLike: TransactionLike): Promise<Transaction> {
+  async getRelatedScripts(
+    txLike: TransactionLike,
+  ): Promise<{ script: Script; cellDeps: CellDepInfo[] }[]> {
     const tx = Transaction.from(txLike);
-    const { script } = await this.getRecommendedAddressObj();
-    await tx.addCellDepsOfKnownScripts(
+
+    const secp256k1 = await this.getAddressObjSecp256k1();
+    const acp = await Script.fromKnownScript(
       this.client,
-      KnownScript.Secp256k1Blake160,
+      KnownScript.AnyoneCanPay,
+      secp256k1.script.args,
+    );
+
+    const scripts: { script: Script; cellDeps: CellDepInfo[] }[] = [];
+    for (const input of tx.inputs) {
+      await input.completeExtraInfos(this.client);
+      if (!input.cellOutput) {
+        throw new Error("Unable to complete input");
+      }
+
+      const { lock } = input.cellOutput;
+      if (scripts.some(({ script }) => script.eq(lock))) {
+        continue;
+      }
+
+      if (lock.eq(secp256k1.script)) {
+        scripts.push({
+          script: lock,
+          cellDeps: (
+            await this.client.getKnownScript(KnownScript.Secp256k1Blake160)
+          ).cellDeps,
+        });
+      } else if (
+        lock.codeHash === acp.codeHash &&
+        lock.hashType === acp.hashType &&
+        lock.args.startsWith(acp.args)
+      ) {
+        scripts.push({
+          script: lock,
+          cellDeps: (await this.client.getKnownScript(KnownScript.AnyoneCanPay))
+            .cellDeps,
+        });
+      }
+    }
+
+    return scripts;
+  }
+
+  async prepareTransaction(txLike: TransactionLike): Promise<Transaction> {
+    const tx = Transaction.from(txLike);
+
+    await Promise.all(
+      (await this.getRelatedScripts(tx)).map(async ({ script, cellDeps }) => {
+        await tx.prepareSighashAllWitness(script, 65, this.client);
+        await tx.addCellDepInfos(this.client, cellDeps);
+      }),
     );
-    await tx.prepareSighashAllWitness(script, 65, this.client);
     return tx;
   }
 }