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

Introduction

Here is the source code for org.sourcepit.b2.internal.generator.BootPomSerializer.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 java.io.File;
import java.io.IOException;
import java.util.Map;

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

import org.apache.maven.model.Model;
import org.apache.maven.model.io.DefaultModelWriter;
import org.apache.maven.plugin.LegacySupport;
import org.apache.maven.project.MavenProject;
import org.sourcepit.b2.execution.IB2Listener;
import org.sourcepit.b2.files.ModuleDirectory;
import org.sourcepit.b2.model.interpolation.layout.IInterpolationLayout;
import org.sourcepit.b2.model.module.AbstractModule;

/**
 * @author Bernd
 */
@Named
public class BootPomSerializer implements IB2Listener {
    @Inject
    private LegacySupport legacySupport;

    @Inject
    private Map<String, IInterpolationLayout> layoutMap;

    @Inject
    private ModulePomBuilder modulePomBuilder;

    @Override
    public void startGeneration(ModuleDirectory moduleDirectory, AbstractModule module) {
        final MavenProject currentProject = legacySupport.getSession().getCurrentProject();
        if (module.getDirectory().equals(currentProject.getBasedir())) {
            try {
                persistB2BootPom(module, currentProject);
                persistModulePomTemplate(module, currentProject);
            } catch (IOException e) {
                throw new IllegalStateException(e);
            }
        }
    }

    private File persistB2BootPom(AbstractModule module, final MavenProject currentProject) throws IOException {
        final File pomFile = createFile(module, "boot-pom.xml");
        final Model model = currentProject.getOriginalModel().clone();
        writeMavenModel(model, pomFile);
        module.setAnnotationData("b2", "bootPom", pomFile.getAbsolutePath());
        return pomFile;
    }

    private void persistModulePomTemplate(AbstractModule module, final MavenProject currentProject)
            throws IOException {
        final Model pomTemplate = modulePomBuilder.buildModulePom(currentProject);
        final File pomFile = createFile(module, "module-pom-template.xml");
        writeMavenModel(pomTemplate, pomFile);
        module.setAnnotationData("maven", "modulePomTemplate", pomFile.getAbsolutePath());
    }

    private File createFile(AbstractModule module, String fileName) throws IOException {
        final IInterpolationLayout layout = layoutMap.get(module.getLayoutId());
        final File pomFile = new File(layout.pathOfMetaDataFile(module, fileName));
        if (!pomFile.exists()) {
            pomFile.getParentFile().mkdirs();
            pomFile.createNewFile();
        }
        return pomFile;
    }

    private void writeMavenModel(final Model model, final File pomFile) throws IOException {
        new DefaultModelWriter().write(pomFile, null, model);
    }
}