on 8a36573 help sign in
bridge/src/store/rollerStore.ts
71 lines 2.5 kB history raw
1import create from 'zustand';
2import { PendingTransaction } from '@urbit/roller-api';
4import { HOUR } from 'lib/utils/roller';
5import { Invite, Invites } from 'lib/types/Invite';
6import Point, { Points, Relationship } from 'lib/types/Point';
7import { toL1Details } from 'lib/utils/point';
9export const EMPTY_POINT = new Point(-1, Relationship.own, toL1Details({}), '');
11export interface RollerStore {
12 nextBatchTime: number;
13 pendingTransactions: PendingTransaction[];
14 point: Point;
15 points: Points;
16 pointList: Point[];
17 invites: Invites;
18 recentlyCompleted: number;
19 inviteGeneratingNum: number;
20 setNextBatchTime: (nextBatchTime: number) => void;
21 setPendingTransactions: (pendingTransactions: PendingTransaction[]) => void;
22 setPoint: (point: number) => void;
23 setPoints: (points: Points) => void;
24 setInvites: (points: number, invites: Invite[]) => void;
25 setInviteGeneratingNum: (numGenerating: number) => void;
26 updateInvite: (point: number, invite: Invite) => void;
29export const useRollerStore = create<RollerStore>(set => ({
30 nextBatchTime: new Date().getTime() + HOUR,
31 pendingTransactions: [],
32 point: EMPTY_POINT,
33 pointList: [],
34 points: {},
35 invites: {},
36 recentlyCompleted: 0,
37 inviteGeneratingNum: 0,
38 setNextBatchTime: (nextBatchTime: number) => set(() => ({ nextBatchTime })),
39 setPendingTransactions: (pendingTransactions: PendingTransaction[]) =>
40 set(state => {
41 const oldPending = state.pendingTransactions.length;
42 if (oldPending > 0 && pendingTransactions.length === 0) {
43 return {
44 ...state,
45 pendingTransactions,
46 recentlyCompleted: oldPending,
47 };
48 }
49 return { ...state, pendingTransactions };
50 }),
51 setPoint: (point: number) =>
52 set(state => ({ point: state.points[point] || EMPTY_POINT })),
53 setPoints: (points: Points) =>
54 set(() => ({ points, pointList: Object.values(points) })),
55 setInvites: (point: number, invites: Invite[]) =>
56 set(state => {
57 const newInvites: Invites = {};
58 newInvites[point] = invites;
59 return { invites: Object.assign(state.invites, newInvites) };
60 }),
61 setInviteGeneratingNum: (inviteGeneratingNum: number) =>
62 set(() => ({ inviteGeneratingNum })),
63 updateInvite: (point: number, invite: Invite) =>
64 set(state => {
65 const newInvites: Invites = {};
66 newInvites[point] = state.invites[point]?.map(i =>
67 i.planet === invite.planet ? invite : i
68 );
69 return { invites: Object.assign(state.invites, newInvites) };
70 }),
71}));