packages/wallet/store/slices/remainingInvites.ts

   1 import { createAction, createSlice } from '@reduxjs/toolkit';
   2
   3 import type { ModelState } from '@proton/account';
   4 import { getInitialModelState } from '@proton/account';
   5 import { createAsyncModelThunk, handleAsyncModel, previousSelector } from '@proton/redux-utilities';
   6
   7 import type { WalletThunkArguments } from '../thunk';
   8
   9 const name = 'remaining_invites' as const;
  10
  11 export interface RemainingInvitesState {
  12     [name]: ModelState<{ available: number; used: number }>;
  13 }
  14
  15 type SliceState = RemainingInvitesState[typeof name];
  16 type Model = NonNullable<SliceState['value']>;
  17
  18 export const selectRemainingInvites = (state: RemainingInvitesState) => {
  19     return state[name];
  20 };
  21
  22 export const decrementAvailableInvites = createAction('decrement available invites', () => ({ payload: {} }));
  23
  24 const modelThunk = createAsyncModelThunk<Model, RemainingInvitesState, WalletThunkArguments>(`${name}/fetch`, {
  25     miss: async ({ extraArgument }) => {
  26         const invites = await extraArgument.walletApi
  27             .clients()
  28             .invite.getRemainingMonthlyInvitation()
  29             .then((data) => ({ available: data.Available, used: data.Used }))
  30             .catch(() => ({ available: 0, used: 0 }));
  31
  32         return invites;
  33     },
  34     previous: previousSelector(selectRemainingInvites),
  35 });
  36
  37 const initialState = getInitialModelState<Model>();
  38
  39 const slice = createSlice({
  40     name,
  41     initialState,
  42     reducers: {},
  43     extraReducers: (builder) => {
  44         handleAsyncModel(builder, modelThunk);
  45         builder.addCase(decrementAvailableInvites, (state) => {
  46             if (state.value && state.value.available > 0) {
  47                 state.value.available--;
  48             }
  49         });
  50     },
  51 });
  52
  53 export const remainingInvitesReducer = { [name]: slice.reducer };
  54 export const remainingInvitesThunk = modelThunk.thunk;