org.tinygroup.codegen.impl.CodeGeneratorDefault Maven / Gradle / Ivy

Go to download
/**
 *  Copyright (c) 1997-2013, www.tinygroup.org ([email protected]).
 *
 *  Licensed under the GPL, Version 3.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.gnu.org/licenses/gpl.html
 *
 *  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 org.tinygroup.codegen.impl;

import java.io.*;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.lang.StringUtils;
import org.tinygroup.codegen.CodeGenerator;
import org.tinygroup.codegen.config.CodeGenMetaData;
import org.tinygroup.codegen.config.MacroDefine;
import org.tinygroup.codegen.config.TemplateDefine;
import org.tinygroup.commons.tools.StringUtil;
import org.tinygroup.context.Context;
import org.tinygroup.context.util.ContextFactory;
import org.tinygroup.docgen.DocumentGenerater;
import org.tinygroup.docgen.config.GenUtilConfig;
import org.tinygroup.docgen.config.StaticClass;
import org.tinygroup.docgen.function.StaticClassFunction;
import org.tinygroup.docgen.impl.DocumentGeneraterImpl;
import org.tinygroup.logger.LogLevel;
import org.tinygroup.logger.Logger;
import org.tinygroup.logger.LoggerFactory;
import org.tinygroup.template.TemplateEngine;
import org.tinygroup.template.impl.TemplateEngineDefault;
import org.tinygroup.template.loader.FileObjectResourceLoader;
import org.tinygroup.vfs.FileObject;
import org.tinygroup.vfs.VFS;

import com.thoughtworks.xstream.XStream;

public class CodeGeneratorDefault implements CodeGenerator{

private static final Logger LOGGER = LoggerFactory.getLogger(CodeGeneratorDefault.class);

	//public static FullContextFileRepository repository;//暂时定为静态
	private DocumentGenerater generater;
	
	
	public CodeGeneratorDefault(){
		generater=new DocumentGeneraterImpl();
		TemplateEngine templateEngine = new TemplateEngineDefault();
		generater.setTemplateGenerater(templateEngine);
	}
	

	/**
	 * @param context
	 *            默认文件需要用到的上下文环境
	 * @throws IOException 
	 */
	public List generate(Context context) throws IOException{
		CodeGenMetaData metaData=context.get(CODE_META_DATA);
		if(metaData==null){
			throw new RuntimeException("代码生成器的元数据不存在");
		}
		addUtilClass();
		List macroDefines=metaData.getMacroDefines();
		for (MacroDefine macroDefine : macroDefines) {
		   LOGGER.logMessage(LogLevel.INFO, "开始加载宏文件路径：{0}",macroDefine.getMacroPath());
		   String macroPath=macroDefine.getMacroPath();
		   if (!StringUtils.startsWith(macroPath, "/")) {
			   macroPath = "/"+macroPath;
		   }
		   FileObject fileObject=VFS.resolveFile((String)context.get(ABSOLUTE_PATH));
		   generater.addMacroFile(fileObject.getFileObject(macroPath));
		   LOGGER.logMessage(LogLevel.INFO, "宏文件路径：{0}，加载完毕",macroDefine.getMacroPath());
		}
		List templateDefines=metaData.getTemplateDefines();
		List fileList=new ArrayList();
		for (TemplateDefine templateDefine : templateDefines) {
			Context newContext=createNewContext(context, templateDefine);
			String templatePath=templateDefine.getTemplatePath();
			LOGGER.logMessage(LogLevel.INFO, "开始加载模板文件路径：{0}",templatePath);
//			if(templatePath.startsWith("/")){
//				templatePath=CodeGenerator.class.getResource(templatePath).getPath();
//			}
			FileObject templateDirObject=VFS.resolveFile((String)context.get(ABSOLUTE_PATH));
			LOGGER.logMessage(LogLevel.INFO, "模板文件路径：{0}，加载完毕",templatePath);
//			String filePath = templatePath;
//			if (!StringUtils.startsWith(templatePath, "/")) {
//				filePath = "/"+templatePath;
//			}
			//repository.addFileObject(templatePath, templateFileObject.getFileObject(filePath));
			FileObject templateFileObject = templateDirObject.getFileObject(templatePath);
			generater.getTemplateGenerater().addResourceLoader(new FileObjectResourceLoader(templateFileObject.getExtName(),null,null,templateDirObject));
			String generateFile=generater.evaluteString(newContext, templateDefine.getFileNameTemplate());
			File file=new File(generateFile);
			if(!file.exists()){
				file.getParentFile().mkdirs();
				file.createNewFile();
			}
			OutputStream outputStream=new FileOutputStream(file);
			try {
				generater.generate(templateFileObject, newContext,new FileOutputStream(file));
			}finally{
				outputStream.close();
			}
			fileList.add(file.getPath());
		}
		return fileList;
		
	}
	
	private void addUtilClass() {
		XStream stream = new XStream();
		stream.setClassLoader(getClass().getClassLoader());
		stream.autodetectAnnotations(true);
		stream.processAnnotations(GenUtilConfig.class);
		GenUtilConfig config=(GenUtilConfig)stream.fromXML(getClass().getResourceAsStream("/codegen.util.xml"));
		if(config.getStaticClasses()!=null){
		   for(StaticClass staticClass:config.getStaticClasses()){
			   try {
				   StaticClassFunction function = new StaticClassFunction(staticClass.getName(),staticClass.getClassName());
				   generater.getTemplateGenerater().addTemplateFunction(function);
			   } catch (Exception e) {
				   LOGGER.logMessage(LogLevel.ERROR, "静态方法类：{0}，实例化失败",staticClass.getClassName());
			   }
		   }
		}
	}

	private Context createNewContext(Context context,TemplateDefine templateDefine){
		Context newContext=ContextFactory.getContext();
		newContext.setParent(context);
		newContext.put(TEMPLATE_FILE, templateDefine);
		String templatePath=templateDefine.getTemplatePath();
		String templateFilePath = templatePath.replaceAll("/", "\\"+File.separator);//把/../..路径转化成系统认知的路径格式
		String path = StringUtil.substringBeforeLast(templateFilePath, File.separator);
		if(path.startsWith(File.separator)){
			path=path.substring(1);
		}
		if(path.trim().length()>0&&!path.endsWith(File.separator)){
			path=path+File.separator;
		}
		newContext.put("templateFilePath", path);
		String fileName=StringUtil.substringAfterLast(templateFilePath, File.separator);
		newContext.put("templateFileName", StringUtil.substringBeforeLast(fileName, "."));
		return newContext;
	}
}