routers subscription.ts

11.11% Statements 1/9
0% Branches 0/8
20% Functions 1/5
11.11% Lines 1/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
38
39
40
41
42
43
44
45
46  
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11x
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  /**
 * Subscription Router — SSE event streams
 */
 
import { z } from 'zod';
import type { ProcedureBuilder } from '../trpc.js';
import {
  eventBusIterable,
  ALL_EVENT_TYPES,
  AGENT_EVENT_TYPES,
  TASK_EVENT_TYPES,
  PAGE_EVENT_TYPES,
} from '../subscriptions.js';
 
export function subscriptionProcedures(publicProcedure: ProcedureBuilder) {
  return {
    onEvent: publicProcedure
      .input(z.object({ lastEventId: z.string().nullish() }).optional())
      .subscription(async function* (opts) {
        const signal = opts.signal ?? new AbortController().signal;
        yield* eventBusIterable(opts.ctx.eventBus, ALL_EVENT_TYPES, signal);
      }),
 
    onAgentUpdate: publicProcedure
      .input(z.object({ lastEventId: z.string().nullish() }).optional())
      .subscription(async function* (opts) {
        const signal = opts.signal ?? new AbortController().signal;
        yield* eventBusIterable(opts.ctx.eventBus, AGENT_EVENT_TYPES, signal);
      }),
 
    onTaskUpdate: publicProcedure
      .input(z.object({ lastEventId: z.string().nullish() }).optional())
      .subscription(async function* (opts) {
        const signal = opts.signal ?? new AbortController().signal;
        yield* eventBusIterable(opts.ctx.eventBus, TASK_EVENT_TYPES, signal);
      }),
 
    onPageUpdate: publicProcedure
      .input(z.object({ lastEventId: z.string().nullish() }).optional())
      .subscription(async function* (opts) {
        const signal = opts.signal ?? new AbortController().signal;
        yield* eventBusIterable(opts.ctx.eventBus, PAGE_EVENT_TYPES, signal);
      }),
  };
}