package com.ibeetl.admin.core.gen;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.beetl.core.GroupTemplate;
import org.beetl.core.Template;

import com.ibeetl.admin.core.gen.model.Attribute;
import com.ibeetl.admin.core.gen.model.Entity;

public class JavaCodeGen  implements AutoGen{
	String basePackage;
	Entity entity;
	String basicFunctionCode = null;
	static String CR = System.getProperty("line.separator");
	
	public JavaCodeGen(String basePackage,Entity entity,String basicFunctionCode) {
		this.basePackage =basePackage;
		this.entity = entity;
		this.basicFunctionCode =basicFunctionCode;
	}
	@Override
	public void make(Target target, Entity entity) {
		
		JavaEntityGen entityGen = new JavaEntityGen(this);
		entityGen.make(target, entity);
		
		JavaDaoGen daoGen = new JavaDaoGen(this);
		daoGen.make(target, entity);
		
		JavaServiceGen serviceGen = new JavaServiceGen(this);
		serviceGen.make(target, entity);
		
		
		JavaControllerGen webGen = new JavaControllerGen(this);
		webGen.make(target, entity);
		
		JavaQueryGen queryGen = new JavaQueryGen(this);
		queryGen.make(target, entity);
	
	}
	
	@Override
	public String getName() {
		return "";
	}
	
}


class JavaEntityGen  implements AutoGen{
	JavaCodeGen gen;
	
	public JavaEntityGen(JavaCodeGen gen){
		this.gen = gen;
		
	}
	@Override
	public void make(Target target, Entity entity) {
		GroupTemplate gt = target.getGroupTemplate();
		Template template = gt.getTemplate("/java/pojo.java");
		template.binding("entity", entity);
		template.binding("target", target);
		template.binding("package", gen.basePackage+".entity");
		template.binding("className", entity.getName());
		List<Map<String,Object>> attrs = new ArrayList<Map<String,Object>>();
		for(Attribute attr:entity.getList()) {
			Map<String,Object> map = new HashMap<String,Object>();
			map.put("comment", attr.getComment());
			map.put("type", attr.getJavaType());
			map.put("name", attr.getName());
			map.put("methodName", BaseTarget.upperFirst(attr.getName()));
			map.put("isId", attr.isId());
			map.put("dictType", attr.getDictType());
			attrs.add(map);
			
		}
		template.binding("attrs", attrs);
		String srcHead ="";
		srcHead+="import java.math.*;"+JavaCodeGen.CR;
		srcHead+="import java.util.Date;"+JavaCodeGen.CR;
		
		srcHead+="import java.sql.Timestamp;"+JavaCodeGen.CR;
		template.binding("imports", srcHead);
		template.binding("comment", entity.getComment());
		String content = template.render();
		target.flush(this, content);
	}

	@Override
	public String getName() {
		return gen.entity.getName()+".java";
	}
	
}


class JavaDaoGen  implements AutoGen{
	JavaCodeGen gen;
	public JavaDaoGen(JavaCodeGen gen){
		this.gen = gen;
	}
	@Override
	public void make(Target target, Entity entity) {
		GroupTemplate gt = target.getGroupTemplate();
		Template template = gt.getTemplate("/java/dao.java");
		template.binding("entity", entity);
		template.binding("target", target);
		template.binding("package", gen.basePackage+".dao");
		template.binding("basePackage", gen.basePackage);
		String content = template.render();
		target.flush(this, content);
	}

	@Override
	public String getName() {
		return gen.entity.getName()+"Dao.java";
	}
	
}


class JavaServiceGen  implements AutoGen{
	JavaCodeGen gen;
	public JavaServiceGen(JavaCodeGen gen){
		this.gen = gen;
	}
	@Override
	public void make(Target target, Entity entity) {
		GroupTemplate gt = target.getGroupTemplate();
		Template template = gt.getTemplate("/java/service.java");
		template.binding("entity", entity);
		template.binding("target", target);
		template.binding("package", gen.basePackage+".service");
		template.binding("basePackage", gen.basePackage);
		String content = template.render();
		target.flush(this, content);
	}

	@Override
	public String getName() {
		return gen.entity.getName()+"Service.java";
	}
	
}


class JavaControllerGen  implements AutoGen{
	JavaCodeGen gen;
	public JavaControllerGen(JavaCodeGen gen){
		this.gen = gen;
	}
	@Override
	public void make(Target target, Entity entity) {
		GroupTemplate gt = target.getGroupTemplate();
		Template template = gt.getTemplate("/java/controller.java");
		template.binding("entity", entity);
		template.binding("target", target);
		template.binding("package", gen.basePackage+".web");
		template.binding("basePackage", gen.basePackage);
		template.binding("basicfunctionCode",gen.basicFunctionCode);
		String content = template.render();
		target.flush(this, content);
	}

	@Override
	public String getName() {
		return gen.entity.getName()+"Controller.java";
	}
	
}


class JavaQueryGen  implements AutoGen{
	JavaCodeGen gen;
	public JavaQueryGen(JavaCodeGen gen){
		this.gen = gen;
	}
	@Override
	public void make(Target target, Entity entity) {
		GroupTemplate gt = target.getGroupTemplate();
		Template template = gt.getTemplate("/java/query.java");
		List<Attribute> list = new ArrayList<Attribute>();
		for(Attribute attr:entity.getList()) {
			if(attr.isShowInQuery()) {
				list.add(attr);
			}
		}
		
		if(list.isEmpty()) {
			list.add(entity.getIdAttribute());
		}
		
		template.binding("entity", entity);
		template.binding("target", target);
		template.binding("package", gen.basePackage+".web.query");
		template.binding("basePackage", gen.basePackage);
		template.binding("attrs", list);
		String content = template.render();
		target.flush(this, content);
	}

	@Override
	public String getName() {
		return gen.entity.getName()+"Query.java";
	}
	
}