gov.nih.nci.sdk.example.generator.util.GeneratorUtil.java Source code

Introduction

Here is the source code for gov.nih.nci.sdk.example.generator.util.GeneratorUtil.java

Source

/*L
 *  Copyright Ekagra Software Technologies Ltd.
 *  Copyright SAIC, SAIC-Frederick
 *
 *  Distributed under the OSI-approved BSD 3-Clause License.
 *  See http://ncip.github.com/cacore-sdk/LICENSE.txt for details.
 */

package gov.nih.nci.sdk.example.generator.util;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.net.MalformedURLException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.antlr.stringtemplate.StringTemplate;
import org.antlr.stringtemplate.StringTemplateGroup;
import org.apache.commons.io.FileUtils;

import gov.nih.nci.sdk.util.EcoreUtil;
import gov.nih.nci.sdk.core.ScriptContext;
import gov.nih.nci.sdk.example.generator.Generator;

public class GeneratorUtil {
    public static void writeFile(String _outputDir, String _fileName, String _content) {
        BufferedWriter bufferedWriter = null;

        try {
            createOutputDir(_outputDir);
            File file = new File(_outputDir, _fileName);
            FileWriter fileWriter = new FileWriter(file);
            bufferedWriter = new BufferedWriter(fileWriter);
            bufferedWriter.write(_content);
            bufferedWriter.flush();
        } catch (Throwable t) {
            throw new RuntimeException(t);
        } finally {
            if (bufferedWriter != null) {
                try {
                    bufferedWriter.close();
                } catch (Throwable t) {
                }
            }
        }
    }

    public static void createOutputDir(String _outputDir) {
        try {
            File file = new File(_outputDir);

            if (!file.exists() == true) {
                FileUtils.forceMkdir(file);
            }
        } catch (Throwable t) {
            throw new RuntimeException(t);
        }
    }

    public static List getFiles(String _dir, String[] _extensions) {
        java.util.logging.Logger.getLogger("DEBUG").info("Directory _dir is: " + _dir);

        List<String> files = new ArrayList();

        try {
            Iterator iter = FileUtils.iterateFiles(new File(_dir), _extensions, false);

            while (iter.hasNext() == true) {
                File file = (File) iter.next();
                files.add(file.getAbsolutePath());
            }
        } catch (Throwable t) {
            throw new RuntimeException(t);
        }

        java.util.logging.Logger.getLogger("DEBUG").info("Returning these files: " + files);

        return files;
    }

    public static String getFiles(String _dir, String[] _extensions, String _separator) {
        java.util.logging.Logger.getLogger("DEBUG").info("Directory _dir is: " + _dir);

        StringBuffer files = new StringBuffer();

        try {
            Iterator iter = FileUtils.iterateFiles(new File(_dir), _extensions, false);

            while (iter.hasNext() == true) {
                File file = (File) iter.next();
                files.append(file.getAbsolutePath()).append(_separator);
            }
        } catch (Throwable t) {
            throw new RuntimeException(t);
        }

        java.util.logging.Logger.getLogger("DEBUG").info("Returning these files: " + files);

        return files.toString().replaceAll(":$", "");
    }

    public static String getJaxbPojoPath(ScriptContext _scriptContext) {
        String s = getGeneratedPath(_scriptContext);
        String jaxbPojoPath = s + File.separator
                + EcoreUtil.determinePackageName(_scriptContext.getFocusDomain()).replace('.', File.separatorChar)
                + File.separator + Generator.JAXBPOJO_PACKAGE_NAME;

        return jaxbPojoPath;
    }

    public static String getPojoPath(ScriptContext _scriptContext) {
        String jaxbPojoPath = getGeneratedPath(_scriptContext) + File.separator
                + EcoreUtil.determinePackageName(_scriptContext.getFocusDomain()).replace('.', File.separatorChar)
                + File.separator + Generator.POJO_PACKAGE_NAME;

        return jaxbPojoPath;
    }

    public static String getServiceImplPath(ScriptContext _scriptContext) {
        String serviceImplPath = getImplPath(_scriptContext) + File.separator
                + EcoreUtil.determinePackageName(_scriptContext.getFocusDomain()).replace('.', File.separatorChar)
                + File.separator + Generator.SERVICE_PACKAGE_NAME;

        return serviceImplPath;
    }

    public static String getServiceClientPath(ScriptContext _scriptContext) {
        return getGeneratedPath(_scriptContext) + File.separator
                + EcoreUtil.determinePackageName(_scriptContext.getFocusDomain()).replace('.', File.separatorChar)
                + File.separator + Generator.SERVICE_PACKAGE_NAME + File.separator
                + Generator.SERVICE_CLIENT_PACKAGE_NAME;
    }

    public static String getServicePath(ScriptContext _scriptContext) {
        return getGeneratedPath(_scriptContext) + File.separator
                + EcoreUtil.determinePackageName(_scriptContext.getFocusDomain()).replace('.', File.separatorChar)
                + File.separator + Generator.SERVICE_PACKAGE_NAME;
    }

    public static String getGeneratedPath(ScriptContext _scriptContext) {

        String returnString = _scriptContext.getProperties().getProperty("PROJECT_ROOT") + File.separator
                + _scriptContext.getProperties().getProperty("PROJECT_SRC") + File.separator
                + Generator.GENERATED_PACKAGE_NAME;

        return returnString;
    }

    public static String getImplPath(ScriptContext _scriptContext) {
        return _scriptContext.getProperties().getProperty("PROJECT_ROOT") + File.separator
                + _scriptContext.getProperties().getProperty("PROJECT_SRC") + File.separator
                + Generator.IMPL_PACKAGE_NAME;
    }

    public static String getServicePackageName(String _fullyQualifiedClassName) {
        return EcoreUtil.determinePackageName(_fullyQualifiedClassName) + "." + Generator.SERVICE_PACKAGE_NAME;
    }

    public static String getJaxbPojoPackageName(String _fullyQualifiedClassName) {
        return EcoreUtil.determinePackageName(_fullyQualifiedClassName) + "." + Generator.JAXBPOJO_PACKAGE_NAME;
    }

    public static String getPojoPackageName(String _fullyQualifiedClassName) {
        return EcoreUtil.determinePackageName(_fullyQualifiedClassName) + "." + Generator.POJO_PACKAGE_NAME;
    }

    public static String getServiceClientPackageName(String _fullyQualifiedClassName) {
        return EcoreUtil.determinePackageName(_fullyQualifiedClassName) + "." + Generator.SERVICE_PACKAGE_NAME + "."
                + Generator.SERVICE_CLIENT_PACKAGE_NAME;
    }

    public static String getClassesPath(ScriptContext _scriptContext) {
        return _scriptContext.getProperties().getProperty("PROJECT_ROOT") + File.separator + "classes";
    }

    public static String convertFirstCharToUpperCase(String _string) {
        String capitalizedString = _string;

        if (_string != null && _string.length() > 0) {
            capitalizedString = _string.substring(0, 1).toUpperCase();

            if (_string.length() > 1) {
                capitalizedString += _string.substring(1, _string.length());
            }
        }

        return capitalizedString;
    }

    public static String convertFirstCharToLowerCase(String _string) {
        String capitalizedString = _string;

        if (_string != null && _string.length() > 0) {
            capitalizedString = _string.substring(0, 1).toLowerCase();

            if (_string.length() > 1) {
                capitalizedString += _string.substring(1, _string.length());
            }
        }

        return capitalizedString;
    }
}