xcandy/src/xcComponents/Calendar/utils.ts

A UI framework for taro.js

import dayjs from 'dayjs'
import { RenderDate } from './interface'

const RENDER_DATE_TOTAL = 6 * 7

function getISODateStr (year: number, month: number, date: number) {
  return `${year}-${month < 10 ? '0' + month : month}-${date < 10 ? '0' + date : date}`
}

function computePreMonthRenderArr (currentRenderDate: dayjs.Dayjs) {
  let renderLen = currentRenderDate.startOf('month').day() - 1
  let renderArr: RenderDate[] = []

  if (renderLen > 0) {
    let preMonth = currentRenderDate.startOf('month').subtract(1, 'day')
    let preMonthYear = preMonth.year()
    let preMonthMonth = preMonth.month() + 1
    let preMonthEndDate = preMonth.date()

    for (let i = 0; i < renderLen; i++) {
      renderArr.unshift({
        date: preMonthEndDate,
        str: dayjs(getISODateStr(preMonthYear, preMonthMonth, preMonthEndDate)).format('YYYY-MM-DD')
      })
      preMonthEndDate -= 1
    }
  }
  return renderArr
}

function computeCurrenMonthRenderArr (currentRenderDate: dayjs.Dayjs) {
  let renderLen = currentRenderDate.endOf('month').date()
  let renderArr: RenderDate[] = []

  let year = currentRenderDate.year()
  let month = currentRenderDate.month() + 1

  for (let i = 1; i <= renderLen; i++) {
    renderArr.push({
      date: i,
      str: dayjs(getISODateStr(year, month, i)).format('YYYY-MM-DD')
    })
  }
  return renderArr
}

function computeNexMonthRenderArr (currentRenderDate: dayjs.Dayjs) {
  let preMonthDateCount = currentRenderDate.startOf('month').day()
  preMonthDateCount = preMonthDateCount > 0 ? preMonthDateCount - 1 : 0

  let currentMonthDateCount = currentRenderDate.endOf('month').date()

  let renderLen = RENDER_DATE_TOTAL - preMonthDateCount - currentMonthDateCount
  let renderArr: RenderDate[] = []
  if (renderLen > 0) {
    let nextMonth = currentRenderDate.endOf('month').add(1, 'day')
    let nextMonthYear = nextMonth.year()
    let nextMonthMonth = nextMonth.month() + 1
    for (let i = 1; i <= renderLen; i++) {
      renderArr.push({
        date: i,
        str: dayjs(getISODateStr(nextMonthYear, nextMonthMonth, i)).format('YYYY-MM-DD')
      })
    }
  }
  return renderArr
}

export { getISODateStr, computePreMonthRenderArr, computeCurrenMonthRenderArr, computeNexMonthRenderArr }