com.appslandia.plum.jsp.FunctionTldGenerator Maven / Gradle / Ivy

Go to download
// The MIT License (MIT)
// Copyright © 2015 AppsLandia. All rights reserved.

// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:

// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

package com.appslandia.plum.jsp;

import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import com.appslandia.common.base.TextBuilder;
import com.appslandia.common.utils.XmlEscaper;

/**
 *
 * @author Loc Ha
 *
 */
public class FunctionTldGenerator {

  public static void main(String[] args) {
    TextBuilder sb = new TextBuilder();

    generateFunction(Functions.class, sb);
    generateFunction(OwaspFunctions.class, sb);
    generateFunction(Pipe.class, sb);

    System.out.println(sb);
  }

  public static void generateFunction(Class clazz, TextBuilder sb) {
    List functionMths = new ArrayList<>();
    for (Method method : clazz.getMethods()) {
      if (Modifier.isPublic(method.getModifiers()) && Modifier.isStatic(method.getModifiers())) {

        if (method.getDeclaredAnnotation(Function.class) != null) {
          functionMths.add(method);
        }
      }
    }
    Collections.sort(functionMths, (m1, m2) -> {
      int compare = m1.getDeclaringClass().getName().compareTo(m2.getDeclaringClass().getName());
      if (compare != 0) {
        return compare;
      }
      return m1.getName().compareTo(m2.getName());
    });

    for (Method functionMth : functionMths) {
      Function function = functionMth.getDeclaredAnnotation(Function.class);
      String signature = functionSignature(functionMth);

      String name = function.name().length() == 0 ? functionMth.getName() : function.name();
      StringBuilder desc = new StringBuilder();

      if (!function.description().isEmpty()) {
        desc.append("description=").append(function.description());
      } else {
        desc.append("description=");
      }

      String mth = functionMth.toString();
      if (mth.startsWith("public static ")) {
        mth = mth.substring("public static ".length());
      }
      desc.append(";method=").append(mth);

      if (sb.length() == 0) {
        sb.appendtab().append("");
        sb.appendln();
      } else {
        sb.appendln(2).appendtab().append("");
        sb.appendln();
      }
      sb.appendtab(2).append("" + XmlEscaper.escapeXml(desc.toString()) + "");
      sb.appendln();
      sb.appendtab(2).append("" + name + "");
      sb.appendln();
      sb.appendtab(2).append("" + clazz.getName() + "");
      sb.appendln();
      sb.appendtab(2).append("" + signature + "");
      sb.appendln();

      sb.appendtab().append("");
    }
  }

  public static String functionSignature(Method method) {
    StringBuilder sb = new StringBuilder();
    sb.append(method.getReturnType().getName()).append(" ");
    sb.append(method.getName()).append('(');
    boolean first = true;
    for (Class paramType : method.getParameterTypes()) {
      if (first) {
        sb.append(paramType.getName());
        first = false;
      } else {
        sb.append(", ").append(paramType.getName());
      }
    }
    sb.append(')');
    return sb.toString();
  }
}