Days.js

/**
 * Persian Calendar Picker Component
 *
 * Copyright 2016 Reza (github.com/rghorbani)
 * Licensed under the terms of the MIT license. See LICENSE file in the project root for terms.
 */

'use strict';

const React = require('react');
const PropTypes = require('prop-types');
const moment = require('moment-jalaali');
const {
  Text,
  View,
} = require('react-native');

const styles = require('./style');
const {
  MAX_ROWS,
  MAX_COLUMNS,
} = require('./util');
const Day = require('./Day');

class Days extends React.Component {
  static propTypes = {
    maxDate: PropTypes.instanceOf(moment),
    minDate: PropTypes.instanceOf(Date),
    date: PropTypes.object.isRequired,
    month: PropTypes.number.isRequired,
    year: PropTypes.number.isRequired,
    onDayChange: PropTypes.func.isRequired,
    selectedDayColor: PropTypes.string,
    selectedDayTextColor: PropTypes.string,
    textStyle: Text.propTypes.style,
    prevDays: PropTypes.string,
  };

  constructor(props) {
    super(props);

    this.state = {
      selectedStates: []
    };

    this.updateSelectedStates = this.updateSelectedStates.bind(this);
    this.onPressDay = this.onPressDay.bind(this);
    this.getCalendarDays = this.getCalendarDays.bind(this);
  }

  componentDidMount() {
    this.updateSelectedStates(this.props.date.jDate());
  }

  // Trigger date change if new props are provided.
  // Typically, when selectedDate is changed programmatically.
  //
  componentWillReceiveProps(newProps) {
    this.updateSelectedStates(newProps.date.jDate());
  }

  updateSelectedStates(day) {
    let selectedStates = [],
      daysInMonth = moment.jDaysInMonth(this.props.year, this.props.month);

    for (let i = 1; i <= daysInMonth; i++) {
      if (i === day) {
        selectedStates.push(true);
      } else {
        selectedStates.push(false);
      }
    }

    this.setState({
      selectedStates: selectedStates
    });
  }

  onPressDay(day) {
    this.updateSelectedStates(day);
    this.props.onDayChange({day: day});
  }

  // Not going to touch this one - I'd look at whether there is a more functional
  // way you can do this using something like `range`, `map`, `partition` and such
  // (see underscore.js), or just break it up into steps: first generate the array for
  // data, then map that into the components
  getCalendarDays() {
    let columns,
      matrix = [],
      month = this.props.month,
      year = this.props.year,
      currentDay = 0,
      thisMonthFirstDay = moment(year + '/' + (month + 1) + '/1', 'jYYYY/jM/jD'),
      dayOfWeek = (thisMonthFirstDay.weekday() - 1) % 7,
      slotsAccumulator = 0;

    for (let i = 0; i < MAX_ROWS; i++ ) { // Week rows
      columns = [];

      for (let j = 0; j < MAX_COLUMNS; j++) { // Day columns
        if (slotsAccumulator >= dayOfWeek) {
          if (currentDay < moment.jDaysInMonth(year, month)) {
            columns.push(<Day
                      key={j}
                      day={currentDay + 1}
                      selected={this.state.selectedStates[currentDay]}
                      date={moment(year + '/' + (month + 1) + '/' + (currentDay + 1), 'jYYYY/jM/jD')}
                      maxDate={this.props.maxDate}
                      minDate={this.props.minDate}
                      onDayChange={this.onPressDay}
                      screenWidth={this.props.screenWidth}
                      selectedDayColor={this.props.selectedDayColor}
                      selectedDayTextColor={this.props.selectedDayTextColor}
                      textStyle={this.props.textStyle}
                      prevDays={this.props.prevDays}
                      />);
            currentDay++;
          } else if (currentDay >= moment.jDaysInMonth(year, month)) {
              columns.push(<Day
                          key={j}
                          day={''}
                          screenWidth={this.props.screenWidth} />);
          }
        } else {
          columns.push(<Day
                        key={j}
                        day={''}
                        screenWidth={this.props.screenWidth} />);
        }

        slotsAccumulator++;
      }
      matrix[i] = [];
      columns.reverse();
      matrix[i].push(<View key={i} style={styles.weekRow}>{columns}</View>);
    }

    return matrix;
  }

  render() {
    return (
      <View style={styles.daysWrapper}>
        {this.getCalendarDays()}
      </View>
    );
  }
}

module.exports = Days;