fairnesscoop / permacoop

import { User } from '../User/User.entity';

import { Leave } from './Leave.entity';

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

import { UserLeavesCollection } from './UserLeavesCollection';

export class LeavesCollection {

    constructor(

        public readonly leaves: LeaveRequest[]

    ){}

    getLeavesByUser(user: User): UserLeavesCollection {

        const userLeaves: LeaveRequest[] = this.leaves.filter((leave: LeaveRequest) => leave.getUser().getId() === user.getId());

        return new UserLeavesCollection(userLeaves);

    }

}