com.spectralogic.ds3contractcomparator.print.htmlprinter.generators.row.HtmlRowGenerator.java Source code

Introduction

Here is the source code for com.spectralogic.ds3contractcomparator.print.htmlprinter.generators.row.HtmlRowGenerator.java

Source

/*
 * ******************************************************************************
 *   Copyright 2016-2017 Spectra Logic Corporation. All Rights Reserved.
 *   Licensed under the Apache License, Version 2.0 (the "License"). You may not use
 *   this file except in compliance with the License. A copy of the License is located at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 *   or in the "license" file accompanying this file.
 *   This file 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.spectralogic.ds3contractcomparator.print.htmlprinter.generators.row;

import com.google.common.collect.ImmutableList;
import com.spectralogic.ds3contractcomparator.print.htmlprinter.models.RowAttributes;
import com.spectralogic.ds3contractcomparator.print.htmlprinter.models.body.rows.NoChangeRow;
import com.spectralogic.ds3contractcomparator.print.htmlprinter.models.body.rows.Row;

import java.lang.reflect.Field;
import java.util.Optional;
import java.util.function.Function;

import static com.spectralogic.ds3contractcomparator.print.utils.HtmlRowGeneratorUtils.getListPropertyFromObject;
import static com.spectralogic.ds3contractcomparator.print.utils.HtmlRowGeneratorUtils.getPropertyValue;
import static com.spectralogic.ds3contractcomparator.print.utils.HtmlRowGeneratorUtils.toFieldIndent;

public class HtmlRowGenerator {

    private HtmlRowGenerator() {
        //pass
    }

    /**
     * Recursively traverse an object using reflection and constructs the rows that
     * represent the object using the specified function.
     */
    public static <T> ImmutableList<Row> createRows(final T object, final int indent,
            Function<RowAttributes, Row> function) {
        final Field[] fields = object.getClass().getDeclaredFields();
        final ImmutableList.Builder<Row> builder = ImmutableList.builder();

        for (final Field field : fields) {
            final String property = field.getName();
            final Optional<String> value = getPropertyValue(object, property);
            final int fieldIndent = toFieldIndent(indent, object, field);
            if (value.isPresent()) {
                if (field.getType() == ImmutableList.class) {
                    builder.add(new NoChangeRow(fieldIndent, property, ""));

                    final ImmutableList<?> objList = getListPropertyFromObject(field, object);
                    objList.forEach(obj -> builder.addAll(createRows(obj, fieldIndent + 1, function)));
                } else {
                    final RowAttributes attributes = new RowAttributes(fieldIndent, property, value.get());
                    builder.add(function.apply(attributes));
                }
            }
        }
        return builder.build();
    }
}