From 154dd1460696d1249fe4ff275bb3c26a39454bfe Mon Sep 17 00:00:00 2001
From: wraeth-eth <104132113+wraeth-eth@users.noreply.github.com>
Date: Wed, 25 Oct 2023 15:51:28 +1100
Subject: [PATCH] allow selection of nfts in pay project (#122)

### Description

Allow the selection of NFTs in the pay project page


[Screen Recording 2023-10-25 at 1.27.33 pm.mov](https://graphite-user-uploaded-assets-prod.s3.amazonaws.com/pinTSPPZHK80UCzWAYgo/2c5972e2-9594-4737-807c-4e028a7b7599.mov)
---
 src/components/Project/ProjectPayPage.tsx     | 47 +++++++++++--------
 .../Project/components/PayRewardCard.tsx      | 31 ++++++++++--
 .../Project/components/ProjectPayForm.tsx     | 23 +++++++--
 .../Project/components/ProjectPayRewards.tsx  | 28 ++++++++++-
 .../Project/providers/ProjectPayContext.tsx   | 19 ++++++++
 .../Project/providers/projectPayReducer.ts    | 27 +++++++++++
 6 files changed, 148 insertions(+), 27 deletions(-)
 create mode 100644 src/components/Project/providers/ProjectPayContext.tsx
 create mode 100644 src/components/Project/providers/projectPayReducer.ts

diff --git a/src/components/Project/ProjectPayPage.tsx b/src/components/Project/ProjectPayPage.tsx
index 83d554ef..c3d628f4 100644
--- a/src/components/Project/ProjectPayPage.tsx
+++ b/src/components/Project/ProjectPayPage.tsx
@@ -1,39 +1,48 @@
 import { useJbProject } from '@/hooks/useJbProject'
 import { ChevronLeftIcon } from '@heroicons/react/24/outline'
 import Link from 'next/link'
+import React from 'react'
 import { twMerge } from 'tailwind-merge'
 import { Button } from '../ui/Button'
 import { ProjectPayForm } from './components/ProjectPayForm'
 import { ProjectPayRewards } from './components/ProjectPayRewards'
+import { projectPayReducer } from './providers/projectPayReducer'
+import { ProjectPayContext } from './providers/ProjectPayContext'
 
 export const ProjectPayPage = () => {
   const { name } = useJbProject()
+  const [state, dispatch] = React.useReducer(projectPayReducer, {
+    nftRewardIds: [],
+  })
+
   return (
-    <div className="block md:flex">
-      {/* Left panel */}
-      <div className="mt-32 flex-1 px-4 pb-14 md:mt-10 md:px-8">
-        <div className="md:mx-auto md:max-w-lg">
-          <BackButton className="hidden md:flex" />
-          <div className="md:mt-12">
-            <h1 className="text-sm font-medium text-gray-500">
-              You are paying
-            </h1>
-            <h2 className="mt-4 font-heading text-2xl font-medium">{name}</h2>
+    <ProjectPayContext.Provider value={{ ...state, dispatch }}>
+      <div className="block md:flex">
+        {/* Left panel */}
+        <div className="mt-32 flex-1 px-4 pb-14 md:mt-10 md:px-8">
+          <div className="md:mx-auto md:max-w-lg">
+            <BackButton className="hidden md:flex" />
+            <div className="md:mt-12">
+              <h1 className="text-sm font-medium text-gray-500">
+                You are paying
+              </h1>
+              <h2 className="mt-4 font-heading text-2xl font-medium">{name}</h2>
 
-            <h3 className="mt-10 font-medium text-gray-800 md:mt-12">
-              Select a reward
-            </h3>
+              <h3 className="mt-10 font-medium text-gray-800 md:mt-12">
+                Select a reward
+              </h3>
 
-            <ProjectPayRewards className="mt-5" />
+              <ProjectPayRewards className="mt-5" />
+            </div>
           </div>
         </div>
-      </div>
 
-      {/* Right panel */}
-      <div className="flex-1 bg-gray-50 px-4 pb-24 pt-12 md:px-8 md:pt-[148px]">
-        <ProjectPayForm className="md:max-w-lg" />
+        {/* Right panel */}
+        <div className="flex-1 bg-gray-50 px-4 pb-24 pt-12 md:px-8 md:pt-[148px]">
+          <ProjectPayForm className="md:max-w-lg" />
+        </div>
       </div>
-    </div>
+    </ProjectPayContext.Provider>
   )
 }
 
diff --git a/src/components/Project/components/PayRewardCard.tsx b/src/components/Project/components/PayRewardCard.tsx
index dab7bc95..e00ebdbd 100644
--- a/src/components/Project/components/PayRewardCard.tsx
+++ b/src/components/Project/components/PayRewardCard.tsx
@@ -6,23 +6,32 @@ import { formatEther } from 'juice-hooks'
 import { RewardDialog } from './RewardDialog'
 import { Button } from '@/components/ui/Button'
 import { Separator } from '@/components/ui/Separator'
+import React from 'react'
+import { CheckIcon } from '@heroicons/react/24/solid'
 
 export type PayRewardCardProps = {
   className?: string
   nft: JB721DelegateTierTier
+  isSelected?: boolean
+  onClick?: () => void
 }
 
 export const PayRewardCard: React.FC<PayRewardCardProps> = ({
   className,
   nft,
+  isSelected,
+  onClick,
 }) => {
   const remaining = nft.remainingQuantity.toString()
   return (
     <div
       className={twMerge(
-        'flex rounded-[10px] border border-gray-200 p-3',
+        'flex cursor-pointer rounded-[10px] border border-gray-200 p-3 shadow-sm transition-all hover:border-gray-300 hover:shadow-md',
+        isSelected &&
+          '-m-[1px] border-2 border-bluebs-500 shadow-md hover:border-bluebs-500',
         className,
       )}
+      onClick={onClick}
     >
       <RewardImage
         className="h-14 w-14 rounded-lg"
@@ -38,16 +47,30 @@ export const PayRewardCard: React.FC<PayRewardCardProps> = ({
         </div>
         <div className="flex h-5 gap-4 text-sm">
           <RewardDialog nft={nft}>
-            <Button size="child" variant="link">
+            <Button className="whitespace-nowrap" size="child" variant="link">
               View details
             </Button>
           </RewardDialog>
           <Separator orientation="vertical" />
-          <span className="text-gray-400">{remaining} remaining</span>
+          <span className="whitespace-nowrap text-gray-400">
+            {remaining} remaining
+          </span>
         </div>
       </div>
 
-      <div className="h-4 w-4 rounded-full border border-gray-200" />
+      <div
+        className={twMerge(
+          'flex h-5 w-5 items-center justify-center rounded-full border',
+          isSelected ? 'border-bluebs-500 bg-bluebs-500' : 'border-gray-200',
+        )}
+      >
+        <CheckIcon
+          className={twMerge(
+            'h-4 w-4 stroke-2 text-white',
+            isSelected ? 'inline' : 'hidden',
+          )}
+        />
+      </div>
     </div>
   )
 }
diff --git a/src/components/Project/components/ProjectPayForm.tsx b/src/components/Project/components/ProjectPayForm.tsx
index fd803e21..f8a33c83 100644
--- a/src/components/Project/components/ProjectPayForm.tsx
+++ b/src/components/Project/components/ProjectPayForm.tsx
@@ -10,10 +10,13 @@ import {
   FormMessage,
 } from '@/components/ui/Form'
 import { zodResolver } from '@hookform/resolvers/zod'
-import { PropsWithChildren, useCallback } from 'react'
+import { PropsWithChildren, useCallback, useMemo } from 'react'
 import { useForm } from 'react-hook-form'
 import { twMerge } from 'tailwind-merge'
 import { z } from 'zod'
+import { useProjectPay } from '../providers/ProjectPayContext'
+import { useJbProject } from '@/hooks/useJbProject'
+import { formatEther } from 'juice-hooks'
 
 const WEI = 1e-18
 
@@ -34,6 +37,9 @@ export type ProjectPayFormProps = {
 export const ProjectPayForm: React.FC<ProjectPayFormProps> = ({
   className,
 }) => {
+  const { nfts } = useJbProject()
+  const { nftRewardIds } = useProjectPay()
+
   const form = useForm<z.infer<typeof formSchema>>({
     resolver: zodResolver(formSchema),
     defaultValues: {
@@ -44,12 +50,23 @@ export const ProjectPayForm: React.FC<ProjectPayFormProps> = ({
     },
   })
 
+  const totalNftSelectionPrice = useMemo(
+    () =>
+      nftRewardIds.reduce((acc, nftId) => {
+        if (!nfts) return acc
+        const nft = nfts.find(nft => nft.id === nftId)
+        if (!nft) return acc
+        return acc + nft.price
+      }, 0n),
+    [nftRewardIds, nfts],
+  )
+
   const onSubmit = useCallback(async (values: z.infer<typeof formSchema>) => {
     console.log(values)
   }, [])
 
   // TODO
-  const total = 0
+  const total = totalNftSelectionPrice
 
   return (
     <Form {...form}>
@@ -105,7 +122,7 @@ export const ProjectPayForm: React.FC<ProjectPayFormProps> = ({
 
         <div className="mt-6 flex justify-between font-medium">
           <div>Total to pay</div>
-          <div>{total} ETH</div>
+          <div>{formatEther(total)} ETH</div>
         </div>
 
         <Button className="mt-2 h-14 w-full" type="submit">
diff --git a/src/components/Project/components/ProjectPayRewards.tsx b/src/components/Project/components/ProjectPayRewards.tsx
index b4efece8..927d7381 100644
--- a/src/components/Project/components/ProjectPayRewards.tsx
+++ b/src/components/Project/components/ProjectPayRewards.tsx
@@ -1,6 +1,8 @@
 import { useJbProject } from '@/hooks/useJbProject'
 import { twMerge } from 'tailwind-merge'
 import { PayRewardCard } from './PayRewardCard'
+import { useProjectPay } from '../providers/ProjectPayContext'
+import { useCallback } from 'react'
 
 export type ProjectPayRewardsProps = {
   className?: string
@@ -10,10 +12,34 @@ export const ProjectPayRewards: React.FC<ProjectPayRewardsProps> = ({
   className,
 }) => {
   const { nfts } = useJbProject()
+  const { dispatch, nftRewardIds: selectedNftIds } = useProjectPay()
+
+  const isSelected = useCallback(
+    (nftId: bigint) => selectedNftIds.includes(nftId),
+    [selectedNftIds],
+  )
+
+  const toggleNft = useCallback(
+    (nftId: bigint) => {
+      if (isSelected(nftId)) {
+        dispatch({ type: 'removeNftReward', id: nftId })
+        return
+      }
+      dispatch({ type: 'addNftReward', id: nftId })
+    },
+    [dispatch, isSelected],
+  )
 
   return (
     <div className={twMerge('flex flex-col gap-3', className)}>
-      {nfts?.map(nft => <PayRewardCard key={nft.id} nft={nft} />)}
+      {nfts?.map(nft => (
+        <PayRewardCard
+          key={nft.id}
+          nft={nft}
+          isSelected={isSelected(nft.id)}
+          onClick={() => toggleNft(nft.id)}
+        />
+      ))}
     </div>
   )
 }
diff --git a/src/components/Project/providers/ProjectPayContext.tsx b/src/components/Project/providers/ProjectPayContext.tsx
new file mode 100644
index 00000000..7e24fcac
--- /dev/null
+++ b/src/components/Project/providers/ProjectPayContext.tsx
@@ -0,0 +1,19 @@
+import React from 'react'
+import {
+  ProjectPayAction,
+  ProjectPayState,
+  projectPayReducer,
+} from './projectPayReducer'
+
+export type ProjectPayContextType = ProjectPayState & {
+  dispatch: React.Dispatch<ProjectPayAction>
+}
+
+export const ProjectPayContext = React.createContext<ProjectPayContextType>({
+  nftRewardIds: [],
+  dispatch: () => {
+    console.error('ProjectPayContext not initialized')
+  },
+})
+
+export const useProjectPay = () => React.useContext(ProjectPayContext)
diff --git a/src/components/Project/providers/projectPayReducer.ts b/src/components/Project/providers/projectPayReducer.ts
new file mode 100644
index 00000000..8a984cdd
--- /dev/null
+++ b/src/components/Project/providers/projectPayReducer.ts
@@ -0,0 +1,27 @@
+export type ProjectPayAction =
+  | { type: 'addNftReward'; id: bigint }
+  | { type: 'removeNftReward'; id: bigint }
+
+export type ProjectPayState = {
+  nftRewardIds: bigint[]
+}
+
+export const projectPayReducer = (
+  state: ProjectPayState,
+  action: ProjectPayAction,
+): ProjectPayState => {
+  switch (action.type) {
+    case 'addNftReward':
+      return {
+        ...state,
+        nftRewardIds: [...state.nftRewardIds, action.id],
+      }
+    case 'removeNftReward':
+      return {
+        ...state,
+        nftRewardIds: state.nftRewardIds.filter(id => id !== action.id),
+      }
+    default:
+      return state
+  }
+}