com.github.jlangch.venice.util.excel.ExcelColumnBuilder Maven / Gradle / Ivy

Go to download
/*   __    __         _
 *   \ \  / /__ _ __ (_) ___ ___
 *    \ \/ / _ \ '_ \| |/ __/ _ \
 *     \  /  __/ | | | | (_|  __/
 *      \/ \___|_| |_|_|\___\___|
 *
 *
 * Copyright 2017-2024 Venice
 *
 * 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.jlangch.venice.util.excel;

import java.util.List;
import java.util.function.Function;

import com.github.jlangch.venice.impl.util.excel.ExcelColumnDef;


public class ExcelColumnBuilder {

    public ExcelColumnBuilder(
            final ExcelSheetWriter excelSheetBuilder,
            final List> columnDefs,
            final String name
    ) {
        this.parentBuilder = excelSheetBuilder;
        this.columnDefs = columnDefs;
        this.name = name;
    }


    public ExcelColumnBuilder id(final String id) {
        this.id = id;
        return this;
    }

    public ExcelColumnBuilder colMapper(final Function mapper) {
        this.mapper = mapper;
        return this;
    }

    public ExcelColumnBuilder colMapper(final String fieldName) {
        this.mapper = e -> ((DataRecord)e).get(fieldName);
        return this;
    }

    public ExcelColumnBuilder widthInPoints(final int width) {
        this.width = width;
        return this;
    }

    public ExcelColumnBuilder skip(final boolean skip) {
        this.skipped = skip;
        return this;
    }

    public ExcelColumnBuilder headerStyle(final String style) {
        this.headerStyle = style;
        return this;
    }

    public ExcelColumnBuilder bodyStyle(final String style) {
        this.bodyStyle = style;
        return this;
    }

    public ExcelColumnBuilder footerStyle(final String style) {
        this.footerStyle = style;
        return this;
    }

    public ExcelColumnBuilder footerTextValue(final String text) {
        this.footerValue = text;
        footerType = ExcelColumnDef.FooterType.TEXT;
        return this;
    }

    public ExcelColumnBuilder footerNumberValue(final Number number) {
        this.footerValue = number;
        footerType = ExcelColumnDef.FooterType.NUMBER;
        return this;
    }

    public ExcelColumnBuilder footerMin() {
        this.footerValue = null;
        footerType = ExcelColumnDef.FooterType.MIN;
        return this;
    }

    public ExcelColumnBuilder footerMax() {
        this.footerValue = null;
        footerType = ExcelColumnDef.FooterType.MAX;
        return this;
    }

    public ExcelColumnBuilder footerAverage() {
        this.footerValue = null;
        footerType = ExcelColumnDef.FooterType.AVERAGE;
        return this;
    }

    public ExcelColumnBuilder footerSum() {
        this.footerValue = null;
        footerType = ExcelColumnDef.FooterType.SUM;
        return this;
    }

    public ExcelSheetWriter end() {
        if (!skipped) {
            columnDefs.add(
                new ExcelColumnDef(
                        id, name, mapper, width,
                        headerStyle, bodyStyle, footerStyle,
                        footerValue, footerType));
        }
        return parentBuilder;
    }


    private final ExcelSheetWriter parentBuilder;
    private final List> columnDefs;
    private Function mapper;
    private String name;
    private String id;
    private Integer width;
    private String headerStyle;
    private String bodyStyle;
    private String footerStyle;
    private Object footerValue;
    private boolean skipped = false;
    private ExcelColumnDef.FooterType footerType = ExcelColumnDef.FooterType.NONE;
}