reactant-share/src/modules/storage.ts

A framework for building shared web applications with Reactant

/* eslint-disable @typescript-eslint/no-empty-interface */
import { injectable, inject } from 'reactant';
import {
  Storage as BaseReactantStorage,
  StorageOptions,
  REHYDRATE,
  getRehydrated,
} from 'reactant-storage';
import type {
  IStorageOptions as IBaseStorageOptions,
  SetStorageOptions,
} from 'reactant-storage';
import { PortDetector } from './portDetector';
import { storageModuleName } from '../constants';

export interface IStorageOptions extends IBaseStorageOptions {
  /**
   * disable client rehydrated
   */
  disableClientRehydrated?: boolean;
}

@injectable({
  name: storageModuleName,
})
class ReactantStorage extends BaseReactantStorage {
  constructor(
    protected portDetector: PortDetector,
    @inject(StorageOptions) public options: IStorageOptions
  ) {
    super(options);
    this.onRehydrated(() => {
      this.portDetector.onServer(() => {
        this.persist();
      });
      this.portDetector.onClient(() => {
        this.pause();
      });
      if (this.portDetector.isServer) {
        this.portDetector.syncToClients();
      }
    });
  }

  /**
   * set module to storage persistent
   */
  override setStorage<T extends object>(
    target: T,
    options: SetStorageOptions<T>
  ) {
    return super.setStorage(
      target,
      this.options.disableClientRehydrated && this.portDetector.isClient
        ? { ...options, blacklist: [], whitelist: [] }
        : options
    );
  }
}

export {
  ReactantStorage as Storage,
  StorageOptions,
  REHYDRATE,
  SetStorageOptions,
  getRehydrated,
};