com.metadave.stow.Stow.java Source code

Introduction

Here is the source code for com.metadave.stow.Stow.java

Source

/*
 * -------------------------------------------------------------------
 *
 *   Copyright (c) 2014 Dave Parfitt
 *
 *   This file is provided to you 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 com.metadave.stow;

import org.apache.commons.cli.*;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.stringtemplate.v4.ST;
import org.stringtemplate.v4.STGroup;
import org.stringtemplate.v4.STGroupFile;
import org.stringtemplate.v4.compiler.CompiledST;
import org.stringtemplate.v4.compiler.FormalArgument;

import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.StringWriter;
import java.net.URL;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

public class Stow {

    static String getNiceName(String id) {
        if (id == null || id.length() == 0) {
            return "";
        }
        return id.substring(0, 1).toUpperCase() + id.substring(1);
    }

    public static void generateObjects(String groupFile, String classPackage, String classPrefix, String dest) {
        if (classPrefix == null) {
            classPrefix = "";
        }

        Set<String> templateNames = new HashSet<String>();
        StowSTGroupFile stg = new StowSTGroupFile(groupFile);
        STGroup outputGroup = new STGroupFile("Stow.stg");

        Map<String, CompiledST> ts = stg.getTemplates();
        int i = 1;
        for (String s : ts.keySet()) {
            CompiledST t = ts.get(s);
            if (t.isAnonSubtemplate) {
                continue;
            }
            String templateId;

            if (!templateNames.contains(t.name.toUpperCase())) {
                templateNames.add(t.name.toUpperCase());
                templateId = t.name;
            } else {
                System.out.println("Adding a unique id to " + t.name);
                templateId = t.name + i;
            }

            STOWSTBean bean = new STOWSTBean(outputGroup);
            bean.addPackage(classPackage);
            bean.addTemplateName(t.name);
            String className = classPrefix + templateId;
            bean.addBeanClass(className);
            if (t.hasFormalArgs) {
                Map<String, FormalArgument> fas = t.formalArguments;
                if (fas != null) {
                    for (String fa : fas.keySet()) {
                        FormalArgument f = fas.get(fa);
                        STOWSTAccessor acc = new STOWSTAccessor(outputGroup);
                        acc.addBeanClass(className);
                        acc.addMethodName(getNiceName(f.name));
                        acc.addParamName(f.name);
                        bean.addAccessor(acc);
                    }
                }
            }

            String outputFileName = dest + File.separator + className + ".java";
            System.out.println("Generating " + outputFileName);
            File f = new File(outputFileName);

            try {
                FileUtils.writeStringToFile(f, bean.getST().render());
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        System.out.println("Finished!");
    }

    public static void main(String args[]) {
        System.out.println("Stow: StringTemplate Object Wrapper");
        System.out.println("(C) 2014 Dave Parfitt");
        System.out.println("Stow uses the Apache 2 license");

        CommandLineParser parser = new BasicParser();

        Options options = new Options();
        //options.addOption( "a", "all", false, "do not hide entries starting with .");

        Option javaPackage = new Option("java_package", "package for generated classes");
        javaPackage.setArgs(1);
        //javaPackage.setRequired(true);

        Option destDir = new Option("dest", "destination directory for generated .java files");
        destDir.setArgs(1);

        Option stgFile = new Option("stg", "StringTemplate4 group file");
        stgFile.setArgs(1);

        Option classPrefix = new Option("class_prefix", "Prefix to use for generated classes");
        classPrefix.setArgs(1);

        //destDir.setRequired(true);

        options.addOption(javaPackage);
        options.addOption(destDir);
        options.addOption(stgFile);
        options.addOption(classPrefix);

        try {
            CommandLine line = parser.parse(options, args);

            if (line.hasOption("java_package") && line.hasOption("dest") && line.hasOption("stg")) {
                generateObjects(line.getOptionValue("stg"), line.getOptionValue("java_package"),
                        line.hasOption("class_prefix") ? line.getOptionValue("class_prefix") : "",
                        line.getOptionValue("dest"));
            } else {
                HelpFormatter formatter = new HelpFormatter();
                formatter.printHelp("stow", options);
            }
        } catch (ParseException exp) {
            System.out.println("Error parsing stow command line:" + exp.getMessage());
        }
    }

}