package org.jboss.windup.rules.apps.javaee.rules; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import org.jboss.windup.config.AbstractRuleProvider; import org.jboss.windup.config.GraphRewrite; import org.jboss.windup.config.loader.RuleLoaderContext; import org.jboss.windup.config.metadata.RuleMetadata; import org.jboss.windup.config.operation.GraphOperation; import org.jboss.windup.config.phase.ReportGenerationPhase; import org.jboss.windup.config.query.Query; import org.jboss.windup.graph.GraphContext; import org.jboss.windup.graph.model.ProjectModel; import org.jboss.windup.graph.model.WindupConfigurationModel; import org.jboss.windup.graph.model.WindupVertexFrame; import org.jboss.windup.graph.model.resource.FileModel; import org.jboss.windup.graph.service.GraphService; import org.jboss.windup.graph.service.WindupConfigurationService; import org.jboss.windup.reporting.model.ApplicationReportModel; import org.jboss.windup.reporting.model.TemplateType; import org.jboss.windup.reporting.model.WindupVertexListModel; import org.jboss.windup.reporting.service.ApplicationReportService; import org.jboss.windup.reporting.service.ReportService; import org.jboss.windup.rules.apps.javaee.model.EjbRemoteServiceModel; import org.jboss.windup.rules.apps.javaee.model.JaxRSWebServiceModel; import org.jboss.windup.rules.apps.javaee.model.JaxWSWebServiceModel; import org.jboss.windup.rules.apps.javaee.model.RMIServiceModel; import org.jboss.windup.rules.apps.javaee.model.RemoteServiceModel; import org.ocpsoft.rewrite.config.Configuration; import org.ocpsoft.rewrite.config.ConfigurationBuilder; import org.ocpsoft.rewrite.context.EvaluationContext; /** * Creates a report of EJB data (eg, a list of EJB session beans). */ @RuleMetadata(phase = ReportGenerationPhase.class, id = "Create Remote Service Report") public class CreateRemoteReportRuleProvider extends AbstractRuleProvider { public static final String TEMPLATE_EJB_REPORT = "/reports/templates/remote.ftl"; public static final String REPORT_DESCRIPTION = "This report displays all remote services references that were found within the application."; @Override public Configuration getConfiguration(RuleLoaderContext ruleLoaderContext) { return ConfigurationBuilder.begin() .addRule() .when(Query.fromType(RemoteServiceModel.class)) .perform(new GraphOperation() { @Override public void perform(GraphRewrite event, EvaluationContext context) { // configuration of current execution WindupConfigurationModel configurationModel = WindupConfigurationService.getConfigurationModel(event.getGraphContext()); for (FileModel inputPath : configurationModel.getInputPaths()) { ProjectModel projectModel = inputPath.getProjectModel(); createReport(event.getGraphContext(), projectModel); } } @Override public String toString() { return "CreateRemoteServiceReport"; } }); } @SuppressWarnings("unchecked") private void createReport(GraphContext context, ProjectModel projectModel) { GraphService<RemoteServiceModel> remoteServices = new GraphService<>(context, RemoteServiceModel.class); List<JaxRSWebServiceModel> jaxRsList = new ArrayList<>(); List<JaxWSWebServiceModel> jaxWsList = new ArrayList<>(); List<EjbRemoteServiceModel> ejbRemoteList = new ArrayList<>(); List<RMIServiceModel> rmiList = new ArrayList<>(); for (RemoteServiceModel remoteServiceModel : remoteServices.findAll()) { if (!remoteServiceModel.isAssociatedWithApplication(projectModel)) continue; if (remoteServiceModel instanceof JaxRSWebServiceModel) { jaxRsList.add((JaxRSWebServiceModel) remoteServiceModel); } else if (remoteServiceModel instanceof JaxWSWebServiceModel) { jaxWsList.add((JaxWSWebServiceModel) remoteServiceModel); } else if (remoteServiceModel instanceof EjbRemoteServiceModel) { ejbRemoteList.add((EjbRemoteServiceModel) remoteServiceModel); } else if (remoteServiceModel instanceof RMIServiceModel) { rmiList.add((RMIServiceModel) remoteServiceModel); } } if (jaxRsList.isEmpty() && jaxWsList.isEmpty() && ejbRemoteList.isEmpty() && rmiList.isEmpty()) return; ApplicationReportService applicationReportService = new ApplicationReportService(context); ApplicationReportModel applicationReportModel = applicationReportService.create(); applicationReportModel.setReportPriority(300); applicationReportModel.setDisplayInApplicationReportIndex(true); applicationReportModel.setReportName("Remote Services"); applicationReportModel.setDescription(REPORT_DESCRIPTION); applicationReportModel.setReportIconClass("glyphicon service-nav-logo"); applicationReportModel.setProjectModel(projectModel); applicationReportModel.setTemplatePath(TEMPLATE_EJB_REPORT); applicationReportModel.setTemplateType(TemplateType.FREEMARKER); GraphService<WindupVertexListModel> listService = new GraphService<>(context, WindupVertexListModel.class); Map<String, WindupVertexFrame> data = new HashMap<>(4); data.put("jaxRsServices", listService.create().addAll(jaxRsList)); data.put("jaxWsServices", listService.create().addAll(jaxWsList)); data.put("ejbRemoteServices", listService.create().addAll(ejbRemoteList)); data.put("rmiServices", listService.create().addAll(rmiList)); applicationReportModel.setRelatedResource(data); ReportService reportService = new ReportService(context); reportService.setUniqueFilename(applicationReportModel, "remotereport_" + projectModel.getName(), "html"); } }