com.factset.protobuf.stach.extensions.v2.StachUtilities Maven / Gradle / Ivy

Go to download
package com.factset.protobuf.stach.extensions.v2;

import com.factset.protobuf.stach.extensions.models.RowSpanSpread;
import com.factset.protobuf.stach.v2.table.RowDefinitionProto;
import com.google.protobuf.InvalidProtocolBufferException;
import com.google.protobuf.Value;
import com.google.protobuf.util.JsonFormat;

import java.util.ArrayList;
import java.util.List;

public class StachUtilities {

    /**
     * @deprecated
     * Returns the respective data type object from the Value object
     *
     * @param value : protobuf Value object input
     * @return returns the respective data type object from the input
     * @throws InvalidProtocolBufferException 
     */
    public static Object valueToObject(Value value) throws InvalidProtocolBufferException {

        return valueToString(value);
    }

    /**
     * Returns the string format of the Value object
     *
     * @param value : protobuf Value object input
     * @return returns the string format of the given Value object
     * @throws InvalidProtocolBufferException
     */
    public static String valueToString(Value value) throws InvalidProtocolBufferException {
        switch (value.getKindCase()) {
            case NULL_VALUE:
                return null;
            case NUMBER_VALUE:
                return "" + value.getNumberValue();
            case STRING_VALUE:
                return value.getStringValue();
            case BOOL_VALUE:
                return Boolean.toString(value.getBoolValue());
            case STRUCT_VALUE:
                return JsonFormat.printer().omittingInsignificantWhitespace().print(value.getStructValue());
            case LIST_VALUE:
                return JsonFormat.printer().omittingInsignificantWhitespace().print(value.getListValue());
            default:
                throw new IllegalArgumentException(String.format("Unsupported protobuf value %s", value));
        }
    }

    /**
     * Returns the index of the element with given id in the list of RowDefinition objects
     *
     * @param rowDefinitionsList : List of RowDefinition objects.
     * @param id   : The id of the RowDefinition object.
     * @return return the index of the object with the given id.
     */
    public static int getIndexOf(List rowDefinitionsList, String id) {
        int pos = 0;
        for (RowDefinitionProto.RowDefinition rowDefinition : rowDefinitionsList) {
            if (id.equalsIgnoreCase(rowDefinition.getId()))
                return pos;
            pos++;
        }
        return -1;
    }

    public static List checkAddRowSpannedItem(int position, int rowIndex, List rowSpanSpreadList) {

        final int pos = position;
        List spannedHeaderList = new ArrayList<>();
        if (rowSpanSpreadList.size() == 0) {
            return null;
        }

        RowSpanSpread rowSpanSpreadOption = filterRowSpanItems(rowSpanSpreadList, position, rowIndex);

        if (rowSpanSpreadOption != null) {
            for (int i = 0; i < rowSpanSpreadOption.getColspan(); i++) {
                spannedHeaderList.add(rowSpanSpreadOption.getValue());
                position++;
            }

            List recursiveSpannedHeaderList = StachUtilities.checkAddRowSpannedItem(position, rowIndex, rowSpanSpreadList);

            if (recursiveSpannedHeaderList != null) {
                spannedHeaderList.addAll(recursiveSpannedHeaderList);
            }
            return spannedHeaderList;
        }
        return null;
    }

    private static RowSpanSpread filterRowSpanItems(List rowSpanSpreadList, int position, int rowIndex){
        for (RowSpanSpread rowSpanSpread: rowSpanSpreadList) {
            if(rowSpanSpread.getPosition() == position && rowIndex < rowSpanSpread.getRowspan()){
                return rowSpanSpread;
            }
        }
        return null;
    }
}