multi: Add ledger ui.

Copy trezor elements and apply to ledger where possible.
decred · Sep 7, 2023 · 4624b6d · 4624b6d
1 parent f615386
commit 4624b6d
Show file tree

Hide file tree

Showing 55 changed files with 783 additions and 131 deletions.
diff --git a/app/actions/DaemonActions.js b/app/actions/DaemonActions.js
@@ -9,6 +9,7 @@ import { stopNotifcations } from "./NotificationActions";
 import { saveSettings, updateStateSettingsChanged } from "./SettingsActions";
 import { rescanCancel, showCantCloseModal } from "./ControlActions";
 import { enableTrezor } from "./TrezorActions";
+import { enableLedger } from "./LedgerActions";
 import {
   DEX_LOGOUT_ATTEMPT,
   DEX_LOGOUT_SUCCESS,
@@ -309,7 +310,7 @@ export const removeWallet = (selectedWallet) => (dispatch) => {
 // selectedWallet = {
 //   label: newWalletName,
 //   value: {
-//    wallet: newWalletName, isWatchingOnly, isTrezor, isNew,
+//    wallet: newWalletName, isWatchingOnly, isTrezor, isLedger, isNew
 //    network: isTestNet ? "testnet" : "mainnet"
 //  }
 // }
@@ -336,7 +337,8 @@ export const createWallet = (selectedWallet) => (dispatch, getState) =>
         dispatch({
           isWatchingOnly: selectedWallet.value.isWatchingOnly,
           createNewWallet: selectedWallet.value.isNew,
-          isTrezor: selectedWallet.value.istrezor,
+          isTrezor: selectedWallet.value.isTrezor,
+          isLedger: selectedWallet.value.isLedger,
           type: WALLETCREATED
         });
         dispatch(setSelectedWallet(selectedWallet));
@@ -509,6 +511,7 @@ export const startWallet =
           confirmDexSeed
         });
         selectedWallet.value.isTrezor && dispatch(enableTrezor());
+        selectedWallet.value.isLedger && dispatch(enableLedger());
         await dispatch(getVersionServiceAttempt());
         await dispatch(openWalletAttempt("", false, selectedWallet));
         return discoverAccountsComplete;

