|
3 | 3 | getSpannedDays, |
4 | 4 | toLocalISODateString, |
5 | 5 | } from "../common/utils/dateUtils"; |
| 6 | +import { persist } from "zustand/middleware"; |
6 | 7 |
|
7 | 8 | interface CalendarEvent { |
8 | 9 | id: string; |
@@ -30,79 +31,92 @@ interface CalendarStoreState { |
30 | 31 | updateEvent(event: CalendarEvent): void; |
31 | 32 | } |
32 | 33 |
|
33 | | -export const useCalendarStore = create<CalendarStoreState>()((set, get) => ({ |
34 | | - events: {}, |
35 | | - _eventIdsByDay: {}, |
36 | | - addEvent(event) { |
37 | | - set((state) => { |
38 | | - const events = { ...state.events }; |
39 | | - const eventIdsByDay = { ...state._eventIdsByDay }; |
40 | | - events[event.id] = event; |
| 34 | +export const useCalendarStore = create<CalendarStoreState>()( |
| 35 | + persist( |
| 36 | + (set, get) => ({ |
| 37 | + events: {}, |
| 38 | + _eventIdsByDay: {}, |
| 39 | + addEvent(event) { |
| 40 | + set((state) => { |
| 41 | + const events = { ...state.events }; |
| 42 | + const eventIdsByDay = { ...state._eventIdsByDay }; |
| 43 | + events[event.id] = event; |
41 | 44 |
|
42 | | - for (const day of getSpannedDays(event.start, event.end)) { |
43 | | - const key = toLocalISODateString(day); |
44 | | - eventIdsByDay[key] = [...(eventIdsByDay[key] ?? []), event.id]; |
45 | | - } |
| 45 | + for (const day of getSpannedDays(event.start, event.end)) { |
| 46 | + const key = toLocalISODateString(day); |
| 47 | + eventIdsByDay[key] = [...(eventIdsByDay[key] ?? []), event.id]; |
| 48 | + } |
46 | 49 |
|
47 | | - return { events, _eventIdsByDay: eventIdsByDay }; |
48 | | - }); |
49 | | - }, |
50 | | - removeEvent(id) { |
51 | | - set((state) => { |
52 | | - if (!state.events[id]) return {}; |
53 | | - const events = { ...state.events }; |
54 | | - const event = state.events[id]; |
55 | | - delete events[id]; |
| 50 | + return { events, _eventIdsByDay: eventIdsByDay }; |
| 51 | + }); |
| 52 | + }, |
| 53 | + removeEvent(id) { |
| 54 | + set((state) => { |
| 55 | + if (!state.events[id]) return {}; |
| 56 | + const events = { ...state.events }; |
| 57 | + const event = state.events[id]; |
| 58 | + delete events[id]; |
56 | 59 |
|
57 | | - const eventIdsByDay = { ...state._eventIdsByDay }; |
| 60 | + const eventIdsByDay = { ...state._eventIdsByDay }; |
58 | 61 |
|
59 | | - for (const day of getSpannedDays(event.start, event.end)) { |
60 | | - const key = toLocalISODateString(day); |
61 | | - if (!eventIdsByDay[key] || !eventIdsByDay[key].includes(id)) continue; |
62 | | - const eventIds = eventIdsByDay[key].filter((eventId) => eventId !== id); |
63 | | - eventIdsByDay[key] = eventIds; |
| 62 | + for (const day of getSpannedDays(event.start, event.end)) { |
| 63 | + const key = toLocalISODateString(day); |
| 64 | + if (!eventIdsByDay[key] || !eventIdsByDay[key].includes(id)) |
| 65 | + continue; |
| 66 | + const eventIds = eventIdsByDay[key].filter( |
| 67 | + (eventId) => eventId !== id, |
| 68 | + ); |
| 69 | + eventIdsByDay[key] = eventIds; |
64 | 70 |
|
65 | | - if (!eventIdsByDay[key].length) { |
66 | | - delete eventIdsByDay[key]; |
67 | | - } |
68 | | - } |
| 71 | + if (!eventIdsByDay[key].length) { |
| 72 | + delete eventIdsByDay[key]; |
| 73 | + } |
| 74 | + } |
69 | 75 |
|
70 | | - return { events, _eventIdsByDay: eventIdsByDay }; |
71 | | - }); |
72 | | - }, |
73 | | - updateEvent(event) { |
74 | | - if (!get().events[event.id]) { |
75 | | - get().addEvent(event); |
76 | | - return; |
77 | | - } |
78 | | - set((state) => { |
79 | | - // the updated event may span different days. |
80 | | - // we could be smarter/more performant about this, but |
81 | | - // lets just be lazy and clear the _eventIdsByDay for |
82 | | - // the current range and add them for the new range. |
| 76 | + return { events, _eventIdsByDay: eventIdsByDay }; |
| 77 | + }); |
| 78 | + }, |
| 79 | + updateEvent(event) { |
| 80 | + if (!get().events[event.id]) { |
| 81 | + get().addEvent(event); |
| 82 | + return; |
| 83 | + } |
| 84 | + set((state) => { |
| 85 | + // the updated event may span different days. |
| 86 | + // we could be smarter/more performant about this, but |
| 87 | + // lets just be lazy and clear the _eventIdsByDay for |
| 88 | + // the current range and add them for the new range. |
83 | 89 |
|
84 | | - const id = event.id; |
85 | | - const events = { ...state.events }; |
86 | | - const eventIdsByDay = { ...state._eventIdsByDay }; |
87 | | - events[id] = event; |
| 90 | + const id = event.id; |
| 91 | + const events = { ...state.events }; |
| 92 | + const eventIdsByDay = { ...state._eventIdsByDay }; |
| 93 | + events[id] = event; |
88 | 94 |
|
89 | | - for (const day of getSpannedDays(event.start, event.end)) { |
90 | | - const key = toLocalISODateString(day); |
91 | | - if (!eventIdsByDay[key] || !eventIdsByDay[key].includes(id)) continue; |
92 | | - const eventIds = eventIdsByDay[key].filter((eventId) => eventId !== id); |
93 | | - eventIdsByDay[key] = eventIds; |
| 95 | + for (const day of getSpannedDays(event.start, event.end)) { |
| 96 | + const key = toLocalISODateString(day); |
| 97 | + if (!eventIdsByDay[key] || !eventIdsByDay[key].includes(id)) |
| 98 | + continue; |
| 99 | + const eventIds = eventIdsByDay[key].filter( |
| 100 | + (eventId) => eventId !== id, |
| 101 | + ); |
| 102 | + eventIdsByDay[key] = eventIds; |
94 | 103 |
|
95 | | - if (!eventIdsByDay[key].length) { |
96 | | - delete eventIdsByDay[key]; |
97 | | - } |
98 | | - } |
| 104 | + if (!eventIdsByDay[key].length) { |
| 105 | + delete eventIdsByDay[key]; |
| 106 | + } |
| 107 | + } |
99 | 108 |
|
100 | | - for (const day of getSpannedDays(event.start, event.end)) { |
101 | | - const key = toLocalISODateString(day); |
102 | | - eventIdsByDay[key] = [...(eventIdsByDay[key] ?? []), event.id]; |
103 | | - } |
| 109 | + for (const day of getSpannedDays(event.start, event.end)) { |
| 110 | + const key = toLocalISODateString(day); |
| 111 | + eventIdsByDay[key] = [...(eventIdsByDay[key] ?? []), event.id]; |
| 112 | + } |
104 | 113 |
|
105 | | - return { _eventIdsByDay: eventIdsByDay, events }; |
106 | | - }); |
107 | | - }, |
108 | | -})); |
| 114 | + return { _eventIdsByDay: eventIdsByDay, events }; |
| 115 | + }); |
| 116 | + }, |
| 117 | + }), |
| 118 | + { |
| 119 | + name: "calendar", |
| 120 | + }, |
| 121 | + ), |
| 122 | +); |
0 commit comments