org.vesalainen.code.Processor Maven / Gradle / Ivy
/*
* Copyright (C) 2014 Timo Vesalainen
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see classnames = new HashSet<>();
private static final AtomicInteger sequence = new AtomicInteger();
private Elements elements;
private Types types;
@Override
public synchronized void init(ProcessingEnvironment pe)
{
super.init(pe);
elements = pe.getElementUtils();
types = pe.getTypeUtils();
}
@Override
public boolean process(Set annotations, RoundEnvironment roundEnv)
{
Messager msg = processingEnv.getMessager();
for (TypeElement te : annotations)
{
for (Element e : roundEnv.getElementsAnnotatedWith(te))
{
TypeElement type = (TypeElement) e;
try
{
msg.printMessage(Diagnostic.Kind.NOTE, "processing", type);
System.err.println("processing "+type);
switch (te.getQualifiedName().toString())
{
case "org.vesalainen.code.BeanProxyClass":
generateBeanProxy(type, processingEnv);
break;
case "org.vesalainen.code.TransactionalSetterClass":
generateTransactionalSetter(type, processingEnv);
break;
case "org.vesalainen.code.PropertyDispatcherClass":
generatePropertyDispatcher(type, processingEnv);
break;
default:
throw new UnsupportedOperationException(te.getQualifiedName().toString()+" not supported");
}
}
catch (Exception ex)
{
ex.printStackTrace();
String m = ex.getMessage();
m = m != null ? m : ex.toString();
msg.printMessage(Diagnostic.Kind.ERROR, m, e);
}
}
}
return true;
}
private void generatePropertyDispatcher(TypeElement cls, ProcessingEnvironment processingEnv) throws IOException
{
PropertyDispatcherClass annotation = cls.getAnnotation(PropertyDispatcherClass.class);
if (annotation == null)
{
throw new IllegalArgumentException("@"+PropertyDispatcherClass.class.getSimpleName()+" missing in cls");
}
TypeElement transactional = elements.getTypeElement(Transactional.class.getCanonicalName());
Filer filer = processingEnv.getFiler();
String value = annotation.value();
JavaFileObject sourceFile = filer.createSourceFile(value);
try (Writer writer = sourceFile.openWriter())
{
CodePrinter mp = new CodePrinter(writer);
int idx = value.lastIndexOf('.');
String classname = value.substring(idx+1);
String pgk = value.substring(0, idx);
List methods = getMethods(cls);
List interfaces = cls.getInterfaces();
mp.println("package "+pgk+";");
mp.println("import org.vesalainen.util.Transactional;");
mp.println("import java.util.Arrays;");
mp.println("import javax.annotation.Generated;");
mp.println("import org.vesalainen.code.PropertySetterDispatcher;");
mp.println("@Generated(");
mp.println("\tvalue=\""+Processor.class.getCanonicalName()+"\"");
mp.println("\t, comments=\"Generated for "+cls+"\"");
Date date = new Date();
mp.println("\t, date=\""+date+"\"");
mp.println(")");
CodePrinter cp = mp.createClass(EnumSet.of(PUBLIC), classname, cls);
Set en = new TreeSet<>();
Integer[] sizes = new Integer[]{0, 0, 0, 0, 0, 0, 0, 0, 0};
for (ExecutableElement m : methods)
{
List parameters = m.getParameters();
TypeMirror returnType = m.getReturnType();
String name = m.getSimpleName().toString();
if (
name.startsWith("set") &&
parameters.size() == 1 &&
returnType.getKind() == VOID
)
{
en.add(getEnum(m));
try
{
VariableElement ve = parameters.get(0);
TypeMirror tm = ve.asType();
TypeKind tk = tm.getKind();
JavaType jt = JavaType.valueOf(tk.name());
sizes[jt.ordinal()]++;
}
catch (IllegalArgumentException ex)
{
}
}
}
cp.print("private enum Prop {");
cp.print(", ", en);
cp.println("};");
cp.println("public "+classname+"()");
cp.println("{");
CodePrinter cons1 = cp.createSub("}");
cons1.print("super(new int[] {");
cons1.print(", ", sizes);
cons1.println("});");
cons1.flush();
cp.println("public "+classname+"(PropertySetterDispatcher dispatcher)");
cp.println("{");
CodePrinter cons2 = cp.createSub("}");
cons2.print("super(new int[] {");
cons2.print(", ", sizes);
cons2.println("}, dispatcher);");
cons2.flush();
for (ExecutableElement m : methods)
{
List parameters = m.getParameters();
TypeMirror returnType = m.getReturnType();
cp.println("@Override");
String name = m.getSimpleName().toString();
EnumSet modifiers = EnumSet.of(PUBLIC);
CodePrinter cm = cp.createMethod(modifiers, m);
if (
name.startsWith("set") &&
parameters.size() == 1 &&
returnType.getKind() == VOID
)
{
String enumname = getEnum(m);
String property = getProperty(m);
VariableElement ve = parameters.get(0);
cm.println("if (observers.containsProperty(\""+property+"\"))");
cm.println("{");
CodePrinter ifobs = cm.createSub("}");
ifobs.println("set(Prop."+enumname+".ordinal(), "+ve.getSimpleName()+");");
ifobs.flush();
}
else
{
if (
name.equals("commit") &&
parameters.size() == 1 &&
"java.lang.String".equals(parameters.get(0).asType().toString()) &&
returnType.getKind() == VOID
)
{
cm.println("try");
cm.println("{");
CodePrinter ct = cm.createSub("}");
ct.println("Arrays.fill(ind, 0);");
ct.println("Prop[] values = Prop.values();");
ct.println("for (int ii=0;ii params = ee.getParameters();
TypeMirror rt = ee.getReturnType();
String sn = ee.getSimpleName().toString();
if (
sn.startsWith("set") &&
params.size() == 1 &&
rt.getKind() == VOID
)
{
JavaType jt = JavaType.valueOf(params.get(0).asType().getKind().name());
String aEnum = getEnum(ee);
String prop = getProperty(sn);
cs.println("case "+aEnum+":");
cs.println("{");
CodePrinter csw = cs.createSub("}");
String code = jt.getCode();
int ordinal = jt.ordinal();
csw.println(code+"[] a = ("+code+"[])arr["+ordinal+"];");
csw.println("observers.set(\""+prop+"\", a[ind["+ordinal+"]]);");
csw.println("ind["+ordinal+"]++;");
csw.println("break;");
csw.flush();
}
}
cs.println("default:");
CodePrinter csw = cs.createSub("");
csw.println("throw new UnsupportedOperationException(\"should not happen\");");
csw.flush();
cs.flush();
cc.flush();
ct.println("for (Transactional tr : transactionalObservers)");
ct.println("{");
CodePrinter ctr = ct.createSub("}");
ctr.println("tr.commit("+parameters.get(0).getSimpleName()+");");
ctr.flush();
ct.println("clear();");
ct.flush();
cm.println("finally");
cm.println("{");
CodePrinter cf = cm.createSub("}");
cf.println("semaphore.release();");
cf.flush();
}
else
{
if (
name.equals("rollback") &&
parameters.size() == 1 &&
"java.lang.String".equals(parameters.get(0).asType().toString()) &&
returnType.getKind() == VOID
)
{
cm.println("try");
cm.println("{");
CodePrinter ct = cm.createSub("}");
ct.println("clear();");
ct.println("for (Transactional tr : transactionalObservers)");
ct.println("{");
CodePrinter ctr = ct.createSub("}");
ctr.println("tr.rollback("+parameters.get(0).getSimpleName()+");");
ctr.flush();
ct.flush();
cm.println("finally");
cm.println("{");
CodePrinter cf = cm.createSub("}");
cf.println("semaphore.release();");
cf.flush();
}
else
{
if (
name.equals("start") &&
parameters.size() == 1 &&
"java.lang.String".equals(parameters.get(0).asType().toString()) &&
returnType.getKind() == VOID
)
{
cm.println("try");
cm.println("{");
CodePrinter cs = cm.createSub("}");
cs.println("semaphore.acquire();");
cs.flush();
cm.println("catch (InterruptedException ex)");
cm.println("{");
CodePrinter cc = cm.createSub("}");
cc.println("throw new IllegalArgumentException(ex);");
cc.flush();
}
else
{
cm.println("throw new UnsupportedOperationException(\"not supported.\");");
}
}
}
}
cm.flush();
}
cp.flush();
}
}
private void generateTransactionalSetter(TypeElement cls, ProcessingEnvironment processingEnv) throws IOException
{
TransactionalSetterClass annotation = cls.getAnnotation(TransactionalSetterClass.class);
if (annotation == null)
{
throw new IllegalArgumentException("@"+TransactionalSetterClass.class.getSimpleName()+" missing in cls");
}
TypeElement transactional = elements.getTypeElement(Transactional.class.getCanonicalName());
Filer filer = processingEnv.getFiler();
String value = annotation.value();
JavaFileObject sourceFile = filer.createSourceFile(value);
try (Writer writer = sourceFile.openWriter())
{
CodePrinter mp = new CodePrinter(writer);
int idx = value.lastIndexOf('.');
String classname = value.substring(idx+1);
String pgk = value.substring(0, idx);
List methods = getMethods(cls);
List interfaces = cls.getInterfaces();
if (interfaces.size() != 1)
{
throw new IllegalArgumentException("interface count != 1 (="+interfaces.size()+")");
}
TypeMirror theInterface = interfaces.get(0);
mp.println("package "+pgk+";");
mp.println("import java.util.Arrays;");
mp.println("import javax.annotation.Generated;");
mp.println("@Generated(");
mp.println("\tvalue=\""+Processor.class.getCanonicalName()+"\"");
mp.println("\t, comments=\"Generated for "+cls+"\"");
Date date = new Date();
mp.println("\t, date=\""+date+"\"");
mp.println(")");
CodePrinter cp = mp.createClass(EnumSet.of(PUBLIC), classname, cls);
Set en = new TreeSet<>();
Integer[] sizes = new Integer[]{0, 0, 0, 0, 0, 0, 0, 0, 0};
for (ExecutableElement m : methods)
{
List parameters = m.getParameters();
TypeMirror returnType = m.getReturnType();
String name = m.getSimpleName().toString();
if (
name.startsWith("set") &&
parameters.size() == 1 &&
returnType.getKind() == VOID
)
{
en.add(getEnum(m));
try
{
VariableElement ve = parameters.get(0);
TypeMirror tm = ve.asType();
TypeKind tk = tm.getKind();
JavaType jt = JavaType.valueOf(tk.name());
sizes[jt.ordinal()]++;
}
catch (IllegalArgumentException ex)
{
}
}
}
cp.print("private enum Prop {");
cp.print(", ", en);
cp.println("};");
cp.println("public "+classname+"()");
cp.println("{");
CodePrinter cons = cp.createSub("}");
cons.print("super(new int[] {");
cons.print(", ", sizes);
cons.println("});");
cons.flush();
for (ExecutableElement m : methods)
{
List parameters = m.getParameters();
TypeMirror returnType = m.getReturnType();
cp.println("@Override");
CodePrinter cm = cp.createMethod(EnumSet.of(PUBLIC), m);
String name = m.getSimpleName().toString();
String enumname = getEnum(m);
if (
name.startsWith("set") &&
parameters.size() == 1 &&
returnType.getKind() == VOID
)
{
VariableElement ve = parameters.get(0);
cm.println("set(Prop."+enumname+".ordinal(), "+ve.getSimpleName()+");");
}
else
{
if (
name.equals("commit") &&
parameters.size() == 1 &&
"java.lang.String".equals(parameters.get(0).asType().toString()) &&
returnType.getKind() == VOID
)
{
cm.println("Arrays.fill(ind, 0);");
cm.println("Prop[] values = Prop.values();");
String iname = theInterface.toString();
cm.println(iname+" interf = ("+iname+")intf;");
cm.println("for (int ii=0;ii params = ee.getParameters();
TypeMirror rt = ee.getReturnType();
String sn = ee.getSimpleName().toString();
if (
sn.startsWith("set") &&
params.size() == 1 &&
rt.getKind() == VOID
)
{
JavaType jt = JavaType.valueOf(params.get(0).asType().getKind().name());
String aEnum = getEnum(ee);
cs.println("case "+aEnum+":");
cs.println("{");
CodePrinter csw = cs.createSub("}");
String code = jt.getCode();
int ordinal = jt.ordinal();
csw.println(code+"[] a = ("+code+"[])arr["+ordinal+"];");
csw.println("interf."+sn+"("+cast(params.get(0).asType())+"a[ind["+ordinal+"]++]);");
csw.println("break;");
csw.flush();
}
}
cs.println("default:");
CodePrinter csw = cs.createSub("");
csw.println("throw new UnsupportedOperationException(\"should not happen\");");
csw.flush();
cs.flush();
cc.flush();
cm.println("clear();");
if (types.isAssignable(theInterface, transactional.asType()))
{
cm.println("interf.commit("+parameters.get(0).getSimpleName()+");");
}
}
else
{
if (
name.equals("rollback") &&
parameters.size() == 1 &&
"java.lang.String".equals(parameters.get(0).asType().toString()) &&
returnType.getKind() == VOID
)
{
cm.println("clear();");
if (types.isAssignable(theInterface, transactional.asType()))
{
String iname = theInterface.toString();
cm.println(iname+" interf = ("+iname+")intf;");
cm.println("interf.rollback("+parameters.get(0).getSimpleName()+");");
}
}
else
{
if (
name.equals("start") &&
parameters.size() == 1 &&
"java.lang.String".equals(parameters.get(0).asType().toString()) &&
returnType.getKind() == VOID
)
{
}
else
{
cm.println("throw new UnsupportedOperationException(\"not supported.\");");
}
}
}
}
cm.flush();
}
cp.flush();
}
}
private void generateBeanProxy(TypeElement cls, ProcessingEnvironment processingEnv) throws IOException
{
BeanProxyClass annotation = cls.getAnnotation(BeanProxyClass.class);
if (annotation == null)
{
throw new IllegalArgumentException("@"+BeanProxyClass.class.getSimpleName()+" missing in cls");
}
Filer filer = processingEnv.getFiler();
String value = annotation.value();
JavaFileObject sourceFile = filer.createSourceFile(value);
try (Writer writer = sourceFile.openWriter())
{
CodePrinter mp = new CodePrinter(writer);
int idx = value.lastIndexOf('.');
String classname = value.substring(idx+1);
String pgk = value.substring(0, idx);
List methods = getMethods(cls);
mp.println("package "+pgk+";");
mp.println("import javax.annotation.Generated;");
mp.println("@Generated(");
mp.println("\tvalue=\""+Processor.class.getCanonicalName()+"\"");
mp.println("\t, comments=\"Generated for "+cls+"\"");
Date date = new Date();
mp.println("\t, date=\""+date+"\"");
mp.println(")");
CodePrinter cp = mp.createClass(EnumSet.of(PUBLIC), classname, cls);
for (ExecutableElement m : methods)
{
List parameters = m.getParameters();
TypeMirror returnType = m.getReturnType();
cp.println("@Override");
CodePrinter cm = cp.createMethod(EnumSet.of(PUBLIC), m);
String name = m.getSimpleName().toString();
String property = getProperty(name);
if (
name.startsWith("set") &&
parameters.size() == 1 &&
returnType.getKind() == VOID
)
{
VariableElement ve = parameters.get(0);
cm.println("set(\""+property+"\", "+ve.getSimpleName()+");");
}
else
{
if (
name.startsWith("get") &&
parameters.isEmpty() &&
returnType.getKind() != VOID
)
{
cm.println("return "+cast(returnType)+getter(returnType.getKind())+"(\""+property+"\");");
}
else
{
cm.println("throw new UnsupportedOperationException(\"not supported.\");");
}
}
cm.flush();
}
cp.flush();
}
}
private List getMethods(TypeElement cls)
{
List list = new ArrayList<>();
for (ExecutableElement m : ElementFilter.methodsIn(elements.getAllMembers(cls)))
{
if (m.getModifiers().contains(ABSTRACT))
{
list.add(m);
}
}
return list;
}
private synchronized String getUniqueClassname(String name)
{
String res = name;
while (classnames.contains(res))
{
res = name+sequence.incrementAndGet();
}
classnames.add(res);
return res;
}
private Set getImports(List methods)
{
Set set = new TreeSet<>();
for (ExecutableElement ee : methods)
{
add(set, ee.getReturnType());
for (TypeMirror tm : ee.getThrownTypes())
{
add(set, tm);
}
for (VariableElement ve : ee.getParameters())
{
add(set, ve.asType());
}
}
return set;
}
private static void add(Set set, TypeMirror tm)
{
if (tm.getKind() == DECLARED)
{
String cl = tm.toString();
int idx = cl.lastIndexOf('.');
if (idx != -1)
{
if ("java.lang".equals(cl.substring(0, idx)))
{
return;
}
}
set.add(cl);
}
}
private String getProperty(String name)
{
return Character.toLowerCase(name.charAt(3))+name.substring(4);
}
private String getEnum(ExecutableElement m)
{
String name = m.getSimpleName().toString();
List parameters = m.getParameters();
if (parameters.size() != 1)
{
throw new IllegalArgumentException(m.toString());
}
VariableElement ve = parameters.get(0);
TypeMirror type = ve.asType();
String typeString = type.toString().replace('.', '_').replace('<', '_').replace('>', '_');
return name.substring(3)+'_'+typeString;
}
private String getProperty(ExecutableElement m)
{
String name = m.getSimpleName().toString();
return Character.toLowerCase(name.charAt(3))+name.substring(4);
}
private String getter(TypeKind kind)
{
return "get"+getTypename(kind);
}
private String getTypename(TypeKind kind)
{
switch (kind)
{
case ARRAY:
case DECLARED:
return "Object";
case BOOLEAN:
return "Boolean";
case BYTE:
return "Byte";
case CHAR:
return "Char";
case DOUBLE:
return "Double";
case FLOAT:
return "Float";
case INT:
return "Int";
case LONG:
return "Long";
case SHORT:
return "Short";
default:
throw new IllegalArgumentException("not type "+kind);
}
}
private String cast(TypeMirror tm)
{
switch (tm.getKind())
{
case ARRAY:
case DECLARED:
return "("+tm.toString()+")";
default:
return "";
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy