Merge pull request #45 from DimensionDev/fix/lens-token

fix: lens token

Author: dudu0506

src/app/api/frames/vote/route.tsx

@@ -34,6 +34,10 @@ export const POST = frames(
         const maxVoteCount = poll.type === POLL_CHOICE_TYPE.Multiple ? poll.multiple_count : PER_USER_VOTE_LIMIT;
 
         if (!currentChoice.is_select && votedLen < maxVoteCount) {
+            if (body.untrustedData.idToken) {
+                body.untrustedData.identityToken = body.untrustedData.idToken;
+                delete body.untrustedData.idToken;
+            }
             const voteResult = await vote(
                 {
                     poll_id: pollId,

src/config/lensFrame.ts

@@ -1,6 +1,8 @@
-import { getLensFrameMessage, isLensFrameActionPayload } from 'frames.js/lens';
+import { isLensFrameActionPayload } from 'frames.js/lens';
 import { openframes } from 'frames.js/middleware';
 
+import { getLensFrameMessage } from '@/helpers/getLensFrameMessage';
+
 export const lensFrame = openframes({
     clientProtocol: {
         id: 'lens',

src/helpers/getLensFrameMessage.ts

@@ -0,0 +1,68 @@
+import { development, FrameVerifySignatureResult, LensClient, production } from '@lens-protocol/client';
+
+import { LensFrameOptions, LensFrameRequest, LensFrameResponse } from '@/types/lens';
+
+type FrameRequest = LensFrameRequest & {
+    untrustedData: {
+        idToken?: string;
+    };
+};
+
+export async function getLensFrameMessage(
+    frameActionPayload: FrameRequest,
+    options?: LensFrameOptions,
+): Promise<
+    LensFrameResponse & {
+        walletAddress(): Promise<string | undefined>;
+    }
+> {
+    const lensClientEnvironment = options?.environment === 'development' ? development : production;
+
+    const lensClient = new LensClient({
+        environment: lensClientEnvironment,
+    });
+
+    const {
+        url,
+        inputText,
+        state,
+        buttonIndex,
+        actionResponse,
+        profileId,
+        pubId,
+        specVersion,
+        deadline,
+        identityToken,
+        idToken,
+    } = frameActionPayload.untrustedData;
+
+    const typedData = await lensClient.frames.createFrameTypedData({
+        url,
+        inputText,
+        state,
+        buttonIndex,
+        actionResponse,
+        profileId,
+        pubId,
+        specVersion,
+        deadline,
+    });
+
+    const response = await lensClient.frames.verifyFrameSignature({
+        identityToken: identityToken || idToken || '',
+        signature: frameActionPayload.trustedData.messageBytes,
+        signedTypedData: typedData,
+    });
+
+    return {
+        ...typedData.value,
+        isValid: response === FrameVerifySignatureResult.Verified,
+        async walletAddress() {
+            const profile = await lensClient.profile.fetch({
+                forProfileId: typedData.value.profileId,
+            });
+
+            return profile?.ownedBy.address;
+        },
+    };
+}