org.sourcepit.b2.internal.generator.ModulePomBuilder.java Source code

Introduction

Here is the source code for org.sourcepit.b2.internal.generator.ModulePomBuilder.java

Source

/*
 * Copyright 2014 Bernd Vogt and others.
 * 
 * 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.sourcepit.b2.internal.generator;

import static org.sourcepit.b2.internal.maven.util.MavenDepenenciesUtils.removeDependencies;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import javax.inject.Inject;
import javax.inject.Named;

import org.apache.maven.model.Activation;
import org.apache.maven.model.BuildBase;
import org.apache.maven.model.Dependency;
import org.apache.maven.model.Model;
import org.apache.maven.model.Profile;
import org.apache.maven.model.inheritance.InheritanceAssembler;
import org.apache.maven.model.merge.ModelMerger;
import org.apache.maven.model.normalization.ModelNormalizer;
import org.apache.maven.project.MavenProject;

import com.google.common.base.Predicate;

@Named
public class ModulePomBuilder {
    @Inject
    private ModelNormalizer modelNormalizer;

    @Inject
    private InheritanceAssembler inheritanceAssembler;

    private static class ProfileMerger extends ModelMerger {
        @Override
        public void mergeModel_Profiles(Model target, Model source, boolean sourceDominant,
                Map<Object, Object> context) {
            final Map<String, Profile> targetProfiles = new LinkedHashMap<String, Profile>();
            for (Profile profile : target.getProfiles()) {
                targetProfiles.put(profile.getId(), profile);
            }
            for (Profile srcProfile : source.getProfiles()) {
                final Profile tgtProfile = targetProfiles.get(srcProfile.getId());
                if (tgtProfile == null) {
                    targetProfiles.put(srcProfile.getId(), srcProfile.clone());
                } else {
                    mergeProfile(tgtProfile, srcProfile, sourceDominant, context);
                }
            }
            target.setProfiles(new ArrayList<Profile>(targetProfiles.values()));
        }

        @Override
        protected void mergeProfile(Profile tgtProfile, Profile srcProfile, boolean sourceDominant,
                Map<Object, Object> context) {
            final Activation tgtActivation = tgtProfile.getActivation();
            final Activation srcActivation = srcProfile.getActivation();
            if (tgtActivation == null) {
                tgtProfile.setActivation(srcActivation);
            } else if (srcActivation != null) {
                if (sourceDominant) {
                    tgtProfile.setActivation(srcActivation);
                }
            }

            mergeModelBase(tgtProfile, srcProfile, sourceDominant, context);

            BuildBase tgtBuild = tgtProfile.getBuild();
            BuildBase srcBuild = srcProfile.getBuild();
            if (tgtBuild == null) {
                tgtProfile.setBuild(srcBuild);
            } else if (srcBuild != null) {
                mergeBuildBase(tgtBuild, srcBuild, sourceDominant, context);
            }
        }
    }

    public Model buildModulePom(MavenProject bootProject) {
        final List<Model> modelHierarchy = cloneModelHierarchy(bootProject);
        for (Model model : modelHierarchy) {
            modelNormalizer.mergeDuplicates(model, null, null);
        }
        for (int i = modelHierarchy.size() - 2; i >= 0; i--) {
            Model parent = modelHierarchy.get(i + 1);
            Model child = modelHierarchy.get(i);
            inheritanceAssembler.assembleModelInheritance(child, parent, null, null);

            // Bug #76: Parent profiles are lost while generating Maven poms
            new ProfileMerger().mergeModel_Profiles(child, parent, false, null);
        }

        Model resultModel = modelHierarchy.get(0);
        modelNormalizer.mergeDuplicates(resultModel, null, null);

        resultModel.setParent(null);

        removeDependencies(resultModel, new Predicate<Dependency>() {
            public boolean apply(Dependency input) {
                return "module".equals(input.getType());
            }
        });

        // poms
        return resultModel;
    }

    private List<Model> cloneModelHierarchy(MavenProject project) {
        List<Model> models = new ArrayList<Model>();
        MavenProject currentProject = project;
        while (currentProject != null) {
            models.add(currentProject.getOriginalModel().clone());
            currentProject = currentProject.getParent();
        }
        return models;
    }
}