import { Repository } from 'typeorm';
import { CommandBus, CommandHandler, ICommandHandler } from '@nestjs/cqrs';
import { InjectRepository } from '@nestjs/typeorm';
import { IPublicOrganization } from '@energyweb/origin-backend-core';
import { GetOrganizationsCommand } from '@energyweb/origin-backend';

import { Beneficiary } from '../beneficiary.entity';
import { GetBeneficiariesCommand } from '../commands';
import { BeneficiaryDTO } from '../dto';

@CommandHandler(GetBeneficiariesCommand)
export class GetBeneficiariesHandler implements ICommandHandler<GetBeneficiariesCommand> {
    constructor(
        @InjectRepository(Beneficiary)
        private readonly repository: Repository<Beneficiary>,
        private readonly commandBus: CommandBus
    ) {}

    public async execute({ organizationId }: GetBeneficiariesCommand): Promise<BeneficiaryDTO[]> {
        const beneficiaries = await this.repository.find({ ownerId: organizationId || null });

        const organizations: IPublicOrganization[] = await this.commandBus.execute(
            new GetOrganizationsCommand({ ids: beneficiaries.map((b) => b.organizationId) })
        );

        return beneficiaries.map((beneficiary) => {
            const organization = organizations.find((org) => org.id === beneficiary.organizationId);

            return BeneficiaryDTO.wrap({
                ...beneficiary,
                organization: organization || null
            });
        });
    }
}