Rate.js 8.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217
  1. import _extends from "@babel/runtime/helpers/esm/extends";
  2. import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
  3. import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
  4. import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
  5. var _excluded = ["prefixCls", "className", "defaultValue", "value", "count", "allowHalf", "allowClear", "keyboard", "character", "characterRender", "disabled", "direction", "tabIndex", "autoFocus", "onHoverChange", "onChange", "onFocus", "onBlur", "onKeyDown", "onMouseLeave"];
  6. import classNames from 'classnames';
  7. import useMergedState from "rc-util/es/hooks/useMergedState";
  8. import KeyCode from "rc-util/es/KeyCode";
  9. import pickAttrs from "rc-util/es/pickAttrs";
  10. import React from 'react';
  11. import Star from "./Star";
  12. import useRefs from "./useRefs";
  13. import { getOffsetLeft } from "./util";
  14. function Rate(props, ref) {
  15. var _props$prefixCls = props.prefixCls,
  16. prefixCls = _props$prefixCls === void 0 ? 'rc-rate' : _props$prefixCls,
  17. className = props.className,
  18. defaultValue = props.defaultValue,
  19. propValue = props.value,
  20. _props$count = props.count,
  21. count = _props$count === void 0 ? 5 : _props$count,
  22. _props$allowHalf = props.allowHalf,
  23. allowHalf = _props$allowHalf === void 0 ? false : _props$allowHalf,
  24. _props$allowClear = props.allowClear,
  25. allowClear = _props$allowClear === void 0 ? true : _props$allowClear,
  26. _props$keyboard = props.keyboard,
  27. keyboard = _props$keyboard === void 0 ? true : _props$keyboard,
  28. _props$character = props.character,
  29. character = _props$character === void 0 ? '★' : _props$character,
  30. characterRender = props.characterRender,
  31. disabled = props.disabled,
  32. _props$direction = props.direction,
  33. direction = _props$direction === void 0 ? 'ltr' : _props$direction,
  34. _props$tabIndex = props.tabIndex,
  35. tabIndex = _props$tabIndex === void 0 ? 0 : _props$tabIndex,
  36. autoFocus = props.autoFocus,
  37. onHoverChange = props.onHoverChange,
  38. onChange = props.onChange,
  39. onFocus = props.onFocus,
  40. onBlur = props.onBlur,
  41. onKeyDown = props.onKeyDown,
  42. onMouseLeave = props.onMouseLeave,
  43. restProps = _objectWithoutProperties(props, _excluded);
  44. var _useRefs = useRefs(),
  45. _useRefs2 = _slicedToArray(_useRefs, 2),
  46. getStarRef = _useRefs2[0],
  47. setStarRef = _useRefs2[1];
  48. var rateRef = React.useRef(null);
  49. // ============================ Ref =============================
  50. var triggerFocus = function triggerFocus() {
  51. if (!disabled) {
  52. var _rateRef$current;
  53. (_rateRef$current = rateRef.current) === null || _rateRef$current === void 0 || _rateRef$current.focus();
  54. }
  55. };
  56. React.useImperativeHandle(ref, function () {
  57. return {
  58. focus: triggerFocus,
  59. blur: function blur() {
  60. if (!disabled) {
  61. var _rateRef$current2;
  62. (_rateRef$current2 = rateRef.current) === null || _rateRef$current2 === void 0 || _rateRef$current2.blur();
  63. }
  64. }
  65. };
  66. });
  67. // =========================== Value ============================
  68. var _useMergedState = useMergedState(defaultValue || 0, {
  69. value: propValue
  70. }),
  71. _useMergedState2 = _slicedToArray(_useMergedState, 2),
  72. value = _useMergedState2[0],
  73. setValue = _useMergedState2[1];
  74. var _useMergedState3 = useMergedState(null),
  75. _useMergedState4 = _slicedToArray(_useMergedState3, 2),
  76. cleanedValue = _useMergedState4[0],
  77. setCleanedValue = _useMergedState4[1];
  78. var getStarValue = function getStarValue(index, x) {
  79. var reverse = direction === 'rtl';
  80. var starValue = index + 1;
  81. if (allowHalf) {
  82. var starEle = getStarRef(index);
  83. var leftDis = getOffsetLeft(starEle);
  84. var width = starEle.clientWidth;
  85. if (reverse && x - leftDis > width / 2) {
  86. starValue -= 0.5;
  87. } else if (!reverse && x - leftDis < width / 2) {
  88. starValue -= 0.5;
  89. }
  90. }
  91. return starValue;
  92. };
  93. // >>>>> Change
  94. var changeValue = function changeValue(nextValue) {
  95. setValue(nextValue);
  96. onChange === null || onChange === void 0 || onChange(nextValue);
  97. };
  98. // =========================== Focus ============================
  99. var _React$useState = React.useState(false),
  100. _React$useState2 = _slicedToArray(_React$useState, 2),
  101. focused = _React$useState2[0],
  102. setFocused = _React$useState2[1];
  103. var onInternalFocus = function onInternalFocus() {
  104. setFocused(true);
  105. onFocus === null || onFocus === void 0 || onFocus();
  106. };
  107. var onInternalBlur = function onInternalBlur() {
  108. setFocused(false);
  109. onBlur === null || onBlur === void 0 || onBlur();
  110. };
  111. // =========================== Hover ============================
  112. var _React$useState3 = React.useState(null),
  113. _React$useState4 = _slicedToArray(_React$useState3, 2),
  114. hoverValue = _React$useState4[0],
  115. setHoverValue = _React$useState4[1];
  116. var onHover = function onHover(event, index) {
  117. var nextHoverValue = getStarValue(index, event.pageX);
  118. if (nextHoverValue !== cleanedValue) {
  119. setHoverValue(nextHoverValue);
  120. setCleanedValue(null);
  121. }
  122. onHoverChange === null || onHoverChange === void 0 || onHoverChange(nextHoverValue);
  123. };
  124. var onMouseLeaveCallback = function onMouseLeaveCallback(event) {
  125. if (!disabled) {
  126. setHoverValue(null);
  127. setCleanedValue(null);
  128. onHoverChange === null || onHoverChange === void 0 || onHoverChange(undefined);
  129. }
  130. if (event) {
  131. onMouseLeave === null || onMouseLeave === void 0 || onMouseLeave(event);
  132. }
  133. };
  134. // =========================== Click ============================
  135. var onClick = function onClick(event, index) {
  136. var newValue = getStarValue(index, event.pageX);
  137. var isReset = false;
  138. if (allowClear) {
  139. isReset = newValue === value;
  140. }
  141. onMouseLeaveCallback();
  142. changeValue(isReset ? 0 : newValue);
  143. setCleanedValue(isReset ? newValue : null);
  144. };
  145. var onInternalKeyDown = function onInternalKeyDown(event) {
  146. var keyCode = event.keyCode;
  147. var reverse = direction === 'rtl';
  148. var step = allowHalf ? 0.5 : 1;
  149. if (keyboard) {
  150. if (keyCode === KeyCode.RIGHT && value < count && !reverse) {
  151. changeValue(value + step);
  152. event.preventDefault();
  153. } else if (keyCode === KeyCode.LEFT && value > 0 && !reverse) {
  154. changeValue(value - step);
  155. event.preventDefault();
  156. } else if (keyCode === KeyCode.RIGHT && value > 0 && reverse) {
  157. changeValue(value - step);
  158. event.preventDefault();
  159. } else if (keyCode === KeyCode.LEFT && value < count && reverse) {
  160. changeValue(value + step);
  161. event.preventDefault();
  162. }
  163. }
  164. onKeyDown === null || onKeyDown === void 0 || onKeyDown(event);
  165. };
  166. // =========================== Effect ===========================
  167. React.useEffect(function () {
  168. if (autoFocus && !disabled) {
  169. triggerFocus();
  170. }
  171. }, []);
  172. // =========================== Render ===========================
  173. // >>> Star
  174. var starNodes = new Array(count).fill(0).map(function (item, index) {
  175. return /*#__PURE__*/React.createElement(Star, {
  176. ref: setStarRef(index),
  177. index: index,
  178. count: count,
  179. disabled: disabled,
  180. prefixCls: "".concat(prefixCls, "-star"),
  181. allowHalf: allowHalf,
  182. value: hoverValue === null ? value : hoverValue,
  183. onClick: onClick,
  184. onHover: onHover,
  185. key: item || index,
  186. character: character,
  187. characterRender: characterRender,
  188. focused: focused
  189. });
  190. });
  191. var classString = classNames(prefixCls, className, _defineProperty(_defineProperty({}, "".concat(prefixCls, "-disabled"), disabled), "".concat(prefixCls, "-rtl"), direction === 'rtl'));
  192. // >>> Node
  193. return /*#__PURE__*/React.createElement("ul", _extends({
  194. className: classString,
  195. onMouseLeave: onMouseLeaveCallback,
  196. tabIndex: disabled ? -1 : tabIndex,
  197. onFocus: disabled ? null : onInternalFocus,
  198. onBlur: disabled ? null : onInternalBlur,
  199. onKeyDown: disabled ? null : onInternalKeyDown,
  200. ref: rateRef
  201. }, pickAttrs(restProps, {
  202. aria: true,
  203. data: true,
  204. attr: true
  205. })), starNodes);
  206. }
  207. export default /*#__PURE__*/React.forwardRef(Rate);