antd-mobile
Version:
<div align="center">
68 lines • 2.4 kB
JavaScript
import dayjs from 'dayjs';
import quarterOfYear from 'dayjs/plugin/quarterOfYear';
dayjs.extend(quarterOfYear);
const precisionRankRecord = {
year: 0,
quarter: 1
};
export function generateDatePickerColumns(selected, min, max, precision, renderLabel, filter) {
const ret = [];
const minYear = min.getFullYear();
const maxYear = max.getFullYear();
const rank = precisionRankRecord[precision];
const selectedYear = parseInt(selected[0]);
const isInMinYear = selectedYear === minYear;
const isInMaxYear = selectedYear === maxYear;
const minDay = dayjs(min);
const maxDay = dayjs(max);
const minQuarter = minDay.quarter();
const maxQuarter = maxDay.quarter();
const generateColumn = (from, to, precision) => {
let column = [];
for (let i = from; i <= to; i++) {
column.push(i);
}
const prefix = selected.slice(0, precisionRankRecord[precision]);
const currentFilter = filter === null || filter === void 0 ? void 0 : filter[precision];
if (currentFilter && typeof currentFilter === 'function') {
column = column.filter(i => currentFilter(i, {
get date() {
const stringArray = [...prefix, i.toString()];
return convertStringArrayToDate(stringArray);
}
}));
}
return column;
};
if (rank >= precisionRankRecord.year) {
const lower = minYear;
const upper = maxYear;
const years = generateColumn(lower, upper, 'year');
ret.push(years.map(v => ({
label: renderLabel('year', v),
value: v.toString()
})));
}
if (rank >= precisionRankRecord.quarter) {
const lower = isInMinYear ? minQuarter : 1;
const upper = isInMaxYear ? maxQuarter : 4;
const quarters = generateColumn(lower, upper, 'quarter');
ret.push(quarters.map(v => ({
label: renderLabel('quarter', v),
value: v.toString()
})));
}
return ret;
}
export function convertDateToStringArray(date) {
if (!date) return [];
const day = dayjs(date);
return [day.year().toString(), day.quarter().toString()];
}
export function convertStringArrayToDate(value) {
var _a, _b;
const yearString = (_a = value[0]) !== null && _a !== void 0 ? _a : '1900';
const quarterString = (_b = value[1]) !== null && _b !== void 0 ? _b : '1';
const day = dayjs().year(parseInt(yearString)).quarter(parseInt(quarterString)).hour(0).minute(0).second(0);
return day.toDate();
}