Carlos Florêncio
22 lines
729 B
TypeScript
22 lines
729 B
TypeScript
import type { StoreApi } from "zustand";
|
|
import type { AppState } from "@/lib/state/store";
|
|
import type { WsHandlers } from "@/lib/ws/handlers/types";
|
|
import type { GitHubRateLimitUpdate, TaskPR } from "@/lib/types/github";
|
|
|
|
export function registerGitHubHandlers(store: StoreApi<AppState>): WsHandlers {
|
|
return {
|
|
"github.task_pr.updated": (message) => {
|
|
const pr = message.payload as TaskPR;
|
|
if (pr.task_id) {
|
|
store.getState().setTaskPR(pr.task_id, pr);
|
|
}
|
|
},
|
|
"github.rate_limit.updated": (message) => {
|
|
const update = message.payload as GitHubRateLimitUpdate;
|
|
if (update?.snapshots?.length) {
|
|
store.getState().applyGitHubRateLimitUpdate(update);
|
|
}
|
|
},
|
|
};
|
|
}
|