CropTraitObservationApiServiceImpl.java

/*
 * Copyright 2026 Global Crop Diversity Trust
 * Licensed under the Apache License, Version 2.0
 * See LICENSE file in project root folder or http://www.apache.org/licenses/LICENSE-2.0
 */

package org.gringlobal.api.v2.facade.impl;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

import org.gringlobal.api.exception.InvalidApiUsageException;
import org.gringlobal.api.model.CropTraitDTO;
import org.gringlobal.api.model.CropTraitObservationDTO;
import org.gringlobal.api.model.InventoryDTO;
import org.gringlobal.api.model.TranslatedCropTraitObservationDTO;
import org.gringlobal.api.v2.facade.CropTraitObservationApiService;
import org.gringlobal.custom.elasticsearch.SearchException;
import org.gringlobal.model.CropTraitObservation;
import org.gringlobal.model.QCropTraitObservation;
import org.gringlobal.persistence.CropTraitRepository;
import org.gringlobal.persistence.InventoryRepository;
import org.gringlobal.persistence.MethodRepository;
import org.gringlobal.service.CropTraitObservationService;
import org.gringlobal.service.filter.CropTraitObservationFilter;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.util.Pair;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import com.querydsl.jpa.impl.JPAQueryFactory;
import org.apache.commons.collections4.CollectionUtils;

@Service
@Transactional(readOnly = true)
public class CropTraitObservationApiServiceImpl extends APIFilteredServiceFacadeImpl<CropTraitObservationService, CropTraitObservationDTO, CropTraitObservation, CropTraitObservationFilter>
	implements CropTraitObservationApiService {

	@Autowired
	private MethodRepository methodRepository;

	@Autowired
	private InventoryRepository inventoryRepository;

	@Autowired
	private CropTraitRepository cropTraitRepository;

	@Autowired
	protected JPAQueryFactory jpaQueryFactory;

	@Override
	protected CropTraitObservation convert(CropTraitObservationDTO source) {
		return mapper.map(source);
	}

	@Override
	protected CropTraitObservationDTO convert(CropTraitObservation source) {
		return mapper.map(source);
	}

	@Override
	public TranslatedCropTraitObservationDTO getTranslated(long id) {
		return mapper.map(service.getTranslated(id));
	}

	@Override
	public Page<TranslatedCropTraitObservationDTO> listTranslated(CropTraitObservationFilter filter, Pageable page) throws SearchException {
		return mapper.map(service.listTranslated(filter, page), mapper::map);
	}

	@Override
	public Page<InventoryDTO> getObservationInventoriesByMethod(Long methodId, Pageable pageable) {
		return mapper.map(service.getObservationInventoriesByMethod(methodId, pageable), mapper::map);
	}

	@Override
	public Page<CropTraitDTO> getObservationTraitsByMethod(Long methodId, Pageable pageable) {
		return mapper.map(service.getObservationTraitsByMethod(methodId, pageable), mapper::map);
	}

	@Override
	@Transactional
	public int ensureObservations(EnsureObservationsRequest requestDTO) {
		var request = mapper.map(requestDTO);
		if (request.methodId == null || CollectionUtils.isEmpty(request.inventoryId) || CollectionUtils.isEmpty(request.cropTraitId)) {
			throw new InvalidApiUsageException("Method id, CropTrait ids and inventory ids must be provided");
		}
		var method = methodRepository.getReferenceById(request.methodId);
		var observationPath = QCropTraitObservation.cropTraitObservation;

		var observations = jpaQueryFactory.selectFrom(observationPath).distinct()
			.where(observationPath.method().id.in(method.getId())
				.and(observationPath.cropTrait().id.in(request.cropTraitId))
				.and(observationPath.inventory().id.in(request.inventoryId))).fetch();

		var idPairs = observations.stream()
			.map(obs -> Pair.of(obs.getCropTrait().getId(), obs.getInventory().getId())).collect(Collectors.toList());

		List<CropTraitObservation> observationsForSave = new ArrayList<>();
		for (var traitId: request.cropTraitId) {
			for (var inventoryId: request.inventoryId) {
				if (!idPairs.contains(Pair.of(traitId, inventoryId))) {
					CropTraitObservation observation = new CropTraitObservation();
					observation.setCropTrait(cropTraitRepository.getReferenceById(traitId));
					observation.setInventory(inventoryRepository.getReferenceById(inventoryId));
					observation.setMethod(method);
					observationsForSave.add(observation);
				}
			}
		}

		observationsForSave.forEach(service::createFast);
		return observationsForSave.size();
	}

	@Override
	public FilteredObservations search(CropTraitObservationFilter filter, Pageable pageable) {
		return mapper.map(service.search(filter, pageable));
	}

}