forked from LedgerHQ/ledger-live-desktop
/
state.js
117 lines (103 loc) · 3.23 KB
/
state.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
// @flow
import { useState, useCallback, useMemo } from "react";
import type { Account, SubAccount } from "@ledgerhq/live-common/lib/types/account";
import { makeEmptyTokenAccount } from "@ledgerhq/live-common/lib/account";
import type { CryptoCurrency, TokenCurrency } from "@ledgerhq/live-common/lib/types/currencies";
type CryptoOrTokenCurrency = TokenCurrency | CryptoCurrency;
export type AccountTuple = {
account: ?Account,
subAccount: ?SubAccount,
};
function getAccountTuplesForCurrency(
currency: CryptoOrTokenCurrency,
allAccounts: Account[],
): AccountTuple[] {
if (currency.type === "TokenCurrency") {
return allAccounts
.filter(account => account.currency.id === currency.parentCurrency.id)
.map(account => ({
account,
subAccount:
(account.subAccounts &&
account.subAccounts.find(
(subAcc: SubAccount) =>
subAcc.type === "TokenAccount" && subAcc.token.id === currency.id,
)) ||
makeEmptyTokenAccount(account, currency),
}));
}
return allAccounts
.filter(account => account.currency.id === currency.id)
.map(account => ({
account,
subAccount: null,
}));
}
const getIdsFromTuple = (accountTuple: AccountTuple) => ({
accountId: accountTuple.account ? accountTuple.account.id : null,
subAccountId: accountTuple.subAccount ? accountTuple.subAccount.id : null,
});
export function useCurrencyAccountSelect(
allCurrencies: CryptoOrTokenCurrency[],
allAccounts: Account[],
) {
const [state, setState] = useState(() => {
const defaultCurrency = allCurrencies[0];
const availableAccounts = getAccountTuplesForCurrency(defaultCurrency, allAccounts);
const { accountId } = availableAccounts.length
? getIdsFromTuple(availableAccounts[0])
: { accountId: null };
return {
currency: defaultCurrency,
accountId,
};
});
const { currency, accountId } = state;
const setCurrency = useCallback(
(currency: ?CryptoOrTokenCurrency) => {
if (currency) {
const availableAccounts = getAccountTuplesForCurrency(currency, allAccounts);
const { accountId } = availableAccounts.length
? getIdsFromTuple(availableAccounts[0])
: { accountId: null };
return setState(currState => ({
...currState,
currency,
accountId,
}));
}
return setState(currState => ({
...currState,
currency,
accountId: null,
}));
},
[allAccounts],
);
const setAccount = useCallback((account: ?Account, subAccount: ?SubAccount) => {
setState(currState => ({
...currState,
accountId: account ? account.id : null,
}));
}, []);
const availableAccounts = useMemo(
() => (currency ? getAccountTuplesForCurrency(currency, allAccounts) : []),
[currency, allAccounts],
);
const { account, subAccount } = useMemo(() => {
return (
availableAccounts.find(tuple => (tuple.account ? tuple.account.id === accountId : false)) || {
account: null,
subAccount: null,
}
);
}, [availableAccounts, accountId]);
return {
availableAccounts,
currency,
account,
subAccount,
setAccount,
setCurrency,
};
}