Add sources field to divined result

XYOracleNetwork · Aug 1, 2023 · c18d0c5 · c18d0c5
1 parent 8498e16
commit c18d0c5
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 7 deletions.
diff --git a/packages/plugins/packages/payloadset/packages/crypto/packages/nft/src/Diviner.ts b/packages/plugins/packages/payloadset/packages/crypto/packages/nft/src/Diviner.ts
@@ -1,4 +1,5 @@
 import { AbstractDiviner } from '@xyo-network/abstract-diviner'
+import { PayloadHasher } from '@xyo-network/core'
 import {
   isNftInfoPayload,
   NftScoreDivinerConfig,
@@ -23,6 +24,14 @@ export const isNftScorePayload = (payload: Payload): payload is NftScorePayload
 export class NftScoreDiviner<TParams extends NftScoreDivinerParams = NftScoreDivinerParams> extends AbstractDiviner<TParams> {
   static override configSchemas = [NftScoreDivinerConfigSchema]
 
-  protected override divineHandler = async (payloads?: Payload[]): Promise<Payload[]> =>
-    (await Promise.all(payloads?.filter(isNftInfoPayload).map(analyzeNft) ?? [])).map(toNftScorePayload)
+  protected override divineHandler = async (payloads?: Payload[]): Promise<Payload[]> => {
+    const nfts = payloads?.filter(isNftInfoPayload) ?? []
+    const results = await Promise.all(
+      nfts.map<Promise<NftScorePayload>>(async (p) => {
+        const [score, sourceHash] = await Promise.all([toNftScorePayload(await analyzeNft(p)), PayloadHasher.hashAsync(p)])
+        return { ...score, sources: [sourceHash] }
+      }),
+    )
+    return results
+  }
 }
diff --git a/packages/plugins/packages/payloadset/packages/crypto/packages/nft/src/spec/Diviner.spec.ts b/packages/plugins/packages/payloadset/packages/crypto/packages/nft/src/spec/Diviner.spec.ts
@@ -21,13 +21,17 @@ describe('NftScoreDiviner', () => {
   })
   test('divine', async () => {
     const diviner = await NftScoreDiviner.create({ account: await HDWallet.random() })
-    const result = await diviner.divine(data)
-    const scores = result.filter(isNftScorePayload)
-    expect(scores.length).toBeGreaterThan(0)
-    expect(result.length).toEqual(scores.length)
-    for (const score of scores) {
+    const scores = (await diviner.divine(data)).filter(isNftScorePayload)
+    expect(scores).toBeArrayOfSize(data.length)
+    for (let i = 0; i < scores.length; i++) {
+      const score = scores[i]
       const wrapped = PayloadWrapper.wrap(score)
       expect(await wrapped.getValid()).toBe(true)
+      const payload = wrapped.payload()
+      expect(payload?.sources).toBeArrayOfSize(1)
+      expect(payload?.sources?.[0]).toBeString()
+      const sourceHash = await PayloadWrapper.wrap(data[i]).hashAsync()
+      expect(payload?.sources?.[0]).toBe(sourceHash)
     }
   })
 })