diff --git a/app/actions/LedgerActions.js b/app/actions/LedgerActions.js
@@ -14,7 +14,6 @@ import {
 const coin = "decred";
 
 import * as selectors from "selectors";
-import * as cfgConstants from "constants/config";
 
 export const LDG_LEDGER_ENABLED = "LDG_LEDGER_ENABLED";
 export const LDG_WALLET_CLOSED = "LDG_WALLET_CLOSED";
@@ -26,18 +25,7 @@ export const LDG_WALLET_CLOSED = "LDG_WALLET_CLOSED";
 // enableLedger only sets a value in the config. Ledger connections are made
 // per action then dropped.
 export const enableLedger = () => (dispatch, getState) => {
-  const walletName = selectors.getWalletName(getState());
-
-  if (walletName) {
-    const config = wallet.getWalletCfg(
-      selectors.isTestNet(getState()),
-      walletName
-    );
-    config.set(cfgConstants.LEDGER, true);
-  }
-
   dispatch({ type: LDG_LEDGER_ENABLED });
-
   connect()(dispatch, getState);
 };
 

diff --git a/app/actions/TrezorActions.js b/app/actions/TrezorActions.js
@@ -9,7 +9,6 @@ import {
   addressPath
 } from "helpers/trezor";
 import { publishTransactionAttempt } from "./ControlActions";
-import * as cfgConstants from "constants/config";
 import {
   MODEL1_DECRED_HOMESCREEN,
   MODELT_DECRED_HOMESCREEN
@@ -49,16 +48,6 @@ export const TRZ_TREZOR_ENABLED = "TRZ_TREZOR_ENABLED";
 // enableTrezor attepts to start a connection with connect if none exist and
 // connect to a trezor device.
 export const enableTrezor = () => (dispatch, getState) => {
-  const walletName = selectors.getWalletName(getState());
-
-  if (walletName) {
-    const config = wallet.getWalletCfg(
-      selectors.isTestNet(getState()),
-      walletName
-    );
-    config.set(cfgConstants.TREZOR, true);
-  }
-
   dispatch({ type: TRZ_TREZOR_ENABLED });
 
   if (!setListeners) {

diff --git a/app/actions/WalletLoaderActions.js b/app/actions/WalletLoaderActions.js
@@ -18,7 +18,7 @@ import {
   getBestBlockHeightAttempt
 } from "./ClientActions";
 import { WALLETREMOVED_FAILED } from "./DaemonActions";
-import { isTestNet, trezorDevice } from "selectors";
+import { isTestNet, trezorDevice, ledgerDevice } from "selectors";
 import { walletrpc as api } from "middleware/walletrpc/api_pb";
 import { push as pushHistory } from "connected-react-router";
 import { stopNotifcations } from "./NotificationActions";
@@ -28,6 +28,7 @@ import * as cfgConstants from "constants/config";
 import { RESCAN_PROGRESS } from "./ControlActions";
 import { stopAccountMixer } from "./AccountMixerActions";
 import { TRZ_WALLET_CLOSED } from "actions/TrezorActions";
+import { LDG_WALLET_CLOSED } from "actions/LedgerActions";
 import { saveSettings, updateStateSettingsChanged } from "./SettingsActions";
 
 const { SyncNotificationType } = api;
@@ -156,7 +157,7 @@ export const CREATEWATCHONLYWALLET_FAILED = "CREATEWATCHONLYWALLET_FAILED";
 export const CREATEWATCHONLYWALLET_SUCCESS = "CREATEWATCHONLYWALLET_SUCCESS";
 
 export const createWatchOnlyWalletRequest =
-  (extendedPubKey, pubPass = "") =>
+  (extendedPubKey, isLedger, isTrezor, pubPass = "") =>
   (dispatch, getState) =>
     new Promise((resolve, reject) => {
       dispatch({ type: CREATEWATCHONLYWALLET_ATTEMPT });
@@ -172,6 +173,12 @@ export const createWatchOnlyWalletRequest =
           } = getState();
           const config = wallet.getWalletCfg(isTestNet(getState()), walletName);
           config.set(cfgConstants.IS_WATCH_ONLY, true);
+          if (isTrezor) {
+            config.set(cfgConstants.TREZOR, true);
+          }
+          if (isLedger) {
+            config.set(cfgConstants.LEDGER, true);
+          }
           config.delete(cfgConstants.DISCOVER_ACCOUNTS);
           wallet.setIsWatchingOnly(true);
           dispatch({ response: {}, type: CREATEWATCHONLYWALLET_SUCCESS });
@@ -269,6 +276,7 @@ const finalCloseWallet = () => async (dispatch, getState) => {
     await wallet.stopWallet();
     dispatch({ type: CLOSEWALLET_SUCCESS });
     if (trezorDevice(getState())) dispatch({ type: TRZ_WALLET_CLOSED });
+    if (ledgerDevice(getState())) dispatch({ type: LDG_WALLET_CLOSED });
     dispatch(pushHistory("/getstarted/initial"));
   } catch (error) {
     dispatch({ error, type: CLOSEWALLET_FAILED });

diff --git a/app/components/SideBar/MenuLinks/Links.jsx b/app/components/SideBar/MenuLinks/Links.jsx
@@ -4,6 +4,7 @@ export const LN_KEY = "ln";
 export const DEX_KEY = "dex";
 export const TICKETS_KEY = "tickets";
 export const GOV_KEY = "governance";
+export const PRIV_KEY = "privacy";
 
 export const linkList = [
   {
@@ -48,6 +49,7 @@ export const linkList = [
     path: "/privacy",
     link: <T id="sidebar.link.privacy" m="Privacy and Security" />,
     icon: "securitycntr",
+    key: PRIV_KEY,
     ariaLabel: "Privacy"
   },
   {

diff --git a/app/components/SideBar/MenuLinks/hooks.js b/app/components/SideBar/MenuLinks/hooks.js
@@ -1,7 +1,14 @@
 import { useEffect, useState, useMemo } from "react";
 import { useSelector } from "react-redux";
 import * as sel from "selectors";
-import { linkList, LN_KEY, DEX_KEY, TICKETS_KEY, GOV_KEY } from "./Links";
+import {
+  linkList,
+  LN_KEY,
+  DEX_KEY,
+  TICKETS_KEY,
+  GOV_KEY,
+  PRIV_KEY
+} from "./Links";
 import { useHistory } from "react-router-dom";
 import { cloneDeep } from "fp";
 
@@ -10,6 +17,7 @@ export function useMenuLinks() {
   const sidebarOnBottom = useSelector(sel.sidebarOnBottom);
   const expandSideBar = useSelector(sel.expandSideBar);
   const isTrezor = useSelector(sel.isTrezor);
+  const isLedger = useSelector(sel.isLedger);
   const lnEnabled = useSelector(sel.lnEnabled);
 
   const newActiveVoteProposalsCount = useSelector(
@@ -47,19 +55,22 @@ export function useMenuLinks() {
       links = links.filter((l) => l.key !== LN_KEY);
     }
     // TODO: Enable ticket purchacing for Trezor.
-    if (isTrezor) {
+    if (isTrezor || isLedger) {
       links = links.filter(
         (l) => l.key !== DEX_KEY && l.key !== TICKETS_KEY && l.key !== GOV_KEY
       );
     }
+    if (isLedger) {
+      links = links.filter((l) => l.key !== PRIV_KEY);
+    }
     return links.map((link) => ({
       ...link,
       notifProp: link.notifProp?.reduce(
         (acc, np) => acc + (notifProps[np] || 0),
         0
       )
     }));
-  }, [notifProps, isTrezor, lnEnabled]);
+  }, [notifProps, isTrezor, lnEnabled, isLedger]);
 
   const [activeTabIndex, setActiveTabIndex] = useState(-1);
   const history = useHistory();

diff --git a/app/components/buttons/SendTransactionButton/SendTransactionButton.jsx b/app/components/buttons/SendTransactionButton/SendTransactionButton.jsx
@@ -16,8 +16,10 @@ const SendTransactionButton = ({
     unsignedTransaction,
     isSendingTransaction,
     isTrezor,
+    isLedger,
     onAttemptSignTransaction,
-    onAttemptSignTransactionTrezor
+    onAttemptSignTransactionTrezor,
+    onAttemptSignTransactionLedger
   } = useSendTransactionButton();
 
   const signTransaction = (privpass) => {
@@ -31,6 +33,12 @@ const SendTransactionButton = ({
     onSubmit?.();
   };
 
+  const signTransactionLedger = () => {
+    if (disabled) return;
+    onAttemptSignTransactionLedger?.(unsignedTransaction);
+    onSubmit?.();
+  };
+
   if (isTrezor) {
     return (
       <KeyBlueButton
@@ -40,6 +48,15 @@ const SendTransactionButton = ({
         {buttonLabel ? buttonLabel : <T id="send.sendBtn" m="Send" />}
       </KeyBlueButton>
     );
+  } else if (isLedger) {
+    return (
+      <KeyBlueButton
+        onClick={signTransactionLedger}
+        disabled={disabled || isSendingTransaction}
+        loading={isSendingTransaction}>
+        {buttonLabel ? buttonLabel : <T id="send.sendBtn" m="Send" />}
+      </KeyBlueButton>
+    );
   } else {
     return (
       <PassphraseModalButton

diff --git a/app/components/buttons/SendTransactionButton/hooks.js b/app/components/buttons/SendTransactionButton/hooks.js
@@ -2,26 +2,32 @@ import { useDispatch, useSelector } from "react-redux";
 import * as sel from "selectors";
 import * as ca from "actions/ControlActions";
 import * as tza from "actions/TrezorActions";
+import * as ldgr from "actions/LedgerActions";
 
 export function useSendTransactionButton() {
   const unsignedTransaction = useSelector(sel.unsignedTransaction);
   const constructTxResponse = useSelector(sel.constructTxResponse);
   const isSendingTransaction = useSelector(sel.isSendingTransaction);
   const isTrezor = useSelector(sel.isTrezor);
+  const isLedger = useSelector(sel.isLedger);
 
   const dispatch = useDispatch();
   const onAttemptSignTransaction = (passphrase, rawTx, acct) => {
     dispatch(ca.signTransactionAttempt(passphrase, rawTx, acct));
   };
   const onAttemptSignTransactionTrezor = (rawUnsigTx, constructTxResponse) =>
     dispatch(tza.signTransactionAttemptTrezor(rawUnsigTx, constructTxResponse));
+  const onAttemptSignTransactionLedger = (rawUnsigTx) =>
+    dispatch(ldgr.signTransactionAttemptLedger(rawUnsigTx));
 
   return {
     unsignedTransaction,
     constructTxResponse,
     isSendingTransaction,
     isTrezor,
+    isLedger,
     onAttemptSignTransaction,
-    onAttemptSignTransactionTrezor
+    onAttemptSignTransactionTrezor,
+    onAttemptSignTransactionLedger
   };
 }
diff --git a/app/components/shared/SendTransaction/Form.jsx b/app/components/shared/SendTransaction/Form.jsx
@@ -22,6 +22,7 @@ const Form = ({
   unsignedRawTx,
   isWatchingOnly,
   isTrezor,
+  isLedger,
   insuficientFunds,
   styles,
   hideDetails,
@@ -71,7 +72,9 @@ const Form = ({
               <T id="send.insuficient.funds" m="Insufficient funds" />
             </div>
           )}
-          {((isTrezor && isWatchingOnly) || !isWatchingOnly) &&
+          {((isTrezor && isWatchingOnly) ||
+            (isLedger && isWatchingOnly) ||
+            !isWatchingOnly) &&
             (getRunningIndicator ? (
               <Tooltip
                 contentClassName={styles.disabledTooltip}
@@ -144,7 +147,7 @@ const Form = ({
         </div>
       </div>
     </div>
-    {unsignedRawTx && isWatchingOnly && !isTrezor && (
+    {unsignedRawTx && isWatchingOnly && !isTrezor && !isLedger && (
       <UnsignedTx
         title={<T id="send.unsignedRawTxTite" m="Unsigned Raw Transaction:" />}
         tx={unsignedRawTx}

diff --git a/app/components/shared/SendTransaction/SendTransaction.jsx b/app/components/shared/SendTransaction/SendTransaction.jsx
@@ -32,6 +32,7 @@ const SendTransaction = ({
     totalSpent,
     notMixedAccounts,
     isTrezor,
+    isLedger,
     isWatchingOnly,
     isConstructingTransaction,
     attemptConstructTransaction,
@@ -394,6 +395,7 @@ const SendTransaction = ({
         willEnter,
         isWatchingOnly,
         isTrezor,
+        isLedger,
         insuficientFunds,
         styles,
         hideDetails,

diff --git a/app/components/shared/SendTransaction/hooks.js b/app/components/shared/SendTransaction/hooks.js
@@ -24,6 +24,7 @@ export function useSendTransaction() {
     shallowEqual
   );
   const isTrezor = useSelector(sel.isTrezor);
+  const isLedger = useSelector(sel.isLedger);
   const isWatchingOnly = useSelector(sel.isWatchingOnly);
   const isConstructingTransaction = useSelector(sel.isConstructingTransaction);
   const constructTxRequestAttempt = useSelector(sel.constructTxRequestAttempt);
@@ -61,6 +62,7 @@ export function useSendTransaction() {
     totalSpent,
     notMixedAccounts,
     isTrezor,
+    isLedger,
     isWatchingOnly,
     isConstructingTransaction,
     attemptConstructTransaction,

diff --git a/app/components/views/GetStartedPage/CreateWalletPage/CreateWalletPage.jsx b/app/components/views/GetStartedPage/CreateWalletPage/CreateWalletPage.jsx
@@ -103,18 +103,16 @@ const CreateWalletPage = ({ createWalletRef, onSendBack }) => {
     cancelCreateWallet
   ]);
 
-  const onCreateWatchOnly = useCallback(() => {
-    createWatchOnlyWalletRequest(walletMasterPubKey)
-      .then(() => sendEvent({ type: "WALLET_CREATED" }))
-      .catch((error) => sendEvent({ type: "ERROR", error }));
-    // we send a continue so we go to loading state
-    sendContinue();
-  }, [
-    createWatchOnlyWalletRequest,
-    sendEvent,
-    sendContinue,
-    walletMasterPubKey
-  ]);
+  const onCreateWatchOnly = useCallback(
+    (isLedger, isTrezor) => {
+      createWatchOnlyWalletRequest(walletMasterPubKey, isLedger, isTrezor)
+        .then(() => sendEvent({ type: "WALLET_CREATED" }))
+        .catch((error) => sendEvent({ type: "ERROR", error }));
+      // we send a continue so we go to loading state
+      sendContinue();
+    },
+    [createWatchOnlyWalletRequest, sendEvent, sendContinue, walletMasterPubKey]
+  );
 
   const getStateComponent = useCallback(() => {
     const { mnemonic, error } = current.context;
@@ -188,7 +186,8 @@ const CreateWalletPage = ({ createWalletRef, onSendBack }) => {
   ]);
 
   useEffect(() => {
-    const { isNew, walletMasterPubKey, mnemonic } = current.context;
+    const { isNew, walletMasterPubKey, mnemonic, isLedger, isTrezor } =
+      current.context;
     switch (current.value) {
       case "createWalletInit":
         setIsNew(isNew);
@@ -222,7 +221,7 @@ const CreateWalletPage = ({ createWalletRef, onSendBack }) => {
         checkIsValid();
         break;
       case "restoreWatchingOnly":
-        onCreateWatchOnly();
+        onCreateWatchOnly(isLedger, isTrezor);
         break;
       case "finished":
         break;

diff --git a/app/components/views/GetStartedPage/CreateWalletPage/hooks.js b/app/components/views/GetStartedPage/CreateWalletPage/hooks.js
@@ -21,8 +21,15 @@ export const useCreateWallet = () => {
   );
   // TODO implement pubpass
   const createWatchOnlyWalletRequest = useCallback(
-    (extendedPubKey, pubPass = "") =>
-      dispatch(wla.createWatchOnlyWalletRequest(extendedPubKey, pubPass)),
+    (extendedPubKey, isLedger, isTrezor, pubPass = "") =>
+      dispatch(
+        wla.createWatchOnlyWalletRequest(
+          extendedPubKey,
+          isLedger,
+          isTrezor,
+          pubPass
+        )
+      ),
     [dispatch]
   );
   const createWalletRequest = useCallback(