src/store/websites.ts


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

import { produce } from 'immer';
import { create } from 'zustand';
import type { DateRange } from '@/lib/types';

const store = create(() => ({}));

export function setWebsiteDateRange(websiteId: string, dateRange: DateRange) {
  store.setState(
    produce(state => {
      if (!state[websiteId]) {
        state[websiteId] = {};
      }

      state[websiteId].dateRange = { ...dateRange, modified: Date.now() };

      return state;
    }),
  );
}

export function setWebsiteDateCompare(websiteId: string, dateCompare: string) {
  store.setState(
    produce(state => {
      if (!state[websiteId]) {
        state[websiteId] = {};
      }

      state[websiteId].dateCompare = dateCompare;

      return state;
    }),
  );
}

export const useWebsites = store;