generator.java

EasyJWeb是基于java技术

package com.easyjf.generator;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;

import com.easyjf.web.tools.generator.TemplateGenerator;

public class Generator {
	private static Generator instance;

	private final static Object keyForEasyJFServer = new Object();

	public static Generator getInstance() {
		if (Generator.instance == null) {
			synchronized (Generator.keyForEasyJFServer) {
				if (Generator.instance == null) {
					Generator.instance = new Generator();
				}
			}
		}
		return Generator.instance;
	}

	private Generator() {
	}

	private String domainName;

	private boolean debug = false;
	
	private Class domainClz=null;

	private String templateDir;

	private String rootDomain;
	/**
	 * 要求三个参数，一个是Action名称，一个是生成方式（有domain、entity）两种，第三个是参数值
	 * 
	 * @param args
	 *            一串用来描述domain组成的简单字符参数数组
	 */
	public void genDomain(String[] args) {
		if (debug) {
			String[] s = {
					"Peoples",
					"domain",
					"age:年岁:Long:150:long#name:名称:String:10#email:电子邮箱地址:string:10#address:家庭住址:string:20" };
			this.doGeneratorDomain(s[0], s[2]);
		} else {
			if (args.length == 3) {
				// / String[] s = { args[0], args[1],args[2] };
				this.doGeneratorDomain(args[0], args[2]);
			} else {
				System.out.println("请指定正确的参数");
				System.out
						.println("如：generator People name:名称:string:10:string#email:电子邮箱地址:string:10#address:家庭住址:string:20");
				System.out.println("本版本支持两个参数");
				System.out.println("参数1：Person 要生成对象的名称");
				System.out
						.println("参数2：\"email:电子邮箱:string:10:string\" 当前对象的相关属性，多个属性之间用\"#\"隔开");
				System.out.println("email Person 的属性名称");
				System.out.println("电子邮箱 Person email属性的别名");
				System.out
						.println("string Person email属性的类型，可选范围为：integer long string ... 等基本数据类型");
				System.out.println("10 Person email属性的最大长度");
				System.out.println("string Person email属性的验证类型");
				System.exit(0);
			}
		}
	}

	private void doGeneratorDomain(String domainName, String attributes) {
		TemplateGenerator tg = new TemplateGenerator();
		File f = new File(this.getPri() + "templates/");
		tg.setTemplateDir(f.getAbsolutePath());
		tg.setTemplateName("/domainTemplate.java");
		tg.setTargetDir(this.getTargetDir() + "domain/");
		tg.setTargetName(this.getDomainName() + ".java");
		tg.setProcess(new DomainTemplateProcess(domainName, attributes));
		tg.generator(false);
		try {
		} catch (Exception e) {
			e.printStackTrace();
		}
		
		String s=tg.getTargetDir()+File.separator+tg.getTargetName();
		
		File dest=new File(s);
		while(true){
			if(dest.exists()) break;
		}
		
		
	}

	private String targetDir;

	private String pri="../";

