fix(platforms): update lens provider to v2 api (#1976)

passportxyz · Dec 8, 2023 · cf791cf · cf791cf
1 parent 7f19729
commit cf791cf
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 13 deletions.
diff --git a/platforms/src/Lens/Providers/lens.ts b/platforms/src/Lens/Providers/lens.ts
@@ -8,13 +8,16 @@ import axios from "axios";
 // ----- Utils
 import { handleProviderAxiosError } from "../../utils/handleProviderAxiosError";
 
-interface DefaultProfile {
+interface Profile {
   id: string;
-  handle: string;
+  fullHandle: string;
+  ownedBy: string;
 }
 
 interface Data {
-  defaultProfile: DefaultProfile;
+  ownedHandles: {
+    items: Profile[];
+  };
 }
 
 interface GraphQlResponse {
@@ -27,26 +30,37 @@ interface LensProfileResponse {
   errors: string[];
 }
 
-const lensApiEndpoint = "https://api.lens.dev";
+const lensApiEndpoint = "https://api-v2.lens.dev/";
 
 async function getLensProfile(userAddress: string): Promise<LensProfileResponse> {
   try {
     const query = `
-      query DefaultProfile {
-        defaultProfile(request: { ethereumAddress: "${userAddress}"}) {
-          id
-          handle
+      query OwnedHandles {
+        ownedHandles(request: {
+          for: "${userAddress}"
+        }) {
+          items {
+            id
+            ownedBy
+            fullHandle
+          }
         }
       }
     `;
     const result: { data: GraphQlResponse } = await axios.post(lensApiEndpoint, {
       query,
     });
 
-    if (result?.data?.data?.defaultProfile?.handle) {
+    const handles = result?.data?.data?.ownedHandles?.items;
+
+    const validHandle = handles?.find(
+      (handle) => handle.ownedBy.toLocaleLowerCase() === userAddress.toLocaleLowerCase()
+    );
+
+    if (validHandle) {
       return {
         valid: true,
-        handle: result?.data?.data?.defaultProfile?.handle,
+        handle: validHandle.fullHandle,
         errors: [],
       };
     } else {

diff --git a/platforms/src/Lens/__tests__/lens.test.ts b/platforms/src/Lens/__tests__/lens.test.ts
@@ -21,9 +21,14 @@ describe("Attempt verification", function () {
     mockedAxios.post.mockResolvedValueOnce({
       data: {
         data: {
-          defaultProfile: {
-            id: MOCK_ADDRESS,
-            handle: MOCK_HANDLE,
+          ownedHandles: {
+            items: [
+              {
+                id: MOCK_ADDRESS,
+                fullHandle: MOCK_HANDLE,
+                ownedBy: MOCK_ADDRESS,
+              },
+            ],
           },
         },
       },
@@ -43,6 +48,31 @@ describe("Attempt verification", function () {
     });
   });
 
+  it("should return false if owner addresses do not match", async () => {
+    mockedAxios.post.mockResolvedValueOnce({
+      data: {
+        data: {
+          ownedHandles: {
+            items: [
+              {
+                id: MOCK_ADDRESS,
+                fullHandle: MOCK_HANDLE,
+                ownedBy: MOCK_FAKE_ADDRESS,
+              },
+            ],
+          },
+        },
+      },
+    });
+
+    const lens = new LensProfileProvider();
+    const verifiedPayload = await lens.verify({
+      address: MOCK_ADDRESS_LOWER,
+    } as RequestPayload);
+
+    expect(verifiedPayload.valid).toEqual(false);
+  });
+
   it("should return false for an address without a lens handle", async () => {
     mockedAxios.post.mockResolvedValueOnce({
       data: {