com.github.camaral.sheeco.processor.PayloadFiller.java Source code

Introduction

Here is the source code for com.github.camaral.sheeco.processor.PayloadFiller.java

Source

/*
 * Copyright 2011-2015 the original author or authors.
 *
 * 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 com.github.camaral.sheeco.processor;

import java.util.ArrayList;
import java.util.List;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.FormulaEvaluator;
import org.apache.poi.ss.usermodel.Row;

import com.github.camaral.sheeco.exceptions.SpreadsheetViolation;
import com.github.camaral.sheeco.type.adapter.InvalidCellFormatException;
import com.github.camaral.sheeco.type.adapter.InvalidCellValueException;

/**
 * @author caio.amaral
 *
 */
public class PayloadFiller {

    /**
     * Reads the spreadsheet row and populate the current Object's native fields
     */
    public static void fillAttributes(final Object payload, final Row row, final PayloadContext<?> ctx) {
        final List<Attribute> attributes = ctx.getPayload().getAttributes();

        final List<SpreadsheetViolation> violations = fillAttributes(payload, row, attributes, ctx.getEvaluator());

        ctx.getViolations().addAll(violations);
    }

    public static <T> List<SpreadsheetViolation> fillAttributes(final T payload, final Row row,
            final List<Attribute> attributes, final FormulaEvaluator evaluator) {
        final List<SpreadsheetViolation> violations = new ArrayList<>();

        for (final Attribute attr : attributes) {
            final Cell cell = row.getCell(attr.getColumnIndex(), Row.CREATE_NULL_AS_BLANK);
            evaluator.evaluateInCell(cell);

            if (cell.getCellType() == Cell.CELL_TYPE_ERROR) {
                violations.add(new SpreadsheetViolation("serializer.spreadsheet.cell.error", cell));
                continue;
            }

            try {
                attr.setValue(payload, cell);
            } catch (final InvalidCellValueException e) {
                violations.add(new SpreadsheetViolation("serializer.spreadsheet.cell.invalid", cell));
                continue;
            } catch (final InvalidCellFormatException e) {
                violations.add(new SpreadsheetViolation("serializer.spreadsheet.cell.type.invalid", cell));
                continue;
            }

        }

        return violations;
    }

    public static void fillElements(final Object payload, final Row row, final PayloadContext<?> ctx) {
        final List<Element> elements = ctx.getPayload().getElements();

        final List<SpreadsheetViolation> violations = fillElements(payload, row, elements, ctx.getEvaluator());

        ctx.getViolations().addAll(violations);
    }

    public static <T> List<SpreadsheetViolation> fillElements(final T payload, final Row row,
            final List<Element> elements, final FormulaEvaluator evaluator) {

        final List<SpreadsheetViolation> violations = new ArrayList<>();

        for (final Element element : elements) {
            Object elemPayload = element.getPayload().newInstance();

            final List<SpreadsheetViolation> stepViolations = fillAttributes(elemPayload, row,
                    element.getPayload().getAttributes(), evaluator);

            violations.addAll(stepViolations);

            element.setValue(payload, elemPayload);

            fillElements(elemPayload, row, element.getPayload().getElements(), evaluator);
        }

        return violations;
    }
}