io.bdeploy.common.cli.data.DataTableBase Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of api Show documentation
Public API including dependencies, ready to be used for integrations and plugins.
The newest version!
package io.bdeploy.common.cli.data;

import java.io.PrintStream;
import java.util.ArrayList;
import java.util.List;

/**
 * Base class for {@link DataTable} implementations.
 * 
 * NOTE: Implementations of this class do not guarantee to be thread save!
 */
abstract class DataTableBase implements DataTable {

    protected final PrintStream output;
    protected final List columns = new ArrayList<>();
    protected final List> rows = new ArrayList<>();
    protected final List footers = new ArrayList<>();
    protected String caption;

    private ExitCode exitCode = ExitCode.OK;

    /**
     * @param output The output to render to.
     */
    protected DataTableBase(PrintStream output) {
        this.output = output;
    }

    /**
     * Renders the table to the predefined {@link #output}.
     */
    @Override
    public final void render() {
        if (columns.isEmpty()) {
            throw new IllegalStateException("No columns found. Ensure the table contains at least one column.");
        }
        int columnCount = columns.size();
        for (int i = 0; i < rows.size(); i++) {
            List cells = rows.get(i);
            int sum = 0;
            boolean infiniteSpanCellFound = false;
            for (DataTableCell cell : cells) {
                if (infiniteSpanCellFound) {
                    throw new IllegalStateException("No cells may be added after a cell with an infinite span.");
                }
                int span = cell.getSpan();
                if (span >= 1) {
                    sum += span;
                } else {
                    cell.setSpan(columnCount - sum);
                    sum++;
                    infiniteSpanCellFound = true;
                }
            }
            if (infiniteSpanCellFound) {
                if (sum > columnCount) {
                    throw new IllegalStateException("There are more cells in row " + (i + 1) + " than there are columns.");
                }
            } else if (sum > columnCount) {
                throw new IllegalStateException("There are more cells in row " + (i + 1) + " than there are columns.");
            } else if (sum < columnCount) {
                throw new IllegalStateException("There are less cells in row " + (i + 1) + " than there are columns.");
            }
        }
        doRender();
    }

    @Override
    public ExitCode getExitCode() {
        return this.exitCode;
    }

    @Override
    public DataTable setExitCode(ExitCode exitCode) {
        this.exitCode = exitCode;
        return this;
    }

    @Override
    public DataTable setCaption(String caption) {
        this.caption = caption;
        return this;
    }

    @Override
    public DataTable column(DataTableColumn column) {
        columns.add(column);
        return this;
    }

    @Override
    public DataTable row(List list) {
        rows.add(list);
        return this;
    }

    @Override
    public DataTableRowBuilder row() {
        return new DataTableRowBuilder(this);
    }

    @Override
    public DataTable addFooter(String footer) {
        if (footer != null) {
            footers.add(footer);
        }
        return this;
    }

    abstract void doRender();
}