@osdk/react-components 0.2.0-beta.2 → 0.2.0-beta.21

package/build/esm/object-table/hooks/useFunctionColumnsData.js

@@ -0,0 +1,270 @@
+/*
+ * Copyright 2026 Palantir Technologies, Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { useOsdkClient } from "@osdk/react";
+import { useEffect, useMemo, useRef, useState } from "react";
+import { createAsyncCellData } from "../utils/AsyncCellData.js";
+export function useFunctionColumnsData(objectSet, objects, columnDefinitions) {
+  const client = useOsdkClient();
+  const [data, setData] = useState({});
+  const abortControllerRef = useRef(null);
+
+  // Function column configurations grouped by unique query definition
+  const functionColumnConfigs = useMemo(() => getFunctionColumnConfigs(columnDefinitions), [columnDefinitions]);
+  const stableObjects = useStableObjects(objects);
+
+  // eslint-disable-next-line react-hooks/exhaustive-deps
+  const stableObjectSet = useMemo(() => objectSet, [JSON.stringify(objectSet)]);
+  useEffect(() => {
+    if (!stableObjects || stableObjects.length === 0 || functionColumnConfigs.length === 0 || !stableObjectSet) {
+      return;
+    }
+
+    // Cancel previous requests
+    if (abortControllerRef.current) {
+      abortControllerRef.current.abort();
+    }
+    const abortController = new AbortController();
+    abortControllerRef.current = abortController;
+    void (async () => {
+      if (!stableObjectSet) {
+        return;
+      }
+      initializeFunctionColumnData(functionColumnConfigs, stableObjects, setData);
+
+      // Process query results as they complete
+      for await (const queryResult of executeQueriesGenerator(functionColumnConfigs, stableObjectSet, client, abortController.signal)) {
+        if (abortController.signal.aborted) {
+          break;
+        }
+        const {
+          config,
+          result,
+          error,
+          aborted
+        } = queryResult;
+        if (!aborted) {
+          processQueryResult(config, result, error, stableObjects, setData);
+        }
+      }
+    })();
+    return () => {
+      abortController.abort();
+      abortControllerRef.current = null;
+    };
+  }, [functionColumnConfigs, stableObjects, stableObjectSet, client]);
+  return data;
+}
+function getFunctionColumnConfigs(columnDefinitions) {
+  if (!columnDefinitions) return [];
+
+  // Group columns by their query definition apiName
+  const configsByApiName = new Map();
+  columnDefinitions.forEach(colDef => {
+    if (colDef.locator.type === "function") {
+      const locator = colDef.locator;
+      const apiName = locator.queryDefinition.apiName;
+      const existingConfig = configsByApiName.get(apiName);
+      if (existingConfig) {
+        // Add this column to the existing config
+        existingConfig.columnIds.push({
+          columnId: String(locator.id),
+          getValue: locator.getValue,
+          getKey: locator.getKey
+        });
+      } else {
+        // Create new config
+        configsByApiName.set(apiName, {
+          queryDefinition: locator.queryDefinition,
+          getParams: locator.getFunctionParams,
+          columnIds: [{
+            columnId: String(locator.id),
+            getValue: locator.getValue,
+            getKey: locator.getKey
+          }]
+        });
+      }
+    }
+  });
+  return Array.from(configsByApiName.values());
+}
+const useStableObjects = objects => {
+  // eslint-disable-next-line react-hooks/exhaustive-deps
+  return useMemo(() => objects, [
+  // eslint-disable-next-line react-hooks/exhaustive-deps
+  JSON.stringify((objects ?? []).map(item => ({
+    $apiName: item.$apiName,
+    $primaryKey: item.$primaryKey
+  })).sort((a, b) => {
+    if (a.$apiName !== b.$apiName) {
+      return a.$apiName.localeCompare(b.$apiName);
+    }
+    return String(a.$primaryKey).localeCompare(String(b.$primaryKey));
+  }))]);
+};
+
+/**
+ * Generator that yields query results as they complete
+ */
+async function* executeQueriesGenerator(functionColumnConfigs, objectSet, client, signal) {
+  // Start all queries in parallel
+  const queryPromises = functionColumnConfigs.map(config => createQueryPromise(config, objectSet, client, signal));
+  const pendingPromises = [...queryPromises];
+
+  // Yield results as they complete using Promise.race
+  while (pendingPromises.length > 0) {
+    const result = await Promise.race(pendingPromises.map((promise, index) => promise.then(result => ({
+      result,
+      index
+    }))));
+    yield result.result;
+
+    // Remove the completed promise from the pending list
+    void pendingPromises.splice(result.index, 1);
+  }
+}
+
+/**
+ * Create a promise that executes a function column query
+ */
+function createQueryPromise(config, objectSet, client, signal) {
+  return new Promise(resolve => {
+    if (signal?.aborted) {
+      resolve({
+        config,
+        result: null,
+        error: null,
+        aborted: true
+      });
+      return;
+    }
+    const params = config.getParams(objectSet);
+    client(config.queryDefinition).executeFunction(params).then(result => {
+      // Check if aborted during execution
+      if (signal?.aborted) {
+        resolve({
+          config,
+          result: null,
+          error: null,
+          aborted: true
+        });
+      } else {
+        resolve({
+          config,
+          result,
+          error: null
+        });
+      }
+    }).catch(error => {
+      // Check if aborted during error handling
+      if (signal?.aborted) {
+        resolve({
+          config,
+          result: null,
+          error: null,
+          aborted: true
+        });
+      } else {
+        resolve({
+          config,
+          result: null,
+          error: error instanceof Error ? error : new Error(String(error))
+        });
+      }
+    });
+  });
+}
+
+/**
+ * Initialize function column data with isLoading state for all columns and objects
+ */
+function initializeFunctionColumnData(functionColumnConfigs, objects, setData) {
+  setData(prev => {
+    const newData = {
+      ...prev
+    };
+    functionColumnConfigs.forEach(config => {
+      config.columnIds.forEach(({
+        columnId
+      }) => {
+        // Initialize column if it doesn't exist
+        if (!newData[columnId]) {
+          newData[columnId] = {};
+        }
+        objects.forEach(obj => {
+          const key = String(obj.$primaryKey);
+          // Only set isLoading state if this object's data doesn't already exist
+          if (!newData[columnId][key]) {
+            newData[columnId][key] = createAsyncCellData({
+              isLoading: true
+            });
+          }
+        });
+      });
+    });
+    return newData;
+  });
+}
+function processQueryResult(config, result, error, objects, setData) {
+  // Batch all updates into a single setState call
+  setData(prev => {
+    const newData = {
+      ...prev
+    };
+    if (error) {
+      // Set error for all objects in all columns that use this query
+      config.columnIds.forEach(({
+        columnId
+      }) => {
+        if (!newData[columnId]) {
+          newData[columnId] = {};
+        }
+        objects.forEach(obj => {
+          const key = String(obj.$primaryKey);
+          newData[columnId][key] = createAsyncCellData({
+            error: error instanceof Error ? error : new Error(String(error)),
+            isLoading: false
+          });
+        });
+      });
+    } else if (result) {
+      // Process the FunctionsMap result
+      objects.forEach(obj => {
+        const key = String(obj.$primaryKey);
+
+        // Process each column that uses this query result
+        config.columnIds.forEach(({
+          columnId,
+          getValue,
+          getKey: columnGetKey
+        }) => {
+          if (!newData[columnId]) {
+            newData[columnId] = {};
+          }
+          const customKey = columnGetKey(obj);
+          const rawData = result[customKey];
+          const cellData = getValue ? getValue(rawData) : rawData;
+          newData[columnId][key] = createAsyncCellData({
+            data: cellData,
+            isLoading: false
+          });
+        });
+      });
+    }
+    return newData;
+  });
+}
+//# sourceMappingURL=useFunctionColumnsData.js.map

