dateUtil.js 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202
  1. import { DECADE_UNIT_DIFF } from '../panels/DecadePanel/index';
  2. export const WEEK_DAY_COUNT = 7;
  3. export function isNullEqual(value1, value2) {
  4. if (!value1 && !value2) {
  5. return true;
  6. }
  7. if (!value1 || !value2) {
  8. return false;
  9. }
  10. return undefined;
  11. }
  12. export function isSameDecade(generateConfig, decade1, decade2) {
  13. const equal = isNullEqual(decade1, decade2);
  14. if (typeof equal === 'boolean') {
  15. return equal;
  16. }
  17. const num1 = Math.floor(generateConfig.getYear(decade1) / 10);
  18. const num2 = Math.floor(generateConfig.getYear(decade2) / 10);
  19. return num1 === num2;
  20. }
  21. export function isSameYear(generateConfig, year1, year2) {
  22. const equal = isNullEqual(year1, year2);
  23. if (typeof equal === 'boolean') {
  24. return equal;
  25. }
  26. return generateConfig.getYear(year1) === generateConfig.getYear(year2);
  27. }
  28. export function getQuarter(generateConfig, date) {
  29. const quota = Math.floor(generateConfig.getMonth(date) / 3);
  30. return quota + 1;
  31. }
  32. export function isSameQuarter(generateConfig, quarter1, quarter2) {
  33. const equal = isNullEqual(quarter1, quarter2);
  34. if (typeof equal === 'boolean') {
  35. return equal;
  36. }
  37. return isSameYear(generateConfig, quarter1, quarter2) && getQuarter(generateConfig, quarter1) === getQuarter(generateConfig, quarter2);
  38. }
  39. export function isSameMonth(generateConfig, month1, month2) {
  40. const equal = isNullEqual(month1, month2);
  41. if (typeof equal === 'boolean') {
  42. return equal;
  43. }
  44. return isSameYear(generateConfig, month1, month2) && generateConfig.getMonth(month1) === generateConfig.getMonth(month2);
  45. }
  46. export function isSameDate(generateConfig, date1, date2) {
  47. const equal = isNullEqual(date1, date2);
  48. if (typeof equal === 'boolean') {
  49. return equal;
  50. }
  51. return generateConfig.getYear(date1) === generateConfig.getYear(date2) && generateConfig.getMonth(date1) === generateConfig.getMonth(date2) && generateConfig.getDate(date1) === generateConfig.getDate(date2);
  52. }
  53. export function isSameTime(generateConfig, time1, time2) {
  54. const equal = isNullEqual(time1, time2);
  55. if (typeof equal === 'boolean') {
  56. return equal;
  57. }
  58. return generateConfig.getHour(time1) === generateConfig.getHour(time2) && generateConfig.getMinute(time1) === generateConfig.getMinute(time2) && generateConfig.getSecond(time1) === generateConfig.getSecond(time2);
  59. }
  60. export function isSameWeek(generateConfig, locale, date1, date2) {
  61. const equal = isNullEqual(date1, date2);
  62. if (typeof equal === 'boolean') {
  63. return equal;
  64. }
  65. return generateConfig.locale.getWeek(locale, date1) === generateConfig.locale.getWeek(locale, date2);
  66. }
  67. export function isEqual(generateConfig, value1, value2) {
  68. return isSameDate(generateConfig, value1, value2) && isSameTime(generateConfig, value1, value2);
  69. }
  70. /** Between in date but not equal of date */
  71. export function isInRange(generateConfig, startDate, endDate, current) {
  72. if (!startDate || !endDate || !current) {
  73. return false;
  74. }
  75. return !isSameDate(generateConfig, startDate, current) && !isSameDate(generateConfig, endDate, current) && generateConfig.isAfter(current, startDate) && generateConfig.isAfter(endDate, current);
  76. }
  77. export function getWeekStartDate(locale, generateConfig, value) {
  78. const weekFirstDay = generateConfig.locale.getWeekFirstDay(locale);
  79. const monthStartDate = generateConfig.setDate(value, 1);
  80. const startDateWeekDay = generateConfig.getWeekDay(monthStartDate);
  81. let alignStartDate = generateConfig.addDate(monthStartDate, weekFirstDay - startDateWeekDay);
  82. if (generateConfig.getMonth(alignStartDate) === generateConfig.getMonth(value) && generateConfig.getDate(alignStartDate) > 1) {
  83. alignStartDate = generateConfig.addDate(alignStartDate, -7);
  84. }
  85. return alignStartDate;
  86. }
  87. export function getClosingViewDate(viewDate, picker, generateConfig) {
  88. let offset = arguments.length > 3 && arguments[3] !== undefined ? arguments[3] : 1;
  89. switch (picker) {
  90. case 'year':
  91. return generateConfig.addYear(viewDate, offset * 10);
  92. case 'quarter':
  93. case 'month':
  94. return generateConfig.addYear(viewDate, offset);
  95. default:
  96. return generateConfig.addMonth(viewDate, offset);
  97. }
  98. }
  99. export function formatValue(value, _ref) {
  100. let {
  101. generateConfig,
  102. locale,
  103. format
  104. } = _ref;
  105. return typeof format === 'function' ? format(value) : generateConfig.locale.format(locale.locale, value, format);
  106. }
  107. export function parseValue(value, _ref2) {
  108. let {
  109. generateConfig,
  110. locale,
  111. formatList
  112. } = _ref2;
  113. if (!value || typeof formatList[0] === 'function') {
  114. return null;
  115. }
  116. return generateConfig.locale.parse(locale.locale, value, formatList);
  117. }
  118. // eslint-disable-next-line consistent-return
  119. export function getCellDateDisabled(_ref3) {
  120. let {
  121. cellDate,
  122. mode,
  123. disabledDate,
  124. generateConfig
  125. } = _ref3;
  126. if (!disabledDate) return false;
  127. // Whether cellDate is disabled in range
  128. const getDisabledFromRange = (currentMode, start, end) => {
  129. let current = start;
  130. while (current <= end) {
  131. let date;
  132. switch (currentMode) {
  133. case 'date':
  134. {
  135. date = generateConfig.setDate(cellDate, current);
  136. if (!disabledDate(date)) {
  137. return false;
  138. }
  139. break;
  140. }
  141. case 'month':
  142. {
  143. date = generateConfig.setMonth(cellDate, current);
  144. if (!getCellDateDisabled({
  145. cellDate: date,
  146. mode: 'month',
  147. generateConfig,
  148. disabledDate
  149. })) {
  150. return false;
  151. }
  152. break;
  153. }
  154. case 'year':
  155. {
  156. date = generateConfig.setYear(cellDate, current);
  157. if (!getCellDateDisabled({
  158. cellDate: date,
  159. mode: 'year',
  160. generateConfig,
  161. disabledDate
  162. })) {
  163. return false;
  164. }
  165. break;
  166. }
  167. }
  168. current += 1;
  169. }
  170. return true;
  171. };
  172. switch (mode) {
  173. case 'date':
  174. case 'week':
  175. {
  176. return disabledDate(cellDate);
  177. }
  178. case 'month':
  179. {
  180. const startDate = 1;
  181. const endDate = generateConfig.getDate(generateConfig.getEndDate(cellDate));
  182. return getDisabledFromRange('date', startDate, endDate);
  183. }
  184. case 'quarter':
  185. {
  186. const startMonth = Math.floor(generateConfig.getMonth(cellDate) / 3) * 3;
  187. const endMonth = startMonth + 2;
  188. return getDisabledFromRange('month', startMonth, endMonth);
  189. }
  190. case 'year':
  191. {
  192. return getDisabledFromRange('month', 0, 11);
  193. }
  194. case 'decade':
  195. {
  196. const year = generateConfig.getYear(cellDate);
  197. const startYear = Math.floor(year / DECADE_UNIT_DIFF) * DECADE_UNIT_DIFF;
  198. const endYear = startYear + DECADE_UNIT_DIFF - 1;
  199. return getDisabledFromRange('year', startYear, endYear);
  200. }
  201. }
  202. }