/*
    * Copyright 2004-2006 the Seasar Foundation and the Others.
    *
    * Licensed under the Apache License, Version 2.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.apache.org/licenses/LICENSE-2.0
    *
   * 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.seasar.tuigwaa.plugin.basic;
  
  import java.io.IOException;
  import java.io.StringWriter;
  import java.util.Arrays;
  import java.util.HashMap;
  import java.util.Map;
  import java.util.Properties;
  
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.apache.velocity.Template;
  import org.apache.velocity.VelocityContext;
  import org.apache.velocity.app.Velocity;
  import org.apache.velocity.exception.MethodInvocationException;
  import org.apache.velocity.exception.ParseErrorException;
  import org.apache.velocity.exception.ResourceNotFoundException;
  import org.apache.velocity.runtime.resource.loader.FileResourceLoader;
  import org.seasar.framework.container.factory.SingletonS2ContainerFactory;
  import org.seasar.tuigwaa.cms.core.CmsRequest;
  import org.seasar.tuigwaa.cms.core.CmsResponse;
  import org.seasar.tuigwaa.plugin.AbstractTgwPlugin;
  import org.seasar.tuigwaa.plugin.PluginException;
  import org.seasar.tuigwaa.plugin.PluginRequest;
  import org.seasar.tuigwaa.plugin.PluginService;
  import org.seasar.tuigwaa.util.TgwContext;
  import org.seasar.tuigwaa.view.DataViewUtils;
  
  import com.isenshi.util.CalendarUtils;
  import com.isenshi.util.CharUtil;
  import com.isenshi.util.ResourceUtils;
  
  public class VelocityPlugin extends AbstractTgwPlugin {
  
  	private Log log = LogFactory.getLog(getClass());
  
  	private static final String ENCODING = "Windows-31J";
  
  	private static final String DIR_VELOCITYFILE = "org/seasar/tuigwaa/plugin/velocity/";
  
  	private PluginService pluginService = (PluginService) getService(PluginService.class);
  
  	private Map templateMap = new HashMap();
  
  	private Map pathMap = new HashMap();
  
  	public VelocityPlugin() {
  		Properties props = new Properties();
  		props.put("resource.loader", "CLASSPATH");
  		props.put("CLASSPATH.resource.loader.class", FileResourceLoader.class
  				.getName());
  		props.put("CLASSPATH.resource.loader.path", ResourceUtils.getPath(""));
  		try {
  			// props.load(ResourceUtils.getInputStream("velocity.properties"));
  			Velocity.init(props);
  		} catch (IOException e) {
  			// e.printStackTrace();
  			log.warn("veloctiy plugin load execpetion" + e.getMessage());
  		} catch (Exception e) {
  			// e.printStackTrace();
  			log.warn("velocity not initialized" + e.getMessage());
  		}
  	}
  
  	public String doHTMLView(CmsRequest request, CmsResponse response,
  			PluginRequest prequest) throws PluginException {
  		return doHTMLViewBindingObject(request, response, prequest, null);
  	}
  
  	public String doHTMLViewBindingObject(CmsRequest request,
  			CmsResponse response, PluginRequest prequest, Object outputDto)
  			throws PluginException {
  
  		String pluginName = prequest.getName();
  		return doHTMLViewBindingObject(request, response, prequest, outputDto,
  				getPath(pluginName));
  	}
  
  	public String doHTMLViewBindingObject(CmsRequest request,
  			CmsResponse response, PluginRequest prequest, Object outputDto,
  			String path) throws PluginException {
  		VelocityContext context = createContext(request, response, prequest,
  				outputDto);
  		Template template = getTemplate(path);
 		return merge(template, context);
 	}
 
 	public void putPath(String pluginName, Class clazz) {
 		String path = clazz.getName().replaceAll("//.", "///");
 		pathMap.put(pluginName, path + ".vm");
 	}
 
 	private String getPath(String pluginName) {
 		String path = (String) pathMap.get(pluginName);
 		if (path == null) {
 			return DIR_VELOCITYFILE + pluginName + ".vm";
 		}
 		return path;
 	}
 
 	private VelocityContext createContext(CmsRequest req, CmsResponse res,
 			PluginRequest prequest, Object obj) {
 		VelocityContext context = new VelocityContext();
 		if (prequest.getArgs() != null) {
 			context.put("args", Arrays.asList(prequest.getArgs()));
 		}
 
 		// Web Appli Setting
 		String message = pluginService.getMessage(getClass());
 		context.put("message", message);
 		context.put("actionPath", getActionPath());
 		context.put("hidden", getHiddenElements(req));
 
 		// Default Setting
 		context.put("pagePath", req.getPage().getResource().getPath());
 		context.put("encodedPagePath", CharUtil.urlEncode(req.getPage()
 				.getResource().getPath()));
 		context.put("mainPagePath", req.getMainPagePath());
 		context.put("encodedMainPagePath", CharUtil.urlEncode(req
 				.getMainPagePath()));
 		context.put("contextPath", TgwContext.getContextPath());
 		context.put("siteName", req.getSiteName());
 		context.put("bean", obj);
 		context.put("request", SingletonS2ContainerFactory.getContainer()
 				.getRequest());
 
 		context.put("charutil", new CharUtil());
 		context.put("dataviewutils", new DataViewUtils());
 		context.put("calendarutils", new CalendarUtils());
 		
 		return context;
 	}
 
 	protected String merge(String path, CmsRequest req, CmsResponse res,
 			PluginRequest preq, Object dto) {
 		Template template = getTemplate(path);
 		VelocityContext context = createContext(req, res, preq, dto);
 		return merge(template, context);
 	}
 
 	protected String merge(Template template, VelocityContext context) {
 		if (template == null) {
 			return "";
 		}
 		StringWriter writer = new StringWriter();
 		try {
 			template.merge(context, writer);
 		} catch (ResourceNotFoundException e) {
 			e.printStackTrace();
 		} catch (ParseErrorException e) {
 			e.printStackTrace();
 		} catch (MethodInvocationException e) {
 			e.printStackTrace();
 		} catch (Exception e) {
 			e.printStackTrace();
 		}
 		String ret = writer.toString();
 		writer.flush();
 		return ret;
 	}
 
 	private Template getTemplate(String path) {
 		Template template = (Template) templateMap.get(path);
 		if (template == null) {
 			try {
 				template = Velocity.getTemplate(path, ENCODING);
 			} catch (ResourceNotFoundException e) {
 				e.printStackTrace();
 			} catch (ParseErrorException e) {
 				e.printStackTrace();
 			} catch (Exception e) {
 				e.printStackTrace();
 			}
 		}
 		return template;
 	}
 }