package cn.ksource.tools.db2java;

import java.io.File;
import java.io.FileOutputStream;
import java.io.Writer;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang.StringUtils;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.transaction.annotation.Transactional;

import cn.ksource.core.dao.BaseDao;
import cn.ksource.core.dao.SqlParameter;
import cn.ksource.core.util.DateUtil;
import cn.ksource.core.util.FreeMarkerUtil;
import cn.ksource.core.util.JdbcUtil;
import freemarker.ext.beans.BeansWrapper;
import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateHashModel;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations={"/applicationContext_core.xml"})
@Transactional
public class Db2Java4Mysql {
	
	@Autowired
	private BaseDao baseDao;
	
	public final static String TABLE_SCHEMA = JdbcUtil.getDbName();
	public final static String FILE_PATH = "d:\\db2java\\";
	public final static String PACKAGE_PATH = "";
	public final static String CLASS_FILE_PATH = FILE_PATH+"classes";
	private List<Map> freeMakerList;
	
	
	/**
	 * @return
	 */
	private List<Map> getFreeMakerList(){
		if (freeMakerList == null || freeMakerList.size() == 0) {
			String sql = "select TABLE_NAME,COLUMN_NAME,COLUMN_TYPE as TYPE,COLUMN_COMMENT from information_schema.columns where TABLE_SCHEMA = '"+ Db2Java4Mysql.TABLE_SCHEMA +"'";
			List<Map> list = baseDao.queryForList(sql,new SqlParameter());
			for (Map map : list) {
				String column_type = map.get("TYPE").toString().toLowerCase();
				if (StringUtils.startsWith(column_type, "char") || StringUtils.startsWith(column_type, "varchar") || StringUtils.startsWith(column_type, "text") || StringUtils.startsWith(column_type, "longtext")) {
					map.put("COLUMN_TYPE", "String");
				} else if (StringUtils.startsWith(column_type, "int") || StringUtils.startsWith(column_type, "tinyint") || StringUtils.startsWith(column_type, "smallint")) {
					map.put("COLUMN_TYPE", "Integer");
				} else if (StringUtils.startsWith(column_type, "bigint")) {
					map.put("COLUMN_TYPE", "Long");
				} else if (StringUtils.startsWith(column_type, "float") || StringUtils.startsWith(column_type, "double") || StringUtils.startsWith(column_type, "decimal")) {
					map.put("COLUMN_TYPE", "Double");
				}
				map.put("COLUMN_LENGTH", !StringUtils.startsWith(column_type,"text") ?
								column_type.substring(column_type.indexOf("(")+1, column_type.length()-1)
								:new Integer(65535));
			}
			freeMakerList = list;
		}
		return this.freeMakerList;
	}
	
	private List<Map> getTableInfoList(){
		List<Map> resultList = new ArrayList<Map>();
		Map bwlMap = new HashMap();
		List<Map> list = getFreeMakerList();
		for (Map map : list) {
			if (!bwlMap.containsKey(map.get("TABLE_NAME"))) {
				resultList.add(map);
				bwlMap.put(map.get("TABLE_NAME"), map);
				List<Map> tempList = new ArrayList<Map>();
				map.put("columns", tempList);
			} else {
				List<Map> tempList = (List<Map>)(((Map)bwlMap.get(map.get("TABLE_NAME"))).get("columns"));
				tempList.add(map);
			}
		}
		return resultList;
	}
	
	/**
	 * ���
	 * @param root
	 * ���ߣ��
	 */
	private Map addPK(){
		
		Map root = new HashMap();
		
		String getPKSQL = "SELECT\n" +
		"	TABLE_NAME,\n" +
		"  COLUMN_NAME,\n" +
		"  REFERENCED_TABLE_NAME as R_TABLE_NAME,\n" +
		"  REFERENCED_COLUMN_NAME as R_COLUMN_NAME\n" +
		" FROM\n" +
		"	information_schema.KEY_COLUMN_USAGE\n" +
		" WHERE\n" +
		//" 	TABLE_NAME = 'ac_role_qx'\n" +
		" POSITION_IN_UNIQUE_CONSTRAINT = 1\n" +
		" and TABLE_SCHEMA = '"+ Db2Java4Mysql.TABLE_SCHEMA +"'";
		
		root.put("MYFK", baseDao.queryForList(getPKSQL,new SqlParameter()));
		return root;
	}
	
