fairnesscoop / permacoop

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

import {Repository} from 'typeorm';

import {IDailyRateRepository} from 'src/Domain/Accounting/Repository/IDailyRateRepository';

import {DailyRate} from 'src/Domain/Accounting/DailyRate.entity';

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

import {Customer} from 'src/Domain/Customer/Customer.entity';

import {Task} from 'src/Domain/Task/Task.entity';

export class DailyRateRepository implements IDailyRateRepository {

  constructor(

    @InjectRepository(DailyRate)

    private readonly repository: Repository<DailyRate>

  ) {}

  public save(dailyRate: DailyRate): Promise<DailyRate> {

    return this.repository.save(dailyRate);

  }

  public findAll(): Promise<DailyRate[]> {

    return this.repository

      .createQueryBuilder('dailyRate')

      .select([

        'dailyRate.id',

        'dailyRate.amount',

        'user.id',

        'user.firstName',

        'user.lastName',

        'user.email',

        'task.id',

        'task.name',

        'customer.id',

        'customer.name'

      ])

      .innerJoin('dailyRate.user', 'user')

      .innerJoin('dailyRate.task', 'task')

      .innerJoin('dailyRate.customer', 'customer')

      .orderBy('user.lastName', 'ASC')

      .addOrderBy('user.firstName', 'ASC')

      .getMany();

  }

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

    return this.repository

      .createQueryBuilder('dailyRate')

      .select([

        'dailyRate.id',

        'dailyRate.amount',

        'user.id',

        'user.firstName',

        'user.lastName',

        'user.email',

        'task.id',

        'task.name',

        'customer.id',

        'customer.name'

      ])

      .innerJoin('dailyRate.user', 'user')

      .innerJoin('dailyRate.task', 'task')

      .innerJoin('dailyRate.customer', 'customer')

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

      .getOne();

  }

  public findOneByUserCustomerAndTask(

    user: User,

    customer: Customer,

    task: Task

  ): Promise<DailyRate | undefined> {

    return this.repository

      .createQueryBuilder('dailyRate')

      .select('dailyRate.id')

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

      .andWhere('dailyRate.customer = :customer', {customer: customer.getId()})

      .andWhere('dailyRate.task = :task', {task: task.getId()})

      .getOne();

  }

}