fairnesscoop / permacoop

import { LeaveRequest, Type } from './LeaveRequest.entity';

export class UserLeavesCollection {

  public paid: LeaveRequest[] = [];

  public unpaid: LeaveRequest[] = [];

  public special: LeaveRequest[] = [];

  public medical: LeaveRequest[] = [];

  constructor(leaves: LeaveRequest[]) {

    this.distributeLeavesByType(leaves);

  }

  private distributeLeavesByType(leaves: LeaveRequest[]): void {

    for (const leave of leaves) {

      switch (leave.getType()) {

        case Type.PAID:

          this.paid.push(leave);

          break;

        case Type.UNPAID:

          this.unpaid.push(leave);

          break;

        case Type.SPECIAL:

          this.special.push(leave);

          break;

        case Type.MEDICAL:

          this.medical.push(leave);

      }

    }

  }

}