{"version":3,"file":"use-calendar.mjs","names":[],"sources":["../../../../../../packages/components/calendar/src/use-calendar.ts"],"sourcesContent":["import { computed, ref } from 'vue'\nimport dayjs from 'dayjs'\nimport { useLocale } from '@element-plus/hooks'\nimport { debugWarn, isArray, isDate } from '@element-plus/utils'\nimport { INPUT_EVENT, UPDATE_MODEL_EVENT } from '@element-plus/constants'\n\nimport type { ComputedRef, SetupContext } from 'vue'\nimport type { Dayjs } from 'dayjs'\nimport type { CalendarDateType, CalendarEmits, CalendarProps } from './calendar'\n\nconst adjacentMonth = (start: Dayjs, end: Dayjs): [Dayjs, Dayjs][] => {\n const firstMonthLastDay = start.endOf('month')\n const lastMonthFirstDay = end.startOf('month')\n\n // Whether the last day of the first month and the first day of the last month is in the same week\n const isSameWeek = firstMonthLastDay.isSame(lastMonthFirstDay, 'week')\n const lastMonthStartDay = isSameWeek\n ? lastMonthFirstDay.add(1, 'week')\n : lastMonthFirstDay\n\n return [\n [start, firstMonthLastDay],\n [lastMonthStartDay.startOf('week'), end],\n ]\n}\n\nconst threeConsecutiveMonth = (start: Dayjs, end: Dayjs): [Dayjs, Dayjs][] => {\n const firstMonthLastDay = start.endOf('month')\n const secondMonthFirstDay = start.add(1, 'month').startOf('month')\n\n // Whether the last day of the first month and the second month is in the same week\n const secondMonthStartDay = firstMonthLastDay.isSame(\n secondMonthFirstDay,\n 'week'\n )\n ? secondMonthFirstDay.add(1, 'week')\n : secondMonthFirstDay\n\n const secondMonthLastDay = secondMonthStartDay.endOf('month')\n const lastMonthFirstDay = end.startOf('month')\n\n // Whether the last day of the second month and the last day of the last month is in the same week\n const lastMonthStartDay = secondMonthLastDay.isSame(lastMonthFirstDay, 'week')\n ? lastMonthFirstDay.add(1, 'week')\n : lastMonthFirstDay\n\n return [\n [start, firstMonthLastDay],\n [secondMonthStartDay.startOf('week'), secondMonthLastDay],\n [lastMonthStartDay.startOf('week'), end],\n ]\n}\n\nexport const useCalendar = (\n props: CalendarProps,\n emit: SetupContext['emit'],\n componentName: string\n) => {\n const { lang } = useLocale()\n\n const selectedDay = ref()\n const now = dayjs().locale(lang.value)\n\n const realSelectedDay = computed({\n get() {\n if (!props.modelValue) return selectedDay.value\n return date.value\n },\n set(val) {\n if (!val) return\n selectedDay.value = val\n const result = val.toDate()\n\n emit(INPUT_EVENT, result)\n emit(UPDATE_MODEL_EVENT, result)\n },\n })\n\n // if range is valid, we get a two-digit array\n const validatedRange = computed(() => {\n if (\n !props.range ||\n !isArray(props.range) ||\n props.range.length !== 2 ||\n props.range.some((item) => !isDate(item))\n )\n return []\n const rangeArrDayjs = props.range.map((_) => dayjs(_).locale(lang.value))\n const [startDayjs, endDayjs] = rangeArrDayjs\n if (startDayjs.isAfter(endDayjs)) {\n debugWarn(componentName, 'end time should be greater than start time')\n return []\n }\n if (startDayjs.isSame(endDayjs, 'month')) {\n // same month\n return calculateValidatedDateRange(startDayjs, endDayjs)\n } else {\n // two months\n if (!startDayjs.add(1, 'month').isSame(endDayjs, 'month')) {\n debugWarn(\n componentName,\n 'start time and end time interval must not exceed two months'\n )\n return []\n }\n return calculateValidatedDateRange(startDayjs, endDayjs)\n }\n })\n\n const date: ComputedRef = computed(() => {\n if (!props.modelValue) {\n return (\n realSelectedDay.value ||\n (validatedRange.value.length ? validatedRange.value[0][0] : now)\n )\n } else {\n return dayjs(props.modelValue).locale(lang.value)\n }\n })\n const prevMonthDayjs = computed(() => date.value.subtract(1, 'month').date(1))\n const nextMonthDayjs = computed(() => date.value.add(1, 'month').date(1))\n const prevYearDayjs = computed(() => date.value.subtract(1, 'year').date(1))\n const nextYearDayjs = computed(() => date.value.add(1, 'year').date(1))\n\n // https://github.com/element-plus/element-plus/issues/3155\n // Calculate the validate date range according to the start and end dates\n const calculateValidatedDateRange = (\n startDayjs: Dayjs,\n endDayjs: Dayjs\n ): [Dayjs, Dayjs][] => {\n const firstDay = startDayjs.startOf('week')\n const lastDay = endDayjs.endOf('week')\n const firstMonth = firstDay.get('month')\n const lastMonth = lastDay.get('month')\n\n // Current month\n if (firstMonth === lastMonth) {\n return [[firstDay, lastDay]]\n }\n // Two adjacent months\n else if ((firstMonth + 1) % 12 === lastMonth) {\n return adjacentMonth(firstDay, lastDay)\n }\n // Three consecutive months (compatible: 2021-01-30 to 2021-02-28)\n else if (\n firstMonth + 2 === lastMonth ||\n (firstMonth + 1) % 11 === lastMonth\n ) {\n return threeConsecutiveMonth(firstDay, lastDay)\n }\n // Other cases\n else {\n debugWarn(\n componentName,\n 'start time and end time interval must not exceed two months'\n )\n return []\n }\n }\n\n const pickDay = (day: Dayjs) => {\n realSelectedDay.value = day\n }\n\n const selectDate = (type: CalendarDateType) => {\n const dateMap: Record = {\n 'prev-month': prevMonthDayjs.value,\n 'next-month': nextMonthDayjs.value,\n 'prev-year': prevYearDayjs.value,\n 'next-year': nextYearDayjs.value,\n today: now,\n }\n\n const day = dateMap[type]\n\n if (!day.isSame(date.value, 'day')) {\n pickDay(day)\n }\n }\n\n const handleDateChange = (date: Dayjs | 'today') => {\n if (date === 'today') {\n selectDate('today')\n } else {\n pickDay(date)\n }\n }\n\n return {\n calculateValidatedDateRange,\n date,\n realSelectedDay,\n pickDay,\n selectDate,\n validatedRange,\n handleDateChange,\n }\n}\n"],"mappings":";;;;;;;AAUA,MAAM,iBAAiB,OAAc,QAAiC;CACpE,MAAM,oBAAoB,MAAM,MAAM,OAAO;CAC7C,MAAM,oBAAoB,IAAI,QAAQ,OAAO;CAI7C,MAAM,oBADa,kBAAkB,OAAO,mBAAmB,MAC5B,IAC/B,kBAAkB,IAAI,GAAG,MAAM,IAC/B;CAEJ,OAAO,CACL,CAAC,OAAO,iBAAiB,GACzB,CAAC,kBAAkB,QAAQ,MAAM,GAAG,GAAG,CACzC;AACF;AAEA,MAAM,yBAAyB,OAAc,QAAiC;CAC5E,MAAM,oBAAoB,MAAM,MAAM,OAAO;CAC7C,MAAM,sBAAsB,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,QAAQ,OAAO;CAGjE,MAAM,sBAAsB,kBAAkB,OAC5C,qBACA,MACF,IACI,oBAAoB,IAAI,GAAG,MAAM,IACjC;CAEJ,MAAM,qBAAqB,oBAAoB,MAAM,OAAO;CAC5D,MAAM,oBAAoB,IAAI,QAAQ,OAAO;CAG7C,MAAM,oBAAoB,mBAAmB,OAAO,mBAAmB,MAAM,IACzE,kBAAkB,IAAI,GAAG,MAAM,IAC/B;CAEJ,OAAO;EACL,CAAC,OAAO,iBAAiB;EACzB,CAAC,oBAAoB,QAAQ,MAAM,GAAG,kBAAkB;EACxD,CAAC,kBAAkB,QAAQ,MAAM,GAAG,GAAG;CACzC;AACF;AAEA,MAAa,eACX,OACA,MACA,kBACG;CACH,MAAM,EAAE,SAAS,UAAU;CAE3B,MAAM,cAAc,IAAW;CAC/B,MAAM,MAAM,MAAM,CAAC,CAAC,OAAO,KAAK,KAAK;CAErC,MAAM,kBAAkB,SAA4B;EAClD,MAAM;GACJ,IAAI,CAAC,MAAM,YAAY,OAAO,YAAY;GAC1C,OAAO,KAAK;EACd;EACA,IAAI,KAAK;GACP,IAAI,CAAC,KAAK;GACV,YAAY,QAAQ;GACpB,MAAM,SAAS,IAAI,OAAO;GAE1B,KAAK,aAAa,MAAM;GACxB,KAAK,oBAAoB,MAAM;EACjC;CACF,CAAC;CAGD,MAAM,iBAAiB,eAAe;EACpC,IACE,CAAC,MAAM,SACP,CAAC,QAAQ,MAAM,KAAK,KACpB,MAAM,MAAM,WAAW,KACvB,MAAM,MAAM,MAAM,SAAS,CAAC,OAAO,IAAI,CAAC,GAExC,OAAO,CAAC;EAEV,MAAM,CAAC,YAAY,YADG,MAAM,MAAM,KAAK,MAAM,MAAM,CAAC,CAAC,CAAC,OAAO,KAAK,KAAK,CAC5B;EAC3C,IAAI,WAAW,QAAQ,QAAQ,GAAG;GAChC,UAAU,eAAe,4CAA4C;GACrE,OAAO,CAAC;EACV;EACA,IAAI,WAAW,OAAO,UAAU,OAAO,GAErC,OAAO,4BAA4B,YAAY,QAAQ;OAClD;GAEL,IAAI,CAAC,WAAW,IAAI,GAAG,OAAO,CAAC,CAAC,OAAO,UAAU,OAAO,GAAG;IACzD,UACE,eACA,6DACF;IACA,OAAO,CAAC;GACV;GACA,OAAO,4BAA4B,YAAY,QAAQ;EACzD;CACF,CAAC;CAED,MAAM,OAA2B,eAAe;EAC9C,IAAI,CAAC,MAAM,YACT,OACE,gBAAgB,UACf,eAAe,MAAM,SAAS,eAAe,MAAM,EAAE,CAAC,KAAK;OAG9D,OAAO,MAAM,MAAM,UAAU,CAAC,CAAC,OAAO,KAAK,KAAK;CAEpD,CAAC;CACD,MAAM,iBAAiB,eAAe,KAAK,MAAM,SAAS,GAAG,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC;CAC7E,MAAM,iBAAiB,eAAe,KAAK,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC;CACxE,MAAM,gBAAgB,eAAe,KAAK,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC;CAC3E,MAAM,gBAAgB,eAAe,KAAK,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC;CAItE,MAAM,+BACJ,YACA,aACqB;EACrB,MAAM,WAAW,WAAW,QAAQ,MAAM;EAC1C,MAAM,UAAU,SAAS,MAAM,MAAM;EACrC,MAAM,aAAa,SAAS,IAAI,OAAO;EACvC,MAAM,YAAY,QAAQ,IAAI,OAAO;EAGrC,IAAI,eAAe,WACjB,OAAO,CAAC,CAAC,UAAU,OAAO,CAAC;OAGxB,KAAK,aAAa,KAAK,OAAO,WACjC,OAAO,cAAc,UAAU,OAAO;OAGnC,IACH,aAAa,MAAM,cAClB,aAAa,KAAK,OAAO,WAE1B,OAAO,sBAAsB,UAAU,OAAO;OAG3C;GACH,UACE,eACA,6DACF;GACA,OAAO,CAAC;EACV;CACF;CAEA,MAAM,WAAW,QAAe;EAC9B,gBAAgB,QAAQ;CAC1B;CAEA,MAAM,cAAc,SAA2B;EAS7C,MAAM,MAAM;GAPV,cAAc,eAAe;GAC7B,cAAc,eAAe;GAC7B,aAAa,cAAc;GAC3B,aAAa,cAAc;GAC3B,OAAO;EAGS,EAAE;EAEpB,IAAI,CAAC,IAAI,OAAO,KAAK,OAAO,KAAK,GAC/B,QAAQ,GAAG;CAEf;CAEA,MAAM,oBAAoB,SAA0B;EAClD,IAAI,SAAS,SACX,WAAW,OAAO;OAElB,QAAQ,IAAI;CAEhB;CAEA,OAAO;EACL;EACA;EACA;EACA;EACA;EACA;EACA;CACF;AACF"}