fairnesscoop / permacoop

import {CommandHandler} from '@nestjs/cqrs';

import {Inject} from '@nestjs/common';

import {CreateQuoteCommand} from './CreateQuoteCommand';

import {IQuoteRepository} from 'src/Domain/Billing/Repository/IQuoteRepository';

import {Quote} from 'src/Domain/Billing/Quote.entity';

import {IProjectRepository} from 'src/Domain/Project/Repository/IProjectRepository';

import {ICustomerRepository} from 'src/Domain/Customer/Repository/ICustomerRepository';

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

import {CustomerNotFoundException} from 'src/Domain/Customer/Exception/CustomerNotFoundException';

import {ProjectNotFoundException} from 'src/Domain/Project/Exception/ProjectNotFoundException';

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

import {QuoteIdGenerator} from 'src/Domain/Billing/QuoteIdGenerator';

@CommandHandler(CreateQuoteCommand)

export class CreateQuoteCommandHandler {

  constructor(

    @Inject('IQuoteRepository')

    private readonly quoteRepository: IQuoteRepository,

    @Inject('ICustomerRepository')

    private readonly customerRepository: ICustomerRepository,

    @Inject('IProjectRepository')

    private readonly projectRepository: IProjectRepository,

    private readonly quoteIdGenerator: QuoteIdGenerator

  ) {}

  public async execute(command: CreateQuoteCommand): Promise<string> {

    const {customerId, status, user, projectId} = command;

    const customer = await this.customerRepository.findOneById(customerId);

    if (!(customer instanceof Customer)) {

      throw new CustomerNotFoundException();

    }

    let project = null;

    if (projectId) {

      project = await this.projectRepository.findOneById(projectId);

      if (!(project instanceof Project)) {

        throw new ProjectNotFoundException();

      }

    }

    const quoteId = await this.quoteIdGenerator.generate();

    const quote = await this.quoteRepository.save(

      new Quote(quoteId, status, user, customer, project)

    );

    return quote.getId();

  }

}