	/**
	 * ��ע��
	 * @return
	 * ���ߣ��
	 */
	public Map getTableComments(){
		String sql = "SELECT\n" +
		"	TABLE_NAME,\n" +
		"	TABLE_COMMENT\n" +
		"FROM\n" +
		"	information_schema.`TABLES`\n" +
		"WHERE\n" +
		"	TABLE_SCHEMA = '"+ Db2Java4Mysql.TABLE_SCHEMA +"'";
		List<Map> list = baseDao.queryForList(sql,new SqlParameter());
		Map rootMap = new HashMap();
		for (Map map : list) {
			rootMap.put(map.get("TABLE_NAME").toString().toUpperCase(), map.get("TABLE_COMMENT"));
		}
		return rootMap;
	}
	
	private void table2Java(Configuration cfg) throws Exception {
		List<Map> list = this.getTableInfoList();

		Map fkMap = addPK();
		Map tableComments = getTableComments();
		
		for (Map map : list) {
			Template template = cfg.getTemplate("bean.ftl");
			
			Map rootMap = new HashMap();
			rootMap.put("DATE", DateUtil.format("yyyy-mm-dd hh:MM", DateUtil.getCurrnetDate12()));
			rootMap.put("TABLENAME", map.get("TABLE_NAME").toString().toUpperCase());
			rootMap.put("columns", map.get("columns"));
			rootMap.put("MYFK", fkMap.get("MYFK"));
			rootMap.put("COMMENTS", tableComments);
			
			File targetFile = new File(FILE_PATH+Db2Java4Mysql.PACKAGE_PATH+map.get("TABLE_NAME").toString().toUpperCase()+".java");
			if (!targetFile.exists()) {
				targetFile.getParentFile().mkdirs();
			}
			Writer writer = new java.io.BufferedWriter(new java.io.OutputStreamWriter(new FileOutputStream(targetFile), "UTF-8"));
			template.process(rootMap, writer);
			
			writer.flush();
			writer.close();
		}
	}
	
	private static void loadCommonStaticClass(Configuration cfg) {
		try {
			BeansWrapper wrapper = BeansWrapper.getDefaultInstance();
			TemplateHashModel staticModels = wrapper.getStaticModels();
//			cfg.setSharedVariable("Constants", (TemplateHashModel) staticModels.get("com.zzsb.Constants"));
			cfg.setSharedVariable("ConvertUtil", (TemplateHashModel) staticModels.get("cn.ksource.core.util.ConvertUtil"));
			cfg.setSharedVariable("FreeMarkerUtil", (TemplateHashModel) staticModels.get("cn.ksource.core.util.FreeMarkerUtil"));
			// root.put("FileUtil", (TemplateHashModel)
			// staticModels.get("com.djwl.core.utils.FileUtil"));
			cfg.setSharedVariable("DateUtil", (TemplateHashModel) staticModels.get("cn.ksource.core.util.DateUtil"));
			cfg.setSharedVariable("StringUtil", (TemplateHashModel) staticModels.get("cn.ksource.core.util.StringUtil"));
			cfg.setSharedVariable("StringUtils", (TemplateHashModel) staticModels.get("org.apache.commons.lang.StringUtils"));
			cfg.setSharedVariable("BwlUtil", (TemplateHashModel) staticModels.get("cn.ksource.core.util.BwlUtil"));
			cfg.setSharedVariable("Math", (TemplateHashModel) staticModels.get("java.lang.Math"));
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	@Test
	public  void create() throws Exception {
		
		System.out.println("db2java ��ʼ��....");
		
		
		Configuration cfg = FreeMarkerUtil.getConfiguration(this.getClass(), "template");
		loadCommonStaticClass(cfg);
		this.table2Java(cfg);
		
		//�������ļ���
		String path = FILE_PATH.replaceAll("/", "\\\\");
        String cmd[]={"explorer.exe",path};   
        Process p = Runtime.getRuntime().exec(cmd);
		//�������ļ���
//        String cmd[]={"cmd /c cd /d " + FILE_PATH,"jar cf bean.jar " + PACKAGE_PATH};   
//		String cmd = "cmd /c cd /d " + FILE_PATH + 
//		" && " +
//		"javac -classpath "+ System.getProperty("java.class.path") +" "+ FILE_PATH+PACKAGE_PATH + "*.java " + 
//		" && " +
//		"jar cf bean-"+TABLE_SCHEMA+"-"+ DateUtil.getCurrentDate14() +".jar " + PACKAGE_PATH+"*.class" +
//		" && " +
//		"jar cf bean-source-"+TABLE_SCHEMA+"-"+ DateUtil.getCurrentDate14() +".jar " + PACKAGE_PATH+"*.java" +
//		" && " +
//		"jar cf bean-all-"+TABLE_SCHEMA+"-"+ DateUtil.getCurrentDate14() +".jar " + PACKAGE_PATH+"*.*" 
//		;
//		System.out.println(cmd);
//        Process p = Runtime.getRuntime().exec(cmd);
//        
//        System.out.println("db2java ִ�гɹ�");
	}
}