m/metamask-extension
1
0
Fork 0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-12-23 09:52:26 +01:00
Code

fix issues relating to race conditions where draftTx does not exist (#15420)

Browse Source 
* fix issues relating to race conditions where draftTx does not exist

* add another fail safe for inflight initializations
This commit is contained in:
Brad Decker 2022-08-02 13:52:09 -05:00 committed by GitHub
parent fb191c865b
commit f425e482e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 59 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
ui/ducks/send/send.js
View File

@ -603,7 +603,7 @@ export const initializeSendState = createAsyncThunk(
// For instance, in the actions.js file we dispatch this action anytime the // For instance, in the actions.js file we dispatch this action anytime the
// chain changes. // chain changes.
if (!draftTransaction) { if (!draftTransaction) {
thunkApi.rejectWithValue( return thunkApi.rejectWithValue(
'draftTransaction not found, possibly not on send flow', 'draftTransaction not found, possibly not on send flow',
); );
} }
@ -678,6 +678,20 @@ export const initializeSendState = createAsyncThunk(
// We have to keep the gas slice in sync with the send slice state // We have to keep the gas slice in sync with the send slice state
// so that it'll be initialized correctly if the gas modal is opened. // so that it'll be initialized correctly if the gas modal is opened.
await thunkApi.dispatch(setCustomGasLimit(gasLimit)); await thunkApi.dispatch(setCustomGasLimit(gasLimit));
// There may be a case where the send has been canceled by the user while
// the gas estimate is being computed. So we check again to make sure that
// a currentTransactionUUID exists and matches the previous tx.
const newState = thunkApi.getState();
if (
newState.send.currentTransactionUUID !== sendState.currentTransactionUUID
) {
return thunkApi.rejectWithValue(
`draftTransaction changed during initialization.
A new initializeSendState action must be dispatched.`,
);
}
return { return {
account, account,
chainId: getCurrentChainId(state), chainId: getCurrentChainId(state),
@ -1396,6 +1410,7 @@ const slice = createSlice({
validateSendState: (state) => { validateSendState: (state) => {
const draftTransaction = const draftTransaction =
state.draftTransactions[state.currentTransactionUUID]; state.draftTransactions[state.currentTransactionUUID];
if (draftTransaction) {
switch (true) { switch (true) {
case Boolean(draftTransaction.amount.error): case Boolean(draftTransaction.amount.error):
case Boolean(draftTransaction.gas.error): case Boolean(draftTransaction.gas.error):
@ -1419,6 +1434,7 @@ const slice = createSlice({
default: default:
draftTransaction.status = SEND_STATUSES.VALID; draftTransaction.status = SEND_STATUSES.VALID;
} }
}
}, },
}, },
extraReducers: (builder) => { extraReducers: (builder) => {
@ -1512,15 +1528,9 @@ const slice = createSlice({
state.selectedAccount.balance = action.payload.account.balance; state.selectedAccount.balance = action.payload.account.balance;
const draftTransaction = const draftTransaction =
state.draftTransactions[state.currentTransactionUUID]; state.draftTransactions[state.currentTransactionUUID];
if (draftTransaction) {
draftTransaction.gas.gasLimit = action.payload.gasLimit; draftTransaction.gas.gasLimit = action.payload.gasLimit;
slice.caseReducers.updateGasFeeEstimates(state, {
payload: {
gasFeeEstimates: action.payload.gasFeeEstimates,
gasEstimateType: action.payload.gasEstimateType,
},
});
draftTransaction.gas.gasTotal = action.payload.gasTotal; draftTransaction.gas.gasTotal = action.payload.gasTotal;
state.gasEstimatePollToken = action.payload.gasEstimatePollToken;
if (action.payload.chainHasChanged) { if (action.payload.chainHasChanged) {
// If the state was reinitialized as a result of the user changing // If the state was reinitialized as a result of the user changing
// the network from the network dropdown, then the selected asset is // the network from the network dropdown, then the selected asset is
@ -1532,6 +1542,14 @@ const slice = createSlice({
state.selectedAccount.balance; state.selectedAccount.balance;
draftTransaction.asset.details = null; draftTransaction.asset.details = null;
} }
}
slice.caseReducers.updateGasFeeEstimates(state, {
payload: {
gasFeeEstimates: action.payload.gasFeeEstimates,
gasEstimateType: action.payload.gasEstimateType,
},
});
state.gasEstimatePollToken = action.payload.gasEstimatePollToken;
if (action.payload.gasEstimatePollToken) { if (action.payload.gasEstimatePollToken) {
state.gasEstimateIsLoading = false; state.gasEstimateIsLoading = false;
} }
@ -2641,7 +2659,11 @@ export function isSendStateInitialized(state) {
* @type {Selector<boolean>} * @type {Selector<boolean>}
*/ */
export function isSendFormInvalid(state) { export function isSendFormInvalid(state) {
return getCurrentDraftTransaction(state).status === SEND_STATUSES.INVALID; const draftTransaction = getCurrentDraftTransaction(state);
if (!draftTransaction) {
return true;
}
return draftTransaction.status === SEND_STATUSES.INVALID;
} }
/** /**