feat(lens): update validation method

.changeset/large-maps-decide.md

@@ -0,0 +1,6 @@
+---
+"@rabbitholegg/questdk-plugin-utils": minor
+"@rabbitholegg/questdk-plugin-lens": minor
+---
+
+update validation method for lens

packages/lens/package.json

@@ -28,8 +28,9 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
+    "@lens-protocol/client": "2.3.0",
     "@rabbitholegg/questdk": "workspace:*",
     "@rabbitholegg/questdk-plugin-utils": "workspace:*",
-    "@lens-protocol/client": "2.3.0"
+    "alchemy-sdk": "3.3.1"
   }
 }

packages/lens/src/Lens.test.ts

@@ -1,5 +1,5 @@
 import { validateCollect } from './Lens'
-import { beforeEach, describe, expect, test, vi, MockedFunction } from 'vitest'
+import { MockedFunction, beforeEach, describe, expect, test, vi } from 'vitest'
 
 vi.mock('./Lens', () => ({
   validateCollect: vi.fn(),

packages/lens/src/Lens.ts

@@ -5,6 +5,7 @@ import {
   type CollectValidationParams,
   type PluginActionValidation,
   type QuestCompletionPayload,
+  ValidationNotValid,
 } from '@rabbitholegg/questdk-plugin-utils'
 import { type Address } from 'viem'
 
@@ -45,14 +46,8 @@ export const validateCollect = async (
     )
     return hasCollected
   } catch (err) {
-    if (err instanceof Error) {
-      const error = new Error(err.message)
-      error.name = 'ValidationNotValid'
-      throw error
-    } else {
-      console.error(err)
-      throw new Error('Unknown error')
-    }
+    console.error('[lens-plugin] Error while validating collect action')
+    throw new ValidationNotValid(err instanceof Error ? err : String(err))
   }
 }
 

packages/lens/src/alchemy.ts

@@ -0,0 +1,9 @@
+import { Alchemy, Network } from 'alchemy-sdk'
+
+const settings = {
+  apiKey: process.env.ALCHEMY_API_KEY,
+  network: Network.MATIC_MAINNET,
+  requestTimeout: 2000,
+}
+
+export const alchemy = new Alchemy(settings)

packages/lens/src/validate.ts

@@ -1,12 +1,13 @@
+import { alchemy } from './alchemy'
 import {
   LensClient,
-  production,
-  type SimpleCollectOpenActionSettingsFragment,
   type MultirecipientFeeCollectOpenActionSettingsFragment,
+  type SimpleCollectOpenActionSettingsFragment,
+  production,
 } from '@lens-protocol/client'
-import { getClient } from './client'
-import { Chains } from '@rabbitholegg/questdk-plugin-utils'
-import { Address } from 'viem'
+import { GetTransfersForOwnerTransferType } from 'alchemy-sdk'
+import axios from 'axios'
+import { Address, zeroAddress } from 'viem'
 
 const lensClient = new LensClient({
   environment: production,
@@ -20,53 +21,96 @@ export async function hasAddressCollectedPost(
   if (collectAddress === null) {
     return false
   }
-  const amountOwned = await checkAddressOwnsCollect(address, collectAddress)
-  return amountOwned > 0n
+  return await checkAddressMintedCollect(address, collectAddress)
 }
 
 export async function getCollectAddress(postId: string) {
-  const result = await lensClient.publication.fetch({
-    forId: postId,
-  })
-  if (!result || result?.__typename === 'Mirror') {
+  try {
+    const result = await lensClient.publication.fetch({
+      forId: postId,
+    })
+    if (!result || result?.__typename === 'Mirror') {
+      return null
+    }
+    const openActions = result.openActionModules
+    if (!openActions || openActions.length === 0) {
+      return null
+    }
+    const collectActions = openActions.filter(
+      (action) =>
+        action.__typename === 'SimpleCollectOpenActionSettings' ||
+        action.__typename === 'MultirecipientFeeCollectOpenActionSettings',
+    ) as Array<
+      | MultirecipientFeeCollectOpenActionSettingsFragment
+      | SimpleCollectOpenActionSettingsFragment
+    >
+
+    const collectNft = collectActions.find((action) => action.collectNft)
+    const collectAddress = collectNft?.collectNft?.toLowerCase()
+    return (collectAddress as Address) ?? null
+  } catch (error) {
+    console.error('Error fetching collect contract address', error)
     return null
   }
-  const openActions = result.openActionModules
-  if (!openActions || openActions.length === 0) {
-    return null
+}
+
+export async function checkAddressMintedCollect(
+  actor: Address,
+  collectAddress: Address,
+) {
+  try {
+    return await checkMintedUsingAlchemy(actor, collectAddress)
+  } catch (err) {
+    console.error('Error while using alchemy api', err)
+    return await checkMintedUsingReservoir(actor, collectAddress)
   }
-  const collectActions = openActions.filter(
-    (action) =>
-      action.__typename === 'SimpleCollectOpenActionSettings' ||
-      action.__typename === 'MultirecipientFeeCollectOpenActionSettings',
-  ) as Array<
-    | MultirecipientFeeCollectOpenActionSettingsFragment
-    | SimpleCollectOpenActionSettingsFragment
-  >
+}
 
-  const collectNft = collectActions.find((action) => action.collectNft)
-  const collectAddress = collectNft?.collectNft?.toLowerCase()
-  return (collectAddress as Address) ?? null
+async function checkMintedUsingAlchemy(
+  actor: Address,
+  collectAddress: Address,
+) {
+  // alchemy will work without an api key, but we risk being rate-limited
+  if (!process.env.ALCHEMY_API_KEY) {
+    console.error('Alchemy API key not found')
+  }
+
+  const options = {
+    contractAddresses: [collectAddress],
+  }
+
+  const transfers = await alchemy.nft.getTransfersForOwner(
+    actor,
+    GetTransfersForOwnerTransferType.TO,
+    options,
+  )
+
+  // only count nfts that originate from zeroAddress
+  const mintedNfts = transfers.nfts.filter((nft) => nft.from === zeroAddress)
+
+  return mintedNfts.length > 0
 }
 
-export async function checkAddressOwnsCollect(
+async function checkMintedUsingReservoir(
   actor: Address,
   collectAddress: Address,
 ) {
-  const client = getClient(Chains.POLYGON_POS)
-  const result = await client.readContract({
-    address: collectAddress as Address,
-    abi: [
-      {
-        inputs: [{ internalType: 'address', name: 'owner', type: 'address' }],
-        name: 'balanceOf',
-        outputs: [{ internalType: 'uint256', name: '', type: 'uint256' }],
-        stateMutability: 'view',
-        type: 'function',
-      },
-    ],
-    functionName: 'balanceOf',
-    args: [actor],
+  const baseUrl = 'https://api-polygon.reservoir.tools/users/activity/v6'
+  const params = new URLSearchParams({
+    users: actor,
+    collection: collectAddress,
+    limit: '1',
+    types: 'mint',
   })
-  return result
+
+  const options = {
+    method: 'GET',
+    url: `${baseUrl}?${params.toString()}`,
+    headers: {
+      accept: '*/*',
+    },
+  }
+
+  const response = await axios.request(options)
+  return response.data?.activities?.length > 0
 }

packages/lens/vite.config.js

@@ -2,7 +2,7 @@
 export default {
   build: {
     rollupOptions: {
-      external: [/@rabbitholegg/, /@lens-protocol/],
+      external: [/@rabbitholegg/, /@lens-protocol/, 'alchemy-sdk'],
     },
     lib: {
       entry: 'src/index.ts',

packages/utils/src/errors/index.ts

@@ -1 +1,4 @@
-export { PluginActionNotImplementedError } from './plugin'
+export {
+  PluginActionNotImplementedError,
+  ValidationNotValid,
+} from './plugin'

packages/utils/src/errors/plugin.ts

@@ -4,3 +4,16 @@ export class PluginActionNotImplementedError extends Error {
     this.name = 'PluginActionNotImplementedError'
   }
 }
+
+export class ValidationNotValid extends Error {
+  constructor(input: string | Error) {
+    if (input instanceof Error) {
+      super(input.message)
+      Object.assign(this, input)
+
+      this.name = 'ValidationNotValid'
+    } else {
+      super(input)
+    }
+  }
+}

packages/utils/src/index.ts

@@ -141,4 +141,7 @@ export {
   PremintActionFormSchema,
 } from './types'
 
-export { PluginActionNotImplementedError } from './errors'
+export {
+  PluginActionNotImplementedError,
+  ValidationNotValid,
+} from './errors'