From 5f95d74e985cb202c46d402b1d1307c40e6b8dd6 Mon Sep 17 00:00:00 2001
From: Iveta <quietbits@gmail.com>
Date: Thu, 5 Sep 2024 17:19:04 -0400
Subject: [PATCH] Fix SEP-6 flow

---
 .../src/components/Sep6Deposit.tsx            | 33 +++++++++----------
 .../src/components/Sep6QuoteModal.tsx         |  6 +++-
 .../src/components/Sep6Withdraw.tsx           | 33 +++++++++----------
 .../src/ducks/sep6Deposit.ts                  | 11 ++++++-
 .../src/ducks/sep6Withdraw.ts                 | 20 ++++++-----
 5 files changed, 59 insertions(+), 44 deletions(-)

diff --git a/packages/demo-wallet-client/src/components/Sep6Deposit.tsx b/packages/demo-wallet-client/src/components/Sep6Deposit.tsx
index 049737f7..ae37aa63 100644
--- a/packages/demo-wallet-client/src/components/Sep6Deposit.tsx
+++ b/packages/demo-wallet-client/src/components/Sep6Deposit.tsx
@@ -297,23 +297,21 @@ export const Sep6Deposit = () => {
                 ))}
               </div>
 
-              {a.country_codes?.length && a.country_codes.length > 1 ? (
-                <div className="Sep6Selection">
-                  <div className="Sep6Selection__label">Country code</div>
-                  {a.country_codes?.map((c) => (
-                    <RadioButton
-                      key={c}
-                      name="deposit-asset-country-code"
-                      id={c}
-                      label={c}
-                      onChange={() => {
-                        setSelectedDepositAssetCountryCode(c);
-                      }}
-                      checked={c === selectedDepositAssetCountryCode}
-                    />
-                  ))}
-                </div>
-              ) : null}
+              <div className="Sep6Selection">
+                <div className="Sep6Selection__label">Country code</div>
+                {a.country_codes?.map((c) => (
+                  <RadioButton
+                    key={c}
+                    name="deposit-asset-country-code"
+                    id={c}
+                    label={c}
+                    onChange={() => {
+                      setSelectedDepositAssetCountryCode(c);
+                    }}
+                    checked={c === selectedDepositAssetCountryCode}
+                  />
+                ))}
+              </div>
             </div>
           ))}
         </div>