package/build/esm/object-table/hooks/useFunctionColumnsData.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useFunctionColumnsData.js","names":["useOsdkClient","useEffect","useMemo","useRef","useState","createAsyncCellData","useFunctionColumnsData","objectSet","objects","columnDefinitions","client","data","setData","abortControllerRef","functionColumnConfigs","getFunctionColumnConfigs","stableObjects","useStableObjects","stableObjectSet","JSON","stringify","length","current","abort","abortController","AbortController","initializeFunctionColumnData","queryResult","executeQueriesGenerator","signal","aborted","config","result","error","processQueryResult","configsByApiName","Map","forEach","colDef","locator","type","apiName","queryDefinition","existingConfig","get","columnIds","push","columnId","String","id","getValue","getKey","set","getParams","getFunctionParams","Array","from","values","map","item","$apiName","$primaryKey","sort","a","b","localeCompare","queryPromises","createQueryPromise","pendingPromises","Promise","race","promise","index","then","splice","resolve","params","executeFunction","catch","Error","prev","newData","obj","key","isLoading","columnGetKey","customKey","rawData","cellData"],"sources":["useFunctionColumnsData.ts"],"sourcesContent":["/*\n * Copyright 2026 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type {\n  ObjectOrInterfaceDefinition,\n  ObjectSet,\n  Osdk,\n  PropertyKeys,\n  QueryDefinition,\n  SimplePropertyDef,\n} from \"@osdk/api\";\nimport { useOsdkClient } from \"@osdk/react\";\nimport { useEffect, useMemo, useRef, useState } from \"react\";\nimport type {\n  ColumnDefinition,\n  ExtractQueryParameters,\n  FunctionColumnLocator,\n} from \"../ObjectTableApi.js\";\nimport {\n  type AsyncCellData,\n  createAsyncCellData,\n} from \"../utils/AsyncCellData.js\";\n\nexport interface FunctionColumnData {\n  [columnId: string]: {\n    [objectPrimaryKey: string]: AsyncCellData;\n  };\n}\n\ntype FunctionColumnConfig<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n  FunctionColumns extends Record<string, QueryDefinition<{}>> = Record<\n    string,\n    never\n  >,\n> = {\n  queryDefinition: QueryDefinition<any>;\n  getParams: (\n    objectSet: ObjectSet<Q>,\n  ) => ExtractQueryParameters<FunctionColumns[keyof FunctionColumns]>;\n  columnIds: Array<{\n    columnId: string;\n    getValue?: (cellData: unknown) => unknown;\n    getKey: (\n      object: Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>,\n    ) => string;\n  }>;\n};\n\ntype QueryResult<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n  FunctionColumns extends Record<string, QueryDefinition<{}>> = Record<\n    string,\n    never\n  >,\n> = {\n  config: FunctionColumnConfig<Q, RDPs, FunctionColumns>;\n  result: Record<string, unknown> | null;\n  error: Error | null;\n  aborted?: boolean;\n};\n\nexport function useFunctionColumnsData<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n  FunctionColumns extends Record<string, QueryDefinition<{}>> = Record<\n    string,\n    never\n  >,\n>(\n  objectSet: ObjectSet<Q, RDPs> | undefined,\n  objects:\n    | Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>[]\n    | undefined,\n  columnDefinitions?: Array<ColumnDefinition<Q, RDPs, FunctionColumns>>,\n): FunctionColumnData {\n  const client = useOsdkClient();\n  const [data, setData] = useState<FunctionColumnData>({});\n  const abortControllerRef = useRef<AbortController | null>(null);\n\n  // Function column configurations grouped by unique query definition\n  const functionColumnConfigs = useMemo(\n    () => getFunctionColumnConfigs(columnDefinitions),\n    [columnDefinitions],\n  );\n\n  const stableObjects = useStableObjects(objects);\n\n  // eslint-disable-next-line react-hooks/exhaustive-deps\n  const stableObjectSet = useMemo(() => objectSet, [JSON.stringify(objectSet)]);\n\n  useEffect(() => {\n    if (\n      !stableObjects || stableObjects.length === 0\n      || functionColumnConfigs.length === 0 || !stableObjectSet\n    ) {\n      return;\n    }\n\n    // Cancel previous requests\n    if (abortControllerRef.current) {\n      abortControllerRef.current.abort();\n    }\n\n    const abortController = new AbortController();\n    abortControllerRef.current = abortController;\n\n    const executeQueries = async () => {\n      if (!stableObjectSet) {\n        return;\n      }\n\n      initializeFunctionColumnData(\n        functionColumnConfigs,\n        stableObjects,\n        setData,\n      );\n\n      // Process query results as they complete\n      for await (\n        const queryResult of executeQueriesGenerator(\n          functionColumnConfigs,\n          stableObjectSet as ObjectSet<Q>,\n          client,\n          abortController.signal,\n        )\n      ) {\n        if (abortController.signal.aborted) {\n          break;\n        }\n\n        const { config, result, error, aborted } = queryResult;\n        if (!aborted) {\n          processQueryResult(config, result, error, stableObjects, setData);\n        }\n      }\n    };\n\n    void executeQueries();\n\n    return () => {\n      abortController.abort();\n      abortControllerRef.current = null;\n    };\n  }, [functionColumnConfigs, stableObjects, stableObjectSet, client]);\n\n  return data;\n}\n\nfunction getFunctionColumnConfigs<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n  FunctionColumns extends Record<string, QueryDefinition<{}>> = Record<\n    string,\n    never\n  >,\n>(\n  columnDefinitions?: Array<ColumnDefinition<Q, RDPs, FunctionColumns>>,\n): Array<FunctionColumnConfig<Q, RDPs, FunctionColumns>> {\n  if (!columnDefinitions) return [];\n\n  // Group columns by their query definition apiName\n  const configsByApiName = new Map<\n    string,\n    FunctionColumnConfig<Q, RDPs, FunctionColumns>\n  >();\n\n  columnDefinitions.forEach((colDef) => {\n    if (colDef.locator.type === \"function\") {\n      const locator = colDef.locator as FunctionColumnLocator<\n        Q,\n        RDPs,\n        FunctionColumns\n      >;\n\n      const apiName = locator.queryDefinition.apiName;\n      const existingConfig = configsByApiName.get(apiName);\n\n      if (existingConfig) {\n        // Add this column to the existing config\n        existingConfig.columnIds.push({\n          columnId: String(locator.id),\n          getValue: locator.getValue,\n          getKey: locator.getKey,\n        });\n      } else {\n        // Create new config\n        configsByApiName.set(apiName, {\n          queryDefinition: locator.queryDefinition,\n          getParams: locator.getFunctionParams,\n          columnIds: [{\n            columnId: String(locator.id),\n            getValue: locator.getValue,\n            getKey: locator.getKey,\n          }],\n        });\n      }\n    }\n  });\n\n  return Array.from(configsByApiName.values());\n}\n\nconst useStableObjects = <\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n>(\n  objects:\n    | Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>[]\n    | undefined,\n):\n  | Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>[]\n  | undefined =>\n{\n  // eslint-disable-next-line react-hooks/exhaustive-deps\n  return useMemo(() => objects, [\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    JSON.stringify(\n      (objects ?? []).map(item => ({\n        $apiName: item.$apiName,\n        $primaryKey: item.$primaryKey,\n      })).sort((a, b) => {\n        if (a.$apiName !== b.$apiName) {\n          return a.$apiName.localeCompare(b.$apiName);\n        }\n        return String(a.$primaryKey).localeCompare(String(b.$primaryKey));\n      }),\n    ),\n  ]);\n};\n\n/**\n * Generator that yields query results as they complete\n */\nasync function* executeQueriesGenerator<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n  FunctionColumns extends Record<string, QueryDefinition<{}>> = Record<\n    string,\n    never\n  >,\n>(\n  functionColumnConfigs: Array<FunctionColumnConfig<Q, RDPs, FunctionColumns>>,\n  objectSet: ObjectSet<Q>,\n  client: ReturnType<typeof useOsdkClient>,\n  signal?: AbortSignal,\n): AsyncGenerator<QueryResult<Q, RDPs, FunctionColumns>, void, unknown> {\n  // Start all queries in parallel\n  const queryPromises = functionColumnConfigs.map(config =>\n    createQueryPromise(config, objectSet, client, signal)\n  ) as Array<Promise<QueryResult<Q, RDPs, FunctionColumns>>>;\n  const pendingPromises = [...queryPromises];\n\n  // Yield results as they complete using Promise.race\n  while (pendingPromises.length > 0) {\n    const result = await Promise.race(\n      pendingPromises.map((promise, index) =>\n        promise.then(result => ({ result, index }))\n      ),\n    );\n\n    yield result.result;\n\n    // Remove the completed promise from the pending list\n    void pendingPromises.splice(result.index, 1);\n  }\n}\n\n/**\n * Create a promise that executes a function column query\n */\nfunction createQueryPromise<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n  FunctionColumns extends Record<string, QueryDefinition<{}>> = Record<\n    string,\n    never\n  >,\n>(\n  config: FunctionColumnConfig<Q, RDPs, FunctionColumns>,\n  objectSet: ObjectSet<Q>,\n  client: ReturnType<typeof useOsdkClient>,\n  signal?: AbortSignal,\n): Promise<QueryResult<Q, RDPs, FunctionColumns>> {\n  return new Promise((resolve) => {\n    if (signal?.aborted) {\n      resolve({ config, result: null, error: null, aborted: true });\n      return;\n    }\n\n    const params = config.getParams(objectSet as ObjectSet<Q>);\n\n    client(config.queryDefinition)\n      .executeFunction(params)\n      .then((result: Record<string, unknown>) => {\n        // Check if aborted during execution\n        if (signal?.aborted) {\n          resolve({ config, result: null, error: null, aborted: true });\n        } else {\n          resolve({ config, result, error: null });\n        }\n      })\n      .catch((error: unknown) => {\n        // Check if aborted during error handling\n        if (signal?.aborted) {\n          resolve({ config, result: null, error: null, aborted: true });\n        } else {\n          resolve({\n            config,\n            result: null,\n            error: error instanceof Error ? error : new Error(String(error)),\n          });\n        }\n      });\n  });\n}\n\n/**\n * Initialize function column data with isLoading state for all columns and objects\n */\nfunction initializeFunctionColumnData<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n  FunctionColumns extends Record<string, QueryDefinition<{}>> = Record<\n    string,\n    never\n  >,\n>(\n  functionColumnConfigs: Array<FunctionColumnConfig<Q, RDPs, FunctionColumns>>,\n  objects: Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>[],\n  setData: React.Dispatch<React.SetStateAction<FunctionColumnData>>,\n): void {\n  setData(prev => {\n    const newData = { ...prev };\n\n    functionColumnConfigs.forEach(config => {\n      config.columnIds.forEach(({ columnId }) => {\n        // Initialize column if it doesn't exist\n        if (!newData[columnId]) {\n          newData[columnId] = {};\n        }\n\n        objects.forEach(obj => {\n          const key = String(obj.$primaryKey);\n          // Only set isLoading state if this object's data doesn't already exist\n          if (!newData[columnId][key]) {\n            newData[columnId][key] = createAsyncCellData({\n              isLoading: true,\n            });\n          }\n        });\n      });\n    });\n\n    return newData;\n  });\n}\n\nfunction processQueryResult<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n  FunctionColumns extends Record<string, QueryDefinition<{}>> = Record<\n    string,\n    never\n  >,\n>(\n  config: FunctionColumnConfig<Q, RDPs, FunctionColumns>,\n  result: Record<string, unknown> | null,\n  error: Error | null,\n  objects: Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>[],\n  setData: React.Dispatch<React.SetStateAction<FunctionColumnData>>,\n): void {\n  // Batch all updates into a single setState call\n  setData(prev => {\n    const newData = { ...prev };\n\n    if (error) {\n      // Set error for all objects in all columns that use this query\n      config.columnIds.forEach(({ columnId }) => {\n        if (!newData[columnId]) {\n          newData[columnId] = {};\n        }\n\n        objects.forEach(obj => {\n          const key = String(obj.$primaryKey);\n          newData[columnId][key] = createAsyncCellData({\n            error: error instanceof Error\n              ? error\n              : new Error(String(error)),\n            isLoading: false,\n          });\n        });\n      });\n    } else if (result) {\n      // Process the FunctionsMap result\n      objects.forEach(obj => {\n        const key = String(obj.$primaryKey);\n\n        // Process each column that uses this query result\n        config.columnIds.forEach(\n          ({ columnId, getValue, getKey: columnGetKey }) => {\n            if (!newData[columnId]) {\n              newData[columnId] = {};\n            }\n\n            const customKey = columnGetKey(obj);\n            const rawData = result[customKey];\n            const cellData = getValue ? getValue(rawData) : rawData;\n\n            newData[columnId][key] = createAsyncCellData({\n              data: cellData,\n              isLoading: false,\n            });\n          },\n        );\n      });\n    }\n\n    return newData;\n  });\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAUA,SAASA,aAAa,QAAQ,aAAa;AAC3C,SAASC,SAAS,EAAEC,OAAO,EAAEC,MAAM,EAAEC,QAAQ,QAAQ,OAAO;AAM5D,SAEEC,mBAAmB,QACd,2BAA2B;AA2ClC,OAAO,SAASC,sBAAsBA,CAQpCC,SAAyC,EACzCC,OAEa,EACbC,iBAAqE,EACjD;EACpB,MAAMC,MAAM,GAAGV,aAAa,CAAC,CAAC;EAC9B,MAAM,CAACW,IAAI,EAAEC,OAAO,CAAC,GAAGR,QAAQ,CAAqB,CAAC,CAAC,CAAC;EACxD,MAAMS,kBAAkB,GAAGV,MAAM,CAAyB,IAAI,CAAC;;EAE/D;EACA,MAAMW,qBAAqB,GAAGZ,OAAO,CACnC,MAAMa,wBAAwB,CAACN,iBAAiB,CAAC,EACjD,CAACA,iBAAiB,CACpB,CAAC;EAED,MAAMO,aAAa,GAAGC,gBAAgB,CAACT,OAAO,CAAC;;EAE/C;EACA,MAAMU,eAAe,GAAGhB,OAAO,CAAC,MAAMK,SAAS,EAAE,CAACY,IAAI,CAACC,SAAS,CAACb,SAAS,CAAC,CAAC,CAAC;EAE7EN,SAAS,CAAC,MAAM;IACd,IACE,CAACe,aAAa,IAAIA,aAAa,CAACK,MAAM,KAAK,CAAC,IACzCP,qBAAqB,CAACO,MAAM,KAAK,CAAC,IAAI,CAACH,eAAe,EACzD;MACA;IACF;;IAEA;IACA,IAAIL,kBAAkB,CAACS,OAAO,EAAE;MAC9BT,kBAAkB,CAACS,OAAO,CAACC,KAAK,CAAC,CAAC;IACpC;IAEA,MAAMC,eAAe,GAAG,IAAIC,eAAe,CAAC,CAAC;IAC7CZ,kBAAkB,CAACS,OAAO,GAAGE,eAAe;IAiC5C,KAAK,CA/BkB,YAAY;MACjC,IAAI,CAACN,eAAe,EAAE;QACpB;MACF;MAEAQ,4BAA4B,CAC1BZ,qBAAqB,EACrBE,aAAa,EACbJ,OACF,CAAC;;MAED;MACA,WACE,MAAMe,WAAW,IAAIC,uBAAuB,CAC1Cd,qBAAqB,EACrBI,eAAe,EACfR,MAAM,EACNc,eAAe,CAACK,MAClB,CAAC,EACD;QACA,IAAIL,eAAe,CAACK,MAAM,CAACC,OAAO,EAAE;UAClC;QACF;QAEA,MAAM;UAAEC,MAAM;UAAEC,MAAM;UAAEC,KAAK;UAAEH;QAAQ,CAAC,GAAGH,WAAW;QACtD,IAAI,CAACG,OAAO,EAAE;UACZI,kBAAkB,CAACH,MAAM,EAAEC,MAAM,EAAEC,KAAK,EAAEjB,aAAa,EAAEJ,OAAO,CAAC;QACnE;MACF;IACF,CAAC,EAEmB,CAAC;IAErB,OAAO,MAAM;MACXY,eAAe,CAACD,KAAK,CAAC,CAAC;MACvBV,kBAAkB,CAACS,OAAO,GAAG,IAAI;IACnC,CAAC;EACH,CAAC,EAAE,CAACR,qBAAqB,EAAEE,aAAa,EAAEE,eAAe,EAAER,MAAM,CAAC,CAAC;EAEnE,OAAOC,IAAI;AACb;AAEA,SAASI,wBAAwBA,CAQ/BN,iBAAqE,EACd;EACvD,IAAI,CAACA,iBAAiB,EAAE,OAAO,EAAE;;EAEjC;EACA,MAAM0B,gBAAgB,GAAG,IAAIC,GAAG,CAG9B,CAAC;EAEH3B,iBAAiB,CAAC4B,OAAO,CAAEC,MAAM,IAAK;IACpC,IAAIA,MAAM,CAACC,OAAO,CAACC,IAAI,KAAK,UAAU,EAAE;MACtC,MAAMD,OAAO,GAAGD,MAAM,CAACC,OAItB;MAED,MAAME,OAAO,GAAGF,OAAO,CAACG,eAAe,CAACD,OAAO;MAC/C,MAAME,cAAc,GAAGR,gBAAgB,CAACS,GAAG,CAACH,OAAO,CAAC;MAEpD,IAAIE,cAAc,EAAE;QAClB;QACAA,cAAc,CAACE,SAAS,CAACC,IAAI,CAAC;UAC5BC,QAAQ,EAAEC,MAAM,CAACT,OAAO,CAACU,EAAE,CAAC;UAC5BC,QAAQ,EAAEX,OAAO,CAACW,QAAQ;UAC1BC,MAAM,EAAEZ,OAAO,CAACY;QAClB,CAAC,CAAC;MACJ,CAAC,MAAM;QACL;QACAhB,gBAAgB,CAACiB,GAAG,CAACX,OAAO,EAAE;UAC5BC,eAAe,EAAEH,OAAO,CAACG,eAAe;UACxCW,SAAS,EAAEd,OAAO,CAACe,iBAAiB;UACpCT,SAAS,EAAE,CAAC;YACVE,QAAQ,EAAEC,MAAM,CAACT,OAAO,CAACU,EAAE,CAAC;YAC5BC,QAAQ,EAAEX,OAAO,CAACW,QAAQ;YAC1BC,MAAM,EAAEZ,OAAO,CAACY;UAClB,CAAC;QACH,CAAC,CAAC;MACJ;IACF;EACF,CAAC,CAAC;EAEF,OAAOI,KAAK,CAACC,IAAI,CAACrB,gBAAgB,CAACsB,MAAM,CAAC,CAAC,CAAC;AAC9C;AAEA,MAAMxC,gBAAgB,GAIpBT,OAEa,IAIf;EACE;EACA,OAAON,OAAO,CAAC,MAAMM,OAAO,EAAE;EAC5B;EACAW,IAAI,CAACC,SAAS,CACZ,CAACZ,OAAO,IAAI,EAAE,EAAEkD,GAAG,CAACC,IAAI,KAAK;IAC3BC,QAAQ,EAAED,IAAI,CAACC,QAAQ;IACvBC,WAAW,EAAEF,IAAI,CAACE;EACpB,CAAC,CAAC,CAAC,CAACC,IAAI,CAAC,CAACC,CAAC,EAAEC,CAAC,KAAK;IACjB,IAAID,CAAC,CAACH,QAAQ,KAAKI,CAAC,CAACJ,QAAQ,EAAE;MAC7B,OAAOG,CAAC,CAACH,QAAQ,CAACK,aAAa,CAACD,CAAC,CAACJ,QAAQ,CAAC;IAC7C;IACA,OAAOZ,MAAM,CAACe,CAAC,CAACF,WAAW,CAAC,CAACI,aAAa,CAACjB,MAAM,CAACgB,CAAC,CAACH,WAAW,CAAC,CAAC;EACnE,CAAC,CACH,CAAC,CACF,CAAC;AACJ,CAAC;;AAED;AACA;AACA;AACA,gBAAgBjC,uBAAuBA,CAQrCd,qBAA4E,EAC5EP,SAAuB,EACvBG,MAAwC,EACxCmB,MAAoB,EACkD;EACtE;EACA,MAAMqC,aAAa,GAAGpD,qBAAqB,CAAC4C,GAAG,CAAC3B,MAAM,IACpDoC,kBAAkB,CAACpC,MAAM,EAAExB,SAAS,EAAEG,MAAM,EAAEmB,MAAM,CACtD,CAA0D;EAC1D,MAAMuC,eAAe,GAAG,CAAC,GAAGF,aAAa,CAAC;;EAE1C;EACA,OAAOE,eAAe,CAAC/C,MAAM,GAAG,CAAC,EAAE;IACjC,MAAMW,MAAM,GAAG,MAAMqC,OAAO,CAACC,IAAI,CAC/BF,eAAe,CAACV,GAAG,CAAC,CAACa,OAAO,EAAEC,KAAK,KACjCD,OAAO,CAACE,IAAI,CAACzC,MAAM,KAAK;MAAEA,MAAM;MAAEwC;IAAM,CAAC,CAAC,CAC5C,CACF,CAAC;IAED,MAAMxC,MAAM,CAACA,MAAM;;IAEnB;IACA,KAAKoC,eAAe,CAACM,MAAM,CAAC1C,MAAM,CAACwC,KAAK,EAAE,CAAC,CAAC;EAC9C;AACF;;AAEA;AACA;AACA;AACA,SAASL,kBAAkBA,CAQzBpC,MAAsD,EACtDxB,SAAuB,EACvBG,MAAwC,EACxCmB,MAAoB,EAC4B;EAChD,OAAO,IAAIwC,OAAO,CAAEM,OAAO,IAAK;IAC9B,IAAI9C,MAAM,EAAEC,OAAO,EAAE;MACnB6C,OAAO,CAAC;QAAE5C,MAAM;QAAEC,MAAM,EAAE,IAAI;QAAEC,KAAK,EAAE,IAAI;QAAEH,OAAO,EAAE;MAAK,CAAC,CAAC;MAC7D;IACF;IAEA,MAAM8C,MAAM,GAAG7C,MAAM,CAACsB,SAAS,CAAC9C,SAAyB,CAAC;IAE1DG,MAAM,CAACqB,MAAM,CAACW,eAAe,CAAC,CAC3BmC,eAAe,CAACD,MAAM,CAAC,CACvBH,IAAI,CAAEzC,MAA+B,IAAK;MACzC;MACA,IAAIH,MAAM,EAAEC,OAAO,EAAE;QACnB6C,OAAO,CAAC;UAAE5C,MAAM;UAAEC,MAAM,EAAE,IAAI;UAAEC,KAAK,EAAE,IAAI;UAAEH,OAAO,EAAE;QAAK,CAAC,CAAC;MAC/D,CAAC,MAAM;QACL6C,OAAO,CAAC;UAAE5C,MAAM;UAAEC,MAAM;UAAEC,KAAK,EAAE;QAAK,CAAC,CAAC;MAC1C;IACF,CAAC,CAAC,CACD6C,KAAK,CAAE7C,KAAc,IAAK;MACzB;MACA,IAAIJ,MAAM,EAAEC,OAAO,EAAE;QACnB6C,OAAO,CAAC;UAAE5C,MAAM;UAAEC,MAAM,EAAE,IAAI;UAAEC,KAAK,EAAE,IAAI;UAAEH,OAAO,EAAE;QAAK,CAAC,CAAC;MAC/D,CAAC,MAAM;QACL6C,OAAO,CAAC;UACN5C,MAAM;UACNC,MAAM,EAAE,IAAI;UACZC,KAAK,EAAEA,KAAK,YAAY8C,KAAK,GAAG9C,KAAK,GAAG,IAAI8C,KAAK,CAAC/B,MAAM,CAACf,KAAK,CAAC;QACjE,CAAC,CAAC;MACJ;IACF,CAAC,CAAC;EACN,CAAC,CAAC;AACJ;;AAEA;AACA;AACA;AACA,SAASP,4BAA4BA,CAQnCZ,qBAA4E,EAC5EN,OAAwE,EACxEI,OAAiE,EAC3D;EACNA,OAAO,CAACoE,IAAI,IAAI;IACd,MAAMC,OAAO,GAAG;MAAE,GAAGD;IAAK,CAAC;IAE3BlE,qBAAqB,CAACuB,OAAO,CAACN,MAAM,IAAI;MACtCA,MAAM,CAACc,SAAS,CAACR,OAAO,CAAC,CAAC;QAAEU;MAAS,CAAC,KAAK;QACzC;QACA,IAAI,CAACkC,OAAO,CAAClC,QAAQ,CAAC,EAAE;UACtBkC,OAAO,CAAClC,QAAQ,CAAC,GAAG,CAAC,CAAC;QACxB;QAEAvC,OAAO,CAAC6B,OAAO,CAAC6C,GAAG,IAAI;UACrB,MAAMC,GAAG,GAAGnC,MAAM,CAACkC,GAAG,CAACrB,WAAW,CAAC;UACnC;UACA,IAAI,CAACoB,OAAO,CAAClC,QAAQ,CAAC,CAACoC,GAAG,CAAC,EAAE;YAC3BF,OAAO,CAAClC,QAAQ,CAAC,CAACoC,GAAG,CAAC,GAAG9E,mBAAmB,CAAC;cAC3C+E,SAAS,EAAE;YACb,CAAC,CAAC;UACJ;QACF,CAAC,CAAC;MACJ,CAAC,CAAC;IACJ,CAAC,CAAC;IAEF,OAAOH,OAAO;EAChB,CAAC,CAAC;AACJ;AAEA,SAAS/C,kBAAkBA,CAQzBH,MAAsD,EACtDC,MAAsC,EACtCC,KAAmB,EACnBzB,OAAwE,EACxEI,OAAiE,EAC3D;EACN;EACAA,OAAO,CAACoE,IAAI,IAAI;IACd,MAAMC,OAAO,GAAG;MAAE,GAAGD;IAAK,CAAC;IAE3B,IAAI/C,KAAK,EAAE;MACT;MACAF,MAAM,CAACc,SAAS,CAACR,OAAO,CAAC,CAAC;QAAEU;MAAS,CAAC,KAAK;QACzC,IAAI,CAACkC,OAAO,CAAClC,QAAQ,CAAC,EAAE;UACtBkC,OAAO,CAAClC,QAAQ,CAAC,GAAG,CAAC,CAAC;QACxB;QAEAvC,OAAO,CAAC6B,OAAO,CAAC6C,GAAG,IAAI;UACrB,MAAMC,GAAG,GAAGnC,MAAM,CAACkC,GAAG,CAACrB,WAAW,CAAC;UACnCoB,OAAO,CAAClC,QAAQ,CAAC,CAACoC,GAAG,CAAC,GAAG9E,mBAAmB,CAAC;YAC3C4B,KAAK,EAAEA,KAAK,YAAY8C,KAAK,GACzB9C,KAAK,GACL,IAAI8C,KAAK,CAAC/B,MAAM,CAACf,KAAK,CAAC,CAAC;YAC5BmD,SAAS,EAAE;UACb,CAAC,CAAC;QACJ,CAAC,CAAC;MACJ,CAAC,CAAC;IACJ,CAAC,MAAM,IAAIpD,MAAM,EAAE;MACjB;MACAxB,OAAO,CAAC6B,OAAO,CAAC6C,GAAG,IAAI;QACrB,MAAMC,GAAG,GAAGnC,MAAM,CAACkC,GAAG,CAACrB,WAAW,CAAC;;QAEnC;QACA9B,MAAM,CAACc,SAAS,CAACR,OAAO,CACtB,CAAC;UAAEU,QAAQ;UAAEG,QAAQ;UAAEC,MAAM,EAAEkC;QAAa,CAAC,KAAK;UAChD,IAAI,CAACJ,OAAO,CAAClC,QAAQ,CAAC,EAAE;YACtBkC,OAAO,CAAClC,QAAQ,CAAC,GAAG,CAAC,CAAC;UACxB;UAEA,MAAMuC,SAAS,GAAGD,YAAY,CAACH,GAAG,CAAC;UACnC,MAAMK,OAAO,GAAGvD,MAAM,CAACsD,SAAS,CAAC;UACjC,MAAME,QAAQ,GAAGtC,QAAQ,GAAGA,QAAQ,CAACqC,OAAO,CAAC,GAAGA,OAAO;UAEvDN,OAAO,CAAClC,QAAQ,CAAC,CAACoC,GAAG,CAAC,GAAG9E,mBAAmB,CAAC;YAC3CM,IAAI,EAAE6E,QAAQ;YACdJ,SAAS,EAAE;UACb,CAAC,CAAC;QACJ,CACF,CAAC;MACH,CAAC,CAAC;IACJ;IAEA,OAAOH,OAAO;EAChB,CAAC,CAAC;AACJ","ignoreList":[]}

package/build/esm/object-table/hooks/useObjectTableData.js

@@ -14,16 +14,17 @@
  * limitations under the License.
  */
 
-import { useOsdkObjects } from "@osdk/react/experimental";
+import { useObjectSet, useOsdkObjects } from "@osdk/react/experimental";
 import { useMemo } from "react";
+import { useFunctionColumnsData } from "./useFunctionColumnsData.js";
 const PAGE_SIZE = 50;
-
 /**
- * This hook is a wrapper around useOsdkObjects
- * It extracts RDP locators from columnDefinitions and calls useObjectSet + withProperties
+ * This hook is a wrapper that conditionally uses either useObjectSet or useOsdkObjects
+ * based on whether an objectSet prop is provided.
+ * It extracts RDP locators from columnDefinitions and applies withProperties
  * to return data containing the derived properties.
  */
-export function useObjectTableData(objectOrInterfaceType, columnDefinitions, filter, sorting) {
+export function useObjectTableData(objectOrInterfaceType, columnDefinitions, filter, sorting, objectSet, objectSetOptions) {
   const orderBy = useMemo(() => {
     if (!sorting || sorting.length === 0) {
       return undefined;
@@ -52,11 +53,61 @@ export function useObjectTableData(objectOrInterfaceType, columnDefinitions, fil
       };
     }, {});
   }, [columnDefinitions]);
-  return useOsdkObjects(objectOrInterfaceType, {
+
+  // When objectSet is provided and it's an object type, use useObjectSet. Otherwise, use useOsdkObjects.
+  const isObjectType = objectOrInterfaceType.type === "object";
+  const shouldUseObjectSet = !!objectSet && isObjectType;
+
+  // When shouldUseObjectSet is true, we know objectSet is defined
+  // and objectOrInterfaceType is an ObjectTypeDefinition
+  const objectSetResult = useObjectSet(shouldUseObjectSet ? objectSet : undefined, {
+    ...objectSetOptions,
+    withProperties: withProperties,
+    where: filter,
+    orderBy,
+    pageSize: PAGE_SIZE,
+    enabled: shouldUseObjectSet
+  });
+  const osdkObjectsResult = useOsdkObjects(objectOrInterfaceType, {
     withProperties,
     pageSize: PAGE_SIZE,
     where: filter,
-    orderBy
+    orderBy,
+    enabled: !shouldUseObjectSet
   });
+
+  // Get the result from the appropriate hook
+  const baseResult = shouldUseObjectSet ? objectSetResult : osdkObjectsResult;
+
+  // Call useFunctionColumnsData to get function column data
+  const functionColumnData = useFunctionColumnsData(objectSetResult.objectSet, baseResult.data, columnDefinitions);
+
+  // Merge function column data into each object
+  const mergedData = useMemo(() => {
+    if (!baseResult.data) return baseResult.data;
+    return baseResult.data.map(obj => {
+      const objKey = String(obj.$primaryKey);
+      const functionData = {};
+
+      // Collect all function column data for this object
+      Object.entries(functionColumnData).forEach(([columnId, columnData]) => {
+        if (columnData[objKey]) {
+          functionData[columnId] = columnData[objKey];
+        }
+      });
+
+      // Return object with function data merged in
+      return {
+        ...obj,
+        ...functionData
+      };
+    });
+  }, [baseResult.data, functionColumnData]);
+
+  // Return the result with merged data
+  return {
+    ...baseResult,
+    data: mergedData
+  };
 }
 //# sourceMappingURL=useObjectTableData.js.map

package/build/esm/object-table/hooks/useObjectTableData.js.map

@@ -1 +1 @@
-{"version":3,"file":"useObjectTableData.js","names":["useOsdkObjects","useMemo","PAGE_SIZE","useObjectTableData","objectOrInterfaceType","columnDefinitions","filter","sorting","orderBy","length","undefined","reduce","acc","sort","id","desc","withProperties","rdpColumns","map","colDef","locator","colLocator","type","cur","creator","pageSize","where"],"sources":["useObjectTableData.ts"],"sourcesContent":["/*\n * Copyright 2025 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type {\n  DerivedProperty,\n  ObjectOrInterfaceDefinition,\n  PropertyKeys,\n  QueryDefinition,\n  SimplePropertyDef,\n  WhereClause,\n} from \"@osdk/api\";\nimport type { UseOsdkListResult } from \"@osdk/react/experimental\";\nimport { useOsdkObjects } from \"@osdk/react/experimental\";\nimport type { SortingState } from \"@tanstack/react-table\";\nimport { useMemo } from \"react\";\nimport type { ColumnDefinition } from \"../ObjectTableApi.js\";\n\nconst PAGE_SIZE = 50;\n\n/**\n * This hook is a wrapper around useOsdkObjects\n * It extracts RDP locators from columnDefinitions and calls useObjectSet + withProperties\n * to return data containing the derived properties.\n */\nexport function useObjectTableData<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<\n    string,\n    never\n  >,\n  FunctionColumns extends Record<string, QueryDefinition<{}>> = Record<\n    string,\n    never\n  >,\n>(\n  objectOrInterfaceType: Q,\n  columnDefinitions?: Array<ColumnDefinition<Q, RDPs, FunctionColumns>>,\n  filter?: WhereClause<Q, RDPs>,\n  sorting?: SortingState,\n): UseOsdkListResult<Q, RDPs> {\n  const orderBy = useMemo(() => {\n    if (!sorting || sorting.length === 0) {\n      return undefined;\n    }\n\n    return sorting.reduce<{ [K in PropertyKeys<Q>]?: \"asc\" | \"desc\" }>(\n      (acc, sort) => {\n        acc[sort.id as PropertyKeys<Q>] = sort.desc ? \"desc\" : \"asc\";\n        return acc;\n      },\n      {},\n    );\n  }, [sorting]);\n\n  // Extract derived properties definition\n  const withProperties = useMemo(() => {\n    if (!columnDefinitions) {\n      return;\n    }\n\n    const rdpColumns = columnDefinitions.map(colDef => colDef.locator).filter(\n      (colLocator) => {\n        return colLocator.type === \"rdp\";\n      },\n    );\n\n    if (!rdpColumns.length) {\n      return;\n    }\n\n    return rdpColumns.reduce<\n      { [K in keyof RDPs]: DerivedProperty.Creator<Q, RDPs[K]> }\n    >(\n      (acc, cur) => {\n        return {\n          ...acc,\n          [cur.id]: cur.creator,\n        };\n      },\n      {} as { [K in keyof RDPs]: DerivedProperty.Creator<Q, RDPs[K]> },\n    );\n  }, [columnDefinitions]);\n\n  return useOsdkObjects<\n    Q,\n    RDPs\n  >(\n    objectOrInterfaceType,\n    {\n      withProperties,\n      pageSize: PAGE_SIZE,\n      where: filter,\n      orderBy,\n    },\n  );\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAWA,SAASA,cAAc,QAAQ,0BAA0B;AAEzD,SAASC,OAAO,QAAQ,OAAO;AAG/B,MAAMC,SAAS,GAAG,EAAE;;AAEpB;AACA;AACA;AACA;AACA;AACA,OAAO,SAASC,kBAAkBA,CAWhCC,qBAAwB,EACxBC,iBAAqE,EACrEC,MAA6B,EAC7BC,OAAsB,EACM;EAC5B,MAAMC,OAAO,GAAGP,OAAO,CAAC,MAAM;IAC5B,IAAI,CAACM,OAAO,IAAIA,OAAO,CAACE,MAAM,KAAK,CAAC,EAAE;MACpC,OAAOC,SAAS;IAClB;IAEA,OAAOH,OAAO,CAACI,MAAM,CACnB,CAACC,GAAG,EAAEC,IAAI,KAAK;MACbD,GAAG,CAACC,IAAI,CAACC,EAAE,CAAoB,GAAGD,IAAI,CAACE,IAAI,GAAG,MAAM,GAAG,KAAK;MAC5D,OAAOH,GAAG;IACZ,CAAC,EACD,CAAC,CACH,CAAC;EACH,CAAC,EAAE,CAACL,OAAO,CAAC,CAAC;;EAEb;EACA,MAAMS,cAAc,GAAGf,OAAO,CAAC,MAAM;IACnC,IAAI,CAACI,iBAAiB,EAAE;MACtB;IACF;IAEA,MAAMY,UAAU,GAAGZ,iBAAiB,CAACa,GAAG,CAACC,MAAM,IAAIA,MAAM,CAACC,OAAO,CAAC,CAACd,MAAM,CACtEe,UAAU,IAAK;MACd,OAAOA,UAAU,CAACC,IAAI,KAAK,KAAK;IAClC,CACF,CAAC;IAED,IAAI,CAACL,UAAU,CAACR,MAAM,EAAE;MACtB;IACF;IAEA,OAAOQ,UAAU,CAACN,MAAM,CAGtB,CAACC,GAAG,EAAEW,GAAG,KAAK;MACZ,OAAO;QACL,GAAGX,GAAG;QACN,CAACW,GAAG,CAACT,EAAE,GAAGS,GAAG,CAACC;MAChB,CAAC;IACH,CAAC,EACD,CAAC,CACH,CAAC;EACH,CAAC,EAAE,CAACnB,iBAAiB,CAAC,CAAC;EAEvB,OAAOL,cAAc,CAInBI,qBAAqB,EACrB;IACEY,cAAc;IACdS,QAAQ,EAAEvB,SAAS;IACnBwB,KAAK,EAAEpB,MAAM;IACbE;EACF,CACF,CAAC;AACH","ignoreList":[]}
+{"version":3,"file":"useObjectTableData.js","names":["useObjectSet","useOsdkObjects","useMemo","useFunctionColumnsData","PAGE_SIZE","useObjectTableData","objectOrInterfaceType","columnDefinitions","filter","sorting","objectSet","objectSetOptions","orderBy","length","undefined","reduce","acc","sort","id","desc","withProperties","rdpColumns","map","colDef","locator","colLocator","type","cur","creator","isObjectType","shouldUseObjectSet","objectSetResult","where","pageSize","enabled","osdkObjectsResult","baseResult","functionColumnData","data","mergedData","obj","objKey","String","$primaryKey","functionData","Object","entries","forEach","columnId","columnData"],"sources":["useObjectTableData.ts"],"sourcesContent":["/*\n * Copyright 2025 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type {\n  DerivedProperty,\n  ObjectOrInterfaceDefinition,\n  ObjectSet,\n  PropertyKeys,\n  QueryDefinition,\n  SimplePropertyDef,\n  WhereClause,\n} from \"@osdk/api\";\nimport type { UseOsdkListResult } from \"@osdk/react/experimental\";\nimport { useObjectSet, useOsdkObjects } from \"@osdk/react/experimental\";\nimport type { SortingState } from \"@tanstack/react-table\";\nimport { useMemo } from \"react\";\nimport type { ColumnDefinition, ObjectSetOptions } from \"../ObjectTableApi.js\";\nimport type { AsyncCellData } from \"../utils/AsyncCellData.js\";\nimport { useFunctionColumnsData } from \"./useFunctionColumnsData.js\";\n\nconst PAGE_SIZE = 50;\n\ntype WithProperties<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<\n    string,\n    never\n  >,\n> = {\n  [K in keyof RDPs]: DerivedProperty.Creator<Q, RDPs[K]>;\n};\n\ninterface UseObjectTableDataResult<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n> extends Omit<UseOsdkListResult<Q, RDPs>, \"isOptimistic\"> {\n  objectSet?: ObjectSet<Q>;\n}\n/**\n * This hook is a wrapper that conditionally uses either useObjectSet or useOsdkObjects\n * based on whether an objectSet prop is provided.\n * It extracts RDP locators from columnDefinitions and applies withProperties\n * to return data containing the derived properties.\n */\nexport function useObjectTableData<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<\n    string,\n    never\n  >,\n  FunctionColumns extends Record<string, QueryDefinition<{}>> = Record<\n    string,\n    never\n  >,\n>(\n  objectOrInterfaceType: Q,\n  columnDefinitions?: Array<ColumnDefinition<Q, RDPs, FunctionColumns>>,\n  filter?: WhereClause<Q, RDPs>,\n  sorting?: SortingState,\n  objectSet?: ObjectSet<Q>,\n  objectSetOptions?: ObjectSetOptions<Q>,\n): UseObjectTableDataResult<Q, RDPs> {\n  const orderBy = useMemo(() => {\n    if (!sorting || sorting.length === 0) {\n      return undefined;\n    }\n\n    return sorting.reduce<{ [K in PropertyKeys<Q>]?: \"asc\" | \"desc\" }>(\n      (acc, sort) => {\n        acc[sort.id as PropertyKeys<Q>] = sort.desc ? \"desc\" : \"asc\";\n        return acc;\n      },\n      {},\n    );\n  }, [sorting]);\n\n  // Extract derived properties definition\n  const withProperties = useMemo(() => {\n    if (!columnDefinitions) {\n      return;\n    }\n\n    const rdpColumns = columnDefinitions.map(colDef => colDef.locator).filter(\n      (colLocator) => {\n        return colLocator.type === \"rdp\";\n      },\n    );\n\n    if (!rdpColumns.length) {\n      return;\n    }\n\n    return rdpColumns.reduce<WithProperties<Q, RDPs>>(\n      (acc, cur) => {\n        return {\n          ...acc,\n          [cur.id]: cur.creator,\n        };\n      },\n      {} as WithProperties<Q, RDPs>,\n    );\n  }, [columnDefinitions]);\n\n  // When objectSet is provided and it's an object type, use useObjectSet. Otherwise, use useOsdkObjects.\n  const isObjectType = objectOrInterfaceType.type === \"object\";\n  const shouldUseObjectSet = !!objectSet && isObjectType;\n\n  // When shouldUseObjectSet is true, we know objectSet is defined\n  // and objectOrInterfaceType is an ObjectTypeDefinition\n  const objectSetResult = useObjectSet(\n    shouldUseObjectSet ? objectSet as ObjectSet<Q, RDPs> : undefined as any,\n    {\n      ...(objectSetOptions as ObjectSetOptions<Q>),\n      withProperties: withProperties as WithProperties<\n        Q,\n        RDPs\n      >,\n      where: filter,\n      orderBy,\n      pageSize: PAGE_SIZE,\n      enabled: shouldUseObjectSet,\n    },\n  );\n\n  const osdkObjectsResult = useOsdkObjects<\n    Q,\n    RDPs\n  >(\n    objectOrInterfaceType,\n    {\n      withProperties,\n      pageSize: PAGE_SIZE,\n      where: filter,\n      orderBy,\n      enabled: !shouldUseObjectSet,\n    },\n  );\n\n  // Get the result from the appropriate hook\n  const baseResult = shouldUseObjectSet ? objectSetResult : osdkObjectsResult;\n\n  // Call useFunctionColumnsData to get function column data\n  const functionColumnData = useFunctionColumnsData<Q, RDPs, FunctionColumns>(\n    objectSetResult.objectSet,\n    baseResult.data,\n    columnDefinitions,\n  );\n\n  // Merge function column data into each object\n  const mergedData = useMemo(() => {\n    if (!baseResult.data) return baseResult.data;\n\n    return baseResult.data.map(obj => {\n      const objKey = String(obj.$primaryKey);\n      const functionData: Record<string, AsyncCellData> = {};\n\n      // Collect all function column data for this object\n      Object.entries(functionColumnData).forEach(([columnId, columnData]) => {\n        if (columnData[objKey]) {\n          functionData[columnId] = columnData[objKey];\n        }\n      });\n\n      // Return object with function data merged in\n      return {\n        ...obj,\n        ...functionData,\n      };\n    });\n  }, [baseResult.data, functionColumnData]);\n\n  // Return the result with merged data\n  return {\n    ...baseResult,\n    data: mergedData,\n  } as UseObjectTableDataResult<Q, RDPs>;\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAYA,SAASA,YAAY,EAAEC,cAAc,QAAQ,0BAA0B;AAEvE,SAASC,OAAO,QAAQ,OAAO;AAG/B,SAASC,sBAAsB,QAAQ,6BAA6B;AAEpE,MAAMC,SAAS,GAAG,EAAE;AAkBpB;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,SAASC,kBAAkBA,CAWhCC,qBAAwB,EACxBC,iBAAqE,EACrEC,MAA6B,EAC7BC,OAAsB,EACtBC,SAAwB,EACxBC,gBAAsC,EACH;EACnC,MAAMC,OAAO,GAAGV,OAAO,CAAC,MAAM;IAC5B,IAAI,CAACO,OAAO,IAAIA,OAAO,CAACI,MAAM,KAAK,CAAC,EAAE;MACpC,OAAOC,SAAS;IAClB;IAEA,OAAOL,OAAO,CAACM,MAAM,CACnB,CAACC,GAAG,EAAEC,IAAI,KAAK;MACbD,GAAG,CAACC,IAAI,CAACC,EAAE,CAAoB,GAAGD,IAAI,CAACE,IAAI,GAAG,MAAM,GAAG,KAAK;MAC5D,OAAOH,GAAG;IACZ,CAAC,EACD,CAAC,CACH,CAAC;EACH,CAAC,EAAE,CAACP,OAAO,CAAC,CAAC;;EAEb;EACA,MAAMW,cAAc,GAAGlB,OAAO,CAAC,MAAM;IACnC,IAAI,CAACK,iBAAiB,EAAE;MACtB;IACF;IAEA,MAAMc,UAAU,GAAGd,iBAAiB,CAACe,GAAG,CAACC,MAAM,IAAIA,MAAM,CAACC,OAAO,CAAC,CAAChB,MAAM,CACtEiB,UAAU,IAAK;MACd,OAAOA,UAAU,CAACC,IAAI,KAAK,KAAK;IAClC,CACF,CAAC;IAED,IAAI,CAACL,UAAU,CAACR,MAAM,EAAE;MACtB;IACF;IAEA,OAAOQ,UAAU,CAACN,MAAM,CACtB,CAACC,GAAG,EAAEW,GAAG,KAAK;MACZ,OAAO;QACL,GAAGX,GAAG;QACN,CAACW,GAAG,CAACT,EAAE,GAAGS,GAAG,CAACC;MAChB,CAAC;IACH,CAAC,EACD,CAAC,CACH,CAAC;EACH,CAAC,EAAE,CAACrB,iBAAiB,CAAC,CAAC;;EAEvB;EACA,MAAMsB,YAAY,GAAGvB,qBAAqB,CAACoB,IAAI,KAAK,QAAQ;EAC5D,MAAMI,kBAAkB,GAAG,CAAC,CAACpB,SAAS,IAAImB,YAAY;;EAEtD;EACA;EACA,MAAME,eAAe,GAAG/B,YAAY,CAClC8B,kBAAkB,GAAGpB,SAAS,GAAyBI,SAAgB,EACvE;IACE,GAAIH,gBAAwC;IAC5CS,cAAc,EAAEA,cAGf;IACDY,KAAK,EAAExB,MAAM;IACbI,OAAO;IACPqB,QAAQ,EAAE7B,SAAS;IACnB8B,OAAO,EAAEJ;EACX,CACF,CAAC;EAED,MAAMK,iBAAiB,GAAGlC,cAAc,CAItCK,qBAAqB,EACrB;IACEc,cAAc;IACda,QAAQ,EAAE7B,SAAS;IACnB4B,KAAK,EAAExB,MAAM;IACbI,OAAO;IACPsB,OAAO,EAAE,CAACJ;EACZ,CACF,CAAC;;EAED;EACA,MAAMM,UAAU,GAAGN,kBAAkB,GAAGC,eAAe,GAAGI,iBAAiB;;EAE3E;EACA,MAAME,kBAAkB,GAAGlC,sBAAsB,CAC/C4B,eAAe,CAACrB,SAAS,EACzB0B,UAAU,CAACE,IAAI,EACf/B,iBACF,CAAC;;EAED;EACA,MAAMgC,UAAU,GAAGrC,OAAO,CAAC,MAAM;IAC/B,IAAI,CAACkC,UAAU,CAACE,IAAI,EAAE,OAAOF,UAAU,CAACE,IAAI;IAE5C,OAAOF,UAAU,CAACE,IAAI,CAAChB,GAAG,CAACkB,GAAG,IAAI;MAChC,MAAMC,MAAM,GAAGC,MAAM,CAACF,GAAG,CAACG,WAAW,CAAC;MACtC,MAAMC,YAA2C,GAAG,CAAC,CAAC;;MAEtD;MACAC,MAAM,CAACC,OAAO,CAACT,kBAAkB,CAAC,CAACU,OAAO,CAAC,CAAC,CAACC,QAAQ,EAAEC,UAAU,CAAC,KAAK;QACrE,IAAIA,UAAU,CAACR,MAAM,CAAC,EAAE;UACtBG,YAAY,CAACI,QAAQ,CAAC,GAAGC,UAAU,CAACR,MAAM,CAAC;QAC7C;MACF,CAAC,CAAC;;MAEF;MACA,OAAO;QACL,GAAGD,GAAG;QACN,GAAGI;MACL,CAAC;IACH,CAAC,CAAC;EACJ,CAAC,EAAE,CAACR,UAAU,CAACE,IAAI,EAAED,kBAAkB,CAAC,CAAC;;EAEzC;EACA,OAAO;IACL,GAAGD,UAAU;IACbE,IAAI,EAAEC;EACR,CAAC;AACH","ignoreList":[]}

package/build/esm/object-table/hooks/useRowSelection.js

@@ -19,6 +19,7 @@ import { getRowId, getRowIdFromPrimaryKey } from "../utils/getRowId.js";
 export function useRowSelection({
   selectionMode = "none",
   selectedRows,
+  isAllSelected: isAllSelectedProp,
   onRowSelection,
   data
 }) {
@@ -28,31 +29,33 @@ export function useRowSelection({
   // Used for shift-click behavior in "multiple" mode
   const [lastSelectedRowIndex, setLastSelectedRowIndex] = useState(null);
   const isControlled = selectedRows !== undefined;
-  const isSelectionEnabled = selectionMode !== "none";
+  const enableRowSelection = selectionMode !== "none";
 
   // Row selection state
   // If controlled mode, return the state from selectedRows prop
   // If uncontrolled, return the internalRowSelection state
   const rowSelectionState = useMemo(() => {
-    if (!isSelectionEnabled) return {};
-    if (isControlled && selectedRows) {
-      return getRowSelectionState(selectedRows);
+    if (!enableRowSelection) return {};
+    if (isControlled) {
+      const selectedRowIds = isAllSelectedProp ? (data ?? []).map(item => item.$primaryKey) : selectedRows;
+      return getRowSelectionState(selectedRowIds);
     }
     return internalRowSelection;
-  }, [isSelectionEnabled, isControlled, selectedRows, internalRowSelection]);
+  }, [enableRowSelection, isControlled, selectedRows, isAllSelectedProp, internalRowSelection, data]);
   const selectedCount = Object.values(rowSelectionState).filter(Boolean).length;
   const totalCount = data?.length ?? 0;
-  const isAllSelected = totalCount > 0 && selectedCount === totalCount;
+  // In controlled mode, use the prop if provided, otherwise calculate based on selected count
+  const isAllSelected = isControlled && isAllSelectedProp !== undefined ? isAllSelectedProp : totalCount > 0 && selectedCount === totalCount;
   const onToggleAll = useCallback(() => {
-    if (!isSelectionEnabled || !data) return;
+    if (!enableRowSelection || !data) return;
     const newSelectedRows = isAllSelected ? [] : data.map(item => item.$primaryKey);
     if (!isControlled) {
       setInternalRowSelection(getRowSelectionState(newSelectedRows));
     }
-    onRowSelection?.(newSelectedRows);
-  }, [isSelectionEnabled, data, isAllSelected, isControlled, onRowSelection]);
+    onRowSelection?.(newSelectedRows, true);
+  }, [enableRowSelection, data, isAllSelected, isControlled, onRowSelection]);
   const onToggleRow = useCallback((rowId, rowIndex, isShiftClick = false) => {
-    if (!isSelectionEnabled || !data) return;
+    if (!enableRowSelection || !data) return;
     let newSelectedRows = [];
     if (selectionMode === "single") {
       newSelectedRows = getSingleSelectionRows({
@@ -61,7 +64,6 @@ export function useRowSelection({
         data,
         rowSelectionState
       });
-      setLastSelectedRowIndex(rowIndex);
     } else {
       // Multiple selection mode
 
@@ -75,6 +77,7 @@ export function useRowSelection({
           lastSelectedRowIndex,
           rowSelectionState
         });
+        setLastSelectedRowIndex(rowIndex);
       } else {
         newSelectedRows = getMultipleSelectionRows({
           rowId,
@@ -82,18 +85,26 @@ export function useRowSelection({
           data,
           rowSelectionState
         });
+        // Only update lastSelectedRowIndex if we're selecting (not deselecting)
+        if (!isCurrentlySelected({
+          rowIndex,
+          data,
+          rowSelectionState
+        })) {
+          setLastSelectedRowIndex(rowIndex);
+        }
       }
-      setLastSelectedRowIndex(rowIndex);
     }
     if (!isControlled) {
       setInternalRowSelection(getRowSelectionState(newSelectedRows));
     }
-    onRowSelection?.(newSelectedRows);
-  }, [isSelectionEnabled, data, selectionMode, lastSelectedRowIndex, isControlled, rowSelectionState, onRowSelection]);
+    onRowSelection?.(newSelectedRows, false);
+  }, [enableRowSelection, data, selectionMode, isControlled, onRowSelection, rowSelectionState, lastSelectedRowIndex]);
   return {
     rowSelection: rowSelectionState,
     isAllSelected,
-    hasSelection: selectedCount > 0,
+    enableRowSelection,
+    hasSelection: isAllSelected || selectedCount > 0,
     onToggleAll,
     onToggleRow
   };
@@ -133,6 +144,15 @@ function getRangeSelectionRows({
   }
   return [];
 }
+function isCurrentlySelected({
+  rowIndex,
+  data,
+  rowSelectionState
+}) {
+  const primaryKey = data[rowIndex].$primaryKey;
+  const currentlySelected = getSelectedPrimaryKeys(rowSelectionState, data);
+  return currentlySelected.includes(primaryKey);
+}
 function getMultipleSelectionRows({
   rowIndex,
   data,
@@ -141,7 +161,11 @@ function getMultipleSelectionRows({
   const primaryKey = data[rowIndex].$primaryKey;
   const currentlySelected = getSelectedPrimaryKeys(rowSelectionState, data);
   // Handles single row toggle in multiple selection mode
-  const newSelectedRows = currentlySelected.includes(primaryKey) ? currentlySelected.filter(i => i !== primaryKey) : [...currentlySelected, primaryKey];
+  const newSelectedRows = isCurrentlySelected({
+    rowIndex,
+    data,
+    rowSelectionState
+  }) ? currentlySelected.filter(i => i !== primaryKey) : [...currentlySelected, primaryKey];
   return newSelectedRows;
 }
 

package/build/esm/object-table/hooks/useRowSelection.js.map

@@ -1 +1 @@
-{"version":3,"file":"useRowSelection.js","names":["useCallback","useMemo","useState","getRowId","getRowIdFromPrimaryKey","useRowSelection","selectionMode","selectedRows","onRowSelection","data","internalRowSelection","setInternalRowSelection","lastSelectedRowIndex","setLastSelectedRowIndex","isControlled","undefined","isSelectionEnabled","rowSelectionState","getRowSelectionState","selectedCount","Object","values","filter","Boolean","length","totalCount","isAllSelected","onToggleAll","newSelectedRows","map","item","$primaryKey","onToggleRow","rowId","rowIndex","isShiftClick","getSingleSelectionRows","getRangeSelectionRows","getMultipleSelectionRows","rowSelection","hasSelection","primaryKey","rowsInRange","getRowsInRange","primaryKeysInRange","currentlySelected","getSelectedPrimaryKeys","forEach","includes","push","i","startIndex","endIndex","start","Math","min","end","max","rows","primaryKeys","reduce","acc","selectionState"],"sources":["useRowSelection.ts"],"sourcesContent":["/*\n * Copyright 2025 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type {\n  ObjectOrInterfaceDefinition,\n  Osdk,\n  PrimaryKeyType,\n} from \"@osdk/api\";\nimport type { RowSelectionState } from \"@tanstack/react-table\";\nimport { useCallback, useMemo, useState } from \"react\";\nimport { getRowId, getRowIdFromPrimaryKey } from \"../utils/getRowId.js\";\n\nexport interface UseRowSelectionProps<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, any> = Record<string, never>,\n> {\n  selectionMode?: \"single\" | \"multiple\" | \"none\";\n  selectedRows?: PrimaryKeyType<Q>[];\n  onRowSelection?: (selectedRowIds: PrimaryKeyType<Q>[]) => void;\n  data: Array<Osdk.Instance<Q, \"$allBaseProperties\", any, RDPs>> | undefined;\n}\n\nexport interface UseRowSelectionResult {\n  rowSelection: RowSelectionState;\n  isAllSelected: boolean;\n  hasSelection: boolean;\n  onToggleAll: () => void;\n  onToggleRow: (\n    rowId: string,\n    rowIndex: number,\n    isShiftClick?: boolean,\n  ) => void;\n}\n\nexport function useRowSelection<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, any> = Record<string, never>,\n>({\n  selectionMode = \"none\",\n  selectedRows,\n  onRowSelection,\n  data,\n}: UseRowSelectionProps<Q, RDPs>): UseRowSelectionResult {\n  // The rowSelectionState in uncontrolled mode\n  const [internalRowSelection, setInternalRowSelection] = useState<\n    RowSelectionState\n  >({});\n\n  // Used for shift-click behavior in \"multiple\" mode\n  const [lastSelectedRowIndex, setLastSelectedRowIndex] = useState<\n    number | null\n  >(null);\n\n  const isControlled = selectedRows !== undefined;\n  const isSelectionEnabled = selectionMode !== \"none\";\n\n  // Row selection state\n  // If controlled mode, return the state from selectedRows prop\n  // If uncontrolled, return the internalRowSelection state\n  const rowSelectionState: RowSelectionState = useMemo(() => {\n    if (!isSelectionEnabled) return {};\n    if (isControlled && selectedRows) {\n      return getRowSelectionState(selectedRows);\n    }\n    return internalRowSelection;\n  }, [\n    isSelectionEnabled,\n    isControlled,\n    selectedRows,\n    internalRowSelection,\n  ]);\n\n  const selectedCount = Object.values(rowSelectionState).filter(Boolean).length;\n  const totalCount = data?.length ?? 0;\n  const isAllSelected = totalCount > 0 && selectedCount === totalCount;\n  const hasSelection = selectedCount > 0;\n\n  const onToggleAll = useCallback(() => {\n    if (!isSelectionEnabled || !data) return;\n\n    const newSelectedRows: PrimaryKeyType<Q>[] = isAllSelected\n      ? []\n      : data.map(item => item.$primaryKey);\n\n    if (!isControlled) {\n      setInternalRowSelection(getRowSelectionState(newSelectedRows));\n    }\n    onRowSelection?.(newSelectedRows);\n  }, [isSelectionEnabled, data, isAllSelected, isControlled, onRowSelection]);\n\n  const onToggleRow = useCallback(\n    (rowId: string, rowIndex: number, isShiftClick: boolean = false) => {\n      if (!isSelectionEnabled || !data) return;\n\n      let newSelectedRows: PrimaryKeyType<Q>[] = [];\n\n      if (selectionMode === \"single\") {\n        newSelectedRows = getSingleSelectionRows({\n          rowId,\n          rowIndex,\n          data,\n          rowSelectionState,\n        });\n        setLastSelectedRowIndex(rowIndex);\n      } else {\n        // Multiple selection mode\n\n        // When user does shiftClick but lastSelectedRowIndex is null,\n        // it is treated as a normal click in multiple selection\n        if (isShiftClick && lastSelectedRowIndex != null) {\n          newSelectedRows = getRangeSelectionRows(\n            { rowId, rowIndex, data, lastSelectedRowIndex, rowSelectionState },\n          );\n        } else {\n          newSelectedRows = getMultipleSelectionRows(\n            { rowId, rowIndex, data, rowSelectionState },\n          );\n        }\n        setLastSelectedRowIndex(rowIndex);\n      }\n      if (!isControlled) {\n        setInternalRowSelection(getRowSelectionState(newSelectedRows));\n      }\n      onRowSelection?.(newSelectedRows);\n    },\n    [\n      isSelectionEnabled,\n      data,\n      selectionMode,\n      lastSelectedRowIndex,\n      isControlled,\n      rowSelectionState,\n      onRowSelection,\n    ],\n  );\n\n  return {\n    rowSelection: rowSelectionState,\n    isAllSelected,\n    hasSelection,\n    onToggleAll,\n    onToggleRow,\n  };\n}\n\ninterface GetSelectedRowsProps<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, any> = Record<string, never>,\n> {\n  rowId: string;\n  rowIndex: number;\n  data: Array<Osdk.Instance<Q, \"$allBaseProperties\", any, RDPs>>;\n  rowSelectionState: RowSelectionState;\n  lastSelectedRowIndex?: number;\n}\n\nfunction getSingleSelectionRows<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, any> = Record<string, never>,\n>(\n  { rowId, rowIndex, data, rowSelectionState }: GetSelectedRowsProps<Q, RDPs>,\n): PrimaryKeyType<Q>[] {\n  const primaryKey = data[rowIndex].$primaryKey;\n  // Toggle row selection in single selection mode\n  const newSelectedRows = rowSelectionState[rowId] ? [] : [primaryKey];\n  return newSelectedRows;\n}\n\nfunction getRangeSelectionRows<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, any> = Record<string, never>,\n>(\n  { lastSelectedRowIndex, rowIndex, data, rowSelectionState }:\n    GetSelectedRowsProps<Q, RDPs>,\n): PrimaryKeyType<Q>[] {\n  // This function is only called if lastSelectedRowIndex is not null\n  // This condition is added for typechecks only\n  if (lastSelectedRowIndex != null) {\n    const rowsInRange = getRowsInRange(data, lastSelectedRowIndex, rowIndex);\n    const primaryKeysInRange = rowsInRange.map(item => item.$primaryKey);\n\n    const currentlySelected = getSelectedPrimaryKeys(rowSelectionState, data);\n\n    // Add all rows in range to selectedRows if not yet selected\n    const newSelectedRows = [...currentlySelected];\n    primaryKeysInRange.forEach(item => {\n      if (!newSelectedRows.includes(item)) {\n        newSelectedRows.push(item);\n      }\n    });\n    return newSelectedRows;\n  }\n  return [];\n}\n\nfunction getMultipleSelectionRows<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, any> = Record<string, never>,\n>(\n  { rowIndex, data, rowSelectionState }: GetSelectedRowsProps<Q, RDPs>,\n): PrimaryKeyType<Q>[] {\n  const primaryKey = data[rowIndex].$primaryKey;\n  const currentlySelected = getSelectedPrimaryKeys(rowSelectionState, data);\n  // Handles single row toggle in multiple selection mode\n  const newSelectedRows = currentlySelected.includes(primaryKey)\n    ? currentlySelected.filter(i => i !== primaryKey)\n    : [...currentlySelected, primaryKey];\n  return newSelectedRows;\n}\n\n/**\n * Builds a range of rows from startIndex to endIndex\n */\nfunction getRowsInRange<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, any> = Record<string, never>,\n>(\n  data: Array<Osdk.Instance<Q, \"$allBaseProperties\", any, RDPs>>,\n  startIndex: number,\n  endIndex: number,\n): Array<Osdk.Instance<Q, \"$allBaseProperties\", any, RDPs>> {\n  const start = Math.min(startIndex, endIndex);\n  const end = Math.max(startIndex, endIndex);\n  const rows: Array<Osdk.Instance<Q, \"$allBaseProperties\", any, RDPs>> = [];\n\n  for (let i = start; i <= end; i++) {\n    const item = data[i];\n    if (item) {\n      rows.push(item);\n    }\n  }\n\n  return rows;\n}\n\n/**\n * Converts an array of primary keys to a RowSelectionState object\n */\nfunction getRowSelectionState<Q extends ObjectOrInterfaceDefinition>(\n  primaryKeys: PrimaryKeyType<Q>[],\n): RowSelectionState {\n  return primaryKeys.reduce<RowSelectionState>(\n    (acc, primaryKey) => {\n      acc[getRowIdFromPrimaryKey(primaryKey)] = true;\n      return acc;\n    },\n    {},\n  );\n}\n\n/**\n * Converts from a RowSelectionState object back to an array of primary keys\n */\nfunction getSelectedPrimaryKeys<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, any> = Record<string, never>,\n>(\n  selectionState: RowSelectionState,\n  data: Array<Osdk.Instance<Q, \"$allBaseProperties\", any, RDPs>>,\n): PrimaryKeyType<Q>[] {\n  return data\n    .filter(item => selectionState[getRowId(item)])\n    .map(item => item.$primaryKey);\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAQA,SAASA,WAAW,EAAEC,OAAO,EAAEC,QAAQ,QAAQ,OAAO;AACtD,SAASC,QAAQ,EAAEC,sBAAsB,QAAQ,sBAAsB;AAwBvE,OAAO,SAASC,eAAeA,CAG7B;EACAC,aAAa,GAAG,MAAM;EACtBC,YAAY;EACZC,cAAc;EACdC;AAC6B,CAAC,EAAyB;EACvD;EACA,MAAM,CAACC,oBAAoB,EAAEC,uBAAuB,CAAC,GAAGT,QAAQ,CAE9D,CAAC,CAAC,CAAC;;EAEL;EACA,MAAM,CAACU,oBAAoB,EAAEC,uBAAuB,CAAC,GAAGX,QAAQ,CAE9D,IAAI,CAAC;EAEP,MAAMY,YAAY,GAAGP,YAAY,KAAKQ,SAAS;EAC/C,MAAMC,kBAAkB,GAAGV,aAAa,KAAK,MAAM;;EAEnD;EACA;EACA;EACA,MAAMW,iBAAoC,GAAGhB,OAAO,CAAC,MAAM;IACzD,IAAI,CAACe,kBAAkB,EAAE,OAAO,CAAC,CAAC;IAClC,IAAIF,YAAY,IAAIP,YAAY,EAAE;MAChC,OAAOW,oBAAoB,CAACX,YAAY,CAAC;IAC3C;IACA,OAAOG,oBAAoB;EAC7B,CAAC,EAAE,CACDM,kBAAkB,EAClBF,YAAY,EACZP,YAAY,EACZG,oBAAoB,CACrB,CAAC;EAEF,MAAMS,aAAa,GAAGC,MAAM,CAACC,MAAM,CAACJ,iBAAiB,CAAC,CAACK,MAAM,CAACC,OAAO,CAAC,CAACC,MAAM;EAC7E,MAAMC,UAAU,GAAGhB,IAAI,EAAEe,MAAM,IAAI,CAAC;EACpC,MAAME,aAAa,GAAGD,UAAU,GAAG,CAAC,IAAIN,aAAa,KAAKM,UAAU;EAGpE,MAAME,WAAW,GAAG3B,WAAW,CAAC,MAAM;IACpC,IAAI,CAACgB,kBAAkB,IAAI,CAACP,IAAI,EAAE;IAElC,MAAMmB,eAAoC,GAAGF,aAAa,GACtD,EAAE,GACFjB,IAAI,CAACoB,GAAG,CAACC,IAAI,IAAIA,IAAI,CAACC,WAAW,CAAC;IAEtC,IAAI,CAACjB,YAAY,EAAE;MACjBH,uBAAuB,CAACO,oBAAoB,CAACU,eAAe,CAAC,CAAC;IAChE;IACApB,cAAc,GAAGoB,eAAe,CAAC;EACnC,CAAC,EAAE,CAACZ,kBAAkB,EAAEP,IAAI,EAAEiB,aAAa,EAAEZ,YAAY,EAAEN,cAAc,CAAC,CAAC;EAE3E,MAAMwB,WAAW,GAAGhC,WAAW,CAC7B,CAACiC,KAAa,EAAEC,QAAgB,EAAEC,YAAqB,GAAG,KAAK,KAAK;IAClE,IAAI,CAACnB,kBAAkB,IAAI,CAACP,IAAI,EAAE;IAElC,IAAImB,eAAoC,GAAG,EAAE;IAE7C,IAAItB,aAAa,KAAK,QAAQ,EAAE;MAC9BsB,eAAe,GAAGQ,sBAAsB,CAAC;QACvCH,KAAK;QACLC,QAAQ;QACRzB,IAAI;QACJQ;MACF,CAAC,CAAC;MACFJ,uBAAuB,CAACqB,QAAQ,CAAC;IACnC,CAAC,MAAM;MACL;;MAEA;MACA;MACA,IAAIC,YAAY,IAAIvB,oBAAoB,IAAI,IAAI,EAAE;QAChDgB,eAAe,GAAGS,qBAAqB,CACrC;UAAEJ,KAAK;UAAEC,QAAQ;UAAEzB,IAAI;UAAEG,oBAAoB;UAAEK;QAAkB,CACnE,CAAC;MACH,CAAC,MAAM;QACLW,eAAe,GAAGU,wBAAwB,CACxC;UAAEL,KAAK;UAAEC,QAAQ;UAAEzB,IAAI;UAAEQ;QAAkB,CAC7C,CAAC;MACH;MACAJ,uBAAuB,CAACqB,QAAQ,CAAC;IACnC;IACA,IAAI,CAACpB,YAAY,EAAE;MACjBH,uBAAuB,CAACO,oBAAoB,CAACU,eAAe,CAAC,CAAC;IAChE;IACApB,cAAc,GAAGoB,eAAe,CAAC;EACnC,CAAC,EACD,CACEZ,kBAAkB,EAClBP,IAAI,EACJH,aAAa,EACbM,oBAAoB,EACpBE,YAAY,EACZG,iBAAiB,EACjBT,cAAc,CAElB,CAAC;EAED,OAAO;IACL+B,YAAY,EAAEtB,iBAAiB;IAC/BS,aAAa;IACbc,YAAY,EAhEOrB,aAAa,GAAG,CAgEvB;IACZQ,WAAW;IACXK;EACF,CAAC;AACH;AAaA,SAASI,sBAAsBA,CAI7B;EAAEH,KAAK;EAAEC,QAAQ;EAAEzB,IAAI;EAAEQ;AAAiD,CAAC,EACtD;EACrB,MAAMwB,UAAU,GAAGhC,IAAI,CAACyB,QAAQ,CAAC,CAACH,WAAW;EAC7C;EACA,MAAMH,eAAe,GAAGX,iBAAiB,CAACgB,KAAK,CAAC,GAAG,EAAE,GAAG,CAACQ,UAAU,CAAC;EACpE,OAAOb,eAAe;AACxB;AAEA,SAASS,qBAAqBA,CAI5B;EAAEzB,oBAAoB;EAAEsB,QAAQ;EAAEzB,IAAI;EAAEQ;AACV,CAAC,EACV;EACrB;EACA;EACA,IAAIL,oBAAoB,IAAI,IAAI,EAAE;IAChC,MAAM8B,WAAW,GAAGC,cAAc,CAAClC,IAAI,EAAEG,oBAAoB,EAAEsB,QAAQ,CAAC;IACxE,MAAMU,kBAAkB,GAAGF,WAAW,CAACb,GAAG,CAACC,IAAI,IAAIA,IAAI,CAACC,WAAW,CAAC;IAEpE,MAAMc,iBAAiB,GAAGC,sBAAsB,CAAC7B,iBAAiB,EAAER,IAAI,CAAC;;IAEzE;IACA,MAAMmB,eAAe,GAAG,CAAC,GAAGiB,iBAAiB,CAAC;IAC9CD,kBAAkB,CAACG,OAAO,CAACjB,IAAI,IAAI;MACjC,IAAI,CAACF,eAAe,CAACoB,QAAQ,CAAClB,IAAI,CAAC,EAAE;QACnCF,eAAe,CAACqB,IAAI,CAACnB,IAAI,CAAC;MAC5B;IACF,CAAC,CAAC;IACF,OAAOF,eAAe;EACxB;EACA,OAAO,EAAE;AACX;AAEA,SAASU,wBAAwBA,CAI/B;EAAEJ,QAAQ;EAAEzB,IAAI;EAAEQ;AAAiD,CAAC,EAC/C;EACrB,MAAMwB,UAAU,GAAGhC,IAAI,CAACyB,QAAQ,CAAC,CAACH,WAAW;EAC7C,MAAMc,iBAAiB,GAAGC,sBAAsB,CAAC7B,iBAAiB,EAAER,IAAI,CAAC;EACzE;EACA,MAAMmB,eAAe,GAAGiB,iBAAiB,CAACG,QAAQ,CAACP,UAAU,CAAC,GAC1DI,iBAAiB,CAACvB,MAAM,CAAC4B,CAAC,IAAIA,CAAC,KAAKT,UAAU,CAAC,GAC/C,CAAC,GAAGI,iBAAiB,EAAEJ,UAAU,CAAC;EACtC,OAAOb,eAAe;AACxB;;AAEA;AACA;AACA;AACA,SAASe,cAAcA,CAIrBlC,IAA8D,EAC9D0C,UAAkB,EAClBC,QAAgB,EAC0C;EAC1D,MAAMC,KAAK,GAAGC,IAAI,CAACC,GAAG,CAACJ,UAAU,EAAEC,QAAQ,CAAC;EAC5C,MAAMI,GAAG,GAAGF,IAAI,CAACG,GAAG,CAACN,UAAU,EAAEC,QAAQ,CAAC;EAC1C,MAAMM,IAA8D,GAAG,EAAE;EAEzE,KAAK,IAAIR,CAAC,GAAGG,KAAK,EAAEH,CAAC,IAAIM,GAAG,EAAEN,CAAC,EAAE,EAAE;IACjC,MAAMpB,IAAI,GAAGrB,IAAI,CAACyC,CAAC,CAAC;IACpB,IAAIpB,IAAI,EAAE;MACR4B,IAAI,CAACT,IAAI,CAACnB,IAAI,CAAC;IACjB;EACF;EAEA,OAAO4B,IAAI;AACb;;AAEA;AACA;AACA;AACA,SAASxC,oBAAoBA,CAC3ByC,WAAgC,EACb;EACnB,OAAOA,WAAW,CAACC,MAAM,CACvB,CAACC,GAAG,EAAEpB,UAAU,KAAK;IACnBoB,GAAG,CAACzD,sBAAsB,CAACqC,UAAU,CAAC,CAAC,GAAG,IAAI;IAC9C,OAAOoB,GAAG;EACZ,CAAC,EACD,CAAC,CACH,CAAC;AACH;;AAEA;AACA;AACA;AACA,SAASf,sBAAsBA,CAI7BgB,cAAiC,EACjCrD,IAA8D,EACzC;EACrB,OAAOA,IAAI,CACRa,MAAM,CAACQ,IAAI,IAAIgC,cAAc,CAAC3D,QAAQ,CAAC2B,IAAI,CAAC,CAAC,CAAC,CAC9CD,GAAG,CAACC,IAAI,IAAIA,IAAI,CAACC,WAAW,CAAC;AAClC","ignoreList":[]}
+{"version":3,"file":"useRowSelection.js","names":["useCallback","useMemo","useState","getRowId","getRowIdFromPrimaryKey","useRowSelection","selectionMode","selectedRows","isAllSelected","isAllSelectedProp","onRowSelection","data","internalRowSelection","setInternalRowSelection","lastSelectedRowIndex","setLastSelectedRowIndex","isControlled","undefined","enableRowSelection","rowSelectionState","selectedRowIds","map","item","$primaryKey","getRowSelectionState","selectedCount","Object","values","filter","Boolean","length","totalCount","onToggleAll","newSelectedRows","onToggleRow","rowId","rowIndex","isShiftClick","getSingleSelectionRows","getRangeSelectionRows","getMultipleSelectionRows","isCurrentlySelected","rowSelection","hasSelection","primaryKey","rowsInRange","getRowsInRange","primaryKeysInRange","currentlySelected","getSelectedPrimaryKeys","forEach","includes","push","i","startIndex","endIndex","start","Math","min","end","max","rows","primaryKeys","reduce","acc","selectionState"],"sources":["useRowSelection.ts"],"sourcesContent":["/*\n * Copyright 2025 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type {\n  ObjectOrInterfaceDefinition,\n  Osdk,\n  PrimaryKeyType,\n  PropertyKeys,\n  SimplePropertyDef,\n} from \"@osdk/api\";\nimport type { RowSelectionState } from \"@tanstack/react-table\";\nimport { useCallback, useMemo, useState } from \"react\";\nimport { getRowId, getRowIdFromPrimaryKey } from \"../utils/getRowId.js\";\n\nexport interface UseRowSelectionProps<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n> {\n  selectionMode?: \"single\" | \"multiple\" | \"none\";\n  selectedRows?: PrimaryKeyType<Q>[];\n  isAllSelected?: boolean;\n  onRowSelection?: (\n    selectedRowIds: PrimaryKeyType<Q>[],\n    isSelectAll: boolean,\n  ) => void;\n  data:\n    | Array<Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>>\n    | undefined;\n}\n\nexport interface UseRowSelectionResult {\n  rowSelection: RowSelectionState;\n  isAllSelected: boolean;\n  hasSelection: boolean;\n  enableRowSelection: boolean;\n  onToggleAll: () => void;\n  onToggleRow: (\n    rowId: string,\n    rowIndex: number,\n    isShiftClick?: boolean,\n  ) => void;\n}\n\nexport function useRowSelection<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n>({\n  selectionMode = \"none\",\n  selectedRows,\n  isAllSelected: isAllSelectedProp,\n  onRowSelection,\n  data,\n}: UseRowSelectionProps<Q, RDPs>): UseRowSelectionResult {\n  // The rowSelectionState in uncontrolled mode\n  const [internalRowSelection, setInternalRowSelection] = useState<\n    RowSelectionState\n  >({});\n\n  // Used for shift-click behavior in \"multiple\" mode\n  const [lastSelectedRowIndex, setLastSelectedRowIndex] = useState<\n    number | null\n  >(null);\n\n  const isControlled = selectedRows !== undefined;\n  const enableRowSelection = selectionMode !== \"none\";\n\n  // Row selection state\n  // If controlled mode, return the state from selectedRows prop\n  // If uncontrolled, return the internalRowSelection state\n  const rowSelectionState: RowSelectionState = useMemo(() => {\n    if (!enableRowSelection) return {};\n    if (isControlled) {\n      const selectedRowIds = isAllSelectedProp\n        ? (data ?? []).map(item => item.$primaryKey)\n        : selectedRows;\n      return getRowSelectionState(selectedRowIds);\n    }\n    return internalRowSelection;\n  }, [\n    enableRowSelection,\n    isControlled,\n    selectedRows,\n    isAllSelectedProp,\n    internalRowSelection,\n    data,\n  ]);\n\n  const selectedCount = Object.values(rowSelectionState).filter(Boolean).length;\n  const totalCount = data?.length ?? 0;\n  // In controlled mode, use the prop if provided, otherwise calculate based on selected count\n  const isAllSelected = isControlled && isAllSelectedProp !== undefined\n    ? isAllSelectedProp\n    : totalCount > 0 && selectedCount === totalCount;\n  const hasSelection = isAllSelected || selectedCount > 0;\n\n  const onToggleAll = useCallback(() => {\n    if (!enableRowSelection || !data) return;\n\n    const newSelectedRows: PrimaryKeyType<Q>[] = isAllSelected\n      ? []\n      : data.map(item => item.$primaryKey);\n\n    if (!isControlled) {\n      setInternalRowSelection(getRowSelectionState(newSelectedRows));\n    }\n    onRowSelection?.(newSelectedRows, true);\n  }, [enableRowSelection, data, isAllSelected, isControlled, onRowSelection]);\n\n  const onToggleRow = useCallback(\n    (rowId: string, rowIndex: number, isShiftClick: boolean = false) => {\n      if (!enableRowSelection || !data) return;\n\n      let newSelectedRows: PrimaryKeyType<Q>[] = [];\n\n      if (selectionMode === \"single\") {\n        newSelectedRows = getSingleSelectionRows({\n          rowId,\n          rowIndex,\n          data,\n          rowSelectionState,\n        });\n      } else {\n        // Multiple selection mode\n\n        // When user does shiftClick but lastSelectedRowIndex is null,\n        // it is treated as a normal click in multiple selection\n        if (isShiftClick && lastSelectedRowIndex != null) {\n          newSelectedRows = getRangeSelectionRows(\n            { rowId, rowIndex, data, lastSelectedRowIndex, rowSelectionState },\n          );\n          setLastSelectedRowIndex(rowIndex);\n        } else {\n          newSelectedRows = getMultipleSelectionRows(\n            { rowId, rowIndex, data, rowSelectionState },\n          );\n          // Only update lastSelectedRowIndex if we're selecting (not deselecting)\n          if (\n            !isCurrentlySelected<Q, RDPs>({ rowIndex, data, rowSelectionState })\n          ) {\n            setLastSelectedRowIndex(rowIndex);\n          }\n        }\n      }\n      if (!isControlled) {\n        setInternalRowSelection(getRowSelectionState(newSelectedRows));\n      }\n      onRowSelection?.(newSelectedRows, false);\n    },\n    [\n      enableRowSelection,\n      data,\n      selectionMode,\n      isControlled,\n      onRowSelection,\n      rowSelectionState,\n      lastSelectedRowIndex,\n    ],\n  );\n\n  return {\n    rowSelection: rowSelectionState,\n    isAllSelected,\n    enableRowSelection,\n    hasSelection,\n    onToggleAll,\n    onToggleRow,\n  };\n}\n\ninterface GetSelectedRowsProps<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n> {\n  rowId: string;\n  rowIndex: number;\n  data: Array<Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>>;\n  rowSelectionState: RowSelectionState;\n  lastSelectedRowIndex?: number;\n}\n\nfunction getSingleSelectionRows<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n>(\n  { rowId, rowIndex, data, rowSelectionState }: GetSelectedRowsProps<Q, RDPs>,\n): PrimaryKeyType<Q>[] {\n  const primaryKey = data[rowIndex].$primaryKey;\n  // Toggle row selection in single selection mode\n  const newSelectedRows = rowSelectionState[rowId] ? [] : [primaryKey];\n  return newSelectedRows;\n}\n\nfunction getRangeSelectionRows<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n>(\n  { lastSelectedRowIndex, rowIndex, data, rowSelectionState }:\n    GetSelectedRowsProps<Q, RDPs>,\n): PrimaryKeyType<Q>[] {\n  // This function is only called if lastSelectedRowIndex is not null\n  // This condition is added for typechecks only\n  if (lastSelectedRowIndex != null) {\n    const rowsInRange = getRowsInRange(data, lastSelectedRowIndex, rowIndex);\n    const primaryKeysInRange = rowsInRange.map(item => item.$primaryKey);\n\n    const currentlySelected = getSelectedPrimaryKeys(rowSelectionState, data);\n\n    // Add all rows in range to selectedRows if not yet selected\n    const newSelectedRows = [...currentlySelected];\n    primaryKeysInRange.forEach(item => {\n      if (!newSelectedRows.includes(item)) {\n        newSelectedRows.push(item);\n      }\n    });\n    return newSelectedRows;\n  }\n  return [];\n}\n\nfunction isCurrentlySelected<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n>(\n  { rowIndex, data, rowSelectionState }: Pick<\n    GetSelectedRowsProps<Q, RDPs>,\n    \"rowIndex\" | \"data\" | \"rowSelectionState\"\n  >,\n): boolean {\n  const primaryKey = data[rowIndex].$primaryKey;\n  const currentlySelected = getSelectedPrimaryKeys(rowSelectionState, data);\n  return currentlySelected.includes(primaryKey);\n}\n\nfunction getMultipleSelectionRows<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n>(\n  { rowIndex, data, rowSelectionState }: GetSelectedRowsProps<Q, RDPs>,\n): PrimaryKeyType<Q>[] {\n  const primaryKey = data[rowIndex].$primaryKey;\n  const currentlySelected = getSelectedPrimaryKeys(rowSelectionState, data);\n  // Handles single row toggle in multiple selection mode\n  const newSelectedRows =\n    isCurrentlySelected<Q, RDPs>({ rowIndex, data, rowSelectionState })\n      ? currentlySelected.filter(i => i !== primaryKey)\n      : [...currentlySelected, primaryKey];\n  return newSelectedRows;\n}\n\n/**\n * Builds a range of rows from startIndex to endIndex\n */\nfunction getRowsInRange<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n>(\n  data: Array<Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>>,\n  startIndex: number,\n  endIndex: number,\n): Array<Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>> {\n  const start = Math.min(startIndex, endIndex);\n  const end = Math.max(startIndex, endIndex);\n  const rows: Array<\n    Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>\n  > = [];\n\n  for (let i = start; i <= end; i++) {\n    const item = data[i];\n    if (item) {\n      rows.push(item);\n    }\n  }\n\n  return rows;\n}\n\n/**\n * Converts an array of primary keys to a RowSelectionState object\n */\nfunction getRowSelectionState<Q extends ObjectOrInterfaceDefinition>(\n  primaryKeys: PrimaryKeyType<Q>[],\n): RowSelectionState {\n  return primaryKeys.reduce<RowSelectionState>(\n    (acc, primaryKey) => {\n      acc[getRowIdFromPrimaryKey(primaryKey)] = true;\n      return acc;\n    },\n    {},\n  );\n}\n\n/**\n * Converts from a RowSelectionState object back to an array of primary keys\n */\nfunction getSelectedPrimaryKeys<\n  Q extends ObjectOrInterfaceDefinition,\n  RDPs extends Record<string, SimplePropertyDef> = Record<string, never>,\n>(\n  selectionState: RowSelectionState,\n  data: Array<Osdk.Instance<Q, \"$allBaseProperties\", PropertyKeys<Q>, RDPs>>,\n): PrimaryKeyType<Q>[] {\n  return data\n    .filter(item => selectionState[getRowId(item)])\n    .map(item => item.$primaryKey);\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAUA,SAASA,WAAW,EAAEC,OAAO,EAAEC,QAAQ,QAAQ,OAAO;AACtD,SAASC,QAAQ,EAAEC,sBAAsB,QAAQ,sBAAsB;AA+BvE,OAAO,SAASC,eAAeA,CAG7B;EACAC,aAAa,GAAG,MAAM;EACtBC,YAAY;EACZC,aAAa,EAAEC,iBAAiB;EAChCC,cAAc;EACdC;AAC6B,CAAC,EAAyB;EACvD;EACA,MAAM,CAACC,oBAAoB,EAAEC,uBAAuB,CAAC,GAAGX,QAAQ,CAE9D,CAAC,CAAC,CAAC;;EAEL;EACA,MAAM,CAACY,oBAAoB,EAAEC,uBAAuB,CAAC,GAAGb,QAAQ,CAE9D,IAAI,CAAC;EAEP,MAAMc,YAAY,GAAGT,YAAY,KAAKU,SAAS;EAC/C,MAAMC,kBAAkB,GAAGZ,aAAa,KAAK,MAAM;;EAEnD;EACA;EACA;EACA,MAAMa,iBAAoC,GAAGlB,OAAO,CAAC,MAAM;IACzD,IAAI,CAACiB,kBAAkB,EAAE,OAAO,CAAC,CAAC;IAClC,IAAIF,YAAY,EAAE;MAChB,MAAMI,cAAc,GAAGX,iBAAiB,GACpC,CAACE,IAAI,IAAI,EAAE,EAAEU,GAAG,CAACC,IAAI,IAAIA,IAAI,CAACC,WAAW,CAAC,GAC1ChB,YAAY;MAChB,OAAOiB,oBAAoB,CAACJ,cAAc,CAAC;IAC7C;IACA,OAAOR,oBAAoB;EAC7B,CAAC,EAAE,CACDM,kBAAkB,EAClBF,YAAY,EACZT,YAAY,EACZE,iBAAiB,EACjBG,oBAAoB,EACpBD,IAAI,CACL,CAAC;EAEF,MAAMc,aAAa,GAAGC,MAAM,CAACC,MAAM,CAACR,iBAAiB,CAAC,CAACS,MAAM,CAACC,OAAO,CAAC,CAACC,MAAM;EAC7E,MAAMC,UAAU,GAAGpB,IAAI,EAAEmB,MAAM,IAAI,CAAC;EACpC;EACA,MAAMtB,aAAa,GAAGQ,YAAY,IAAIP,iBAAiB,KAAKQ,SAAS,GACjER,iBAAiB,GACjBsB,UAAU,GAAG,CAAC,IAAIN,aAAa,KAAKM,UAAU;EAGlD,MAAMC,WAAW,GAAGhC,WAAW,CAAC,MAAM;IACpC,IAAI,CAACkB,kBAAkB,IAAI,CAACP,IAAI,EAAE;IAElC,MAAMsB,eAAoC,GAAGzB,aAAa,GACtD,EAAE,GACFG,IAAI,CAACU,GAAG,CAACC,IAAI,IAAIA,IAAI,CAACC,WAAW,CAAC;IAEtC,IAAI,CAACP,YAAY,EAAE;MACjBH,uBAAuB,CAACW,oBAAoB,CAACS,eAAe,CAAC,CAAC;IAChE;IACAvB,cAAc,GAAGuB,eAAe,EAAE,IAAI,CAAC;EACzC,CAAC,EAAE,CAACf,kBAAkB,EAAEP,IAAI,EAAEH,aAAa,EAAEQ,YAAY,EAAEN,cAAc,CAAC,CAAC;EAE3E,MAAMwB,WAAW,GAAGlC,WAAW,CAC7B,CAACmC,KAAa,EAAEC,QAAgB,EAAEC,YAAqB,GAAG,KAAK,KAAK;IAClE,IAAI,CAACnB,kBAAkB,IAAI,CAACP,IAAI,EAAE;IAElC,IAAIsB,eAAoC,GAAG,EAAE;IAE7C,IAAI3B,aAAa,KAAK,QAAQ,EAAE;MAC9B2B,eAAe,GAAGK,sBAAsB,CAAC;QACvCH,KAAK;QACLC,QAAQ;QACRzB,IAAI;QACJQ;MACF,CAAC,CAAC;IACJ,CAAC,MAAM;MACL;;MAEA;MACA;MACA,IAAIkB,YAAY,IAAIvB,oBAAoB,IAAI,IAAI,EAAE;QAChDmB,eAAe,GAAGM,qBAAqB,CACrC;UAAEJ,KAAK;UAAEC,QAAQ;UAAEzB,IAAI;UAAEG,oBAAoB;UAAEK;QAAkB,CACnE,CAAC;QACDJ,uBAAuB,CAACqB,QAAQ,CAAC;MACnC,CAAC,MAAM;QACLH,eAAe,GAAGO,wBAAwB,CACxC;UAAEL,KAAK;UAAEC,QAAQ;UAAEzB,IAAI;UAAEQ;QAAkB,CAC7C,CAAC;QACD;QACA,IACE,CAACsB,mBAAmB,CAAU;UAAEL,QAAQ;UAAEzB,IAAI;UAAEQ;QAAkB,CAAC,CAAC,EACpE;UACAJ,uBAAuB,CAACqB,QAAQ,CAAC;QACnC;MACF;IACF;IACA,IAAI,CAACpB,YAAY,EAAE;MACjBH,uBAAuB,CAACW,oBAAoB,CAACS,eAAe,CAAC,CAAC;IAChE;IACAvB,cAAc,GAAGuB,eAAe,EAAE,KAAK,CAAC;EAC1C,CAAC,EACD,CACEf,kBAAkB,EAClBP,IAAI,EACJL,aAAa,EACbU,YAAY,EACZN,cAAc,EACdS,iBAAiB,EACjBL,oBAAoB,CAExB,CAAC;EAED,OAAO;IACL4B,YAAY,EAAEvB,iBAAiB;IAC/BX,aAAa;IACbU,kBAAkB;IAClByB,YAAY,EAtEOnC,aAAa,IAAIiB,aAAa,GAAG,CAsExC;IACZO,WAAW;IACXE;EACF,CAAC;AACH;AAaA,SAASI,sBAAsBA,CAI7B;EAAEH,KAAK;EAAEC,QAAQ;EAAEzB,IAAI;EAAEQ;AAAiD,CAAC,EACtD;EACrB,MAAMyB,UAAU,GAAGjC,IAAI,CAACyB,QAAQ,CAAC,CAACb,WAAW;EAC7C;EACA,MAAMU,eAAe,GAAGd,iBAAiB,CAACgB,KAAK,CAAC,GAAG,EAAE,GAAG,CAACS,UAAU,CAAC;EACpE,OAAOX,eAAe;AACxB;AAEA,SAASM,qBAAqBA,CAI5B;EAAEzB,oBAAoB;EAAEsB,QAAQ;EAAEzB,IAAI;EAAEQ;AACV,CAAC,EACV;EACrB;EACA;EACA,IAAIL,oBAAoB,IAAI,IAAI,EAAE;IAChC,MAAM+B,WAAW,GAAGC,cAAc,CAACnC,IAAI,EAAEG,oBAAoB,EAAEsB,QAAQ,CAAC;IACxE,MAAMW,kBAAkB,GAAGF,WAAW,CAACxB,GAAG,CAACC,IAAI,IAAIA,IAAI,CAACC,WAAW,CAAC;IAEpE,MAAMyB,iBAAiB,GAAGC,sBAAsB,CAAC9B,iBAAiB,EAAER,IAAI,CAAC;;IAEzE;IACA,MAAMsB,eAAe,GAAG,CAAC,GAAGe,iBAAiB,CAAC;IAC9CD,kBAAkB,CAACG,OAAO,CAAC5B,IAAI,IAAI;MACjC,IAAI,CAACW,eAAe,CAACkB,QAAQ,CAAC7B,IAAI,CAAC,EAAE;QACnCW,eAAe,CAACmB,IAAI,CAAC9B,IAAI,CAAC;MAC5B;IACF,CAAC,CAAC;IACF,OAAOW,eAAe;EACxB;EACA,OAAO,EAAE;AACX;AAEA,SAASQ,mBAAmBA,CAI1B;EAAEL,QAAQ;EAAEzB,IAAI;EAAEQ;AAGlB,CAAC,EACQ;EACT,MAAMyB,UAAU,GAAGjC,IAAI,CAACyB,QAAQ,CAAC,CAACb,WAAW;EAC7C,MAAMyB,iBAAiB,GAAGC,sBAAsB,CAAC9B,iBAAiB,EAAER,IAAI,CAAC;EACzE,OAAOqC,iBAAiB,CAACG,QAAQ,CAACP,UAAU,CAAC;AAC/C;AAEA,SAASJ,wBAAwBA,CAI/B;EAAEJ,QAAQ;EAAEzB,IAAI;EAAEQ;AAAiD,CAAC,EAC/C;EACrB,MAAMyB,UAAU,GAAGjC,IAAI,CAACyB,QAAQ,CAAC,CAACb,WAAW;EAC7C,MAAMyB,iBAAiB,GAAGC,sBAAsB,CAAC9B,iBAAiB,EAAER,IAAI,CAAC;EACzE;EACA,MAAMsB,eAAe,GACnBQ,mBAAmB,CAAU;IAAEL,QAAQ;IAAEzB,IAAI;IAAEQ;EAAkB,CAAC,CAAC,GAC/D6B,iBAAiB,CAACpB,MAAM,CAACyB,CAAC,IAAIA,CAAC,KAAKT,UAAU,CAAC,GAC/C,CAAC,GAAGI,iBAAiB,EAAEJ,UAAU,CAAC;EACxC,OAAOX,eAAe;AACxB;;AAEA;AACA;AACA;AACA,SAASa,cAAcA,CAIrBnC,IAA0E,EAC1E2C,UAAkB,EAClBC,QAAgB,EACsD;EACtE,MAAMC,KAAK,GAAGC,IAAI,CAACC,GAAG,CAACJ,UAAU,EAAEC,QAAQ,CAAC;EAC5C,MAAMI,GAAG,GAAGF,IAAI,CAACG,GAAG,CAACN,UAAU,EAAEC,QAAQ,CAAC;EAC1C,MAAMM,IAEL,GAAG,EAAE;EAEN,KAAK,IAAIR,CAAC,GAAGG,KAAK,EAAEH,CAAC,IAAIM,GAAG,EAAEN,CAAC,EAAE,EAAE;IACjC,MAAM/B,IAAI,GAAGX,IAAI,CAAC0C,CAAC,CAAC;IACpB,IAAI/B,IAAI,EAAE;MACRuC,IAAI,CAACT,IAAI,CAAC9B,IAAI,CAAC;IACjB;EACF;EAEA,OAAOuC,IAAI;AACb;;AAEA;AACA;AACA;AACA,SAASrC,oBAAoBA,CAC3BsC,WAAgC,EACb;EACnB,OAAOA,WAAW,CAACC,MAAM,CACvB,CAACC,GAAG,EAAEpB,UAAU,KAAK;IACnBoB,GAAG,CAAC5D,sBAAsB,CAACwC,UAAU,CAAC,CAAC,GAAG,IAAI;IAC9C,OAAOoB,GAAG;EACZ,CAAC,EACD,CAAC,CACH,CAAC;AACH;;AAEA;AACA;AACA;AACA,SAASf,sBAAsBA,CAI7BgB,cAAiC,EACjCtD,IAA0E,EACrD;EACrB,OAAOA,IAAI,CACRiB,MAAM,CAACN,IAAI,IAAI2C,cAAc,CAAC9D,QAAQ,CAACmB,IAAI,CAAC,CAAC,CAAC,CAC9CD,GAAG,CAACC,IAAI,IAAIA,IAAI,CAACC,WAAW,CAAC;AAClC","ignoreList":[]}