245 lines
7.6 KiB
JavaScript
245 lines
7.6 KiB
JavaScript
"use client";
|
|
|
|
import _toConsumableArray from "@babel/runtime/helpers/esm/toConsumableArray";
|
|
import * as React from 'react';
|
|
import { useEvent } from '@rc-component/util';
|
|
import pickAttrs from "@rc-component/util/es/pickAttrs";
|
|
import { clsx } from 'clsx';
|
|
import { useMergeSemantic } from '../../_util/hooks';
|
|
import { getMergedStatus } from '../../_util/statusUtils';
|
|
import { devUseWarning } from '../../_util/warning';
|
|
import { useComponentConfig } from '../../config-provider/context';
|
|
import useSize from '../../config-provider/hooks/useSize';
|
|
import { FormItemInputContext } from '../../form/context';
|
|
import useStyle from '../style/otp';
|
|
import OTPInput from './OTPInput';
|
|
function strToArr(str) {
|
|
return (str || '').split('');
|
|
}
|
|
const Separator = props => {
|
|
const {
|
|
index,
|
|
prefixCls,
|
|
separator,
|
|
className: semanticClassName,
|
|
style: semanticStyle
|
|
} = props;
|
|
const separatorNode = typeof separator === 'function' ? separator(index) : separator;
|
|
if (!separatorNode) {
|
|
return null;
|
|
}
|
|
return /*#__PURE__*/React.createElement("span", {
|
|
className: clsx(`${prefixCls}-separator`, semanticClassName),
|
|
style: semanticStyle
|
|
}, separatorNode);
|
|
};
|
|
const OTP = /*#__PURE__*/React.forwardRef((props, ref) => {
|
|
const {
|
|
prefixCls: customizePrefixCls,
|
|
length = 6,
|
|
size: customSize,
|
|
defaultValue,
|
|
value,
|
|
onChange,
|
|
formatter,
|
|
separator,
|
|
variant,
|
|
disabled,
|
|
status: customStatus,
|
|
autoFocus,
|
|
mask,
|
|
type,
|
|
autoComplete,
|
|
onInput,
|
|
onFocus,
|
|
inputMode,
|
|
classNames,
|
|
styles,
|
|
className,
|
|
style,
|
|
...restProps
|
|
} = props;
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
const warning = devUseWarning('Input.OTP');
|
|
process.env.NODE_ENV !== "production" ? warning(!(typeof mask === 'string' && mask.length > 1), 'usage', '`mask` prop should be a single character.') : void 0;
|
|
}
|
|
const {
|
|
classNames: contextClassNames,
|
|
styles: contextStyles,
|
|
getPrefixCls,
|
|
direction,
|
|
style: contextStyle,
|
|
className: contextClassName
|
|
} = useComponentConfig('otp');
|
|
const prefixCls = getPrefixCls('otp', customizePrefixCls);
|
|
const mergedProps = {
|
|
...props,
|
|
length
|
|
};
|
|
const [mergedClassNames, mergedStyles] = useMergeSemantic([contextClassNames, classNames], [contextStyles, styles], {
|
|
props: mergedProps
|
|
});
|
|
const domAttrs = pickAttrs(restProps, {
|
|
aria: true,
|
|
data: true,
|
|
attr: true
|
|
});
|
|
// ========================= Root =========================
|
|
// Style
|
|
const [hashId, cssVarCls] = useStyle(prefixCls);
|
|
// ========================= Size =========================
|
|
const mergedSize = useSize(ctx => customSize ?? ctx);
|
|
// ======================== Status ========================
|
|
const formContext = React.useContext(FormItemInputContext);
|
|
const mergedStatus = getMergedStatus(formContext.status, customStatus);
|
|
const proxyFormContext = React.useMemo(() => ({
|
|
...formContext,
|
|
status: mergedStatus,
|
|
hasFeedback: false,
|
|
feedbackIcon: null
|
|
}), [formContext, mergedStatus]);
|
|
// ========================= Refs =========================
|
|
const containerRef = React.useRef(null);
|
|
const inputsRef = React.useRef({});
|
|
React.useImperativeHandle(ref, () => ({
|
|
focus: () => {
|
|
inputsRef.current[0]?.focus();
|
|
},
|
|
blur: () => {
|
|
for (let i = 0; i < length; i += 1) {
|
|
inputsRef.current[i]?.blur();
|
|
}
|
|
},
|
|
nativeElement: containerRef.current
|
|
}));
|
|
// ======================= Formatter ======================
|
|
const internalFormatter = txt => formatter ? formatter(txt) : txt;
|
|
// ======================== Values ========================
|
|
const [valueCells, setValueCells] = React.useState(() => strToArr(internalFormatter(defaultValue || '')));
|
|
React.useEffect(() => {
|
|
if (value !== undefined) {
|
|
setValueCells(strToArr(value));
|
|
}
|
|
}, [value]);
|
|
const triggerValueCellsChange = useEvent(nextValueCells => {
|
|
setValueCells(nextValueCells);
|
|
if (onInput) {
|
|
onInput(nextValueCells);
|
|
}
|
|
// Trigger if all cells are filled
|
|
if (onChange && nextValueCells.length === length && nextValueCells.every(c => c) && nextValueCells.some((c, index) => valueCells[index] !== c)) {
|
|
onChange(nextValueCells.join(''));
|
|
}
|
|
});
|
|
const patchValue = useEvent((index, txt) => {
|
|
let nextCells = _toConsumableArray(valueCells);
|
|
// Fill cells till index
|
|
for (let i = 0; i < index; i += 1) {
|
|
if (!nextCells[i]) {
|
|
nextCells[i] = '';
|
|
}
|
|
}
|
|
if (txt.length <= 1) {
|
|
nextCells[index] = txt;
|
|
} else {
|
|
nextCells = nextCells.slice(0, index).concat(strToArr(txt));
|
|
}
|
|
nextCells = nextCells.slice(0, length);
|
|
// Clean the last empty cell
|
|
for (let i = nextCells.length - 1; i >= 0; i -= 1) {
|
|
if (nextCells[i]) {
|
|
break;
|
|
}
|
|
nextCells.pop();
|
|
}
|
|
// Format if needed
|
|
const formattedValue = internalFormatter(nextCells.map(c => c || ' ').join(''));
|
|
nextCells = strToArr(formattedValue).map((c, i) => {
|
|
if (c === ' ' && !nextCells[i]) {
|
|
return nextCells[i];
|
|
}
|
|
return c;
|
|
});
|
|
return nextCells;
|
|
});
|
|
// ======================== Change ========================
|
|
const onInputChange = (index, txt) => {
|
|
const nextCells = patchValue(index, txt);
|
|
const nextIndex = Math.min(index + txt.length, length - 1);
|
|
if (nextIndex !== index && nextCells[index] !== undefined) {
|
|
inputsRef.current[nextIndex]?.focus();
|
|
}
|
|
triggerValueCellsChange(nextCells);
|
|
};
|
|
const onInputActiveChange = nextIndex => {
|
|
inputsRef.current[nextIndex]?.focus();
|
|
};
|
|
// ======================== Focus ========================
|
|
const onInputFocus = (event, index) => {
|
|
// keep focus on the first empty cell
|
|
for (let i = 0; i < index; i += 1) {
|
|
if (!inputsRef.current[i]?.input?.value) {
|
|
inputsRef.current[i]?.focus();
|
|
break;
|
|
}
|
|
}
|
|
onFocus?.(event);
|
|
};
|
|
// ======================== Render ========================
|
|
const inputSharedProps = {
|
|
variant,
|
|
disabled,
|
|
status: mergedStatus,
|
|
mask,
|
|
type,
|
|
inputMode,
|
|
autoComplete
|
|
};
|
|
return /*#__PURE__*/React.createElement("div", {
|
|
...domAttrs,
|
|
ref: containerRef,
|
|
className: clsx(className, prefixCls, {
|
|
[`${prefixCls}-sm`]: mergedSize === 'small',
|
|
[`${prefixCls}-lg`]: mergedSize === 'large',
|
|
[`${prefixCls}-rtl`]: direction === 'rtl'
|
|
}, cssVarCls, hashId, contextClassName, mergedClassNames.root),
|
|
style: {
|
|
...mergedStyles.root,
|
|
...contextStyle,
|
|
...style
|
|
},
|
|
role: "group"
|
|
}, /*#__PURE__*/React.createElement(FormItemInputContext.Provider, {
|
|
value: proxyFormContext
|
|
}, Array.from({
|
|
length
|
|
}).map((_, index) => {
|
|
const key = `otp-${index}`;
|
|
const singleValue = valueCells[index] || '';
|
|
return /*#__PURE__*/React.createElement(React.Fragment, {
|
|
key: key
|
|
}, /*#__PURE__*/React.createElement(OTPInput, {
|
|
ref: inputEle => {
|
|
inputsRef.current[index] = inputEle;
|
|
},
|
|
index: index,
|
|
size: mergedSize,
|
|
htmlSize: 1,
|
|
className: clsx(mergedClassNames.input, `${prefixCls}-input`),
|
|
style: mergedStyles.input,
|
|
onChange: onInputChange,
|
|
value: singleValue,
|
|
onActiveChange: onInputActiveChange,
|
|
autoFocus: index === 0 && autoFocus,
|
|
onFocus: event => onInputFocus(event, index),
|
|
...inputSharedProps
|
|
}), index < length - 1 && (/*#__PURE__*/React.createElement(Separator, {
|
|
separator: separator,
|
|
index: index,
|
|
prefixCls: prefixCls,
|
|
className: clsx(mergedClassNames.separator),
|
|
style: mergedStyles.separator
|
|
})));
|
|
})));
|
|
});
|
|
export default OTP; |