com.nbsaas.codegen.command.RouterPageCommand Maven / Gradle / Ivy

Go to download
package com.nbsaas.codegen.command;

import com.nbsaas.codegen.templates.elementuiForm.ElementUIFormDir;
import freemarker.cache.ClassTemplateLoader;
import freemarker.template.Configuration;
import org.apache.commons.chain.Command;
import org.apache.commons.chain.Context;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.FileWriter;

public class RouterPageCommand implements Command {

    private Logger logger = LoggerFactory.getLogger(getClass());

    @Override
    public boolean execute(Context context) throws Exception {
        boolean isView = (Boolean) context.get("isView");
        if (!isView) {
            return false;
        }
        logger.info("Page接口和实现生成");

        Class classdir = (Class) context.get("dir");
        ClassTemplateLoader ctl = new ClassTemplateLoader(classdir, "/" + classdir.getName().replace(classdir.getSimpleName(), "").replace(".", "/"));
        Configuration config = new Configuration(Configuration.DEFAULT_INCOMPATIBLE_IMPROVEMENTS);
        config.setTemplateLoader(ctl);

        File view = (File) context.get("view");
        if (!view.exists()) {
            view.mkdirs();
        }
        Class entity = (Class) context.get("entity");
        File dir = new File(view, entity.getSimpleName().toLowerCase());
        if (!dir.exists()) {
            dir.mkdirs();
        }

        if (classdir.equals(ElementUIFormDir.class)) {
            handle(dir, config, context, "componentAdd.ftl", "componentAdd.html");
            handle(dir, config, context, "componentList.ftl", "componentList.html");
            handle(dir, config, context, "componentShow.ftl", "componentShow.html");
            handle(dir, config, context, "componentUpdate.ftl", "componentUpdate.html");
            handle(dir, config, context, "page_list.ftl", "list.html");

        } else {
            Boolean catalog = (Boolean) context.get("catalog");
            if (catalog) {
                handle(dir, config, context, "page_catalog_list.html", "list.html");
            } else {
                handle(dir, config, context, "page_list.html", "list.html");
            }
            handle(dir, config, context, "component.html", "component.html");
        }


        return false;
    }

    private void handle(File dir, Configuration config, Context context, String template, String out) {
        try {
            FileWriter component = new FileWriter(new File(dir, out));
            config.getTemplate(template).process(context, component);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}