BRLConditionColumnDefinitionBuilder.java

/*
 * Copyright 2016 Red Hat, Inc. and/or its affiliates.
 *
 * 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.drools.workbench.screens.guided.dtable.client.widget.table.popovers.definitions;

import java.util.ArrayList;
import java.util.List;
import javax.enterprise.context.Dependent;
import javax.inject.Inject;

import org.drools.workbench.models.datamodel.oracle.DataType;
import org.drools.workbench.models.guided.dtable.shared.model.BRLConditionColumn;
import org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn;
import org.drools.workbench.models.guided.dtable.shared.model.BaseColumn;
import org.drools.workbench.models.guided.dtable.shared.model.DTCellValue52;
import org.drools.workbench.models.guided.dtable.shared.model.GuidedDecisionTable52;
import org.drools.workbench.screens.guided.dtable.client.widget.table.GuidedDecisionTableView;
import org.drools.workbench.screens.guided.dtable.client.widget.table.utilities.ColumnUtilities;
import org.drools.workbench.screens.guided.dtable.service.GuidedDecisionTableEditorService;
import org.jboss.errai.common.client.api.Caller;
import org.uberfire.client.callbacks.Callback;

@Dependent
public class BRLConditionColumnDefinitionBuilder extends BaseColumnDefinitionBuilder {

    @Inject
    public BRLConditionColumnDefinitionBuilder( final Caller<GuidedDecisionTableEditorService> service ) {
        super( service );
    }

    @Override
    public Class getSupportedColumnType() {
        return BRLConditionVariableColumn.class;
    }

    @Override
    public void generateDefinition( final GuidedDecisionTableView.Presenter dtPresenter,
                                    final BaseColumn column,
                                    final Callback<String> afterGenerationCallback ) {
        if ( !( column instanceof BRLConditionVariableColumn ) ) {
            return;
        }

        final GuidedDecisionTable52 existingModel = dtPresenter.getModel();
        final BRLConditionVariableColumn brlVariableColumn = (BRLConditionVariableColumn) column;
        final BRLConditionColumn brlColumn = existingModel.getBRLColumn( brlVariableColumn );

        final GuidedDecisionTable52 partialModel = new GuidedDecisionTable52();
        final ColumnUtilities columnUtilities = new ColumnUtilities( existingModel,
                                                                     dtPresenter.getDataModelOracle() );
        partialModel.getConditions().add( brlColumn );
        partialModel.getData().add( makeRowData( columnUtilities,
                                                 brlColumn ) );

        generateDefinitionOnServer( partialModel,
                                    dtPresenter.getCurrentPath(),
                                    ( String drl ) -> afterGenerationCallback.callback( getLHS( drl ) ) );
    }

    private List<DTCellValue52> makeRowData( final ColumnUtilities columnUtilities,
                                             final BRLConditionColumn brlColumn ) {
        final List<DTCellValue52> row = new ArrayList<>();
        row.add( new DTCellValue52( 1 ) );
        row.add( new DTCellValue52( "desc" ) );

        for ( BRLConditionVariableColumn brlVariableColumn : brlColumn.getChildColumns() ) {
            final DataType.DataTypes dataType = columnUtilities.getDataType( brlVariableColumn );
            row.add( makeCell( dataType ) );
        }

        return row;
    }

}