001package io.konik.csv.processor;
002
003import io.konik.zugferd.unece.codes.DocumentCode;
004import org.slf4j.Logger;
005import org.slf4j.LoggerFactory;
006import org.supercsv.cellprocessor.ift.CellProcessor;
007import org.supercsv.util.CsvContext;
008
009/**
010 * Custom {@link CellProcessor} for {@link io.konik.zugferd.unece.codes.DocumentCode}.
011 */
012public final class DocumentCodeProcessor implements CellProcessor {
013
014        private static final Logger log = LoggerFactory.getLogger(DocumentCodeProcessor.class);
015
016        @Override
017        public Object execute(Object value, CsvContext context) {
018
019                if (value instanceof String) {
020                        String code = (String) value;
021                        try {
022                                return DocumentCode.getByCode(code);
023                        } catch (IllegalArgumentException e) {
024                                log.warn("DocumentCode for value {} does not exist", value);
025                        }
026                }
027
028                if (value instanceof DocumentCode) {
029                        DocumentCode code = (DocumentCode) value;
030                        return code.getCode();
031                }
032
033                return null;
034        }
035}