vuikit/core/helpers/date/matrix.js

A Vuejs component library based on UIkit

/**
* Vuikit 0.7.0
* (c) 2018 Miljan Aleksic
* @license MIT
*/

import { range } from 'vuikit/core/util'

import getDate from 'date-fns/get_date'
import addDays from 'date-fns/add_days'
import getYear from 'date-fns/get_year'
import getMonth from 'date-fns/get_month'
import isSameMonth from 'date-fns/is_same_month'
import startOfWeek from 'date-fns/start_of_week'

const rows = range(6)
const cols = range(7)

/**
 * Returns a two-dimensional array with calendar represented dates
 *
 * @date  Date or Object
 * @plain Boolean - Whetever the dates should be set as raw numbers
 */
export default function ({ year, month, weekStartsOn } = {
  year: getYear(Date.now()),
  month: getMonth(Date.now()),
  weekStartsOn: 0
}, plain = false) {
  const matrix = []
  const date = arguments[0] instanceof Date
    ? arguments[0]
    : new Date(year, month)
  let curDate = startOfWeek(date, { weekStartsOn })

  rows.forEach(row => {
    const week = []
    cols.forEach(col => {
      // when plain return a raw date re
      if (plain) {
        week.push(isSameMonth(curDate, date)
          ? getDate(curDate)
          : -getDate(curDate)
        )
      } else {
        week.push(curDate)
      }
      curDate = addDays(curDate, 1)
    })

    matrix.push(week)
  })

  return matrix
}