	private void init(String[] args) {
		String domainName=args[0];
		
		if (debug) {
			this.setPri("");
		} 
		
		try{
			Class clz=Class.forName(domainName);
			this.domainClz=clz;
			this.rootDomain=this.domainClz.getName().substring(0,this.domainClz.getPackage().getName().lastIndexOf('.'));
			System.out.println(this.getPri());
			this.setTargetDir(this.getPri()+"src/main/java/"+this.rootDomain.replace('.', '/')+"/");
			this.setDomainName(clz.getSimpleName());
		}
		catch(Exception e)
		{
			e.printStackTrace();
			this.setTargetDir(this.getPri() + "src/main/java/org/easyjf/cms/");
			this.setDomainName(domainName.substring(0, 1).toUpperCase()
					+ domainName.substring(1));
			//生成DOMAIN
			this.genDomain(args);
			this.rootDomain="org.easyjf.cms";
		}
		
	}	
	public void doGenerator(String[] args) throws Exception
	{
		
		this.init(args);//初始化参数，在需要的时候还会根据参数生成Domain
		
		File f = new File(this.getTargetDir() + "domain/"
				+ this.getDomainName() + ".java");
		String targetFile = f.getAbsolutePath().replace("bin\\..\\src", "src");
		
		File idao = new File(this.getTargetDir() + "dao/I"
				+ this.getDomainName() + "DAO.java");
		File imvc = new File(this.getTargetDir() + "mvc/"
				+this.getDomainName() + "Action.java");
		File is = new File(this.getTargetDir() + "service/I"
				+ this.getDomainName() + "Service.java");

		File isi = new File(this.getTargetDir() + "service/impl/"
				+ this.getDomainName() + "ServiceImpl.java");
		
	
			Runtime.getRuntime().exec("cmd /c compile.bat " + targetFile);
			// 参数１为ＤＯＭＡＩＮ名 ２为包名
			String arg1=this.rootDomain+".domain." + this.getDomainName();
			String arg2=this.rootDomain;		
			String[] arg = {arg1,arg2};
			this.waitFor(arg1);
			AllGenerator gen = new AllGenerator(arg);
			if(this.templateDir!=null)gen.templateDir=this.templateDir;
			gen.setPri(this.pri);
			gen.gener();			
			Thread.sleep(1000);
			System.out.println(idao.getAbsolutePath());
			Runtime.getRuntime().exec(
					"cmd /c compile.bat " + idao.getAbsolutePath());
			this.waitFor(this.process(idao.getAbsolutePath()));
			Runtime.getRuntime().exec(
					"cmd /c compile.bat " + is.getAbsolutePath());
			this.waitFor(this.process(is.getAbsolutePath()));
			Runtime.getRuntime().exec(
					"cmd /c compile.bat " + isi.getAbsolutePath());			
			this.waitFor(this.process(isi.getAbsolutePath()));
			Runtime.getRuntime().exec(
					"cmd /c compile.bat " + imvc.getAbsolutePath());
			
			System.out.println("编译完成");
			
			File mvc=new File(this.getPri() + "src/main/java/mvc-app.xml");
			File dao = new File(this.getPri() + "src/main/java/dao.xml");
			File service = new File(this.getPri() + "src/main/java/service.xml");
			File application=new File(this.getPri() + "src/main/java/application.xml");
			File jpabase=new File(this.getPri() + "src/main/java/jpa-base.xml");
			File log4j=new File(this.getPri() + "src/main/java/log4j.properties");
			File persistence=new File(this.getPri() + "src/main/java/persistence.xml");
			
			String cls=this.getPri() + "src/main/webapp/WEB-INF/classes/";
			String webinf=this.getPri() + "src/main/webapp/WEB-INF/";
			
			System.out.println("开始配置文件");
			try{
			this.copyFile(application, new File(cls+"application.xml"));
			this.copyFile(jpabase, new File(cls+"jpa-base.xml"));
			this.copyFile(log4j, new File(cls+"log4j.properties"));
			this.copyFile(persistence, new File(cls+"persistence.xml"));
			this.copyFile(dao, new File(cls+"dao.xml"));
			this.copyFile(service, new File(cls+"service.xml"));
			this.copyFile(mvc, new File(webinf+"mvc-app.xml"));
			}
			catch(Exception e)
			{
				System.out.println("拷贝配置文件出错!");
				e.printStackTrace();
			}
			/*
			File targetDaoFile = new File(
					"../src/main/webapp/WEB-INF/classes/dao.xml");
			File targetServiceFile = new File(
					"../src/main/webapp/WEB-INF/classes/service.xml");
			if (dao.exists() && service.exists()) {				
				generator.copyFile(dao, targetDaoFile);
				generator.copyFile(service, targetServiceFile);
				generator.copyFile(
						new File("../src/main/java/mvc-app.xml"), new File(
								"../src/main/webapp/WEB-INF/mvc-app.xml"));
			}
			*/
			
			
			
			
		
	}
	/**
	 * 基于EasyJWeb+Jpa+Spring的Crud应用生成 useage generator ActionName domainType
	 * argumentvalue 如generator Person domain com.easyjweb.demo.domain.Person
	 * 
	 * @param args
	 */
	public static void main(String[] args) {
		try {
		Generator generator = Generator.getInstance();
		generator.doGenerator(args);
		System.out.println("配置文件添加完成，准备启动服务器");

		
		Thread.sleep(3000);
		Runtime.getRuntime().exec("cmd /c start db.bat");
		Runtime.getRuntime().exec("cmd /c start web.bat");
		
		Thread.sleep(13000);
		System.out.println("正在打开默认浏览器");

		Runtime.getRuntime().exec(
				"cmd /c start explorer http://127.0.0.1:82/ejf/"
						+ args[0].toLowerCase());
		System.out.println("完成");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
private void waitFor(String className){
	boolean haveClass=false; 
	int t=0;
	while(!haveClass && t<5){
	try{
		System.out.println(className);
		Class.forName(className);
		haveClass=true;
	}
	catch(java.lang.ClassNotFoundException e){
		try {
			Thread.sleep(1000);
			t++;
		} catch (Exception e1) {
			e1.printStackTrace();
		}
	}
	}
}

//   \. \\
private String process(String s){	
	String s1=this.rootDomain.replace('.',File.separator.charAt(0));	
	String daoClass=s.substring(s.indexOf(s1));
	return daoClass.substring(0,daoClass.length()-5).replaceAll("\\\\", ".");
}

	private void copyFile(File source, File target) throws Exception {
		FileInputStream in = new FileInputStream(source);
		FileOutputStream out = new FileOutputStream(target);
		byte[] a = new byte[2000];
		int len = in.read(a);
		while (len != -1) {
			out.write(a, 0, len);
			len = in.read(a);
		}
	}

	public String getDomainName() {
		return domainName;
	}

	public void setDomainName(String domainName) {
		this.domainName = domainName;
	}

	public String getTargetDir() {
		return targetDir;
	}

	public void setTargetDir(String targetDir) {
		this.targetDir = targetDir;
	}

	public String getPri() {
		return pri;
	}

	public void setPri(String pri) {
		this.pri = pri;
	}

	public void setTemplateDir(String templateDir) {
		this.templateDir = templateDir;
	}
	
}