package Web;

import java.io.File;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.FileWriter;
import java.util.regex.Pattern;
import java.io.IOException;
import java.util.ArrayList;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.regex.Matcher;

/**
 * 编译模版成Java代码
 *
 * @author Ma
 */
public class TemplateCompile {

 /**
 * 模版目录
 */
 String TemplatePath = Options.TemplatePath;
 protected ArrayList<String> block;
 protected ArrayList<String> blockContent;
 protected boolean isBlock = false;
 protected boolean isExtends = false;//标记是否是有继承关系
 private BufferedWriter writer;

 public TemplateCompile() {
 this.blockContent = new ArrayList<String>();
 this.block = new ArrayList<String>();
 this.TemplatePath = Options.TemplatePath;
 }

 /**
 * public static void main(String[] args) { TemplateCompile self = new
 * TemplateCompile(); File tplDir = new File(self.TemplatePath); //判断目录是否存在
 * tplDir.isDirectory();
 *
 * String[] fileList = tplDir.list();
 *
 * for (int i = 0; i < fileList.length; i++) { String string = fileList[i];
 * if (string.matches(".+\\.java")) {//不处理Java文件 continue; }
 * self.start(string); System.out.println(string);//@todo 后期处理子目录的问题 } }
 *
 */
 public int start(String fileName) {
 File tplFile = new File(this.TemplatePath + fileName);
 if (tplFile.exists()) {
 try {
 BufferedReader reader = new BufferedReader(new FileReader(tplFile));
 String line = null;
 //编译文件Start

 File compilerFile = this.getFile(fileName);//获取Java文件
 if (compilerFile.exists()) {
 //如果文件存在就比较时间
 if (compilerFile.lastModified() > tplFile.lastModified()) {
 //如果最后修改的时候小于编译后的Java文件的最后日期
 //直接返回
 //return 0;注释掉,每次都重新编译
 }
 } else {
 //如果文件不存在就创建
 compilerFile.createNewFile();
 }

 this.writer = new BufferedWriter(new FileWriter(compilerFile));
 if (!compilerFile.isFile()) {
 ;//说明创建失败
 }
 //编译文件END
 String fatherTpl = "Web.BaseTemplate";//默认继承
 line = reader.readLine();//分析第一行是否继承
 Pattern pat = Pattern.compile("\\{\\%\\s*extends\\s*\\s(\\w+)\\s*\\%\\}");
 Matcher mat = pat.matcher(line);
 if (mat.find()) {
 fatherTpl = mat.group(1) + "Template";
 line = line.replace(mat.group(), "");
 this.isExtends = true;
 //遇到继承,就去解析父模版
 TemplateCompile templateCompile = new TemplateCompile();
 int r = templateCompile.start(mat.group(1) + ".html");
 }
 //识别继承标识Start
 Pattern blockPat = Pattern.compile("\\{\\%\\s*block\\s*(\\w+)\\s*\\%\\}");
 //识别继承标识END


 //writer.write("package Web.Template" + this.getPackage(fileName) + ";\npublic class "
 writer.write("package Template" + this.getPackage(fileName) + ";\npublic class "
 + this.getfileName(fileName)
 + "Template extends " + fatherTpl + "{");
 if (!this.isExtends) {
 writer.write("\n\tpublic String display(){\n\t\tString str= \n");
 }
 while (line != null) {
 line = line.trim();
 if (line.length() < 1) {
 line = reader.readLine();//读取下一行
 continue;
 }
 //检查是否是block Start
 Matcher blockMat = blockPat.matcher(line);
 if (blockMat.find()) {
 this.block.add(blockMat.group(1));
 this.blockContent.add("");
 this.isBlock = true;
 line = line.replace(blockMat.group(), "\"\"+this." + blockMat.group(1) + "()");
 this.writerFile(line, true);
 line = reader.readLine();//读取下一行
 continue;
 //System.out.println(blockMat.group(1));
 }

 //检查是否是block End
 if (this.isBlock && line.matches("\\{\\%\\s*end\\s*\\%\\}")) {
 this.isBlock = false;
 line = reader.readLine();//读取下一行
 continue;

 }



 //检查if 语句
 if (line.matches(".*\\{\\%\\s*if\\s*\\(.+\\)\\s*\\%\\}.*")) {

 line = line.replace("\"", "\\\"");
 line = line.replace("{%", "\";");
 line = line.replace("%}", "{str=str");
 line = "\"" + line + "\n";

 if (this.isBlock) {
 this.writerBlock(line, true);
 } else {
 this.writerFile(line, true);
 }
 line = reader.readLine();//读取下一行
 continue;
 }
 //检查endif 语句
 if (line.matches(".*\\{\\%\\s*endif\\s*\\%\\}.*")) {
 line = line.replaceAll("\\{\\%\\s*endif\\s*\\%\\}", "\";}str=str+\"");
 line = "\"" + line + "\"\n";
 if (this.isBlock) {
 this.writerBlock(line, true);
 } else {
 this.writerFile(line, true);
 }
 line = reader.readLine();//读取下一行
 continue;
 }
 //检查elseif 语句
 if (line.matches(".*\\{\\%\\s*elseif\\s*\\(.+\\)\\s*\\%\\}.*")) {
 line = line.replace("{%", "\";}");
 line = line.replace("%}", "{str=str");
 line = line.replace("elseif", "else if");
 line = "\"" + line + "\n";

 if (this.isBlock) {
 this.writerBlock(line, true);
 } else {
 this.writerFile(line, true);
 }
 line = reader.readLine();//读取下一行
 continue;
 }
 //检查else 语句
 if (line.matches(".*\\{\\%\\s*else\\s*\\%\\}.*")) {
 line = line.replaceAll("\\{\\%\\s*else\\s*\\%\\}", "\";}else{str=str");

 line = "\"" + line + "\n";
 if (this.isBlock) {
 this.writerBlock(line, true);
 } else {
 this.writerFile(line, true);
 }
 line = reader.readLine();//读取下一行
 continue;
 }
 //检查for 语句
 if (line.matches(".*\\{\\%\\s*for\\s*\\(.+\\)\\s*\\%\\}.*")) {
 line = line.replace("{%", "\";");
 line = line.replace("%}", "{str=str");
 line = "\"" + line + "\n";
 if (this.isBlock) {
 this.writerBlock(line, true);
 } else {
 this.writerFile(line, true);
 }
 line = reader.readLine();//读取下一行
 continue;
 }

 //检查endfor 语句
 if (line.matches(".*\\{\\%\\s*endfor\\s*\\%\\}.*")) {
 line = line.replaceAll("\\{\\%\\s*endfor\\s*\\%\\}", "\";}str=str");
 line = "\"" + line + "\n";
 if (this.isBlock) {
 this.writerBlock(line, true);
 } else {
 this.writerFile(line, true);
 }
 line = reader.readLine();//读取下一行
 continue;
 }


 if (this.isBlock) {
 this.writerBlock(line);
 } else {
 this.writerFile(line);
 }


 writer.newLine();
 //writer.newLine();
 line = reader.readLine();//读取下一行
 }

 reader.close();
 this.flush();
 this.writer.flush();
 this.writer.close();
 return 0;

 } catch (FileNotFoundException ex) {
 Logger.getLogger(TemplateCompile.class.getName()).log(Level.SEVERE, null, ex);
 } catch (IOException ex) {
 Logger.getLogger(TemplateCompile.class.getName()).log(Level.SEVERE, null, ex);
 }


 } else {
 return 1;
 }
 return 0;
 }

