org.beangle.struts2.freemarker.TemplateFinder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of beangle-struts2-convention Show documentation
There is a newer version: 3.5.3
Show newest version
/*
 * Beangle, Agile Java/Scala Development Scaffold and Toolkit
 *
 * Copyright (c) 2005-2013, Beangle Software.
 *
 * Beangle is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Beangle is distributed in the hope that it will be useful.
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Beangle.  If not, see .
 */
package org.beangle.struts2.freemarker;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.Set;

import org.beangle.commons.collection.CollectUtils;
import org.beangle.struts2.convention.route.ViewMapper;

import freemarker.cache.TemplateLoader;
import freemarker.template.Configuration;

/**
 * Try to find template
 * 
 * @author chaostone
 */
public class TemplateFinder {

  private final TemplateLoader loader;

  private final Configuration configuration;

  private final ViewMapper viewMapper;

  private Set missings = CollectUtils.newHashSet();

  public TemplateFinder(TemplateLoader loader, ViewMapper viewMapper) {
    super();
    this.loader = loader;
    this.configuration = null;
    this.viewMapper = viewMapper;
  }

  public TemplateFinder(Configuration configuration, ViewMapper viewMapper) {
    super();
    this.loader = null;
    this.configuration = configuration;
    this.viewMapper = viewMapper;
  }

  public String find(Class actionClass, String method, String viewName, String extention) {
    if (null != loader) return findByLoader(actionClass, method, viewName, extention);
    else return findByConfig(actionClass, method, viewName, extention);
  }

  private String findByConfig(Class actionClass, String method, String viewName, String extention) {
    String path = null;
    Class superClass = actionClass;
    Object source = null;
    do {
      StringBuilder buf = new StringBuilder();
      buf.append(viewMapper.getViewPath(superClass.getName(), method, viewName));
      buf.append('.').append(extention);
      path = buf.toString();
      String templateName = path;
      if (path.charAt(0) != '/') templateName = "/" + templateName;
      if (!missings.contains(templateName)) {
        try {
          source = configuration.getTemplate(templateName);
        } catch (FileNotFoundException e) {
          missings.add(templateName);
        } catch (IOException e) {
          break;
        }
      }
      superClass = superClass.getSuperclass();
    } while (null == source && !superClass.equals(Object.class) && !superClass.isPrimitive());
    return (null == source) ? null : path;
  }

  private String findByLoader(Class actionClass, String method, String viewName, String extention) {
    String path = null;
    Class superClass = actionClass;
    Object source = null;
    do {
      StringBuilder buf = new StringBuilder();
      buf.append(viewMapper.getViewPath(superClass.getName(), method, viewName));
      buf.append('.').append(extention);
      path = buf.toString();
      String templateName = path;
      if (path.charAt(0) == '/') templateName = templateName.substring(1);
      if (!missings.contains(templateName)) {
        try {
          source = loader.findTemplateSource(templateName);
          if (null != source) loader.closeTemplateSource(source);
          else missings.add(templateName);
        } catch (IOException e) {
          break;
        }
      }
      superClass = superClass.getSuperclass();
    } while (null == source && !superClass.equals(Object.class) && !superClass.isPrimitive());
    return (null == source) ? null : path;
  }
}