28 lines
996 B
TypeScript
28 lines
996 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, TaskCIAutomationOptions, 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.task_ci_options.updated": (message) => {
|
|
const options = message.payload as TaskCIAutomationOptions;
|
|
if (options.task_id) {
|
|
store.getState().setTaskCIAutomationOptions(options.task_id, options);
|
|
}
|
|
},
|
|
"github.rate_limit.updated": (message) => {
|
|
const update = message.payload as GitHubRateLimitUpdate;
|
|
if (update?.snapshots?.length) {
|
|
store.getState().applyGitHubRateLimitUpdate(update);
|
|
}
|
|
},
|
|
};
|
|
}
|