net.sourceforge.mavenhippo.BeanCreator.java Source code

Introduction

Here is the source code for net.sourceforge.mavenhippo.BeanCreator.java

Source

/*
 *    Copyright 2013 Ebrahim Aharpour
 *
 *   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.
 *   
 *   Partially sponsored by Smile B.V
 */
package net.sourceforge.mavenhippo;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import net.sourceforge.mavenhippo.FileManager.FileManagerException;
import net.sourceforge.mavenhippo.gen.BeanGenerator;
import net.sourceforge.mavenhippo.model.ContentTypeBean;
import net.sourceforge.mavenhippo.model.ContentTypeBean.ContentTypeException;
import net.sourceforge.mavenhippo.model.HippoBeanClass;

import org.apache.commons.lang3.StringUtils;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.logging.Log;

import freemarker.template.TemplateException;

/**
 * @author Ebrahim Aharpour
 * 
 */
public class BeanCreator {

    private final ContentTypeDefinitionFinder contentTypeDefinitionFinder;
    private final BeanGenerator generator;
    private final FileManager fileManager;
    private final File sourceRoot;

    public BeanCreator(BeanGeneratorConfig config, Map<String, HippoBeanClass> beansOnClassPath,
            Map<String, HippoBeanClass> beansInProject, Map<String, String> namespaces,
            ClassLoader projectClassLoader) throws FileManagerException, ContentTypeException {
        contentTypeDefinitionFinder = new ContentTypeDefinitionFinder(config.namespaceFolder,
                config.maximumDepthOfScan, config.log, namespaces);
        this.sourceRoot = config.sourceRoot;
        this.fileManager = new FileManager(sourceRoot, config.log, true);
        Map<String, ContentTypeBean> mixins = getMixins();
        this.generator = new BeanGenerator(beansOnClassPath, beansInProject, config.packageToSearch,
                config.basePackage, namespaces.keySet(), mixins, projectClassLoader);
    }

    public void createBeans() throws MojoExecutionException {

        Map<String, ContentTypeBean> toBeGenerated = getBeansToBeGenerate();
        for (Iterator<Entry<String, ContentTypeBean>> nodeTypeIterator = toBeGenerated.entrySet()
                .iterator(); nodeTypeIterator.hasNext();) {
            createBean(nodeTypeIterator.next().getValue());
        }

    }

    private void createBean(ContentTypeBean contentType) throws MojoExecutionException {
        try {
            String[] packages = generator.getPackage(contentType);
            String className = generator.getClassName(contentType);
            File pack = fileManager.getPackage(packages);
            File classFile = fileManager.getClassFile(pack, className);
            PrintWriter writer = new PrintWriter(classFile);
            writer.print(generator.generateBean(contentType));
            writer.close();
        } catch (ContentTypeException e) {
            throw new MojoExecutionException(e.getLocalizedMessage(), e);
        } catch (TemplateException e) {
            throw new MojoExecutionException(e.getLocalizedMessage(), e);
        } catch (IOException e) {
            throw new MojoExecutionException(e.getLocalizedMessage(), e);
        }
    }

    private Map<String, ContentTypeBean> getBeansToBeGenerate() throws ContentTypeException {
        Map<String, ContentTypeBean> result = new HashMap<String, ContentTypeBean>();
        List<ContentTypeBean> contentTypeBeans = contentTypeDefinitionFinder.getContentTypeBeans();
        for (ContentTypeBean contentTypeBean : contentTypeBeans) {
            if (!contentTypeBean.isMixin()) {
                result.put(contentTypeBean.getFullyQualifiedName(), contentTypeBean);
            }
        }
        return result;
    }

    private Map<String, ContentTypeBean> getMixins() throws ContentTypeException {
        Map<String, ContentTypeBean> result = new HashMap<String, ContentTypeBean>();
        List<ContentTypeBean> contentTypeBeans = contentTypeDefinitionFinder.getContentTypeBeans();
        for (ContentTypeBean contentTypeBean : contentTypeBeans) {
            if (contentTypeBean.isMixin()) {
                result.put(contentTypeBean.getFullyQualifiedName(), contentTypeBean);
            }
        }
        return result;
    }

    public static class BeanGeneratorConfig {
        public BeanGeneratorConfig(Log log, String namespaceLocation, String[] basePackage, String packageToSearch,
                File sourceRoot, int maximumDepthOfScan) {
            if (log == null) {
                throw new IllegalArgumentException("Argument log is required");
            }
            if (StringUtils.isBlank(namespaceLocation)) {
                throw new IllegalArgumentException("Argument namespaceLocation is required");
            } else {
                this.namespaceFolder = new File(namespaceLocation);
                if (!namespaceFolder.exists() || !namespaceFolder.isDirectory()) {
                    throw new IllegalArgumentException("Argument namespaceLocation must point to a folder.");
                }
            }
            this.log = log;
            this.basePackage = basePackage != null ? basePackage.clone() : new String[0];
            this.packageToSearch = packageToSearch;
            this.sourceRoot = sourceRoot;
            this.maximumDepthOfScan = maximumDepthOfScan;
        }

        private final Log log;
        private final File namespaceFolder;
        private final String[] basePackage;
        private final String packageToSearch;
        private final File sourceRoot;
        private final int maximumDepthOfScan;
    }

}