From 3cfa294cde2a9be9ad41109095a96fd3ef1d2d37 Mon Sep 17 00:00:00 2001
From: saml33 <slam.uke@gmail.com>
Date: Wed, 4 Jan 2023 13:38:17 +1100
Subject: [PATCH] hide settle funds button for unowned account

---
 components/trade/OpenOrders.tsx      | 80 ++++++++++++++--------------
 components/trade/UnsettledTrades.tsx | 56 ++++++++++---------
 2 files changed, 73 insertions(+), 63 deletions(-)

diff --git a/components/trade/OpenOrders.tsx b/components/trade/OpenOrders.tsx
index 3af6ed47..6573c50a 100644
--- a/components/trade/OpenOrders.tsx
+++ b/components/trade/OpenOrders.tsx
@@ -473,46 +473,48 @@ const OpenOrders = () => {
                     </div>
                   )}
                 </div>
-                <div className="flex items-center space-x-3 pl-8">
-                  <div className="flex items-center space-x-2">
-                    {modifyOrderId !== o.orderId.toString() ? (
-                      <>
-                        <IconButton
-                          onClick={() => showEditOrderForm(o, tickSize)}
-                        >
-                          <PencilIcon className="h-4 w-4" />
-                        </IconButton>
-                        <IconButton
-                          disabled={cancelId === o.orderId.toString()}
-                          onClick={() =>
-                            o instanceof PerpOrder
-                              ? handleCancelPerpOrder(o)
-                              : handleCancelSerumOrder(o)
-                          }
-                        >
-                          {cancelId === o.orderId.toString() ? (
-                            <Loading className="h-4 w-4" />
-                          ) : (
-                            <TrashIcon className="h-4 w-4" />
-                          )}
-                        </IconButton>
-                      </>
-                    ) : (
-                      <>
-                        <IconButton onClick={() => modifyOrder(o)}>
-                          {loadingModifyOrder ? (
-                            <Loading className="h-4 w-4" />
-                          ) : (
-                            <CheckIcon className="h-4 w-4" />
-                          )}
-                        </IconButton>
-                        <IconButton onClick={cancelEditOrderForm}>
-                          <XMarkIcon className="h-4 w-4" />
-                        </IconButton>
-                      </>
-                    )}
+                {connected ? (
+                  <div className="flex items-center space-x-3 pl-8">
+                    <div className="flex items-center space-x-2">
+                      {modifyOrderId !== o.orderId.toString() ? (
+                        <>
+                          <IconButton
+                            onClick={() => showEditOrderForm(o, tickSize)}
+                          >
+                            <PencilIcon className="h-4 w-4" />
+                          </IconButton>
+                          <IconButton
+                            disabled={cancelId === o.orderId.toString()}
+                            onClick={() =>
+                              o instanceof PerpOrder
+                                ? handleCancelPerpOrder(o)
+                                : handleCancelSerumOrder(o)
+                            }
+                          >
+                            {cancelId === o.orderId.toString() ? (
+                              <Loading className="h-4 w-4" />
+                            ) : (
+                              <TrashIcon className="h-4 w-4" />
+                            )}
+                          </IconButton>
+                        </>
+                      ) : (
+                        <>
+                          <IconButton onClick={() => modifyOrder(o)}>
+                            {loadingModifyOrder ? (
+                              <Loading className="h-4 w-4" />
+                            ) : (
+                              <CheckIcon className="h-4 w-4" />
+                            )}
+                          </IconButton>
+                          <IconButton onClick={cancelEditOrderForm}>
+                            <XMarkIcon className="h-4 w-4" />
+                          </IconButton>
+                        </>
+                      )}
+                    </div>
                   </div>
-                </div>
+                ) : null}
               </div>
             )
           })
diff --git a/components/trade/UnsettledTrades.tsx b/components/trade/UnsettledTrades.tsx
index c4f82b05..306fc5d7 100644
--- a/components/trade/UnsettledTrades.tsx
+++ b/components/trade/UnsettledTrades.tsx
@@ -14,6 +14,7 @@ import useMangoGroup from 'hooks/useMangoGroup'
 import { PerpMarket, PerpPosition } from '@blockworks-foundation/mango-v4'
 import TableMarketName from './TableMarketName'
 import useMangoAccount from 'hooks/useMangoAccount'
+import { useWallet } from '@solana/wallet-adapter-react'
 
 const UnsettledTrades = ({
   unsettledSpotBalances,
@@ -28,6 +29,7 @@ const UnsettledTrades = ({
   const { width } = useViewport()
   const showTableView = width ? width > breakpoints.md : false
   const { mangoAccount } = useMangoAccount()
+  const { connected } = useWallet()
 
   const handleSettleSerumFunds = useCallback(async (mktAddress: string) => {
     const client = mangoStore.getState().client
@@ -144,7 +146,7 @@ const UnsettledTrades = ({
           <TrHead>
             <Th className="bg-th-bkg-1 text-left">{t('market')}</Th>
             <Th className="bg-th-bkg-1 text-right">{t('trade:amount')}</Th>
-            <Th className="bg-th-bkg-1 text-right" />
+            {connected ? <Th className="bg-th-bkg-1 text-right" /> : null}
           </TrHead>
         </thead>
         <tbody>
@@ -180,22 +182,24 @@ const UnsettledTrades = ({
                     ) : null}
                   </div>
                 </Td>
-                <Td>
-                  <div className="flex justify-end">
-                    <Tooltip content={t('trade:settle-funds')}>
-                      <IconButton
-                        onClick={() => handleSettleSerumFunds(mktAddress)}
-                        size="small"
-                      >
-                        {settleMktAddress === mktAddress ? (
-                          <Loading className="h-4 w-4" />
-                        ) : (
-                          <CheckIcon className="h-4 w-4" />
-                        )}
-                      </IconButton>
-                    </Tooltip>
-                  </div>
-                </Td>
+                {connected ? (
+                  <Td>
+                    <div className="flex justify-end">
+                      <Tooltip content={t('trade:settle-funds')}>
+                        <IconButton
+                          onClick={() => handleSettleSerumFunds(mktAddress)}
+                          size="small"
+                        >
+                          {settleMktAddress === mktAddress ? (
+                            <Loading className="h-4 w-4" />
+                          ) : (
+                            <CheckIcon className="h-4 w-4" />
+                          )}
+                        </IconButton>
+                      </Tooltip>
+                    </div>
+                  </Td>
+                ) : null}
               </TrBody>
             )
           })}
@@ -264,13 +268,17 @@ const UnsettledTrades = ({
                     </span>
                   </span>
                 ) : null}
-                <IconButton onClick={() => handleSettleSerumFunds(mktAddress)}>
-                  {settleMktAddress === mktAddress ? (
-                    <Loading className="h-4 w-4" />
-                  ) : (
-                    <CheckIcon className="h-4 w-4" />
-                  )}
-                </IconButton>
+                {connected ? (
+                  <IconButton
+                    onClick={() => handleSettleSerumFunds(mktAddress)}
+                  >
+                    {settleMktAddress === mktAddress ? (
+                      <Loading className="h-4 w-4" />
+                    ) : (
+                      <CheckIcon className="h-4 w-4" />
+                    )}
+                  </IconButton>
+                ) : null}
               </div>
             </div>
           )