org.broadleafcommerce.common.util.TableCreator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of broadleaf-common Show documentation
A collection of classes shared by broadleaf profile, cms, admin, and core.
There is a newer version: 3.1.15-GA
/*
 * Copyright 2008-2013 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 org.broadleafcommerce.common.util;

import org.apache.commons.lang3.StringUtils;


public class TableCreator {
    
    protected Col[] cols;
    protected int rowWidth;
    protected StringBuilder sb;
    protected int globalRowHeaderWidth = 20;
    
    public TableCreator(Col[] cols) {
        this.cols = cols;
        this.rowWidth = calculateRowWidth();
        this.sb = new StringBuilder("\r\n");
        
        addSeparator();
        addRow(cols);
        addSeparator();
    }
    
    protected int calculateRowWidth() {
        int length = 1;
        for (Col col : cols) {
            length += col.width + 3;
        }
        return length;
    }
    
    public TableCreator addSeparator() {
        sb.append(StringUtils.leftPad("", rowWidth, '-')).append("\r\n");
        return this;
    }

    public TableCreator addRow(Col[] cols) {
        String[] data = new String[cols.length];
        for (int i = 0; i < cols.length; i++) {
            data[i] = cols[i].title;
        }
        return addRow(data);
    }
    
    public TableCreator addRow(Object[] data) {
        if (data.length != cols.length) {
            throw new IllegalArgumentException("Wrong number of data elements. Needed [" + cols.length + "] " +
            		"but received [" + data.length + "]");
        }
        
        sb.append('|');
        
        for (int i = 0; i < data.length; i++) {
            String trimmed = StringUtils.left(String.valueOf(data[i]), cols[i].width);
            sb.append(' ').append(StringUtils.rightPad(trimmed, cols[i].width)).append(" |");
        }
        
        sb.append("\r\n");
        return this;
    }

    
    public TableCreator addRow(String rowHeader, Object rowData) {
        String trimmed = StringUtils.left(rowHeader, globalRowHeaderWidth);
        sb.append("| ")
            .append(StringUtils.rightPad(trimmed, globalRowHeaderWidth))
            .append(StringUtils.rightPad(String.valueOf(rowData), rowWidth - globalRowHeaderWidth - 3))
            .append("|\r\n");
        return this;
    }
    
    public TableCreator withGlobalRowHeaderWidth(int width) {
        this.globalRowHeaderWidth = width;
        return this;
    }
    
    public String toString() {
        return sb.toString();
    }
    
    public static class Col {
        
        String title;
        int width;
        
        public Col(String title) {
            this.title = title;
            this.width = title.length();
        }

        public Col(String title, int width) {
            this.title = title;
            this.width = width;
        }

    }

}