cn.dreampie.routebind.RouteBind Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jfinal-routebind Show documentation
jfinal routebind plugins
The newest version!
package cn.dreampie.routebind;

import cn.dreampie.ClassSearchKit;
import com.google.common.base.Preconditions;
import com.google.common.collect.Lists;
import com.jfinal.config.Routes;
import com.jfinal.core.Controller;
import com.jfinal.kit.StrKit;
import com.jfinal.log.Logger;

import java.util.List;

/**
 * Created by wangrenhui on 14-1-2.
 */
public class RouteBind extends Routes {

  private boolean autoScan = true;

  private List> excludeClasses = Lists.newArrayList();
  private List> includeClasses = Lists.newArrayList();
  private List includeClassPaths = Lists.newArrayList();
  private List excludeClassPaths = Lists.newArrayList();
//  private boolean includeAllJarsInLib;

//  private List includeJars = Lists.newArrayList();

  protected final Logger logger = Logger.getLogger(getClass());

  private String suffix = "Controller";

  public RouteBind autoScan(boolean autoScan) {
    this.autoScan = autoScan;
    return this;
  }

  @SuppressWarnings({"rawtypes", "unchecked"})
  public RouteBind addExcludeClasses(Class... clazzes) {
    for (Class clazz : clazzes) {
      excludeClasses.add(clazz);
    }
    return this;
  }

  @SuppressWarnings("rawtypes")
  public RouteBind addExcludeClasses(List> clazzes) {
    if (clazzes != null) {
      excludeClasses.addAll(clazzes);
    }
    return this;
  }

  public RouteBind addExcludePaths(String... paths) {
    for (String path : paths) {
      excludeClassPaths.add(path);
    }
    return this;
  }

  @SuppressWarnings({"rawtypes", "unchecked"})
  public RouteBind addIncludeClasses(Class... clazzes) {
    for (Class clazz : clazzes) {
      includeClasses.add(clazz);
    }
    return this;
  }

  @SuppressWarnings("rawtypes")
  public RouteBind addIncludeClasses(List> clazzes) {
    if (clazzes != null) {
      includeClasses.addAll(clazzes);
    }
    return this;
  }

  public RouteBind addIncludePaths(String... paths) {
    for (String path : paths) {
      includeClassPaths.add(path);
    }
    return this;
  }


  @Override
  @SuppressWarnings({"rawtypes", "unchecked"})
  public void config() {
    List> controllerClasses = ClassSearchKit.of(Controller.class).includepaths(includeClassPaths).search();
    ControllerKey controllerKey = null;
    for (Class controller : controllerClasses) {
      if (excludeClasses.contains(controller)) {
        continue;
      }
      controllerKey = (ControllerKey) controller.getAnnotation(ControllerKey.class);
      if (controllerKey == null) {
        if (!autoScan) {
          continue;
        }
        this.add(controllerKey(controller), controller);
        logger.debug("routes.add(" + controllerKey(controller) + ", " + controller.getName() + ")");
      } else if (StrKit.isBlank(controllerKey.path())) {
        this.add(controllerKey.value(), controller);
        logger.debug("routes.add(" + controllerKey.value() + ", " + controller.getName() + ")");
      } else {
        this.add(controllerKey.value(), controller, controllerKey.path());
        logger.debug("routes.add(" + controllerKey.value() + ", " + controller + ","
            + controllerKey.path() + ")");
      }
    }
  }

  private String controllerKey(Class clazz) {
    Preconditions.checkArgument(clazz.getSimpleName().endsWith(suffix),
        " does not has a @ControllerKey annotation and it's name is not end with " + suffix);
    String simpleName = clazz.getSimpleName();
    String controllerKey = "/";
    if (!simpleName.equalsIgnoreCase(suffix)) {
      controllerKey += StrKit.firstCharToLowerCase(simpleName.replace(suffix, ""));
    }
    return controllerKey;
  }

//  public AutoBindRoutes includeAllJarsInLib(boolean includeAllJarsInLib) {
//    this.includeAllJarsInLib = includeAllJarsInLib;
//    return this;
//  }

  public RouteBind suffix(String suffix) {
    this.suffix = suffix;
    return this;
  }

}