io.lightlink.excel.reader.DataReadingExcelStreamVisitor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lightlink-core Show documentation
There is a newer version: 1.2.4
package io.lightlink.excel.reader;

/*
 * #%L
 * LightLink Core
 * %%
 * Copyright (C) 2015 - 2016 Vitaliy Shevchuk
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser 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 General Lesser Public License for more details.
 * 
 * You should have received a copy of the GNU General Lesser Public
 * License along with this program.  If not, see
 * .
 * #L%
 */

import io.lightlink.excel.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class DataReadingExcelStreamVisitor extends AbstractExcelStreamVisitor {

    public static final Logger LOG = LoggerFactory.getLogger(DataReadingExcelStreamVisitor.class);

    private Map targets;
    private Map data = new HashMap();

    public DataReadingExcelStreamVisitor(Map targets) {
        this.targets = targets;
    }

    protected void handleCell(RowNode rowNode, RowPrintCallback rowPrintCallback, int i, CellNode cell) {
        List cells = rowNode.getCells();
        MappingBean mappingBean;
        String cellDecodedValue = cell.getDecodedValue();

        //is mapping here ?
        MappingBean mappingHere = targets.get(cell.getCoordinates());
        if (mappingHere != null) {
            putData(cellDecodedValue, mappingHere);
        } else {
            // is line mapping on the left ?
            MappingBean mappingLeft = targets.get(left(cell.getCoordinates()));

            if (mappingLeft != null && mappingLeft instanceof LineMappingBean) {
                List value = (List) data.get(mappingLeft.getProperty());
                Object lastEl = value.get(value.size() - 1);
                if (lastEl instanceof List) // is RegionMappingBean ?
                    value = (List) lastEl;
                value.add(cellDecodedValue);
                targets.put(cell.getCoordinates(), mappingLeft);
            } else {
                MappingBean mappingAbove = targets.get(above(cell.getCoordinates()));

                if (mappingAbove != null && mappingAbove instanceof RegionMappingBean) {
                    List value = (List) data.get(mappingAbove.getProperty());
                    List line = new ArrayList();
                    value.add(line);
                    line.add(cellDecodedValue);
                    targets.put(cell.getCoordinates(), mappingAbove);
                }
            }

        }

    }

    private Object putData(String cellDecodedValue, MappingBean mappingHere) {
        String property = mappingHere.getProperty();
        Map map = data;
        int dotPos;
        while ((dotPos = property.indexOf(".")) != -1) {
            Map newMap = new HashMap();
            String containerProp = property.substring(0, dotPos);
            if (map.containsKey(containerProp)) {
                Object container = map.get(containerProp);
                if (container instanceof Map){
                    map = (Map) container;
                } else {
                    throw new IllegalArgumentException("Map expected for '" + container + "' in '" + mappingHere.getProperty()
                            + "'. Found class:" + container.getClass() + " value:" + container);
                }
            } else {
                map.put(containerProp, newMap);
                map = newMap;
            }
            property = property.substring(dotPos + 1);
        }

        return map.put(property, mappingHere.convertData(cellDecodedValue));
    }

    public Map getData() {
        return data;
    }

    private String left(String coordinates) {
        int colNum = ExcelUtils.toExcelColumnNumber(coordinates.replaceAll("[0-9]", ""));
        if (colNum == 0)
            return "";

        return ExcelUtils.toExcelColumnName(colNum - 1) + coordinates.replaceAll("[A-Za-z]", "");
    }

    private String above(String coordinates) {
        int rowNum = Integer.parseInt(coordinates.replaceAll("[A-Za-z]", "")) - 1;
        if (rowNum == 0)
            return "";
        else
            return coordinates.replaceAll("[0-9]", "") + (rowNum);
    }


}