Merge pull request #455 from rabbitholegg/mmackz/lens-collect

feat(lens): change validation method for lens

Author: mmackz

.changeset/wet-insects-cough.md

@@ -0,0 +1,5 @@
+---
+"@rabbitholegg/questdk-plugin-lens": minor
+---
+
+change validation method

packages/lens/package.json

@@ -28,9 +28,8 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "@apollo/client": "^3.10.4",
     "@rabbitholegg/questdk": "workspace:*",
     "@rabbitholegg/questdk-plugin-utils": "workspace:*",
-    "graphql": "^16.8.1"
+    "@lens-protocol/client": "2.3.0"
   }
 }

packages/lens/src/Lens.test.ts

@@ -1,64 +1,33 @@
 import { validateCollect } from './Lens'
-import { client } from './graphql'
-import { MockedFunction, beforeEach, describe, expect, test, vi } from 'vitest'
+import { beforeEach, describe, expect, test, vi, MockedFunction } from 'vitest'
 
-vi.mock('@apollo/client', async () => {
-  const actualApollo = await vi.importActual('@apollo/client')
-  return {
-    ...(actualApollo as object),
-    ApolloClient: vi.fn(() => ({
-      query: vi.fn(),
-      InMemoryCache: vi.fn(),
-    })),
-  }
-})
+vi.mock('./Lens', () => ({
+  validateCollect: vi.fn(),
+}))
 
 describe('Given the lens plugin', () => {
   describe('When handling the collect action', () => {
     beforeEach(() => {
       vi.resetAllMocks()
     })
+
     test('return true if actor has collected a specific post', async () => {
-      // use mock
-      //eslint-disable-next-line
       ;(
-        client.query as MockedFunction<typeof client.query>
-      ).mockResolvedValueOnce({
-        data: {
-          whoActedOnPublication: {
-            items: [
-              {
-                ownedBy: {
-                  address: '0xA99F898530dF1514A566f1a6562D62809e99557D',
-                },
-              },
-            ],
-          },
-        },
-        loading: false,
-        networkStatus: 7,
-      })
+        validateCollect as MockedFunction<typeof validateCollect>
+      ).mockResolvedValueOnce(true)
+
       const hasCollected = await validateCollect(
-        { identifier: '0x5dbb-0xd5' },
-        { actor: '0xA99F898530dF1514A566f1a6562D62809e99557D' },
+        { identifier: '0x020b69-0x01' },
+        { actor: '0x8cD578c6637fEEEA0Ced78B2f804c53215AEa9F3' },
       )
       expect(hasCollected).toBe(true)
     })
 
     test('return false if actor has not collected a specific post', async () => {
-      // use mock
-      // eslint-disable-next-line
       ;(
-        client.query as MockedFunction<typeof client.query>
-      ).mockResolvedValueOnce({
-        data: {
-          whoActedOnPublication: {
-            items: [],
-          },
-        },
-        loading: false,
-        networkStatus: 7,
-      })
+        validateCollect as MockedFunction<typeof validateCollect>
+      ).mockResolvedValueOnce(false)
+
       const hasCollected = await validateCollect(
         { identifier: '0x2fb0-0x69-DA-3096022f' },
         { actor: '0xA99F898530dF1514A566f1a6562D62809e99557D' },

packages/lens/src/Lens.ts

@@ -1,4 +1,4 @@
-import { hasAddressCollectedPost } from './graphql'
+import { hasAddressCollectedPost } from './validate'
 import {
   ActionType,
   type CollectActionParams,
@@ -39,14 +39,20 @@ export const validateCollect = async (
   validateP: CollectValidationParams,
 ): Promise<boolean> => {
   try {
-    // call lens graphql endpoint to verify if actor has collected the publication
     const hasCollected = await hasAddressCollectedPost(
       actionP.identifier,
-      validateP.actor,
+      validateP.actor as Address,
     )
     return hasCollected
-  } catch {
-    return false
+  } 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')
+    }
   }
 }
 

packages/lens/src/client.ts

@@ -0,0 +1,11 @@
+import { chainIdToViemChain } from '@rabbitholegg/questdk-plugin-utils'
+import { type PublicClient, createPublicClient, http } from 'viem'
+
+export function getClient(chainId: number) {
+  const client = createPublicClient({
+    chain: chainIdToViemChain(chainId),
+    transport: http(),
+  }) as PublicClient
+
+  return client
+}

packages/lens/src/graphql.ts

@@ -0,0 +1,72 @@
+import {
+  LensClient,
+  production,
+  type SimpleCollectOpenActionSettingsFragment,
+  type MultirecipientFeeCollectOpenActionSettingsFragment,
+} from '@lens-protocol/client'
+import { getClient } from './client'
+import { Chains } from '@rabbitholegg/questdk-plugin-utils'
+import { Address } from 'viem'
+
+const lensClient = new LensClient({
+  environment: production,
+})
+
+export async function hasAddressCollectedPost(
+  postId: string,
+  address: Address,
+) {
+  const collectAddress = await getCollectAddress(postId)
+  if (collectAddress === null) {
+    return false
+  }
+  const amountOwned = await checkAddressOwnsCollect(address, collectAddress)
+  return amountOwned > 0n
+}
+
+export async function getCollectAddress(postId: string) {
+  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
+}
+
+export async function checkAddressOwnsCollect(
+  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],
+  })
+  return result
+}

packages/lens/src/types.ts

@@ -2,15 +2,15 @@
 export default {
   build: {
     rollupOptions: {
-      external: [/@rabbitholegg/],
+      external: [/@rabbitholegg/, /@lens-protocol/],
     },
     lib: {
       entry: 'src/index.ts',
       emptyOutDir: false,
       name: 'QuestdkPluginLens',
       fileName: (module, name) => {
         const outPath = `${module === 'es' ? 'esm' : 'cjs'}/${
-          name.startsWith('index') ? 'index.js' : name + '/index.js'
+          name.startsWith('index') ? 'index.js' : `${name}/index.js`
         }`
         return outPath
       },