63 lines
2.0 KiB
JavaScript
63 lines
2.0 KiB
JavaScript
|
import { useRef } from 'react';
|
||
|
|
import warning from "rc-util/es/warning";
|
||
|
|
/**
|
||
|
|
* Keep input cursor in the correct position if possible.
|
||
|
|
* Is this necessary since we have `formatter` which may mass the content?
|
||
|
|
*/
|
||
|
|
export default function useCursor(input, focused) {
|
||
|
|
var selectionRef = useRef(null);
|
||
|
|
function recordCursor() {
|
||
|
|
// Record position
|
||
|
|
try {
|
||
|
|
var start = input.selectionStart,
|
||
|
|
end = input.selectionEnd,
|
||
|
|
value = input.value;
|
||
|
|
var beforeTxt = value.substring(0, start);
|
||
|
|
var afterTxt = value.substring(end);
|
||
|
|
selectionRef.current = {
|
||
|
|
start: start,
|
||
|
|
end: end,
|
||
|
|
value: value,
|
||
|
|
beforeTxt: beforeTxt,
|
||
|
|
afterTxt: afterTxt
|
||
|
|
};
|
||
|
|
} catch (e) {
|
||
|
|
// Fix error in Chrome:
|
||
|
|
// Failed to read the 'selectionStart' property from 'HTMLInputElement'
|
||
|
|
// http://stackoverflow.com/q/21177489/3040605
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Restore logic:
|
||
|
|
* 1. back string same
|
||
|
|
* 2. start string same
|
||
|
|
*/
|
||
|
|
function restoreCursor() {
|
||
|
|
if (input && selectionRef.current && focused) {
|
||
|
|
try {
|
||
|
|
var value = input.value;
|
||
|
|
var _selectionRef$current = selectionRef.current,
|
||
|
|
beforeTxt = _selectionRef$current.beforeTxt,
|
||
|
|
afterTxt = _selectionRef$current.afterTxt,
|
||
|
|
start = _selectionRef$current.start;
|
||
|
|
var startPos = value.length;
|
||
|
|
if (value.endsWith(afterTxt)) {
|
||
|
|
startPos = value.length - selectionRef.current.afterTxt.length;
|
||
|
|
} else if (value.startsWith(beforeTxt)) {
|
||
|
|
startPos = beforeTxt.length;
|
||
|
|
} else {
|
||
|
|
var beforeLastChar = beforeTxt[start - 1];
|
||
|
|
var newIndex = value.indexOf(beforeLastChar, start - 1);
|
||
|
|
if (newIndex !== -1) {
|
||
|
|
startPos = newIndex + 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
input.setSelectionRange(startPos, startPos);
|
||
|
|
} catch (e) {
|
||
|
|
warning(false, "Something warning of cursor restore. Please fire issue about this: ".concat(e.message));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return [recordCursor, restoreCursor];
|
||
|
|
}
|