fairnesscoop / permacoop

import { InjectRepository } from '@nestjs/typeorm';

import { Repository } from 'typeorm';

import { IHolidayRequestRepository } from 'src/Domain/HumanResource/Holiday/Repository/IHolidayRequestRepository';

import { MAX_ITEMS_PER_PAGE } from 'src/Application/Common/Pagination';

import { HolidayRequest, HolidayStatus } from 'src/Domain/HumanResource/Holiday/HolidayRequest.entity';

import { User } from 'src/Domain/HumanResource/User/User.entity';

export class HolidayRequestRepository implements IHolidayRequestRepository {

  constructor(

    @InjectRepository(HolidayRequest)

    private readonly repository: Repository<HolidayRequest>

  ) {}

  public save(holidayRequest: HolidayRequest): Promise<HolidayRequest> {

    return this.repository.save(holidayRequest);

  }

  public findOneById(id: string): Promise<HolidayRequest | undefined> {

    return this.repository

      .createQueryBuilder('holidayRequest')

      .select(['holidayRequest.id', 'holidayRequest.status', 'user.id'])

      .where('holidayRequest.id = :id', {id})

      .innerJoin('holidayRequest.user', 'user')

      .getOne();

  }

  public findExistingHolidayRequestsByUserAndPeriod(

    user: User,

    startDate: string,

    endDate: string

  ): Promise<HolidayRequest | undefined> {

    return this.repository

      .createQueryBuilder('holidayRequest')

      .select('holidayRequest.id')

      .where('holidayRequest.user = :id', {id: user.getId()})

      .andWhere(

        '(holidayRequest.startDate BETWEEN :startDate AND :endDate OR holidayRequest.endDate BETWEEN :startDate AND :endDate)',

        {

          startDate,

          endDate

        }

      )

      .andWhere('(holidayRequest.status = :accepted OR holidayRequest.status = :pending)', {

        accepted: HolidayStatus.ACCEPTED,

        pending: HolidayStatus.PENDING

      })

      .getOne();

  }

  public findHolidayRequests(page: number): Promise<[HolidayRequest[], number]> {

    return this.repository

      .createQueryBuilder('holidayRequest')

      .select([

        'holidayRequest.id',

        'holidayRequest.type',

        'holidayRequest.status',

        'holidayRequest.startDate',

        'holidayRequest.startsAllDay',

        'holidayRequest.endDate',

        'holidayRequest.endsAllDay',

        'user.id',

        'user.firstName',

        'user.lastName'

      ])

      .innerJoin('holidayRequest.user', 'user')

      .orderBy('holidayRequest.startDate', 'ASC')

      .limit(MAX_ITEMS_PER_PAGE)

      .offset((page - 1) * MAX_ITEMS_PER_PAGE)

      .getManyAndCount();

  }

}