001package org.apache.maven.plugins.enforcer;
002
003/*
004 * Licensed to the Apache Software Foundation (ASF) under one
005 * or more contributor license agreements.  See the NOTICE file
006 * distributed with this work for additional information
007 * regarding copyright ownership.  The ASF licenses this file
008 * to you under the Apache License, Version 2.0 (the
009 * "License"); you may not use this file except in compliance
010 * with the License.  You may obtain a copy of the License at
011 *
012 *  http://www.apache.org/licenses/LICENSE-2.0
013 *
014 * Unless required by applicable law or agreed to in writing,
015 * software distributed under the License is distributed on an
016 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
017 * KIND, either express or implied.  See the License for the
018 * specific language governing permissions and limitations
019 * under the License.
020 */
021
022import java.util.ArrayList;
023import java.util.Collection;
024import java.util.HashSet;
025import java.util.LinkedHashMap;
026import java.util.List;
027import java.util.Map;
028import java.util.Set;
029import java.util.regex.Pattern;
030
031import org.apache.maven.artifact.Artifact;
032import org.apache.maven.enforcer.rule.api.EnforcerRuleException;
033import org.apache.maven.enforcer.rule.api.EnforcerRuleHelper;
034import org.apache.maven.project.MavenProject;
035import org.codehaus.plexus.component.configurator.expression.ExpressionEvaluationException;
036
037/**
038 * 
039 * @author Robert Scholte
040 * @since 1.3
041 */
042public class RequireSameVersions
043    extends AbstractNonCacheableEnforcerRule
044{
045    private boolean uniqueVersions;
046
047    private Set<String> dependencies = new HashSet<String>();
048
049    private Set<String> plugins = new HashSet<String>();
050
051    private Set<String> buildPlugins = new HashSet<String>();
052
053    private Set<String> reportPlugins = new HashSet<String>();
054
055    public void execute( EnforcerRuleHelper helper )
056        throws EnforcerRuleException
057    {
058        // get the project
059        MavenProject project = null;
060        try
061        {
062            project = (MavenProject) helper.evaluate( "${project}" );
063        }
064        catch ( ExpressionEvaluationException eee )
065        {
066            throw new EnforcerRuleException( "Unable to retrieve the MavenProject: ", eee );
067        }
068
069        // consider including profile based artifacts
070        Map<String, List<String>> versionMembers = new LinkedHashMap<String, List<String>>();
071
072        Set<String> buildPluginSet = new HashSet<String>( buildPlugins );
073        buildPluginSet.addAll( plugins );
074        Set<String> reportPluginSet = new HashSet<String>( reportPlugins );
075        reportPluginSet.addAll( plugins );
076
077        versionMembers.putAll( collectVersionMembers( project.getArtifacts(), dependencies, " (dependency)" ) );
078        versionMembers.putAll( collectVersionMembers( project.getPluginArtifacts(), buildPlugins, " (buildPlugin)" ) );
079        versionMembers.putAll( collectVersionMembers( project.getReportArtifacts(), reportPlugins, " (reportPlugin)" ) );
080
081        if ( versionMembers.size() > 1 )
082        {
083            StringBuilder builder = new StringBuilder( "Found entries with different versions\n" );
084            for ( Map.Entry<String, List<String>> entry : versionMembers.entrySet() )
085            {
086                builder.append( "Entries with version " ).append( entry.getKey() ).append( '\n' );
087                for ( String conflictId : entry.getValue() )
088                {
089                    builder.append( "- " ).append( conflictId ).append( '\n' );
090                }
091            }
092            throw new EnforcerRuleException( builder.toString() );
093        }
094    }
095
096    private Map<String, List<String>> collectVersionMembers( Set<Artifact> artifacts, Collection<String> patterns,
097                                                             String source )
098    {
099        Map<String, List<String>> versionMembers = new LinkedHashMap<String, List<String>>();
100        
101        List<Pattern> regExs = new ArrayList<Pattern>();
102        for ( String pattern : patterns )
103        {
104            String regex = pattern.replace( ".", "\\." ).replace( "*", ".*" ).replace( ":", "\\:" ).replace( '?', '.' );
105
106            // pattern is groupId[:artifactId[:type[:classifier]]]
107            regExs.add( Pattern.compile( regex  + "(\\:.+)?" ) );
108        }
109        
110        for ( Artifact artifact : artifacts )
111        {
112            for ( Pattern regEx: regExs )
113            {
114                if ( regEx.matcher( artifact.getDependencyConflictId() ).matches() )
115                {
116                    String version = uniqueVersions ? artifact.getVersion() : artifact.getBaseVersion();
117                    if ( !versionMembers.containsKey( version ) )
118                    {
119                        versionMembers.put( version, new ArrayList<String>() );
120                    }
121                    versionMembers.get( version ).add( artifact.getDependencyConflictId() + source );
122                }
123            }
124        }
125        return versionMembers;
126    }
127
128}