InventoryApiServiceImpl.java

/*
 * Copyright 2024 Global Crop Diversity Trust
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.gringlobal.api.v2.facade.impl;

import java.io.OutputStreamWriter;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
import javax.servlet.http.HttpServletResponse;

import org.genesys.blocks.auditlog.model.filters.AuditLogFilter;
import org.genesys.blocks.auditlog.service.AuditTrailService;
import org.genesys.blocks.model.filters.NumberFilter;

import org.gringlobal.model.AccessionInvAnnotation;
import org.gringlobal.model.AccessionInvAttach;
import org.gringlobal.model.AccessionInvGroupMap;
import org.gringlobal.model.AccessionInvName;
import org.gringlobal.model.InventoryAction;
import org.gringlobal.model.InventoryExtra;
import org.gringlobal.model.InventoryMaintenancePolicy;
import org.gringlobal.model.InventoryQualityStatus;
import org.gringlobal.model.InventoryViability;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.Pageable;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.web.servlet.mvc.method.annotation.StreamingResponseBody;

import org.gringlobal.api.FilteredPage;
import org.gringlobal.api.model.AuditLogDTO;
import org.gringlobal.api.model.InventoryDTO;
import org.gringlobal.api.model.InventoryDetailsDTO;
import org.gringlobal.api.model.OrderRequestDTO;
import org.gringlobal.api.v2.facade.InventoryApiService;
import org.gringlobal.model.Inventory;
import org.gringlobal.service.CRUDService;
import org.gringlobal.service.InventoryService;
import org.gringlobal.service.filter.InventoryActionFilter;
import org.gringlobal.service.filter.InventoryFilter;

@Service
@Transactional(readOnly = true)
public class InventoryApiServiceImpl extends APIFilteredServiceFacadeImpl<InventoryService, InventoryDTO, Inventory, InventoryFilter> implements InventoryApiService {

	@Autowired
	private AuditTrailService auditTrailService;

	@Override
	protected InventoryDTO convert(Inventory source) {
		return mapper.map(source);
	}

	@Override
	protected Inventory convert(InventoryDTO source) {
		return mapper.map(source);
	}

	@Override
	public ResponseEntity<StreamingResponseBody> generateLabelsPDF(CRUDService.LabelConfig labelConfig, List<Long> ids, HttpServletResponse response) throws Exception {
		return service.generateLabelsPDF(labelConfig, ids, LocaleContextHolder.getLocale(), response);
	}

	@Override
	public ResponseEntity<StreamingResponseBody> generateLabels(CRUDService.LabelConfig labelConfig, List<Long> ids) {
		return ResponseEntity.ok().contentType(MediaType.TEXT_PLAIN).body(out -> {
			var w = new OutputStreamWriter(out, "UTF8");
			service.generateLabels(labelConfig, ids, w);
			w.flush();
		});
	}

	@Override
	public ResponseEntity<StreamingResponseBody> generateLabels(Map<Long, CRUDService.LabelConfig> labelConfig) {
		return ResponseEntity.ok().contentType(MediaType.TEXT_PLAIN).body(out -> {
			var w = new OutputStreamWriter(out, "UTF8");
			service.generateLabels(labelConfig, w);
			w.flush();
		});
	}

	@Override
	@Transactional
	public void shareAttachment(Long attachId, List<Long> inventoryIds) {
		service.shareAttachment(attachId, inventoryIds);
	}

	@Override
	public InventoryDetailsDTO getInventoryDetails(String barcode) {
		return mapper.map(service.getInventoryDetails(service.getByBarcode(barcode)));
	}

	@Override
	public InventoryDetailsDTO getInventoryDetails(long inventoryId) {
		return mapper.map(service.getInventoryDetails(service.get(inventoryId)));
	}

	@Override
	@Transactional
	public void discardMaterial(Map<Long, Integer> discardQuantities) {
		service.discardMaterial(discardQuantities);
	}

	@Override
	@Transactional
	public InventoryDTO setInventoryQuantity(InventoryQuantityRequestDTO requestDTO) {
		return mapper.map(service.setInventoryQuantity(mapper.map(requestDTO)));
	}

	@Override
	@Transactional
	public List<InventoryDTO> assignLocation(AssignLocationRequestDTO assignLocationRequest) {
		return mapper.map(service.assignLocation(mapper.map(assignLocationRequest)), mapper::map);
	}

	@Override
	@Transactional
	public List<InventoryDTO> assignLocations(List<AssignLocationRequestDTO> assignLocationRequests) {
		return mapper.map(service.assignLocations(mapper.map(assignLocationRequests, mapper::map)), mapper::map);
	}

	@Override
	public Page<ComparedSitesResponseDTO> compareSites(CompareSitesRequestDTO requestDTO, Pageable page) {
		return mapper.map(service.compareSites(requestDTO.filter, requestDTO.sites, page), mapper::map);
	}

	@Override
	@Transactional
	public String assignBarcode(Long inventoryId) {
		return service.assignBarcode(service.get(inventoryId));
	}

	@Override
	public Page<AuditLogDTO> listAuditLogs(Long inventoryId, Pageable page) {
		var inventory = service.get(inventoryId);
		var extra = inventory.getExtra();
		var inventoryMaintenancePolicy = inventory.getInventoryMaintenancePolicy();
		var actions = inventory.getActions();
		var quality = inventory.getQuality();
		var viability = inventory.getViability();
		var annotation = inventory.getAnnotations();
		var names = inventory.getNames();
		var attachments = inventory.getAttachments();
		
		List<AuditLogFilter> filters = new LinkedList<>();

		filters.add(new AuditLogFilter().entityId(new NumberFilter<Long>().eq(Set.of(inventory.getId()))).classname(Inventory.class.getName()));
		if (extra != null) {
			filters.add(new AuditLogFilter().entityId(new NumberFilter<Long>().eq(Set.of(extra.getId()))).classname(extra.getClass().getName()));
		}
		if (inventoryMaintenancePolicy != null) {
			filters.add((new AuditLogFilter().entityId(new NumberFilter<Long>().eq(Set.of(inventoryMaintenancePolicy.getId()))).classname(InventoryMaintenancePolicy.class.getName())));
		}
		if (!CollectionUtils.isEmpty(actions)) {
			filters.add((new AuditLogFilter()
				.entityId(new NumberFilter<Long>().eq(actions.stream().map(InventoryAction::getId).collect(Collectors.toSet())))
				.classname(InventoryAction.class.getName())));
		}
		if (!CollectionUtils.isEmpty(quality)) {
			filters.add((new AuditLogFilter()
				.entityId(new NumberFilter<Long>().eq(quality.stream().map(InventoryQualityStatus::getId).collect(Collectors.toSet())))
				.classname(InventoryQualityStatus.class.getName())));
		}
		if (!CollectionUtils.isEmpty(viability)) {
			filters.add((new AuditLogFilter()
				.entityId(new NumberFilter<Long>().eq(viability.stream().map(InventoryViability::getId).collect(Collectors.toSet())))
				.classname(InventoryViability.class.getName())));
		}
		if (!CollectionUtils.isEmpty(annotation)) {
			filters.add((new AuditLogFilter()
				.entityId(new NumberFilter<Long>().eq(annotation.stream().map(AccessionInvAnnotation::getId).collect(Collectors.toSet())))
				.classname(AccessionInvAnnotation.class.getName())));
		}
		if (!CollectionUtils.isEmpty(names)) {
			filters.add((new AuditLogFilter()
				.entityId(new NumberFilter<Long>().eq(names.stream().map(AccessionInvName::getId).collect(Collectors.toSet())))
				.classname(AccessionInvName.class.getName())));
		}
		if (!CollectionUtils.isEmpty(attachments)) {
			filters.add((new AuditLogFilter()
				.entityId(new NumberFilter<Long>().eq(attachments.stream().map(AccessionInvAttach::getId).collect(Collectors.toSet())))
				.classname(AccessionInvAttach.class.getName())));
		}
		AuditLogFilter orFilter = null;
		for (AuditLogFilter f : filters) {
			orFilter = (orFilter == null) ? f : f.OR(orFilter);
		}

		return mapper.map(auditTrailService.listAuditLogs(orFilter, page), mapper::map);
	}

	@Override
	@Transactional
	public List<InventoryDTO> splitInventory(SplitInventoryRequestDTO request) {
		return mapper.map(service.splitInventory(mapper.map(request)), mapper::map);
	}

	@Override
	public Map<?, ?> inventoryOverview(String groupBy, InventoryFilter filter) {
		return service.inventoryOverview(groupBy, filter);
	}

	@Override
	public Page<AggregatedInventoryQuantityDTO> aggregateQuantity(InventoryFilter filter, Pageable pageable) {
		return mapper.map(service.aggregateQuantity(filter, pageable), mapper::map);
	}

	@Override
	@Transactional
	public OrderRequestDTO multiplicationOrder(MultiplicationOrderRequestDTO request) {
		return mapper.map(service.multiplicationOrder(
			mapper.mapInfo(request.site), request.inventoryIds, mapper.mapInfo(request.cooperator), request.orderTypeCode, request.intendedUseCode
		));
	}

	@Override
	@Transactional(readOnly = true)
	public FilteredPage<InventoryHarvestDTO, InventoryActionFilter> listInventoryHarvest(InventoryActionFilter filter, Pageable page) {
		var result = service.listInventoryHarvest(filter, page);
		return new FilteredPage<>(result.filterCode, result.filter, new PageImpl<>(mapper.map(result.page.getContent(), mapper::map), result.page.getPageable(), result.page.getTotalElements()));
	}
	
	@Override
	@Transactional
	public InventoryHarvestDTO createHarvestedInventory(InventoryHarvestRequest request) {
		return mapper.map(service.createHarvestedInventory(mapper.mapInfo(request.getPlantedInventory()), mapper.mapInfo(request.getSite()), mapper.mapInfo(request.getInventoryMantenancePolicy()), request.getInventoryNumberPart1(), request.getInventoryNumberPart3(), request.getQuantityOnHand(), request.getQuantityOnHandUnitCode(), request.getContainerTypeCode(), mapper.mapInfo(request.getProductionGeography()), request.getStorageLocationPart1(), request.getStorageLocationPart2(), request.getStorageLocationPart3(), request.getStorageLocationPart4(), request.isFinalBatch()));
	}

}