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 }