 /**
 * 创建编译后的文件
 *
 * @param fileName
 * @return
 */
 protected File getFile(String fileName) {
 fileName = this.getFilePath(fileName) + "/" + this.getfileName(fileName);
 String javaTplName = fileName + "Template.java";
 File JavaFile = new File(this.TemplatePath + javaTplName);

 return JavaFile;
 }

 /**
 * 获取文件名字,不包括扩展名字
 *
 * @param fileName
 * @return
 */
 protected String getfileName(String fileName) {
 //获取文件名字,不包含扩展名
 int index = fileName.lastIndexOf(".html");
 fileName = fileName.substring(0, index);
 index = fileName.lastIndexOf("/");
 fileName = fileName.substring(index + 1);
 return fileName;
 }

 /**
 * 获取文件的路径
 *
 * @param fileName
 * @return
 */
 protected String getFilePath(String fileName) {
 int index = fileName.lastIndexOf(".html");
 fileName = fileName.substring(0, index);
 index = fileName.lastIndexOf("/");
 String path = "";
 if (index > 0) {
 path = fileName.substring(0, index);
 }
 return path;
 }

 protected String getPackage(String fileName) {
 String path = this.getFilePath(fileName);
 if (path.equals("")) {
 path = "";
 } else {
 path = "." + path.replace("/", ".");
 }
 return path;
 }

