/*
* This program is part of the OpenLMIS logistics management information system platform software.
* Copyright © 2013 VillageReach
*
* This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more details.
* You should have received a copy of the GNU Affero General Public License along with this program. If not, see http://www.gnu.org/licenses. For additional information contact info@OpenLMIS.org.
*/
package org.openlmis.upload.parser;
import lombok.NoArgsConstructor;
import org.openlmis.upload.Importable;
import org.openlmis.upload.RecordHandler;
import org.openlmis.upload.exception.UploadException;
import org.openlmis.upload.model.AuditFields;
import org.openlmis.upload.model.ModelClass;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;
import org.supercsv.exception.SuperCsvCellProcessorException;
import org.supercsv.exception.SuperCsvConstraintViolationException;
import org.supercsv.exception.SuperCsvException;
import org.supercsv.util.CsvContext;
import java.io.IOException;
import java.io.InputStream;
/**
* This class has logic to invoke corresponding respective record handler to parse data from input stream into the corresponding model.
*/
@Component
@NoArgsConstructor
public class CSVParser {
@Transactional
public int process(InputStream inputStream, ModelClass modelClass, RecordHandler recordHandler, AuditFields auditFields)
throws UploadException {
CsvBeanReader csvBeanReader = null;
String[] headers = null;
try {
csvBeanReader = new CsvBeanReader(modelClass, inputStream);
headers = csvBeanReader.getHeaders();
csvBeanReader.validateHeaders();
Importable importedModel;
while ((importedModel = csvBeanReader.readWithCellProcessors()) != null) {
recordHandler.execute(importedModel, csvBeanReader.getRowNumber(), auditFields);
}
recordHandler.postProcess(auditFields);
} catch (SuperCsvConstraintViolationException constraintException) {
if (constraintException.getMessage().contains("^\\d{1,2}/\\d{1,2}/\\d{4}$")) {
createHeaderException("incorrect.date.format", headers, constraintException);
}
createHeaderException("missing.mandatory", headers, constraintException);
} catch (SuperCsvCellProcessorException processorException) {
createHeaderException("incorrect.data.type", headers, processorException);
} catch (SuperCsvException superCsvException) {
if (csvBeanReader.length() > headers.length) {
throw new UploadException("incorrect.file.format");
}
createDataException("column.do.not.match", headers, superCsvException);
} catch (IOException e) {
throw new UploadException(e.getStackTrace().toString());
}
return csvBeanReader.getRowNumber() - 1;
}
private void createHeaderException(String error, String[] headers, SuperCsvException exception) {
CsvContext csvContext = exception.getCsvContext();
String header = headers[csvContext.getColumnNumber() - 1];
Integer rowNum = csvContext.getRowNumber() - 1;
throw new UploadException(error, header, "of Record No. ", rowNum.toString());
}
private void createDataException(String error, String[] headers, SuperCsvException exception) {
CsvContext csvContext = exception.getCsvContext();
Integer rowNum = csvContext.getRowNumber() - 1;
throw new UploadException(error, headers.toString(), "in Record No. ", rowNum.toString(), csvContext.getRowSource().toString());
}
}