@@ -458,6 +456,7 @@ export const Sep6Deposit = () => {
   ) {
     return (
       <Sep6QuoteModal
+        type="deposit"
         quote={sep6Deposit.data.quote}
         onClose={handleClose}
         onSubmit={handleSubmitWithQuote}
diff --git a/packages/demo-wallet-client/src/components/Sep6QuoteModal.tsx b/packages/demo-wallet-client/src/components/Sep6QuoteModal.tsx
index f54a75f2..c194563e 100644
--- a/packages/demo-wallet-client/src/components/Sep6QuoteModal.tsx
+++ b/packages/demo-wallet-client/src/components/Sep6QuoteModal.tsx
@@ -3,12 +3,14 @@ import { CSS_MODAL_PARENT_ID } from "demo-wallet-shared/build/constants/settings
 import { AnchorQuote } from "types/types";
 
 type Sep6QuoteModalProps = {
+  type: "deposit" | "withdrawal";
   quote: AnchorQuote;
   onClose: () => void;
   onSubmit: (event: React.MouseEvent<HTMLButtonElement, MouseEvent>) => void;
 };
 
 export const Sep6QuoteModal = ({
+  type,
   quote,
   onClose,
   onSubmit,
@@ -66,7 +68,9 @@ export const Sep6QuoteModal = ({
 
   return (
     <Modal visible onClose={onClose} parentId={CSS_MODAL_PARENT_ID}>
-      <Modal.Heading>SEP-6 Deposit Quote</Modal.Heading>
+      <Modal.Heading>{`SEP-6 ${
+        type === "deposit" ? "Deposit" : "Withdrawal"
+      } Quote`}</Modal.Heading>
       <Modal.Body>
         <p>
           These prices are indicative. The actual price will be calculated at
diff --git a/packages/demo-wallet-client/src/components/Sep6Withdraw.tsx b/packages/demo-wallet-client/src/components/Sep6Withdraw.tsx
index b56d3616..c035aa38 100644
--- a/packages/demo-wallet-client/src/components/Sep6Withdraw.tsx
+++ b/packages/demo-wallet-client/src/components/Sep6Withdraw.tsx
@@ -316,23 +316,21 @@ export const Sep6Withdraw = () => {
                 ))}
               </div>
 
-              {a.country_codes?.length && a.country_codes.length > 1 ? (
-                <div className="Sep6Selection">
-                  <div className="Sep6Selection__label">Country code</div>
-                  {a.country_codes?.map((c) => (
-                    <RadioButton
-                      key={c}
-                      name="withdraw-asset-country-code"
-                      id={c}
-                      label={c}
-                      onChange={() => {
-                        setSelectedWithdrawAssetCountryCode(c);
-                      }}
-                      checked={c === selectedWithdrawAssetCountryCode}
-                    />
-                  ))}
-                </div>
-              ) : null}
+              <div className="Sep6Selection">
+                <div className="Sep6Selection__label">Country code</div>
+                {a.country_codes?.map((c) => (
+                  <RadioButton
+                    key={c}
+                    name="withdraw-asset-country-code"
+                    id={c}
+                    label={c}
+                    onChange={() => {
+                      setSelectedWithdrawAssetCountryCode(c);
+                    }}
+                    checked={c === selectedWithdrawAssetCountryCode}
+                  />
+                ))}
+              </div>
             </div>
           ))}
         </div>
@@ -502,6 +500,7 @@ export const Sep6Withdraw = () => {
   ) {
     return (
       <Sep6QuoteModal
+        type="withdrawal"
         quote={sep6Withdraw.data.quote}
         onClose={handleClose}
         onSubmit={handleSubmitWithQuote}
diff --git a/packages/demo-wallet-client/src/ducks/sep6Deposit.ts b/packages/demo-wallet-client/src/ducks/sep6Deposit.ts
index 2d628a68..ace85874 100644
--- a/packages/demo-wallet-client/src/ducks/sep6Deposit.ts
+++ b/packages/demo-wallet-client/src/ducks/sep6Deposit.ts
@@ -122,7 +122,7 @@ export const initiateDepositAction = createAsyncThunk<
         });
 
         depositAssets = quotesResult.assets.filter(
-          (a) => a.asset !== buyAsset && a.buy_delivery_methods,
+          (a) => a.asset !== buyAsset && a.sell_delivery_methods,
         );
 
         log.instruction({
@@ -131,6 +131,15 @@ export const initiateDepositAction = createAsyncThunk<
         });
       }
 
+      const actionType =
+        supportsQuotes && depositAssets && depositAssets?.length > 0
+          ? AnchorActionType.DEPOSIT_EXCHANGE
+          : AnchorActionType.DEPOSIT;
+
+      log.instruction({
+        title: `Selected ${actionType} path`,
+      });
+
       // Get either deposit or deposit-exchange asset data
       const assetInfoData =
         infoData[
diff --git a/packages/demo-wallet-client/src/ducks/sep6Withdraw.ts b/packages/demo-wallet-client/src/ducks/sep6Withdraw.ts
index 04488706..fec4a420 100644
--- a/packages/demo-wallet-client/src/ducks/sep6Withdraw.ts
+++ b/packages/demo-wallet-client/src/ducks/sep6Withdraw.ts
@@ -122,7 +122,7 @@ export const initiateWithdrawAction = createAsyncThunk<
         });
 
         withdrawAssets = quotesResult.assets.filter(
-          (a) => a.asset !== sellAsset && a.sell_delivery_methods,
+          (a) => a.asset !== sellAsset && a.buy_delivery_methods,
         );
 
         log.instruction({
@@ -131,13 +131,17 @@ export const initiateWithdrawAction = createAsyncThunk<
         });
       }
 
-      // Get either deposit or deposit-exchange asset data
-      const assetInfoData =
-        infoData[
-          supportsQuotes && withdrawAssets && withdrawAssets?.length > 0
-            ? AnchorActionType.WITHDRAW_EXCHANGE
-            : AnchorActionType.WITHDRAWAL
-        ]?.[assetCode];
+      const actionType =
+        supportsQuotes && withdrawAssets && withdrawAssets?.length > 0
+          ? AnchorActionType.WITHDRAW_EXCHANGE
+          : AnchorActionType.WITHDRAWAL;
+
+      log.instruction({
+        title: `Selected ${actionType} path`,
+      });
+
+      // Get either withdarw or withdraw-exchange asset data
+      const assetInfoData = infoData[actionType]?.[assetCode];
 
       // This is unlikely
       if (!assetInfoData) {