• Home
• com.aventstack
• extentreports
• 5.0.3
• source code
• Table.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.aventstack.extentreports.markuputils.Table Maven / Gradle / Ivy

package com.aventstack.extentreports.markuputils;

import java.lang.reflect.Array;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import com.aventstack.extentreports.annotations.MarkupIgnore;
import com.aventstack.extentreports.markuputils.util.ReflectionUtil;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Getter;
import lombok.ToString;

@Getter
@Builder
@AllArgsConstructor
@ToString
class Table implements Markup {
    private static final long serialVersionUID = 7780527162407673294L;

    private String[][] data;
    private Object object;
    @Builder.Default
    private String cssClass = "";
    private String[] cssClasses;

    @Override
    public String getMarkup() {
        if (cssClasses != null && cssClasses.length > 0)
            Arrays.asList(cssClasses).forEach(x -> cssClass += " " + x);
        if (data != null)
            return fromData();
        try {
            return fromObject();
        } catch (IllegalArgumentException | IllegalAccessException e) {
            e.printStackTrace();
            return "";
        }
    }

    private String fromData() {
        StringBuilder sb = new StringBuilder();
        sb.append("");
        for (int row = 0; row < data.length; row++) {
            sb.append("");
            for (int col = 0; col < data[row].length; col++) {
                sb.append("");
            }
            sb.append("");
        }
        sb.append("
" + data[row][col] + "
");
        return sb.toString();
    }

    private String fromObject() throws IllegalArgumentException, IllegalAccessException {
        if (object == null)
            return "";
        final List fieldList = ReflectionUtil.getFieldsIgnoringAnnotation(object.getClass(), MarkupIgnore.class);
        final StringBuilder sb = new StringBuilder();
        final StringBuilder sb2 = new StringBuilder();
        sb.append("");
        sb.append("");
        final List list = Stream.of("").collect(Collectors.toList());
        final List columnRowCount = Stream.of(0).collect(Collectors.toList());
        if (object instanceof Map) {
            appendMapItems(object, list, sb2, columnRowCount);
        } else if (object instanceof List || object instanceof Set) {
            appendListItems(object, list, sb2, columnRowCount);
        } else if (object instanceof String) {
            list.set(0, "");
        } else if (object.getClass().isArray()) {
            appendArrayItems(object, list, sb2, columnRowCount);
        } else if (!fieldList.isEmpty()) {
            sb.append("");
            fieldList.forEach(x -> sb.append(""));
            sb.append("");
            for (Field f : fieldList) {
                if (f != null && !f.isAccessible()) {
                    f.setAccessible(true);
                }
                if (f.getType().isAssignableFrom(Map.class)) {
                    appendMapItems(f.get(object), list, sb2, columnRowCount);
                } else if (f.getType().isAssignableFrom(List.class) || f.getType().isAssignableFrom(Set.class)) {
                    appendListItems(f.get(object), list, sb2, columnRowCount);
                } else if (f.getType().isArray()) {
                    appendArrayItems(f.get(object), list, sb2, columnRowCount);
                } else {
                    list.set(0, list.get(0) + "");
                    columnRowCount.set(columnRowCount.size() - 1, 1);
                }
                columnRowCount.add(0);
            }
        } else {
            list.set(0, "");
        }
        list.forEach(x -> sb.append("" + x + ""));
        sb.append("");
        sb.append("
" + object.toString() + "
" + x.getName() + "
" + f.get(object).toString() + "
" + object.toString() + "
");
        return sb.toString();
    }

    @SuppressWarnings("unchecked")
    private void appendMapItems(Object object, List list, StringBuilder sb, List columnRowCount) {
        Map map = (Map) object;
        for (int ix = map.size(); ix >= list.size(); ix--)
            list.add("");
        columnRowCount.set(columnRowCount.size() - 1, map.size());
        int row = 0;
        for (Map.Entry entry : map.entrySet()) {
            sb.setLength(0);
            if (columnRowCount.size() > 1 && row + 1 > columnRowCount.get(columnRowCount.size() - 2))
                sb.append("");
            sb.append("" + entry.getKey() + ":" + entry.getValue() + "");
            list.set(row, list.get(row) + sb.toString());
            row++;
        }
    }

    @SuppressWarnings("unchecked")
    private void appendListItems(Object object, List list, StringBuilder sb, List columnRowCount) {
        int row = 0;
        Collection