31 lines
1.3 KiB
JavaScript
31 lines
1.3 KiB
JavaScript
import { useCallback, useState } from 'react';
|
|
/**
|
|
* @title multipleSelect hooks
|
|
* @description multipleSelect by hold down shift key
|
|
*/
|
|
export const useMultipleSelect = getKey => {
|
|
const [prevSelectedIndex, setPrevSelectedIndex] = useState(null);
|
|
const multipleSelect = useCallback((currentSelectedIndex, data, selectedKeys) => {
|
|
const configPrevSelectedIndex = prevSelectedIndex !== null && prevSelectedIndex !== void 0 ? prevSelectedIndex : currentSelectedIndex;
|
|
// add/delete the selected range
|
|
const startIndex = Math.min(configPrevSelectedIndex || 0, currentSelectedIndex);
|
|
const endIndex = Math.max(configPrevSelectedIndex || 0, currentSelectedIndex);
|
|
const rangeKeys = data.slice(startIndex, endIndex + 1).map(getKey);
|
|
const shouldSelected = rangeKeys.some(rangeKey => !selectedKeys.has(rangeKey));
|
|
const changedKeys = [];
|
|
rangeKeys.forEach(item => {
|
|
if (shouldSelected) {
|
|
if (!selectedKeys.has(item)) {
|
|
changedKeys.push(item);
|
|
}
|
|
selectedKeys.add(item);
|
|
} else {
|
|
selectedKeys.delete(item);
|
|
changedKeys.push(item);
|
|
}
|
|
});
|
|
setPrevSelectedIndex(shouldSelected ? endIndex : null);
|
|
return changedKeys;
|
|
}, [prevSelectedIndex]);
|
|
return [multipleSelect, setPrevSelectedIndex];
|
|
}; |