CropTraitObservationDataApiServiceImpl.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.CropTraitObservationDataDTO;
import org.gringlobal.api.model.InventoryDTO;
import org.gringlobal.api.v2.facade.CropTraitObservationApiService;
import org.gringlobal.api.v2.facade.CropTraitObservationDataApiService;
import org.gringlobal.model.CropTraitObservationData;
import org.gringlobal.model.QCropTraitObservationData;
import org.gringlobal.persistence.CropTraitRepository;
import org.gringlobal.persistence.InventoryRepository;
import org.gringlobal.persistence.MethodRepository;
import org.gringlobal.service.CropTraitObservationDataService;
import org.gringlobal.service.filter.CropTraitObservationDataFilter;

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 CropTraitObservationDataApiServiceImpl
	extends APIFilteredServiceFacadeImpl<CropTraitObservationDataService, CropTraitObservationDataDTO, CropTraitObservationData, CropTraitObservationDataFilter>
	implements CropTraitObservationDataApiService {

	@Autowired
	private MethodRepository methodRepository;

	@Autowired
	private InventoryRepository inventoryRepository;

	@Autowired
	private CropTraitRepository cropTraitRepository;

	@Autowired
	protected JPAQueryFactory jpaQueryFactory;

	@Override
	protected CropTraitObservationData convert(CropTraitObservationDataDTO source) {
		return mapper.map(source);
	}

	@Override
	protected CropTraitObservationDataDTO convert(CropTraitObservationData source) {
		return mapper.map(source);
	}


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

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

	@Override
	@Transactional
	public int ensureObservationData(CropTraitObservationApiService.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 observationDataPath = QCropTraitObservationData.cropTraitObservationData;
		var observations = jpaQueryFactory.selectFrom(observationDataPath).distinct()
			.where(observationDataPath.method().id.in(method.getId())
				.and(observationDataPath.cropTrait().id.in(request.cropTraitId))
				.and(observationDataPath.inventory().id.in(request.inventoryId))).fetch();

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

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

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

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