packages/account/vpn/serversCount.ts

   1 import { createSlice } from '@reduxjs/toolkit';
   2
   3 import type { ProtonThunkArguments } from '@proton/redux-shared-store-types';
   4 import { createAsyncModelThunk, handleAsyncModel, previousSelector } from '@proton/redux-utilities';
   5 import type { VPNServersCountData } from '@proton/shared/lib/interfaces';
   6 import { defaultVPNServersCountData, getVPNServersCountData } from '@proton/shared/lib/vpn/serversCount';
   7
   8 import { getInitialModelState } from '../initialModelState';
   9 import type { ModelState } from '../interface';
  10
  11 const name = 'vpnServersCount' as const;
  12
  13 export interface VPNServersCountState {
  14     [name]: ModelState<VPNServersCountData>;
  15 }
  16
  17 type SliceState = VPNServersCountState[typeof name];
  18 type Model = NonNullable<SliceState['value']>;
  19
  20 export const selectVpnServersCount = (state: VPNServersCountState) => state[name];
  21
  22 const modelThunk = createAsyncModelThunk<Model, VPNServersCountState, ProtonThunkArguments>(`${name}/fetch`, {
  23     miss: async ({ extraArgument }) => {
  24         return getVPNServersCountData(extraArgument.api);
  25     },
  26     previous: previousSelector(selectVpnServersCount),
  27 });
  28
  29 const initialState = getInitialModelState<Model>(defaultVPNServersCountData);
  30 const slice = createSlice({
  31     name,
  32     initialState,
  33     reducers: {},
  34     extraReducers: (builder) => {
  35         handleAsyncModel(builder, modelThunk);
  36     },
  37 });
  38
  39 export const vpnServersCountReducer = { [name]: slice.reducer };
  40 export const vpnServersCountThunk = modelThunk.thunk;