feat(web3mail): Add API routes for storage ability

prisma/schema.prisma

@@ -63,6 +63,7 @@ model User {
   // Custom
   walletAddress String?   @unique
   generatedMessage String?
+  protectedDataAddress String? @unique
 
   twitterId String?       @unique
   twitterUsername String?

src/app/api/mailing/retrieveProtectedDataAddress/route.ts

@@ -0,0 +1,33 @@
+import { NextResponse, NextRequest } from "next/server";
+import { getServerSession } from "next-auth";
+import { nextAuthOptions } from "@/app/api/auth/[...nextauth]/options";
+import { prisma } from "@/lib/db";
+
+export const GET = async () => {
+    const session = await getServerSession(nextAuthOptions);
+
+    // If not session return error
+    if (!session) {
+        return NextResponse.json({
+            success: false,
+            entries: 0,
+            error: "Twitter account not connected",
+        });
+    }
+
+    // Fetch user
+    const user = await prisma.user.findFirst({
+        where: {
+            id: session.user.id,
+        },
+        select: {
+            protectedDataAddress: true,
+        }
+    });
+    if (!user || !user?.protectedDataAddress) {
+        return NextResponse.json({ success: false, error: "User already has no protected data address" });
+    }
+
+    // Return success
+    return NextResponse.json({ success: true, address: user.protectedDataAddress });
+}

src/app/api/mailing/saveProtectedDataAddress/route.ts

@@ -0,0 +1,50 @@
+import { NextResponse, NextRequest } from "next/server";
+import { getServerSession } from "next-auth";
+import { nextAuthOptions } from "@/app/api/auth/[...nextauth]/options";
+import { prisma } from "@/lib/db";
+
+export const POST = async (request: NextRequest) => {
+    const session = await getServerSession(nextAuthOptions);
+
+    // If not session return error
+    if (!session) {
+        return NextResponse.json({
+            success: false,
+            entries: 0,
+            error: "Twitter account not connected",
+        });
+    }
+
+    // Retrieve body data
+    const data = await request.json();
+    const protectedDataAddress = data.protectedDataAddress;
+
+    // Ensure user has no protected data address registered
+    const userExists = await prisma.user.count({
+        where: {
+            id: session.user.id,
+            protectedDataAddress: null
+        },
+    });
+    if (!userExists) {
+        return NextResponse.json({ success: false, error: "User already has protected data address" });
+    }
+
+    // Ensure there is a protectedDataAddress
+    if (!protectedDataAddress) {
+        return NextResponse.json({ success: false, error: "Missing protected data address" });
+    }
+
+    // Update the user's protectedDataAddress
+    await prisma.user.update({
+        where: {
+            id: session.user.id,
+        },
+        data: {
+            protectedDataAddress
+        },
+    });
+
+    // Return success
+    return NextResponse.json({ success: true });
+}

src/app/app/mail/page.tsx

@@ -1,5 +1,6 @@
-'use client'
+"use client";
 import { type NextPage } from "next";
+
 import React, { useEffect, useState } from 'react';
 import { useSwitchChain, useChainId, useAccount } from 'wagmi'
 import ExplanationCard from "@/components/mail/ExplanationCard";
@@ -9,13 +10,15 @@ import ProtectedDataBox from "@/components/mail/ProtectedDataBox";
 import { type Address } from "@iexec/web3mail";
 import { checkIsProtected, handleSubmitProtection, checkAppIsGrantedAccess } from "@/components/mail/utils/utils";
 
+
 const Page: NextPage = () => {
   const currentChainId = useChainId();
   const { switchChain } = useSwitchChain();
   const { address } = useAccount();
   const userAddress = address as string;
   const desiredChainId = 134; // For iExec sidechain.
 
+
   const [protectedDataAddress, setProtectedDataAddress] = useState<string>('');
   const [isProtected, setIsProtected] = useState(false);
   const [hasJustRegistered, setHasJustRegistered] = useState(false);
@@ -63,7 +66,7 @@ const Page: NextPage = () => {
     } catch (err) {
       setError(`Failed to check protection status: ${(err as Error).message}`);
     }
-  }
+  };
 
   const handleSubmitProtectionWrapper = async (email: string) => {
     try {
@@ -76,7 +79,7 @@ const Page: NextPage = () => {
     } catch (err) {
       setError(`Failed to submit protection: ${(err as Error).message}`);
     }
-  }
+  };
 
   const checkAppIsGrantedAccessWrapper = async () => {
     try {
@@ -113,4 +116,4 @@ const Page: NextPage = () => {
   );
 };
 
-export default Page;
+export default Page;