qlik-oss/sn-table

import { useCallback, useEffect, useState } from "react";
import { Column, TableLayout } from "../../types";
import { TableContext, useContextSelector } from "../context";
import modelStore from "../utils/model-store";

const SELECTION_ACTIONS_ENABLED_DEFAULT_STATUS: Record<string, boolean> = {
  canSelectAll: false,
  canClearSelections: false,
  canSelectPossible: false,
  canSelectAlternative: false,
  canSelectExcluded: false,
};

export interface UseFieldSelectionOutput {
  fieldInstance: EngineAPI.IField | null;
  selectionActionsEnabledStatus: Record<string, boolean>;
  resetSelectionActionsEnabledStatus: () => void;
  updateSelectionActionsEnabledStatus: (layout: TableLayout) => void;
}

export const checkStateCountByKey = <T>(keys: (keyof T)[], obj: T): boolean =>
  keys.some((key) => (obj[key] as number) > 0);

export const getListBoxSessionObject = (qLibraryId: string, qStateName = "$") => ({
  qInfo: {
    qType: "tableListbox",
  },
  qListObjectDef: {
    qLibraryId,
    qStateName,
  },
});

const useFieldSelection = (column: Column, openMenuDropdown: boolean): UseFieldSelectionOutput => {
  const { app, layout } = useContextSelector(TableContext, (value) => value.baseProps);
  const [fieldInstance, setFieldInstance] = useState<EngineAPI.IField | null>(null);
  const [selectionActionsEnabledStatus, setSelectionActionsEnabledStatus] = useState(
    SELECTION_ACTIONS_ENABLED_DEFAULT_STATUS,
  );

  useEffect(() => {
    if (!app || !app.getField || !column || !column.isDim || !openMenuDropdown) return;
    const { qLibraryId, fieldId } = column;
    if (qLibraryId) {
      const key = `${app.id}-${qLibraryId}`;
      if (modelStore.get(key)) {
        setFieldInstance(modelStore.get(key) as EngineAPI.IField);
      } else {
        app.createSessionObject(getListBoxSessionObject(qLibraryId, layout.qStateName)).then((listboxSessionObject) => {
          const newFieldInstance = {
            selectAll: () => listboxSessionObject.selectListObjectAll("/qListObjectDef"),
            clear: () => listboxSessionObject.clearSelections("/qListObjectDef").then((result) => result),
            selectPossible: () => listboxSessionObject.selectListObjectPossible("/qListObjectDef"),
            selectAlternative: () => listboxSessionObject.selectListObjectAlternative("/qListObjectDef"),
            selectExcluded: () => listboxSessionObject.selectListObjectExcluded("/qListObjectDef"),
          } as EngineAPI.IField;
          setFieldInstance(newFieldInstance);
          modelStore.set(key, newFieldInstance);
        });
      }
    } else {
      app.getField(fieldId, layout.qStateName).then(setFieldInstance);
    }
  }, [app, column, layout.qStateName, openMenuDropdown]);

  const resetSelectionActionsEnabledStatus = useCallback(
    () => setSelectionActionsEnabledStatus(SELECTION_ACTIONS_ENABLED_DEFAULT_STATUS),
    [],
  );
  const updateSelectionActionsEnabledStatus = (latestLayout: TableLayout) => {
    const dimInfo = latestLayout.qHyperCube.qDimensionInfo.find((dim) => dim.qFallbackTitle === column.label);
    if (!dimInfo) return;
    setSelectionActionsEnabledStatus({
      canSelectAll: checkStateCountByKey(["qOption", "qAlternative", "qExcluded", "qDeselected"], dimInfo.qStateCounts),
      canClearSelections: checkStateCountByKey(["qSelected"], dimInfo.qStateCounts),
      canSelectPossible: checkStateCountByKey(["qOption"], dimInfo.qStateCounts),
      canSelectAlternative: checkStateCountByKey(["qAlternative"], dimInfo.qStateCounts),
      canSelectExcluded: checkStateCountByKey(["qAlternative", "qExcluded"], dimInfo.qStateCounts),
    });
  };

  return {
    fieldInstance,
    selectionActionsEnabledStatus,
    resetSelectionActionsEnabledStatus,
    updateSelectionActionsEnabledStatus,
  };
};

export default useFieldSelection;