File size: 958 Bytes
c2b7eb3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import type {
  EntityId,
  EntityState,
  EntityStateAdapter,
  EntityStateFactory,
} from './models'

export function getInitialEntityState<T, Id extends EntityId>(): EntityState<
  T,
  Id
> {
  return {
    ids: [],
    entities: {} as Record<Id, T>,
  }
}

export function createInitialStateFactory<T, Id extends EntityId>(
  stateAdapter: EntityStateAdapter<T, Id>,
): EntityStateFactory<T, Id> {
  function getInitialState(
    state?: undefined,
    entities?: readonly T[] | Record<Id, T>,
  ): EntityState<T, Id>
  function getInitialState<S extends object>(
    additionalState: S,
    entities?: readonly T[] | Record<Id, T>,
  ): EntityState<T, Id> & S
  function getInitialState(
    additionalState: any = {},
    entities?: readonly T[] | Record<Id, T>,
  ): any {
    const state = Object.assign(getInitialEntityState(), additionalState)
    return entities ? stateAdapter.setAll(state, entities) : state
  }

  return { getInitialState }
}