import {
  GridImperativeAPI,
  OnCellsRenderedData,
  ScrollerGridProps,
  useSearchParamsWithCache
} from "@etsoo/react";
import React from "react";
import { GridUtils } from "../GridUtils";
import { ExtendUtils } from "@etsoo/shared";

export function gridCacheKeyGenerator(cacheKey: string) {
  return `${cacheKey}-grid-scroll`;
}

export function useGridCacheInitLoad<T extends object>(
  cacheKey: string | undefined,
  cacheMinutes: number
): ScrollerGridProps<T>["onInitLoad"] {
  // Reference
  const ref = React.useRef<boolean>(null);

  // Search data
  const searchData = useSearchParamsWithCache(cacheKey);

  // Avoid repeatedly load from cache
  if (ref.current || !cacheKey) return undefined;

  // Cache data
  const cacheData = GridUtils.getCacheData<T>(cacheKey, cacheMinutes);
  if (cacheData) {
    const { rows, state } = cacheData;

    GridUtils.mergeSearchData(state, searchData);

    // Update flag value
    ref.current = true;

    return (ref: GridImperativeAPI) => {
      // Scroll position
      const scrollData = sessionStorage.getItem(
        gridCacheKeyGenerator(cacheKey)
      );
      if (scrollData) {
        const data = JSON.parse(scrollData) as OnCellsRenderedData;
        ExtendUtils.waitFor(
          () =>
            ref.scrollToCell({
              rowIndex: data.rowStartIndex,
              columnIndex: data.columnStartIndex
            }),
          100
        );
      }

      // Return cached rows and state
      return [rows, state];
    };
  }

  return undefined;
}