001 /* 002 * Sonar, open source software quality management tool. 003 * Copyright (C) 2008-2011 SonarSource 004 * mailto:contact AT sonarsource DOT com 005 * 006 * Sonar is free software; you can redistribute it and/or 007 * modify it under the terms of the GNU Lesser General Public 008 * License as published by the Free Software Foundation; either 009 * version 3 of the License, or (at your option) any later version. 010 * 011 * Sonar is distributed in the hope that it will be useful, 012 * but WITHOUT ANY WARRANTY; without even the implied warranty of 013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 014 * Lesser General Public License for more details. 015 * 016 * You should have received a copy of the GNU Lesser General Public 017 * License along with Sonar; if not, write to the Free Software 018 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02 019 */ 020 package org.sonar.plugins.design.batch; 021 022 import org.apache.maven.artifact.Artifact; 023 import org.apache.maven.artifact.factory.ArtifactFactory; 024 import org.apache.maven.artifact.metadata.ArtifactMetadataSource; 025 import org.apache.maven.artifact.repository.ArtifactRepository; 026 import org.apache.maven.artifact.resolver.ArtifactCollector; 027 import org.apache.maven.shared.dependency.tree.DependencyNode; 028 import org.apache.maven.shared.dependency.tree.DependencyTreeBuilder; 029 import org.apache.maven.shared.dependency.tree.DependencyTreeBuilderException; 030 import org.apache.maven.shared.dependency.tree.filter.AncestorOrSelfDependencyNodeFilter; 031 import org.apache.maven.shared.dependency.tree.filter.DependencyNodeFilter; 032 import org.apache.maven.shared.dependency.tree.filter.StateDependencyNodeFilter; 033 import org.apache.maven.shared.dependency.tree.traversal.BuildingDependencyNodeVisitor; 034 import org.apache.maven.shared.dependency.tree.traversal.CollectingDependencyNodeVisitor; 035 import org.apache.maven.shared.dependency.tree.traversal.DependencyNodeVisitor; 036 import org.apache.maven.shared.dependency.tree.traversal.FilteringDependencyNodeVisitor; 037 import org.sonar.api.batch.SupportedEnvironment; 038 import org.sonar.api.batch.Sensor; 039 import org.sonar.api.batch.SensorContext; 040 import org.sonar.api.batch.SonarIndex; 041 import org.sonar.api.design.Dependency; 042 import org.sonar.api.resources.Library; 043 import org.sonar.api.resources.Project; 044 import org.sonar.api.resources.Resource; 045 import org.sonar.api.utils.SonarException; 046 047 @SupportedEnvironment("maven") 048 public class MavenDependenciesSensor implements Sensor { 049 050 private ArtifactRepository localRepository; 051 private ArtifactFactory artifactFactory; 052 private ArtifactMetadataSource artifactMetadataSource; 053 private ArtifactCollector artifactCollector; 054 private DependencyTreeBuilder treeBuilder; 055 private SonarIndex index; 056 057 public MavenDependenciesSensor(ArtifactRepository localRepository, ArtifactFactory artifactFactory, ArtifactMetadataSource artifactMetadataSource, ArtifactCollector artifactCollector, DependencyTreeBuilder treeBuilder, SonarIndex index) { 058 this.localRepository = localRepository; 059 this.artifactFactory = artifactFactory; 060 this.artifactMetadataSource = artifactMetadataSource; 061 this.artifactCollector = artifactCollector; 062 this.index = index; 063 this.treeBuilder = treeBuilder; 064 } 065 066 public boolean shouldExecuteOnProject(Project project) { 067 return true; 068 } 069 070 public void analyse(final Project project, final SensorContext context) { 071 try { 072 DependencyNode root = treeBuilder.buildDependencyTree(project.getPom(), localRepository, artifactFactory, artifactMetadataSource, null, artifactCollector); 073 074 DependencyNodeVisitor visitor = new BuildingDependencyNodeVisitor(new DependencyNodeVisitor() { 075 public boolean visit(DependencyNode node) { 076 return true; 077 } 078 079 public boolean endVisit(DependencyNode node) { 080 if (node.getParent() != null && node.getParent() != node) { 081 saveDependency(node, context); 082 } 083 return true; 084 } 085 }); 086 087 // mode verbose OFF : do not show the same lib many times 088 DependencyNodeFilter filter = StateDependencyNodeFilter.INCLUDED; 089 090 CollectingDependencyNodeVisitor collectingVisitor = new CollectingDependencyNodeVisitor(); 091 DependencyNodeVisitor firstPassVisitor = new FilteringDependencyNodeVisitor(collectingVisitor, filter); 092 root.accept(firstPassVisitor); 093 094 DependencyNodeFilter secondPassFilter = new AncestorOrSelfDependencyNodeFilter(collectingVisitor.getNodes()); 095 visitor = new FilteringDependencyNodeVisitor(visitor, secondPassFilter); 096 097 root.accept(visitor); 098 099 } catch (DependencyTreeBuilderException e) { 100 throw new SonarException("Can not load the graph of dependencies of the project " + project.getKey(), e); 101 } 102 } 103 104 protected void saveDependency(DependencyNode node, SensorContext context) { 105 Resource from = (node.getParent().getParent() == null) ? index.getProject() : toResource(node.getParent().getArtifact(), context); 106 Resource to = toResource(node.getArtifact(), context); 107 Dependency dependency = new Dependency(from, to); 108 dependency.setUsage(node.getArtifact().getScope()); 109 dependency.setWeight(1); 110 context.saveDependency(dependency); 111 } 112 113 protected static Resource toResource(Artifact artifact, SensorContext context) { 114 Project project = Project.createFromMavenIds(artifact.getGroupId(), artifact.getArtifactId()); 115 Resource result = context.getResource(project); 116 if (result == null || !((Project) result).getAnalysisVersion().equals(artifact.getBaseVersion())) { 117 Library lib = new Library(project.getKey(), artifact.getBaseVersion()); 118 context.saveResource(lib); 119 result = context.getResource(lib); 120 } 121 return result; 122 } 123 124 @Override 125 public String toString() { 126 return "Maven dependencies"; 127 } 128 }