/*
* 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.kie.workbench.common.services.refactoring.backend.server.drl;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import java.util.HashMap;
import java.util.List;
import javax.enterprise.context.ApplicationScoped;
import org.drools.compiler.compiler.DrlParser;
import org.drools.compiler.compiler.DroolsError;
import org.drools.compiler.lang.descr.PackageDescr;
import org.drools.workbench.models.commons.backend.oracle.ProjectDataModelOracleImpl;
import org.drools.workbench.models.datamodel.oracle.DataType;
import org.drools.workbench.models.datamodel.oracle.FieldAccessorsAndMutators;
import org.drools.workbench.models.datamodel.oracle.ModelField;
import org.drools.workbench.models.datamodel.oracle.ProjectDataModelOracle;
import org.guvnor.common.services.project.model.Package;
import org.guvnor.common.services.project.model.Project;
import org.kie.workbench.common.services.refactoring.backend.server.TestIndexer;
import org.kie.workbench.common.services.refactoring.backend.server.indexing.DefaultIndexBuilder;
import org.kie.workbench.common.services.refactoring.backend.server.indexing.ErrorMessageUtilities;
import org.kie.workbench.common.services.refactoring.backend.server.indexing.PackageDescrIndexVisitor;
import org.kie.workbench.common.services.refactoring.backend.server.indexing.drools.AbstractDrlFileIndexer;
import org.kie.workbench.common.services.refactoring.backend.server.util.KObjectUtil;
import org.kie.workbench.common.services.shared.project.KieProjectService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.uberfire.backend.server.util.Paths;
import org.uberfire.ext.metadata.model.KObject;
import org.uberfire.io.IOService;
import org.uberfire.java.nio.file.Path;
@ApplicationScoped
public class TestPackageNameDrlFileIndexer
extends AbstractDrlFileIndexer
implements TestIndexer<TestDrlFileTypeDefinition> {
private static final Logger logger = LoggerFactory.getLogger( TestPackageNameDrlFileIndexer.class );
private TestDrlFileTypeDefinition type;
@Override
public void setIOService( final IOService ioService ) {
this.ioService = ioService;
}
@Override
public void setProjectService( final KieProjectService projectService ) {
this.projectService = projectService;
}
@Override
public void setResourceTypeDefinition( final TestDrlFileTypeDefinition type ) {
this.type = type;
}
@Override
public boolean supportsPath( final Path path ) {
return type.accept( Paths.convert( path ) );
}
@Override
public KObject toKObject( final Path path ) {
KObject index = null;
try {
final String drl = ioService.readAllString( path );
final DrlParser drlParser = new DrlParser();
final PackageDescr packageDescr = drlParser.parse( true,
drl );
if ( drlParser.hasErrors() ) {
final List<DroolsError> errors = drlParser.getErrors();
logger.warn( ErrorMessageUtilities.makeErrorMessage( path,
errors.toArray( new DroolsError[ errors.size() ] ) ) );
return index;
}
if ( packageDescr == null ) {
logger.warn( ErrorMessageUtilities.makeErrorMessage( path ) );
return index;
}
final ProjectDataModelOracle dmo = getProjectDataModelOracle( path );
final Project project = projectService.resolveProject( Paths.convert( path ) );
// This is the reason we're overriding toKObject and not using fillDrlIndextBuilder(..)
final Package pkg = mock( Package.class );
when( pkg.getPackageName() ).thenReturn( packageDescr.getName() );
final DefaultIndexBuilder builder = new DefaultIndexBuilder( Paths.convert(path).getFileName(),
project,
pkg );
final PackageDescrIndexVisitor visitor = new PackageDescrIndexVisitor( dmo,
builder,
packageDescr );
visitor.visit();
addReferencedResourcesToIndexBuilder(builder, visitor);
index = KObjectUtil.toKObject( path,
builder.build() );
} catch ( Exception e ) {
logger.error( "Unable to index '" + path.toUri().toString() + "'.",
e );
}
return index;
}
@Override
protected ProjectDataModelOracle getProjectDataModelOracle( Path path ) {
final ProjectDataModelOracle dmo = new ProjectDataModelOracleImpl();
dmo.addProjectModelFields( new HashMap<String, ModelField[]>() {{
put( "org.kie.workbench.common.services.refactoring.backend.server.drl.classes.Applicant",
new ModelField[]{ new ModelField( "age",
"java.lang.Integer",
ModelField.FIELD_CLASS_TYPE.REGULAR_CLASS,
ModelField.FIELD_ORIGIN.DECLARED,
FieldAccessorsAndMutators.ACCESSOR,
DataType.TYPE_NUMERIC_INTEGER ) } );
put( "org.kie.workbench.common.services.refactoring.backend.server.drl.classes.Mortgage",
new ModelField[]{ new ModelField( "amount",
"java.lang.Integer",
ModelField.FIELD_CLASS_TYPE.REGULAR_CLASS,
ModelField.FIELD_ORIGIN.DECLARED,
FieldAccessorsAndMutators.ACCESSOR,
DataType.TYPE_NUMERIC_INTEGER ) } );
put( "org.kie.workbench.common.services.refactoring.backend.server.drl.classes.Mortgage",
new ModelField[]{ new ModelField( "applicant",
"org.kie.workbench.common.services.refactoring.backend.server.drl.classes.Applicant",
ModelField.FIELD_CLASS_TYPE.REGULAR_CLASS,
ModelField.FIELD_ORIGIN.DECLARED,
FieldAccessorsAndMutators.ACCESSOR,
"org.kie.workbench.common.services.refactoring.backend.server.drl.classes.Applicant" ) } );
put( "org.kie.workbench.common.services.refactoring.backend.server.drl.classes.Bank",
new ModelField[]{ new ModelField( "mortgage",
"org.kie.workbench.common.services.refactoring.backend.server.drl.classes.Mortgage",
ModelField.FIELD_CLASS_TYPE.REGULAR_CLASS,
ModelField.FIELD_ORIGIN.DECLARED,
FieldAccessorsAndMutators.ACCESSOR,
"org.kie.workbench.common.services.refactoring.backend.server.drl.classes.Mortgage" ) } );
}} );
return dmo;
}
/* (non-Javadoc)
* @see org.kie.workbench.common.services.refactoring.backend.server.indexing.AbstractFileIndexer#fillIndexBuilder(org.uberfire.java.nio.file.Path)
*/
@Override
protected DefaultIndexBuilder fillIndexBuilder(Path path) throws Exception {
// not used here because we're also overriding toKObject(Path)
return null;
}
}