git.iko.soy/bridge
on 04765bb help sign in
bridge/src/store/wallet.js
160 lines 4.2 kB history raw
1import React, {
2 createContext,
3 forwardRef,
4 useCallback,
5 useContext,
6 useState,
7} from 'react';
8import Maybe from 'folktale/maybe';
9import { includes } from 'lodash';
11import {
12 WALLET_TYPES,
13 DEFAULT_HD_PATH,
14 addressFromSecp256k1Public,
15 walletFromMnemonic,
16} from 'lib/wallet';
17import { BRIDGE_ERROR } from 'lib/error';
19export const WalletContext = createContext(null);
21const kDefaultWalletType = WALLET_TYPES.TICKET;
23// NB (jtobin):
24//
25// Note that the 'wallet' prop has type depending on the 'walletType' prop.
26// These could be bound together in a single structure (so that
27// 'walletType' acted more explicitly as a data constructor of sorts) but
28// it doesn't necessarily help us much, since we branch on 'walletType'
29// before setting 'wallet'.
30//
31// Wallets are always wrapped in Maybe. For most authentication
32// mechanisms, Maybe wraps a BIP32 HD wallet; for Ethereum private keys,
33// JSON keystore files, and Metamask authentication, it wraps an
34// 'EthereumWallet'.
35function _useWallet(
36 initialWallet = Maybe.Nothing(),
37 initialMnemonic = Maybe.Nothing()
38) {
39 const [walletType, _setWalletType] = useState(kDefaultWalletType);
40 const [walletHdPath, setWalletHdPath] = useState(DEFAULT_HD_PATH);
41 const [wallet, _setWallet] = useState(initialWallet);
42 const [urbitWallet, _setUrbitWallet] = useState(Maybe.Nothing());
43 const [authMnemonic, setAuthMnemonic] = useState(initialMnemonic);
44 const [networkSeed, setNetworkSeed] = useState(Maybe.Nothing());
45 const [networkRevision, setNetworkRevision] = useState(Maybe.Nothing());
47 const setWalletType = useCallback(
48 walletType => {
49 if (!includes(WALLET_TYPES, walletType)) {
50 throw BRIDGE_ERROR.INVALID_WALLET_TYPE;
51 }
53 _setWalletType(walletType);
54 },
55 [_setWalletType]
56 );
58 const setWallet = useCallback(
59 wallet => {
60 // force that public addresses are derived for each wallet
61 wallet.map(wal => {
62 wal.address = wal.address || addressFromSecp256k1Public(wal.publicKey);
63 return wal;
64 });
66 _setWallet(wallet);
67 },
68 [_setWallet]
69 );
71 const setUrbitWallet = useCallback(
72 urbitWallet => {
73 if (Maybe.Just.hasInstance(urbitWallet)) {
74 // when an urbit wallet is set, also derive
75 // a normal bip32 wallet using the ownership address
76 const wallet = walletFromMnemonic(
77 urbitWallet.value.ownership.seed,
78 DEFAULT_HD_PATH,
79 urbitWallet.value.meta.passphrase
80 );
82 // manually set the public address
83 wallet.map(wal => {
84 wal.address = urbitWallet.value.ownership.keys.address;
85 return wal;
86 });
88 setWallet(wallet);
89 } else {
90 setWallet(Maybe.Nothing());
91 }
93 _setUrbitWallet(urbitWallet);
94 },
95 [setWallet]
96 );
98 const resetWallet = useCallback(() => {
99 _setWalletType(kDefaultWalletType);
100 setWalletHdPath(DEFAULT_HD_PATH);
101 _setWallet(Maybe.Nothing());
102 _setUrbitWallet(Maybe.Nothing());
103 setAuthMnemonic(Maybe.Nothing());
104 setNetworkSeed(Maybe.Nothing());
105 setNetworkRevision(Maybe.Nothing());
106 }, [
107 _setWalletType,
108 setWalletHdPath,
109 _setWallet,
110 _setUrbitWallet,
111 setAuthMnemonic,
112 setNetworkSeed,
113 setNetworkRevision,
114 ]);
116 return {
117 //
118 walletType,
119 setWalletType,
120 //
121 walletHdPath,
122 setWalletHdPath,
123 //
124 wallet,
125 setWallet,
126 //
127 urbitWallet,
128 setUrbitWallet,
129 authMnemonic,
130 setAuthMnemonic,
131 //
132 networkSeed,
133 setNetworkSeed,
134 networkRevision,
135 setNetworkRevision,
136 //
137 resetWallet,
138 };
141export function WalletProvider({ initialWallet, initialMnemonic, children }) {
142 const wallet = _useWallet(initialWallet, initialMnemonic);
144 return (
145 <WalletContext.Provider value={wallet}>{children}</WalletContext.Provider>
146 );
149// Hook version
150export function useWallet() {
151 return useContext(WalletContext);
154// HOC version
155export const withWallet = Component =>
156 forwardRef((props, ref) => (
157 <WalletContext.Consumer>
158 {wallet => <Component ref={ref} {...wallet} {...props} />}
159 </WalletContext.Consumer>
160 ));