@react-aria/selection 3.22.0 → 3.23.0

package/src/utils.ts

@@ -10,7 +10,9 @@
  * governing permissions and limitations under the License.
  */
 
-import {isAppleDevice} from '@react-aria/utils';
+import {Collection, Key} from '@react-types/shared';
+import {isAppleDevice, useId} from '@react-aria/utils';
+import {RefObject} from 'react';
 
 interface Event {
   altKey: boolean,
@@ -23,3 +25,23 @@ export function isNonContiguousSelectionModifier(e: Event) {
   // On Windows and Ubuntu, Alt + Space has a system wide meaning.
   return isAppleDevice() ? e.altKey : e.ctrlKey;
 }
+
+export function getItemElement(collectionRef: RefObject<HTMLElement | null>, key: Key) {
+  let selector = `[data-key="${CSS.escape(String(key))}"]`;
+  let collection = collectionRef.current?.dataset.collection;
+  if (collection) {
+    selector = `[data-collection="${CSS.escape(collection)}"]${selector}`;
+  }
+  return collectionRef.current?.querySelector(selector);
+}
+
+const collectionMap = new WeakMap();
+export function useCollectionId(collection: Collection<any>) {
+  let id = useId();
+  collectionMap.set(collection, id);
+  return id;
+}
+
+export function getCollectionId(collection: Collection<any>) {
+  return collectionMap.get(collection)!;
+}