AFFiNE-Mirror/packages/component/src/components/page-list/use-all-page-setting.ts

import type { DBSchema } from 'idb';
import { openDB } from 'idb';
import type { IDBPDatabase } from 'idb/build/entry';
import { useAtom } from 'jotai';
import { atomWithReset } from 'jotai/utils';
import { useCallback } from 'react';
import useSWRImmutable from 'swr/immutable';
import { NIL } from 'uuid';

import { evalFilterList } from './filter';
import type { Filter, VariableMap, View } from './filter/shared-types';

type PersistenceView = View;

export interface PageViewDBV1 extends DBSchema {
  view: {
    key: PersistenceView['id'];
    value: PersistenceView;
  };
}

const pageViewDBPromise: Promise<IDBPDatabase<PageViewDBV1>> =
  environment.isServer
    ? // never resolve in SSR
      new Promise<any>(() => {})
    : openDB<PageViewDBV1>('page-view', 1, {
        upgrade(database) {
          database.createObjectStore('view', {
            keyPath: 'id',
          });
        },
      });

const currentViewAtom = atomWithReset<View>({
  name: 'default',
  id: NIL,
  filterList: [],
});

export const useAllPageSetting = () => {
  const { data: savedViews, mutate } = useSWRImmutable(
    ['affine', 'page-view'],
    {
      fetcher: async () => {
        const db = await pageViewDBPromise;
        const t = db.transaction('view').objectStore('view');
        return await t.getAll();
      },
      suspense: true,
      fallbackData: [],
      revalidateOnMount: true,
    }
  );

  const [currentView, setCurrentView] = useAtom(currentViewAtom);

  const createView = useCallback(
    async (view: View) => {
      if (view.id === NIL) {
        return;
      }
      const db = await pageViewDBPromise;
      const t = db.transaction('view', 'readwrite').objectStore('view');
      await t.put(view);
      await mutate();
    },
    [mutate]
  );

  return {
    currentView,
    savedViews: savedViews as View[],

    // actions
    createView,
    setCurrentView,
  };
};
export const filterByFilterList = (filterList: Filter[], varMap: VariableMap) =>
  evalFilterList(filterList, varMap);