All files / src/hooks/notifications notificationSubscriptions.ts

100% Statements 9/9
100% Branches 0/0
100% Functions 6/6
100% Lines 9/9

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                    12245x 375x     12245x           1714x 192x     1714x           12245x 3x     12245x    
import { ReadyState } from "react-use-websocket";
import {
  NotificationMessageType,
  useNotificationConnection,
} from "@/hooks/notifications/useNotificationConnection.ts";
import { TaskRead } from "@/api/apiStore.gen.ts";
 
export function useTaskCompletedSubscription(
  onMessage: (message: TaskRead) => void,
): ReadyState {
  const { readyState } = useNotificationConnection((message) => {
    onMessage(message as TaskRead);
  }, NotificationMessageType.TASK_COMPLETED);
 
  return readyState;
}
 
export function useTaskUpdatedSubscription(
  onMessage: (message: TaskRead) => void,
): ReadyState {
  const { readyState } = useNotificationConnection((message) => {
    onMessage(message as TaskRead);
  }, NotificationMessageType.TASK_UPDATED);
 
  return readyState;
}
 
export function useTaskCancelledSubscription(
  onMessage: (message: TaskRead) => void,
): ReadyState {
  const { readyState } = useNotificationConnection((message) => {
    onMessage(message as TaskRead);
  }, NotificationMessageType.TASK_CANCELLED);
 
  return readyState;
}