 /**
 * 写入bolck,自动添加引号
 *
 * @param line
 * @return
 */
 protected String writerBlock(String line) {
 String r = "";
 //首先把所有变量修改{{}}
 if (line.matches(".*\\{\\{.+\\}\\}.*")) {
 Pattern pattern = Pattern.compile(".*\\{\\{(\\w+):(\\w+):?(\\w+)?\\}\\}.*");
 Matcher matcher = pattern.matcher(line);
 if (matcher.find()) {
 line = line.replace("\"", "\\\"");
 if (matcher.group(3) == null) {
 line = line.replaceFirst("\\{\\{(.+):(.+)\\}\\}", "\"+this.getArray(\""
 + matcher.group(1) + "\")["
 + matcher.group(2) + "]+\"");
 } else {//说明是引用2维数组
 line = line.replaceFirst("\\{\\{(.+):(.+):(.+)\\}\\}", "\"+this.getArray2(\""
 + matcher.group(1) + "\")["
 + matcher.group(2) + "][" + matcher.group(3) + "]+\"");
 }
 } else {
 line = line.replace("{{", "\"+this.get(\"");
 line = line.replace("}}", "\")+\"");
 line = line.replace("\"", "\\\"");
 }
 r = this.blockContent.set(this.block.size() - 1, this.blockContent.get(this.block.size() - 1) + "+\n" + "\"" + line + "\"");
 } else {
 r = this.blockContent.set(this.block.size() - 1, this.blockContent.get(this.block.size() - 1) + "+\n" + "\"" + line.replace("\"", "\\\"") + "\"");
 }
 return r;
 }

 /**
 * 写入 bolck,不写入引号
 *
 * @param line
 * @param eval
 * @return
 */
 protected String writerBlock(String line, boolean eval) {

 //首先把所有变量修改{{}}
 Pattern pat = Pattern.compile("\\(.*\\$([_a-zA-Z][_a-zA-Z0-9]*):?(\\w*).*\\)");
 Matcher mat = pat.matcher(line);

 if (mat.find()) {
 String param = mat.group(1);
 String valueType = "";
 if (mat.group(2).length() > 0) {
 valueType = mat.group(2).replaceFirst(mat.group(2).substring(0, 1), mat.group(2).substring(0, 1).toUpperCase());
 }
 line = line.replaceAll("\\$([_a-zA-Z][_a-zA-Z0-9]*):?(\\w*)", "this.get"
 + valueType
 + "(\"" + param + "\")");
 }
 String r = this.blockContent.set(this.block.size() - 1, this.blockContent.get(this.block.size() - 1) + "+\n" + line);
 return r;
 }

 /**
 * 写入引号
 *
 * @param line
 */
 protected void writerFile(String line) {
 line = line.replace("\"", "\\\"");
 //首先把所有变量修改{{}}
 if (line.matches(".*\\{\\{.+\\}\\}.*")) {
 Pattern pattern = Pattern.compile(".*\\{\\{(\\w+):(\\w+)\\}\\}.*");
 Matcher matcher = pattern.matcher(line);
 if (matcher.matches()) {

 line = line.replaceAll("\\{\\{(.+):(.+)\\}\\}", "\"+this.getArray(\""
 + matcher.group(1) + "\")["
 + matcher.group(2) + "]+\"");
 } else {
 line = line.replace("{{", "\"+this.get(\"");
 line = line.replace("}}", "\")+\"");
 }
 }
 try {
 writer.write("\"" + line + "\"+");

 } catch (IOException ex) {
 Logger.getLogger(TemplateCompile.class.getName()).log(Level.SEVERE, null, ex);
 }
 }

 /**
 * 不写入引号
 *
 * @param line
 * @param eval
 */
 protected void writerFile(String line, boolean eval) {

 //首先把所有变量修改{{}}
 Pattern pat = Pattern.compile("\\(.*\\$([_a-zA-Z][_a-zA-Z0-9]*):?(\\w*).*\\)");
 Matcher mat = pat.matcher(line);

 if (mat.find()) {
 String param = mat.group(1);
 String valueType = "";
 if (mat.group(2).length() > 0) {
 valueType = mat.group(2).replaceFirst(mat.group(2).substring(0, 1), mat.group(2).substring(0, 1).toUpperCase());
 }
 line = line.replaceAll("\\$([_a-zA-Z][_a-zA-Z0-9]*):?(\\w*)", "this.get"
 + valueType
 + "(\"" + param + "\")");
 }
 try {
 if (!this.isExtends) {
 writer.write("" + line + "+");
 }
 } catch (IOException ex) {
 Logger.getLogger(TemplateCompile.class.getName()).log(Level.SEVERE, null, ex);
 }
 }

 /**
 * 处理结尾,把block写在里面
 */
 protected void flush() {
 try {
 if (!this.isExtends) {
 this.writer.write("\"\";return str;\n}");
 }
 for (int i = 0; i < this.block.size(); i++) {
 try {
 this.writer.write("public String " + this.block.get(i) + " (){ String str = \"\"" + this.blockContent.get(i) + ";return str;}");

 } catch (IOException ex) {
 Logger.getLogger(TemplateCompile.class.getName()).log(Level.SEVERE, null, ex);
 }

 }
 this.writer.write("\n}");
 } catch (IOException ex) {
 Logger.getLogger(TemplateCompile.class.getName()).log(Level.SEVERE, null, ex);
 }
 }
}