chore: add assertInInjectionContext

Author: crutchcorn

projects/angular-redux/src/lib/inject-dispatch.ts

@@ -1,12 +1,10 @@
-import type { Action, Dispatch, UnknownAction } from 'redux'
-import { ReduxProvider } from './provider'
-import { inject } from '@angular/core'
+import type { Dispatch, UnknownAction } from 'redux'
+import { assertInInjectionContext } from '@angular/core'
+import { injectStore } from './inject-store'
 
 // TODO: Add `withTypes` support
 export function injectDispatch<AppDispatch extends Dispatch<UnknownAction> = Dispatch<UnknownAction>>(): AppDispatch {
-  const context = inject(ReduxProvider)
-  // TODO: assertInInjectionContext
-  const { store } = context
-
+  assertInInjectionContext(injectDispatch)
+  const store = injectStore();
   return store.dispatch as AppDispatch
 }

projects/angular-redux/src/lib/inject-selector.ts

@@ -1,5 +1,5 @@
 import { EqualityFn } from './types'
-import { effect, inject, Signal, signal } from '@angular/core'
+import { assertInInjectionContext, effect, inject, Signal, signal } from '@angular/core'
 import { ReduxProvider } from './provider'
 
 export interface UseSelectorOptions<Selected = unknown> {
@@ -13,6 +13,7 @@ export function injectSelector<TState = unknown, Selected = unknown>(
   selector: (state: TState) => Selected,
   equalityFnOrOptions?: EqualityFn<Selected> | UseSelectorOptions<Selected>,
 ): Signal<Selected> {
+  assertInInjectionContext(injectSelector)
   const reduxContext = inject(ReduxProvider);
 
   // const { equalityFn = refEquality } =

projects/angular-redux/src/lib/inject-store.ts

@@ -1,10 +1,10 @@
-import { inject } from "@angular/core";
+import { assertInInjectionContext, inject } from '@angular/core'
 import { ReduxProvider } from './provider'
 import type { Store, Action, UnknownAction } from 'redux'
 
 export function injectStore<A extends Action<string> = UnknownAction, S = unknown>(): Store<S, A> {
+  assertInInjectionContext(injectStore)
   const context = inject(ReduxProvider)
-  // TODO: assertInInjectionContext
   const { store } = context
   return store
 }