All files / packages/hooks/src useEffectService.ts

100% Statements 8/8
100% Branches 2/2
100% Functions 3/3
100% Lines 8/8

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55            1x                         1x       8x                                   10x   10x 2x       8x 6x          
import React, { createContext, useContext, useMemo } from "react";
 
import type { Context } from "effect";
import { ManagedRuntime } from "effect";
 
// Create a context to hold the ManagedRuntime
const EffectRuntimeContext = createContext<ManagedRuntime.ManagedRuntime<
  unknown,
  unknown
> | null>(null);
 
export interface EffectProviderProps<R, E> {
  runtime: ManagedRuntime.ManagedRuntime<R, E>;
  children: React.ReactNode;
}
 
/**
 * Provides an Effect Runtime to the application tree.
 */
export const EffectProvider = <R, E>({
  runtime,
  children,
}: EffectProviderProps<R, E>) => {
  return React.createElement(
    EffectRuntimeContext.Provider,
    {
      value: runtime as unknown as ManagedRuntime.ManagedRuntime<
        unknown,
        unknown
      >,
    },
    children,
  );
};
 
/**
 * Hook to access a service from the Effect Runtime.
 */
export function useEffectService<Identifier, Service>(
  tag: Context.Tag<Identifier, Service>,
): Service {
  const runtime = useContext(EffectRuntimeContext);
 
  if (!runtime) {
    throw new Error("useEffectService must be used within an EffectProvider");
  }
 
  // Resolve the service once per runtime/tag pair.
  return useMemo(() => {
    return (
      runtime as ManagedRuntime.ManagedRuntime<Identifier, Service>
    ).runSync(tag);
  }, [runtime, tag]);
}