Merge pull request #173 from ashuralyk/chore/tiny-complete

chore: tiny complete for core & spore

Author: Hanssen0

.changeset/blue-pans-hope.md

@@ -0,0 +1,5 @@
+---
+"@ckb-ccc/spore": minor
+---
+
+support search spore/cluster under customized lock script

.changeset/gentle-crews-stare.md

@@ -0,0 +1,5 @@
+---
+"@ckb-ccc/core": minor
+---
+
+Add treatment to uncompatible XUDT data format

packages/core/src/ckb/transaction.ts

@@ -701,11 +701,13 @@ export class WitnessArgs extends mol.Entity.Base<
 }
 
 /**
+ * Convert a bytes to a num.
+ *
  * @public
  */
 export function udtBalanceFrom(dataLike: BytesLike): Num {
   const data = bytesFrom(dataLike).slice(0, 16);
-  return numFromBytes(data);
+  return data.length === 0 ? Zero : numFromBytes(data);
 }
 
 export const RawTransaction = mol.table({

packages/spore/src/cluster/index.ts

@@ -234,3 +234,61 @@ export async function* findSporeClustersBySigner({
     }
   }
 }
+
+/**
+ * Search on-chain clusters under the specified lock or not
+ *
+ * @param client the client to search clusters
+ * @param lock the lock of clusters
+ */
+export async function* findSporeClusters({
+  client,
+  lock,
+  order,
+  limit,
+  scriptInfos,
+}: {
+  client: ccc.Client;
+  lock?: ccc.ScriptLike;
+  order?: "asc" | "desc";
+  limit?: number;
+  scriptInfos?: SporeScriptInfoLike[];
+}): AsyncGenerator<{
+  cell: ccc.Cell;
+  cluster: ccc.Cell;
+  clusterData: ClusterDataView;
+  scriptInfo: SporeScriptInfo;
+}> {
+  for (const scriptInfo of scriptInfos ??
+    Object.values(getClusterScriptInfos(client))) {
+    if (!scriptInfo) {
+      continue;
+    }
+
+    for await (const cluster of client.findCells(
+      {
+        script: {
+          ...scriptInfo,
+          args: "",
+        },
+        scriptType: "type",
+        scriptSearchMode: "prefix",
+        withData: true,
+        filter: lock
+          ? {
+              script: lock,
+            }
+          : undefined,
+      },
+      order,
+      limit,
+    )) {
+      yield {
+        cell: cluster,
+        cluster,
+        clusterData: unpackToRawClusterData(cluster.outputData),
+        scriptInfo: SporeScriptInfo.from(scriptInfo),
+      };
+    }
+  }
+}

packages/spore/src/spore/index.ts

@@ -295,3 +295,72 @@ export async function* findSporesBySigner({
     }
   }
 }
+
+/**
+ * Search on-chain spores under the specified lock or not, if cluster provided, filter spores belonging to this cluster
+ *
+ * @param client the client to search spores
+ * @param lock the lock of spores
+ * @param clusterId the cluster that spores belong to. "" to find public spores
+ * @param scriptInfos the deployed script infos of spores
+ * @returns specified spore cells
+ */
+export async function* findSpores({
+  client,
+  lock,
+  clusterId,
+  scriptInfos,
+  limit,
+  order,
+}: {
+  client: ccc.Client;
+  lock?: ccc.ScriptLike;
+  order?: "asc" | "desc";
+  limit?: number;
+  clusterId?: ccc.HexLike;
+  scriptInfos?: SporeScriptInfoLike[];
+}): AsyncGenerator<{
+  cell: ccc.Cell;
+  spore: ccc.Cell;
+  sporeData: SporeDataView;
+  scriptInfo: SporeScriptInfo;
+}> {
+  for (const scriptInfo of scriptInfos ??
+    Object.values(getSporeScriptInfos(client))) {
+    if (!scriptInfo) {
+      continue;
+    }
+    for await (const spore of client.findCells(
+      {
+        script: {
+          ...scriptInfo,
+          args: [],
+        },
+        scriptType: "type",
+        scriptSearchMode: "prefix",
+        withData: true,
+        filter: lock
+          ? {
+              script: lock,
+            }
+          : undefined,
+      },
+      order,
+      limit,
+    )) {
+      const sporeData = unpackToRawSporeData(spore.outputData);
+      if (
+        !clusterId ||
+        (clusterId === "" && !sporeData.clusterId) ||
+        sporeData.clusterId === ccc.hexFrom(clusterId)
+      ) {
+        yield {
+          cell: spore,
+          spore,
+          sporeData,
+          scriptInfo: SporeScriptInfo.from(scriptInfo),
+        };
+      }
+    }
+  }
+}